Searched refs:AlwaysRewrite (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | mkl_eager_op_rewrite.cc | 68 static bool AlwaysRewrite(EagerOperation* op) { return true; } in AlwaysRewrite() function in tensorflow::MklEagerOpRewrite
89 InsertMKLEagerOps({"AvgPool", AlwaysRewrite, CreateGenericMklOp}); in MklEagerOpRewrite()
90 InsertMKLEagerOps({"AvgPoolGrad", AlwaysRewrite, CreateGenericMklOp}); in MklEagerOpRewrite()
91 InsertMKLEagerOps({"AvgPool3D", AlwaysRewrite, CreateGenericMklOp}); in MklEagerOpRewrite()
92 InsertMKLEagerOps({"AvgPool3DGrad", AlwaysRewrite, CreateGenericMklOp}); in MklEagerOpRewrite()
93 InsertMKLEagerOps({"BatchMatMul", AlwaysRewrite, CreateGenericMklOp}); in MklEagerOpRewrite()
94 InsertMKLEagerOps({"BatchMatMulV2", AlwaysRewrite, CreateGenericMklOp}); in MklEagerOpRewrite()
95 InsertMKLEagerOps({"Conv2D", AlwaysRewrite, CreateGenericMklOp}); in MklEagerOpRewrite()
99 InsertMKLEagerOps({"Conv3D", AlwaysRewrite, CreateGenericMklOp}); in MklEagerOpRewrite()
105 {"DepthwiseConv2dNative", AlwaysRewrite, CreateGenericMklOp}); in MklEagerOpRewrite()
[all …]
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | mkl_layout_pass.cc | 381 CopyAttrsAll, AlwaysRewrite, GetRewriteCause()}); in MklLayoutRewritePass()
390 CopyAttrsAll, AlwaysRewrite, GetRewriteCause()}); in MklLayoutRewritePass()
393 CopyAttrsAll, AlwaysRewrite, GetRewriteCause()}); in MklLayoutRewritePass()
396 CopyAttrsAll, AlwaysRewrite, GetRewriteCause()}); in MklLayoutRewritePass()
399 CopyAttrsAll, AlwaysRewrite, GetRewriteCause()}); in MklLayoutRewritePass()
408 CopyAttrsAll, AlwaysRewrite, GetRewriteCause()}); in MklLayoutRewritePass()
415 CopyAttrsAll, AlwaysRewrite, kRewriteForOpNameChange}); in MklLayoutRewritePass()
418 CopyAttrsConvCheckConstFilter, AlwaysRewrite, GetRewriteCause()}); in MklLayoutRewritePass()
422 CopyAttrsConvCheckConstFilter, AlwaysRewrite, in MklLayoutRewritePass()
426 CopyAttrsConv, AlwaysRewrite, GetRewriteCause()}); in MklLayoutRewritePass()
[all …]
|