| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | name_utils_test.cc | 34 EXPECT_EQ(name_utils::DatasetDebugString(ConcatenateDatasetOp::kDatasetType), in TEST()
38 EXPECT_EQ(name_utils::DatasetDebugString(RangeDatasetOp::kDatasetType, in TEST()
45 EXPECT_EQ(name_utils::DatasetDebugString(ShuffleDatasetOp::kDatasetType, in TEST()
52 name_utils::DatasetDebugString(ParallelInterleaveDatasetOp::kDatasetType, in TEST()
58 EXPECT_EQ(name_utils::OpName(RangeDatasetOp::kDatasetType), "RangeDataset"); in TEST()
59 EXPECT_EQ(name_utils::OpName(ConcatenateDatasetOp::kDatasetType, in TEST()
65 name_utils::OpName(ParallelInterleaveDatasetOp::kDatasetType, params), in TEST()
|
| 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()
116 CheckDatasetTypeString(name_utils::OpName(ZipDatasetOp::kDatasetType))); in TEST_F()
181 ZipDatasetOp::kDatasetType, dataset_params.iterator_prefix()))); in ITERATOR_OUTPUT_SHAPES_TEST_P()
|
| D | text_line_dataset_op.cc | 30 /* static */ constexpr const char* const TextLineDatasetOp::kDatasetType; member in tensorflow::data::TextLineDatasetOp
55 name_utils::IteratorPrefix(TextLineDatasetOp::kDatasetType, prefix)}); in MakeIteratorInternal()
70 return name_utils::DatasetDebugString(kDatasetType); in DebugString()
115 name_utils::OpName(TextLineDatasetOp::kDatasetType)); in GetNextInternal()
278 metrics::RecordTFDataFilename(kDatasetType, filenames[i]); in MakeDataset()
|
| D | tf_record_dataset_op.cc | 34 /* static */ constexpr const char* const TFRecordDatasetOp::kDatasetType; member in tensorflow::data::TFRecordDatasetOp
70 this, name_utils::IteratorPrefix(kDatasetType, prefix)}); in MakeIteratorInternal()
85 return name_utils::DatasetDebugString(kDatasetType); in DebugString()
129 metrics::GetTFDataBytesReadCounter(kDatasetType); in GetNextInternal()
295 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 | 82 return ShuffleAndRepeatDatasetOp::kDatasetType; in dataset_type()
84 return ShuffleDatasetOp::kDatasetType; in dataset_type()
389 ShuffleDatasetOp::kDatasetType)}, in DATASET_NODE_NAME_TEST_P()
392 name_utils::OpName(ShuffleAndRepeatDatasetOp::kDatasetType)}}; in DATASET_NODE_NAME_TEST_P()
448 ShuffleDatasetOp::kDatasetType, dataset_params.iterator_prefix()))); in TEST_F()
|
| D | fixed_length_record_dataset_op.cc | 31 FixedLengthRecordDatasetOp::kDatasetType; member in tensorflow::data::FixedLengthRecordDatasetOp
72 this, name_utils::IteratorPrefix(kDatasetType, prefix, params)}); in MakeIteratorInternal()
75 this, name_utils::IteratorPrefix(kDatasetType, prefix, params)}); in MakeIteratorInternal()
93 return name_utils::DatasetDebugString(kDatasetType, params); in DebugString()
146 metrics::GetTFDataBytesReadCounter(kDatasetType); in GetNextInternal()
260 metrics::GetTFDataBytesReadCounter(kDatasetType); in GetNextInternal()
474 metrics::RecordTFDataFilename(kDatasetType, filenames[i]); in MakeDataset()
|
| D | flat_map_dataset_op_test.cc | 67 return FlatMapDatasetOp::kDatasetType; in dataset_type()
143 name_utils::OpName(FlatMapDatasetOp::kDatasetType))); in TEST_F()
180 FlatMapDatasetOp::kDatasetType, dataset_params.iterator_prefix()))); in TEST_F()
|
| D | prefetch_dataset_op_test.cc | 69 return PrefetchDatasetOp::kDatasetType; in dataset_type()
243 name_utils::OpName(PrefetchDatasetOp::kDatasetType))); in TEST_F()
290 PrefetchDatasetOp::kDatasetType, dataset_params.iterator_prefix()))); in TEST_F()
|
| D | text_line_dataset_op_test.cc | 63 return TextLineDatasetOp::kDatasetType; in dataset_type()
177 name_utils::OpName(TextLineDatasetOp::kDatasetType))); in TEST_F()
215 TextLineDatasetOp::kDatasetType, dataset_params.iterator_prefix()))); in TEST_F()
|
| D | sparse_tensor_slice_dataset_op_test.cc | 24 constexpr char kDatasetType[] = "SparseTensorSlice"; variable
58 string dataset_type() const override { return kDatasetType; } in dataset_type()
191 TF_ASSERT_OK(CheckDatasetTypeString(name_utils::OpName(kDatasetType))); in TEST_F()
300 kDatasetType, dataset_params.iterator_prefix()))); in ITERATOR_OUTPUT_SHAPES_TEST_P()
|
| D | filter_dataset_op_test.cc | 72 string dataset_type() const override { return FilterDatasetOp::kDatasetType; } in dataset_type()
193 name_utils::OpName(FilterDatasetOp::kDatasetType))); in TEST_F()
244 FilterDatasetOp::kDatasetType, dataset_params.iterator_prefix()))); in ITERATOR_OUTPUT_SHAPES_TEST_P()
|
| D | tf_record_dataset_op_test.cc | 57 return TFRecordDatasetOp::kDatasetType; in dataset_type()
205 name_utils::OpName(TFRecordDatasetOp::kDatasetType))); in TEST_F()
242 TFRecordDatasetOp::kDatasetType, dataset_params.iterator_prefix()))); in TEST_F()
|
| D | tensor_slice_dataset_op.cc | 31 /* static */ constexpr const char* const TensorSliceDatasetOp::kDatasetType; member in tensorflow::data::TensorSliceDatasetOp
58 this, name_utils::IteratorPrefix(kDatasetType, prefix)}); in MakeIteratorInternal()
75 return name_utils::DatasetDebugString(kDatasetType); in DebugString()
|
| 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 | zip_dataset_op.cc | 27 /* static */ constexpr const char* const ZipDatasetOp::kDatasetType; member in tensorflow::data::ZipDatasetOp
60 this, name_utils::IteratorPrefix(kDatasetType, prefix)}); in MakeIteratorInternal()
72 return name_utils::DatasetDebugString(kDatasetType); in DebugString()
|
| D | map_dataset_op_test.cc | 118 CheckDatasetTypeString(name_utils::OpName(MapDatasetOp::kDatasetType))); in TEST_F()
161 MapDatasetOp::kDatasetType, dataset_params.iterator_prefix()))); in TEST_F()
|
| D | range_dataset_op_test.cc | 69 CheckDatasetTypeString(name_utils::OpName(RangeDatasetOp::kDatasetType))); in TEST_F()
120 RangeDatasetOp::kDatasetType, range_dataset_params.iterator_prefix()))); in TEST_F()
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| 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 | sampling_dataset_op_test.cc | 62 return SamplingDatasetOp::kDatasetType; in dataset_type()
132 SamplingDatasetOp::kDatasetType)}}; in DATASET_NODE_NAME_TEST_P()
190 SamplingDatasetOp::kDatasetType, in ITERATOR_OUTPUT_SHAPES_TEST_P()
|
| 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 | 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 | 72 return RandomDatasetOp::kDatasetType; in dataset_type()
160 RandomDatasetOp::kDatasetType)}}; in DatasetTypeStringTestCases()
214 RandomDatasetOp::kDatasetType, kIteratorPrefix)}}; in IteratorOutputPrefixTestCases()
|
| D | unique_dataset_op_test.cc | 52 string dataset_type() const override { return UniqueDatasetOp::kDatasetType; } in dataset_type()
191 name_utils::OpName(UniqueDatasetOp::kDatasetType))); in TEST_F()
244 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()
|