Home
last modified time | relevance | path

Searched refs:IsMklLayoutDependentOp (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/graph/
Dmkl_graph_util.h200 static inline bool IsMklLayoutDependentOp(const string& op_name, DataType T) { in IsMklLayoutDependentOp() function
235 static inline bool IsMklLayoutDependentOp(const string& op_name, in IsMklLayoutDependentOp() function
301 return IsMklLayoutDependentOp(op_name, T) || IsMklNameChangeOp(op_name, T); in IsMklOp()
/external/tensorflow/tensorflow/core/common_runtime/
Dmkl_tfconversion_pass.cc97 return mkl_op_registry::IsMklLayoutDependentOp(op_name, T); in IsMklSupportedOp()
Dmkl_layout_pass.cc1463 return mkl_op_registry::IsMklLayoutDependentOp( in CheckForMklOp()
1781 !mkl_op_registry::IsMklLayoutDependentOp(csinfo_.mkl_fused_conv2d, T)) { in FusedConv2DRewrite()
1809 !mkl_op_registry::IsMklLayoutDependentOp( in FusedDepthwiseConv2DRewrite()
2194 mkl_op_registry::IsMklLayoutDependentOp(n->type_string(), T)) { in GetNodeProducingMklTensor()
3687 mkl_op_registry::IsMklLayoutDependentOp( in CheckForQuantizedNodeRewrite()
3692 mkl_op_registry::IsMklLayoutDependentOp( in CheckForQuantizedNodeRewrite()
3947 !mkl_op_registry::IsMklLayoutDependentOp(n->type_string(), T)) { in FixMklMetaDataEdges()
3972 mkl_op_registry::IsMklLayoutDependentOp(e_src->type_string(), T)) { in FixMklMetaDataEdges()