Searched refs:BatchDatasetOp (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | batch_dataset_op.cc | 37 /* static */ constexpr const char* const BatchDatasetOp::kDatasetType;
38 /* static */ constexpr const char* const BatchDatasetOp::kInputDataset;
39 /* static */ constexpr const char* const BatchDatasetOp::kBatchSize;
40 /* static */ constexpr const char* const BatchDatasetOp::kDropRemainder;
41 /* static */ constexpr const char* const BatchDatasetOp::kParallelCopy;
42 /* static */ constexpr const char* const BatchDatasetOp::kOutputTypes;
43 /* static */ constexpr const char* const BatchDatasetOp::kOutputShapes;
48 class BatchDatasetOp::Dataset : public DatasetBase {
256 BatchDatasetOp::BatchDatasetOp(OpKernelConstruction* ctx) in BatchDatasetOp() function in tensorflow::data::BatchDatasetOp
264 void BatchDatasetOp::MakeDataset(OpKernelContext* ctx, DatasetBase* input, in MakeDataset()
[all …]
|
| D | batch_dataset_op.h | 23 class BatchDatasetOp : public UnaryDatasetOpKernel {
33 explicit BatchDatasetOp(OpKernelConstruction* ctx);
|
| D | batch_dataset_op_test.cc | 164 name_utils::OpName(BatchDatasetOp::kDatasetType, params))); in TEST_F()
241 BatchDatasetOp::kDatasetType, batch_dataset_params.iterator_prefix(), in ITERATOR_OUTPUT_SHAPES_TEST_P()
|
| D | dataset_test_base.cc | 1019 *input_names = {BatchDatasetOp::kInputDataset, BatchDatasetOp::kBatchSize, in GetInputNames()
1020 BatchDatasetOp::kDropRemainder}; in GetInputNames()
1025 *attr_vector = {{BatchDatasetOp::kParallelCopy, parallel_copy_}, in GetAttributes()
1026 {BatchDatasetOp::kOutputTypes, output_dtypes_}, in GetAttributes()
1027 {BatchDatasetOp::kOutputShapes, output_shapes_}}; in GetAttributes()
1032 return BatchDatasetOp::kDatasetType; in dataset_type()
|