Searched refs:MatchesAnyVersion (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | dataset_utils_test.cc | 38 TEST(DatasetUtilsTest, MatchesAnyVersion) { in TEST() argument
39 EXPECT_TRUE(MatchesAnyVersion("BatchDataset", "BatchDataset")); in TEST()
40 EXPECT_TRUE(MatchesAnyVersion("BatchDataset", "BatchDatasetV2")); in TEST()
41 EXPECT_TRUE(MatchesAnyVersion("BatchDataset", "BatchDatasetV3")); in TEST()
42 EXPECT_FALSE(MatchesAnyVersion("BatchDataset", "BatchDatasetXV3")); in TEST()
43 EXPECT_FALSE(MatchesAnyVersion("BatchDataset", "BatchV2Dataset")); in TEST()
44 EXPECT_FALSE(MatchesAnyVersion("BatchDataset", "PaddedBatchDataset")); in TEST()
|
| D | dataset_utils.h | 283 bool MatchesAnyVersion(StringPiece op_prefix, StringPiece op_to_match);
|
| D | hash_utils.cc | 62 if (MatchesAnyVersion(type, node.op())) { in IsNodeOfType()
|
| D | dataset_utils.cc | 523 bool MatchesAnyVersion(StringPiece op_prefix, StringPiece op_to_match) { in MatchesAnyVersion() function
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | compute_batch_size_op.cc | 68 if (MatchesAnyVersion(dataset_op, node.op())) return true; in IsDatasetNodeOfType()
|
| D | assert_next_dataset_op.cc | 105 if (!MatchesAnyVersion(dataset()->transformations_[i], in Initialize()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| D | auto_shard.cc | 131 if (tensorflow::data::MatchesAnyVersion(/*op_prefix=*/dataset_op_name, in IsDatasetNodeOfType()
|