Searched refs:IsDatasetOp (Results 1 – 6 of 6) sorted by relevance
67 inline bool IsDatasetOp(absl::string_view tf_op_type) { in IsDatasetOp() function70 inline bool IsDatasetOp(const TfOp& tf_op) { in IsDatasetOp() function
281 return IsInfeedEnqueueOp(category) || IsDatasetOp(category) || in IsInputOp()293 DCHECK(IsDatasetOp(category)); in CategorizeInputOp()
744 data::DatasetOpKernel::IsDatasetOp(&src->op_def()) && in ColocateResourceAndRefEdges()745 data::DatasetOpKernel::IsDatasetOp(&dst->op_def())) { in ColocateResourceAndRefEdges()
665 bool DatasetOpKernel::IsDatasetOp(const OpDef* op_def) { in IsDatasetOp() function in tensorflow::data::DatasetOpKernel
1163 static bool IsDatasetOp(const OpDef* op_def);
86 if (data::DatasetOpKernel::IsDatasetOp(&op_def)) { in KernelCacheEnabled()