Searched refs:BatchDatasetParams (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | batch_dataset_op_test.cc | 26 BatchDatasetParams BatchDatasetParams1() { in BatchDatasetParams1()
27 return BatchDatasetParams(RangeDatasetParams(0, 12, 1), in BatchDatasetParams1()
38 BatchDatasetParams BatchDatasetParams2() { in BatchDatasetParams2()
39 return BatchDatasetParams(RangeDatasetParams(0, 12, 1), in BatchDatasetParams2()
50 BatchDatasetParams BatchDatasetParams3() { in BatchDatasetParams3()
51 return BatchDatasetParams(RangeDatasetParams(0, 10, 1), in BatchDatasetParams3()
62 BatchDatasetParams BatchDatasetParams4() { in BatchDatasetParams4()
63 return BatchDatasetParams(RangeDatasetParams(0, 10, 1), in BatchDatasetParams4()
74 BatchDatasetParams BatchDatasetParams5() { in BatchDatasetParams5()
75 return BatchDatasetParams(RangeDatasetParams(0, 10, 1), in BatchDatasetParams5()
[all …]
|
| D | map_dataset_op_test.cc | 56 BatchDatasetParams(RangeDatasetParams(10, 0, -3), in MapDatasetParams2()
|
| D | dataset_test_base.cc | 1010 std::vector<Tensor> BatchDatasetParams::GetInputTensors() const { in GetInputTensors()
1017 Status BatchDatasetParams::GetInputNames( in GetInputNames()
1024 Status BatchDatasetParams::GetAttributes(AttributeVector* attr_vector) const { in GetAttributes()
1031 string BatchDatasetParams::dataset_type() const { in dataset_type()
|
| D | dataset_test_base.h | 205 class BatchDatasetParams : public DatasetParams {
208 BatchDatasetParams(T input_dataset_params, int64 batch_size, in BatchDatasetParams() function
|