Searched refs:IsDatasetOp (Results 1 – 9 of 9) sorted by relevance
137 TEST(DatasetTest, IsDatasetOp) { in TEST() argument140 EXPECT_FALSE(DatasetOpKernel::IsDatasetOp(op_def)); in TEST()144 EXPECT_FALSE(DatasetOpKernel::IsDatasetOp(op_def)); in TEST()149 EXPECT_FALSE(DatasetOpKernel::IsDatasetOp(op_def)); in TEST()156 EXPECT_TRUE(DatasetOpKernel::IsDatasetOp(op_def)); in TEST()
1348 static bool IsDatasetOp(const OpDef& op_def);
1013 bool DatasetOpKernel::IsDatasetOp(const OpDef& op_def) { in IsDatasetOp() function in tensorflow::data::DatasetOpKernel
72 inline bool IsDatasetOp(absl::string_view tf_op_type) { in IsDatasetOp() function75 inline bool IsDatasetOp(const TfOp& tf_op) { in IsDatasetOp() function
287 return IsInfeedEnqueueOp(category) || IsDatasetOp(category) || in IsInputOp()299 DCHECK(IsDatasetOp(category)); in CategorizeInputOp()
766 data::DatasetOpKernel::IsDatasetOp(src->op_def()) && in ColocateResourceAndRefEdges()767 data::DatasetOpKernel::IsDatasetOp(dst->op_def())) { in ColocateResourceAndRefEdges()
625 if (data::DatasetOpKernel::IsDatasetOp(*op_def)) { in NodeMayIntroduceNondeterminismWhenAsync()
364 if (DatasetOpKernel::IsDatasetOp(reg->op_def) && attr_key == "metadata") in HashNodeNonInput()
104 if (data::DatasetOpKernel::IsDatasetOp(op_def)) { in KernelCacheEnabled()