@class MLNeuralNetworkEngine;
@protocol MLBatchProvider;

@interface MLComputeBatchDataSource : NSObject <MLComputeDataProvider>

@property (readonly, nonatomic) id<MLBatchProvider> batchProvider;
@property (readonly, nonatomic) MLNeuralNetworkEngine *nnEngine;
@property (readonly, nonatomic) BOOL useForPrediction;
@property (nonatomic) unsigned long long batchSize;

- (void).cxx_destruct;
- (unsigned long long)numberOfBatches;
- (id)batchAtIndex:(unsigned long long)a0 error:(id *)a1;
- (id)initWithBatchProvider:(id)a0 batchSize:(unsigned long long)a1 forPrediction:(BOOL)a2 neuralNetworkEngine:(id)a3 error:(id *)a4;
- (id)mlcDataSourceAtIndex:(long long)a0 error:(id *)a1;
- (unsigned long long)sizeOfBatchAtIndex:(unsigned long long)a0;

@end