| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | assert_prev_dataset_op_test.cc | 84 return AssertPrevDatasetOp::kDatasetType; in dataset_type()
103 {GetTransformation(TakeDatasetOp::kDatasetType)}, in AssertPrevDatasetParams1()
119 {GetTransformation(TakeDatasetOp::kDatasetType), in AssertPrevDatasetParams2()
120 GetTransformation(RangeDatasetOp::kDatasetType)}, in AssertPrevDatasetParams2()
140 {GetTransformation(TakeDatasetOp::kDatasetType), in AssertPrevDatasetParams2WithAttrs()
141 GetTransformation(TensorSliceDatasetOp::kDatasetType, in AssertPrevDatasetParams2WithAttrs()
173 {GetTransformation(TakeDatasetOp::kDatasetType), in ShortAssertPrevDatasetParams()
174 GetTransformation(RangeDatasetOp::kDatasetType), in ShortAssertPrevDatasetParams()
209 name_utils::OpName(AssertPrevDatasetOp::kDatasetType))); in TEST_F()
246 AssertPrevDatasetOp::kDatasetType, dataset_params.iterator_prefix()))); in TEST_F()
|
| D | assert_next_dataset_op_test.cc | 62 return AssertNextDatasetOp::kDatasetType; in dataset_type()
80 /*transformations=*/{TakeDatasetOp::kDatasetType}, in AssertNextDatasetParams1()
96 {TakeDatasetOp::kDatasetType, RangeDatasetOp::kDatasetType}, in AssertNextDatasetParams2()
126 {TakeDatasetOp::kDatasetType, RangeDatasetOp::kDatasetType, "Whoops"}, in ShortAssertNextDatasetParams()
154 name_utils::OpName(AssertNextDatasetOp::kDatasetType))); in TEST_F()
191 AssertNextDatasetOp::kDatasetType, dataset_params.iterator_prefix()))); in TEST_F()
|
| D | threadpool_dataset_op.h | 27 static constexpr const char* const kDatasetType =
51 static constexpr const char* const kDatasetType = "PrivateThreadPoolDataset";
|
| D | sampling_dataset_op_test.cc | 63 return SamplingDatasetOp::kDatasetType; in dataset_type()
133 SamplingDatasetOp::kDatasetType)}}; in DATASET_NODE_NAME_TEST_P()
191 SamplingDatasetOp::kDatasetType, in ITERATOR_OUTPUT_SHAPES_TEST_P()
|
| D | assert_next_dataset_op.cc | 29 /* static */ constexpr const char* const AssertNextDatasetOp::kDatasetType; member in tensorflow::data::experimental::AssertNextDatasetOp
53 this, name_utils::IteratorPrefix(kDatasetType, prefix)}); in MakeIteratorInternal()
62 return name_utils::DatasetDebugString(kDatasetType); in DebugString()
|
| D | random_dataset_op_test.cc | 74 return RandomDatasetOp::kDatasetType; in dataset_type()
162 RandomDatasetOp::kDatasetType)}}; in DatasetTypeStringTestCases()
216 RandomDatasetOp::kDatasetType, kIteratorPrefix)}}; in IteratorOutputPrefixTestCases()
|
| D | assert_cardinality_dataset_op.cc | 31 AssertCardinalityDatasetOp::kDatasetType; member in tensorflow::data::experimental::AssertCardinalityDatasetOp
57 this, name_utils::IteratorPrefix(kDatasetType, prefix)}); in MakeIteratorInternal()
66 return name_utils::DatasetDebugString(kDatasetType); in DebugString()
|
| D | unique_dataset_op_test.cc | 53 string dataset_type() const override { return UniqueDatasetOp::kDatasetType; } in dataset_type()
192 name_utils::OpName(UniqueDatasetOp::kDatasetType))); in TEST_F()
245 UniqueDatasetOp::kDatasetType, dataset_params.iterator_prefix()))); in ITERATOR_OUTPUT_SHAPES_TEST_P()
|
| D | lmdb_dataset_op_test.cc | 57 string dataset_type() const override { return LMDBDatasetOp::kDatasetType; } in dataset_type()
215 LMDBDatasetOp::kDatasetType)}}; in DatasetTypeStringTestCases()
271 LMDBDatasetOp::kDatasetType, kIteratorPrefix)}}; in IteratorOutputPrefixTestCases()
|
| D | io_ops.h | 77 static constexpr const char* const kDatasetType = "SaveV2"; variable
111 static constexpr const char* const kDatasetType = "Load"; variable
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | zip_dataset_op_test.cc | 50 absl::StrCat(ZipDatasetOp::kDatasetType, "_", i)); in GetInputNames()
64 string dataset_type() const override { return ZipDatasetOp::kDatasetType; } in dataset_type()
117 CheckDatasetTypeString(name_utils::OpName(ZipDatasetOp::kDatasetType))); in TEST_F()
182 ZipDatasetOp::kDatasetType, dataset_params.iterator_prefix()))); in ITERATOR_OUTPUT_SHAPES_TEST_P()
|
| D | tf_record_dataset_op.cc | 35 /* static */ constexpr const char* const TFRecordDatasetOp::kDatasetType; member in tensorflow::data::TFRecordDatasetOp
72 this, name_utils::IteratorPrefix(kDatasetType, prefix)}); in MakeIteratorInternal()
87 return name_utils::DatasetDebugString(kDatasetType); in DebugString()
131 metrics::GetTFDataBytesReadCounter(kDatasetType); in GetNextInternal()
298 metrics::RecordTFDataFilename(kDatasetType, filenames[i]); in MakeDataset()
|
| D | text_line_dataset_op.cc | 31 /* static */ constexpr const char* const TextLineDatasetOp::kDatasetType; member in tensorflow::data::TextLineDatasetOp
56 name_utils::IteratorPrefix(TextLineDatasetOp::kDatasetType, prefix)}); in MakeIteratorInternal()
71 return name_utils::DatasetDebugString(kDatasetType); in DebugString()
116 name_utils::OpName(TextLineDatasetOp::kDatasetType)); in GetNextInternal()
280 metrics::RecordTFDataFilename(kDatasetType, filenames[i]); in MakeDataset()
|
| D | shuffle_dataset_op.h | 42 static constexpr const char* const kDatasetType = "Shuffle";
60 static constexpr const char* const kDatasetType = "ShuffleAndRepeat";
|
| D | shuffle_dataset_op_test.cc | 86 return ShuffleAndRepeatDatasetOp::kDatasetType; in dataset_type()
88 return ShuffleDatasetOp::kDatasetType; in dataset_type()
394 ShuffleDatasetOp::kDatasetType)}, in DATASET_NODE_NAME_TEST_P()
397 name_utils::OpName(ShuffleAndRepeatDatasetOp::kDatasetType)}}; in DATASET_NODE_NAME_TEST_P()
453 ShuffleDatasetOp::kDatasetType, dataset_params.iterator_prefix()))); in TEST_F()
|
| D | fixed_length_record_dataset_op.cc | 32 FixedLengthRecordDatasetOp::kDatasetType; member in tensorflow::data::FixedLengthRecordDatasetOp
73 this, name_utils::IteratorPrefix(kDatasetType, prefix, params)}); in MakeIteratorInternal()
76 this, name_utils::IteratorPrefix(kDatasetType, prefix, params)}); in MakeIteratorInternal()
94 return name_utils::DatasetDebugString(kDatasetType, params); in DebugString()
147 metrics::GetTFDataBytesReadCounter(kDatasetType); in GetNextInternal()
263 metrics::GetTFDataBytesReadCounter(kDatasetType); in GetNextInternal()
479 metrics::RecordTFDataFilename(kDatasetType, filenames[i]); in MakeDataset()
|
| D | flat_map_dataset_op_test.cc | 68 return FlatMapDatasetOp::kDatasetType; in dataset_type()
162 name_utils::OpName(FlatMapDatasetOp::kDatasetType))); in TEST_F()
199 FlatMapDatasetOp::kDatasetType, dataset_params.iterator_prefix()))); in TEST_F()
|
| D | prefetch_dataset_op_test.cc | 68 return PrefetchDatasetOp::kDatasetType; in dataset_type()
242 name_utils::OpName(PrefetchDatasetOp::kDatasetType))); in TEST_F()
289 PrefetchDatasetOp::kDatasetType, dataset_params.iterator_prefix()))); in TEST_F()
|
| D | text_line_dataset_op_test.cc | 64 return TextLineDatasetOp::kDatasetType; in dataset_type()
178 name_utils::OpName(TextLineDatasetOp::kDatasetType))); in TEST_F()
216 TextLineDatasetOp::kDatasetType, dataset_params.iterator_prefix()))); in TEST_F()
|
| D | sparse_tensor_slice_dataset_op_test.cc | 28 constexpr char kDatasetType[] = "SparseTensorSlice"; variable
62 string dataset_type() const override { return kDatasetType; } in dataset_type()
195 TF_ASSERT_OK(CheckDatasetTypeString(name_utils::OpName(kDatasetType))); in TEST_F()
304 kDatasetType, dataset_params.iterator_prefix()))); in ITERATOR_OUTPUT_SHAPES_TEST_P()
|
| D | options_dataset_op_test.cc | 95 name_utils::OpName(OptionsDatasetOp::kDatasetType))); in TEST_F()
132 RangeDatasetOp::kDatasetType, dataset_params.iterator_prefix()))); in TEST_F()
|
| D | shard_dataset_op_test.cc | 61 string dataset_type() const override { return ShardDatasetOp::kDatasetType; } in dataset_type()
236 CheckDatasetTypeString(name_utils::OpName(ShardDatasetOp::kDatasetType))); in TEST_F()
287 ShardDatasetOp::kDatasetType, dataset_params.iterator_prefix()))); in TEST_F()
|
| D | generator_dataset_op.cc | 34 /* static */ constexpr const char* const GeneratorDatasetOp::kDatasetType; member in tensorflow::data::GeneratorDatasetOp
65 this, name_utils::IteratorPrefix(kDatasetType, prefix)}); in MakeIteratorInternal()
75 return name_utils::DatasetDebugString(kDatasetType); in DebugString()
|
| D | tf_record_dataset_op_test.cc | 58 return TFRecordDatasetOp::kDatasetType; in dataset_type()
206 name_utils::OpName(TFRecordDatasetOp::kDatasetType))); in TEST_F()
243 TFRecordDatasetOp::kDatasetType, dataset_params.iterator_prefix()))); in TEST_F()
|
| /external/tensorflow/tensorflow/core/data/ |
| D | root_dataset.cc | 36 constexpr char kDatasetType[] = "Root"; variable
291 : DatasetBase(DatasetContext({name_utils::OpName(kDatasetType), in RootDataset()
292 name_utils::OpName(kDatasetType)})), in RootDataset()
300 : DatasetBase(DatasetContext({name_utils::OpName(kDatasetType), in RootDataset()
301 name_utils::OpName(kDatasetType)})), in RootDataset()
313 Iterator::Params{this, name_utils::IteratorPrefix(kDatasetType, prefix)}); in MakeIteratorInternal()
325 return name_utils::DatasetDebugString(kDatasetType); in DebugString()
|