Searched refs:GetMklOpName (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/graph/ |
| D | mkl_graph_util.h | 165 inline string GetMklOpName(const string& name) { in GetMklOpName() function
320 bool result = (0 == op_name.compare(GetMklOpName("Add")) || in IsMklElementWiseOp()
321 0 == op_name.compare(GetMklOpName("AddV2")) || in IsMklElementWiseOp()
322 0 == op_name.compare(GetMklOpName("Sub")) || in IsMklElementWiseOp()
323 0 == op_name.compare(GetMklOpName("Mul")) || in IsMklElementWiseOp()
324 0 == op_name.compare(GetMklOpName("Maximum")) || in IsMklElementWiseOp()
325 0 == op_name.compare(GetMklOpName("SquaredDifference"))); in IsMklElementWiseOp()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | mkl_layout_pass.cc | 380 rinfo_.push_back({csinfo_.addn, mkl_op_registry::GetMklOpName(csinfo_.addn), in MklLayoutRewritePass()
382 rinfo_.push_back({csinfo_.add, mkl_op_registry::GetMklOpName(csinfo_.add), in MklLayoutRewritePass()
386 {csinfo_.add_v2, mkl_op_registry::GetMklOpName(csinfo_.add_v2), in MklLayoutRewritePass()
389 mkl_op_registry::GetMklOpName(csinfo_.avg_pool), in MklLayoutRewritePass()
392 mkl_op_registry::GetMklOpName(csinfo_.avg_pool_grad), in MklLayoutRewritePass()
395 mkl_op_registry::GetMklOpName(csinfo_.avg_pool3d), in MklLayoutRewritePass()
398 mkl_op_registry::GetMklOpName(csinfo_.avg_pool3d_grad), in MklLayoutRewritePass()
401 mkl_op_registry::GetMklOpName(csinfo_.batch_matmul), in MklLayoutRewritePass()
404 mkl_op_registry::GetMklOpName(csinfo_.batch_matmul_v2), in MklLayoutRewritePass()
407 mkl_op_registry::GetMklOpName(csinfo_.concat), in MklLayoutRewritePass()
[all …]
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | mkl_eager_op_rewrite.cc | 230 mkl_op_registry::GetMklOpName(op_name), dt)); in SlowCheckIfKernelRegistered()
|