Home
last modified time | relevance | path

Searched refs:name_utils (Results 1 – 25 of 87) sorted by relevance

1234

/external/tensorflow/tensorflow/core/kernels/data/
Dname_utils_test.cc28 EXPECT_EQ(name_utils::ArgsToString({}), ""); in TEST()
29 EXPECT_EQ(name_utils::ArgsToString({"a"}), "(a)"); in TEST()
30 EXPECT_EQ(name_utils::ArgsToString({"1", "2", "3"}), "(1, 2, 3)"); in TEST()
34 EXPECT_EQ(name_utils::DatasetDebugString(ConcatenateDatasetOp::kDatasetType), in TEST()
36 name_utils::DatasetDebugStringParams range_params; in TEST()
38 EXPECT_EQ(name_utils::DatasetDebugString(RangeDatasetOp::kDatasetType, in TEST()
42 name_utils::DatasetDebugStringParams shuffle_params; in TEST()
45 EXPECT_EQ(name_utils::DatasetDebugString(ShuffleDatasetOp::kDatasetType, in TEST()
49 name_utils::DatasetDebugStringParams parallel_interleave_params; in TEST()
52 name_utils::DatasetDebugString(ParallelInterleaveDatasetOp::kDatasetType, in TEST()
[all …]
Dskip_dataset_op_test.cc34 name_utils::IteratorPrefix(input_dataset_params.dataset_type(), in SkipDatasetParams()
144 CheckDatasetTypeString(name_utils::OpName(SkipDatasetOp::kDatasetType))); in TEST_F()
191 name_utils::IteratorPrefix("FiniteSkip", in IteratorPrefixTestCases()
194 /*expected_iterator_prefix=*/name_utils::IteratorPrefix( in IteratorPrefixTestCases()
197 /*expected_iterator_prefix=*/name_utils::IteratorPrefix( in IteratorPrefixTestCases()
200 /*expected_iterator_prefix=*/name_utils::IteratorPrefix( in IteratorPrefixTestCases()
203 /*expected_iterator_prefix=*/name_utils::IteratorPrefix( in IteratorPrefixTestCases()
Dtake_dataset_op_test.cc91 CheckDatasetTypeString(name_utils::OpName(TakeDatasetOp::kDatasetType))); in TEST_F()
153 /*expected_iterator_prefix=*/name_utils::IteratorPrefix( in ITERATOR_OUTPUT_SHAPES_TEST_P()
156 /*expected_iterator_prefix=*/name_utils::IteratorPrefix( in ITERATOR_OUTPUT_SHAPES_TEST_P()
159 /*expected_iterator_prefix=*/name_utils::IteratorPrefix( in ITERATOR_OUTPUT_SHAPES_TEST_P()
162 /*expected_iterator_prefix=*/name_utils::IteratorPrefix( in ITERATOR_OUTPUT_SHAPES_TEST_P()
DBUILD34 ":name_utils",
69 ":name_utils",
138 ":name_utils",
195 ":name_utils",
253 ":name_utils",
291 ":name_utils",
326 ":name_utils",
366 ":name_utils",
413 ":name_utils",
478 ":name_utils",
[all …]
Dbatch_dataset_op_test.cc161 name_utils::OpNameParams params; in TEST_F()
164 name_utils::OpName(BatchDatasetOp::kDatasetType, params))); in TEST_F()
238 name_utils::IteratorPrefixParams params; in ITERATOR_OUTPUT_SHAPES_TEST_P()
240 TF_ASSERT_OK(CheckIteratorPrefix(name_utils::IteratorPrefix( in ITERATOR_OUTPUT_SHAPES_TEST_P()
Dparallel_map_dataset_op_test.cc47 name_utils::IteratorPrefixParams params; in ParallelMapDatasetParams()
49 iterator_prefix_ = name_utils::IteratorPrefix( in ParallelMapDatasetParams()
311 name_utils::OpNameParams params; in TEST_F()
314 name_utils::OpName(ParallelMapDatasetOp::kDatasetType, params))); in TEST_F()
363 name_utils::IteratorPrefixParams params; in TEST_F()
366 name_utils::IteratorPrefix(ParallelMapDatasetOp::kDatasetType, in TEST_F()
Dparallel_batch_dataset_op_test.cc42 name_utils::IteratorPrefix(input_dataset_params.dataset_type(), in ParallelBatchDatasetParams()
250 name_utils::OpNameParams params; in TEST_F()
253 name_utils::OpName(ParallelBatchDatasetOp::kDatasetType, params))); in TEST_F()
348 name_utils::IteratorPrefixParams params; in ITERATOR_OUTPUT_SHAPES_TEST_P()
350 TF_ASSERT_OK(CheckIteratorPrefix(name_utils::IteratorPrefix( in ITERATOR_OUTPUT_SHAPES_TEST_P()
Drepeat_dataset_op_test.cc38 name_utils::IteratorPrefix(input_dataset_params.dataset_type(), in RepeatDatasetParams()
189 name_utils::OpName(RepeatDatasetOp::kDatasetType))); in TEST_F()
264 name_utils::IteratorPrefix( in ITERATOR_OUTPUT_SHAPES_TEST_P()
267 /*expected_iterator_prefix=*/name_utils::IteratorPrefix( in ITERATOR_OUTPUT_SHAPES_TEST_P()
270 /*expected_iterator_prefix=*/name_utils::IteratorPrefix( in ITERATOR_OUTPUT_SHAPES_TEST_P()
Dfixed_length_record_dataset_op_test.cc200 name_utils::OpNameParams params; in TEST_F()
203 name_utils::OpName(FixedLengthRecordDatasetOp::kDatasetType, params))); in TEST_F()
239 name_utils::IteratorPrefixParams iterator_prefix_params; in TEST_F()
241 TF_ASSERT_OK(CheckIteratorPrefix(name_utils::IteratorPrefix( in TEST_F()
Dzip_dataset_op_test.cc40 name_utils::IteratorPrefix(input_dataset_params[0].dataset_type(), in ZipDatasetParams()
116 CheckDatasetTypeString(name_utils::OpName(ZipDatasetOp::kDatasetType))); in TEST_F()
180 TF_ASSERT_OK(CheckIteratorPrefix(name_utils::IteratorPrefix( in ITERATOR_OUTPUT_SHAPES_TEST_P()
Dbatch_dataset_op.cc91 name_utils::IteratorPrefixParams params; in MakeIteratorInternal()
94 this, name_utils::IteratorPrefix(kDatasetType, prefix, params)}); in MakeIteratorInternal()
106 name_utils::DatasetDebugStringParams params; in DebugString()
109 return name_utils::DatasetDebugString(kDatasetType, params); in DebugString()
Dflat_map_dataset_op_test.cc41 name_utils::IteratorPrefix(input_dataset_params.dataset_type(), in FlatMapDatasetParams()
143 name_utils::OpName(FlatMapDatasetOp::kDatasetType))); in TEST_F()
179 TF_ASSERT_OK(CheckIteratorPrefix(name_utils::IteratorPrefix( in TEST_F()
Dcache_dataset_ops_test.cc38 name_utils::IteratorPrefix(input_dataset_params.dataset_type(), in CacheDatasetParams()
212 CheckDatasetTypeString(name_utils::OpName(CacheDatasetOp::kDatasetType))); in TEST_F()
276 name_utils::IteratorPrefixParams iterator_prefix_params; in ITERATOR_OUTPUT_SHAPES_TEST_P()
279 TF_ASSERT_OK(CheckIteratorPrefix(name_utils::IteratorPrefix( in ITERATOR_OUTPUT_SHAPES_TEST_P()
Drepeat_dataset_op.cc54 this, name_utils::IteratorPrefix(kForeverRepeat, prefix)}); in MakeIteratorInternal()
57 this, name_utils::IteratorPrefix(kEmptyRepeat, prefix)}); in MakeIteratorInternal()
60 this, name_utils::IteratorPrefix(kFiniteRepeat, prefix)}); in MakeIteratorInternal()
72 return name_utils::DatasetDebugString(RepeatDatasetOp::kDatasetType); in DebugString()
Dparallel_interleave_dataset_op_test.cc49 name_utils::IteratorPrefixParams params; in ParallelInterleaveDatasetParams()
51 iterator_prefix_ = name_utils::IteratorPrefix( in ParallelInterleaveDatasetParams()
616 name_utils::OpNameParams params; in TEST_F()
619 name_utils::OpName(ParallelInterleaveDatasetOp::kDatasetType, params))); in TEST_F()
677 name_utils::IteratorPrefixParams params; in TEST_F()
680 name_utils::IteratorPrefix(ParallelInterleaveDatasetOp::kDatasetType, in TEST_F()
Dprefetch_dataset_op_test.cc41 name_utils::IteratorPrefix(input_dataset_params.dataset_type(), in PrefetchDatasetParams()
243 name_utils::OpName(PrefetchDatasetOp::kDatasetType))); in TEST_F()
289 TF_ASSERT_OK(CheckIteratorPrefix(name_utils::IteratorPrefix( in TEST_F()
Dshard_dataset_op_test.cc36 name_utils::IteratorPrefix(input_dataset_params.dataset_type(), in ShardDatasetParams()
236 CheckDatasetTypeString(name_utils::OpName(ShardDatasetOp::kDatasetType))); in TEST_F()
286 TF_ASSERT_OK(CheckIteratorPrefix(name_utils::IteratorPrefix( in TEST_F()
Dfilter_dataset_op_test.cc41 name_utils::IteratorPrefix(input_dataset_params.dataset_type(), in FilterDatasetParams()
193 name_utils::OpName(FilterDatasetOp::kDatasetType))); in TEST_F()
243 TF_ASSERT_OK(CheckIteratorPrefix(name_utils::IteratorPrefix( in ITERATOR_OUTPUT_SHAPES_TEST_P()
/external/tensorflow/tensorflow/core/kernels/data/experimental/
DBUILD22 "//tensorflow/core/kernels/data:name_utils",
34 "//tensorflow/core/kernels/data:name_utils",
138 "//tensorflow/core/kernels/data:name_utils",
174 "//tensorflow/core/kernels/data:name_utils",
219 "//tensorflow/core/kernels/data:name_utils",
292 "//tensorflow/core/kernels/data:name_utils",
344 "//tensorflow/core/kernels/data:name_utils",
400 "//tensorflow/core/kernels/data:name_utils",
432 "//tensorflow/core/kernels/data:name_utils",
493 "//tensorflow/core/kernels/data:name_utils",
[all …]
Dparallel_interleave_dataset_op_test.cc49 name_utils::IteratorPrefixParams params; in ParallelInterleaveDatasetParams()
51 iterator_prefix_ = name_utils::IteratorPrefix( in ParallelInterleaveDatasetParams()
418 name_utils::OpNameParams params; in TEST_F()
421 name_utils::OpName(ParallelInterleaveDatasetOp::kDatasetType, params))); in TEST_F()
469 name_utils::IteratorPrefixParams params; in TEST_F()
472 name_utils::IteratorPrefix(ParallelInterleaveDatasetOp::kDatasetType, in TEST_F()
Dsampling_dataset_op_test.cc37 name_utils::IteratorPrefix(input_dataset_params.dataset_type(), in SamplingDatasetParams()
131 /*expected_dataset_type_string=*/name_utils::OpName( in DATASET_NODE_NAME_TEST_P()
189 /*expected_iterator_prefix=*/name_utils::IteratorPrefix( in ITERATOR_OUTPUT_SHAPES_TEST_P()
Dassert_next_dataset_op_test.cc38 name_utils::IteratorPrefix(input_dataset_params.dataset_type(), in AssertNextDatasetParams()
154 name_utils::OpName(AssertNextDatasetOp::kDatasetType))); in TEST_F()
190 TF_ASSERT_OK(CheckIteratorPrefix(name_utils::IteratorPrefix( in TEST_F()
Drebatch_dataset_op.cc75 name_utils::IteratorPrefixParams params; in MakeIteratorInternal()
77 this, name_utils::IteratorPrefix(kDatasetTypeV1, prefix, params)}); in MakeIteratorInternal()
89 name_utils::DatasetDebugStringParams params; in DebugString()
91 return name_utils::DatasetDebugString(kDatasetTypeV1, params); in DebugString()
332 name_utils::IteratorPrefixParams params; in MakeIteratorInternal()
334 this, name_utils::IteratorPrefix(kDatasetTypeV2, prefix, params)}); in MakeIteratorInternal()
346 return name_utils::DatasetDebugString(kDatasetTypeV2); in DebugString()
Dunique_dataset_op_test.cc34 name_utils::IteratorPrefix(input_dataset_params.dataset_type(), in UniqueDatasetParams()
191 name_utils::OpName(UniqueDatasetOp::kDatasetType))); in TEST_F()
243 TF_ASSERT_OK(CheckIteratorPrefix(name_utils::IteratorPrefix( in ITERATOR_OUTPUT_SHAPES_TEST_P()
/external/tensorflow/tensorflow/compiler/mlir/
DBUILD52 name = "name_utils",
53 srcs = ["utils/name_utils.cc"],
54 hdrs = ["utils/name_utils.h"],
66 ":name_utils",

1234