Searched refs:IsMklOp (Results 1 – 2 of 2) sorted by relevance
300 static inline bool IsMklOp(const string& op_name, DataType T) { in IsMklOp() function304 static inline bool IsMklOp(const Node* n) { in IsMklOp() function306 return GetNodeAttr(n->def(), "T", &T).ok() && IsMklOp(n->type_string(), T); in IsMklOp()317 if (!IsMklOp(op_name, T)) { in IsMklElementWiseOp()
1498 mkl_op_registry::IsMklOp( in RewriteIfAtleastOneMklInput()1502 if (mkl_op_registry::IsMklOp(e->src())) { in RewriteIfAtleastOneMklInput()2494 mkl_op_registry::IsMklOp( in AddWorkSpaceEdgeIfNeeded()2521 mkl_op_registry::IsMklOp( in AddWorkSpaceEdgeIfNeeded()3752 !mkl_op_registry::IsMklOp(mkl_op_registry::GetMklOpName(n->type_string()), in CheckForNodeRewrite()