Searched refs:IsMklOp (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/graph/ |
| D | mkl_graph_util.h | 94 static inline bool IsMklOp(const string& op_name, DataType T) { in IsMklOp() function
111 static inline bool IsMklOp(const string& op_name, DataType Tinput, in IsMklOp() function
130 if (!IsMklOp(op_name, T)) { in IsMklElementWiseOp()
|
| D | mkl_layout_pass.cc | 1175 return mkl_op_registry::IsMklOp( in CheckForMklOp()
1317 !mkl_op_registry::IsMklOp(csinfo_.mkl_fused_conv2d, T)) { in FusedConv2DRewrite()
1684 mkl_op_registry::IsMklOp(n->type_string(), T)) { in GetNodeProducingMklTensor()
1938 mkl_op_registry::IsMklOp( in AddWorkSpaceEdgeIfNeeded()
1965 mkl_op_registry::IsMklOp( in AddWorkSpaceEdgeIfNeeded()
3244 if (mkl_op_registry::IsMklOp(mkl_op_registry::GetMklOpName(n->type_string()), in CheckForQuantizedNodeRewrite()
3281 !mkl_op_registry::IsMklOp(mkl_op_registry::GetMklOpName(n->type_string()), in CheckForNodeRewrite()
3305 if (mkl_op_registry::IsMklOp(parent->src()->type_string(), T)) { in CheckForNodeRewrite()
3514 !mkl_op_registry::IsMklOp(n->type_string(), T)) { in FixMklMetaDataEdges()
3539 mkl_op_registry::IsMklOp(e_src->type_string(), T)) { in FixMklMetaDataEdges()
|
| D | mkl_tfconversion_pass.cc | 96 return mkl_op_registry::IsMklOp(op_name, T); in IsMklSupportedOp()
|