Searched refs:GetMklOpName (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/graph/ |
| D | mkl_graph_util.h | 85 inline string GetMklOpName(const string& name) { in GetMklOpName() function
133 bool result = (0 == op_name.compare(GetMklOpName("Add")) || in IsMklElementWiseOp()
134 0 == op_name.compare(GetMklOpName("Sub")) || in IsMklElementWiseOp()
135 0 == op_name.compare(GetMklOpName("Mul")) || in IsMklElementWiseOp()
136 0 == op_name.compare(GetMklOpName("Maximum")) || in IsMklElementWiseOp()
137 0 == op_name.compare(GetMklOpName("SquaredDifference"))); in IsMklElementWiseOp()
|
| D | mkl_layout_pass.cc | 338 rinfo_.push_back({csinfo_.addn, mkl_op_registry::GetMklOpName(csinfo_.addn), in MklLayoutRewritePass()
340 rinfo_.push_back({csinfo_.add, mkl_op_registry::GetMklOpName(csinfo_.add), in MklLayoutRewritePass()
343 mkl_op_registry::GetMklOpName(csinfo_.avg_pool), in MklLayoutRewritePass()
346 mkl_op_registry::GetMklOpName(csinfo_.avg_pool_grad), in MklLayoutRewritePass()
349 mkl_op_registry::GetMklOpName(csinfo_.avg_pool3d), in MklLayoutRewritePass()
352 mkl_op_registry::GetMklOpName(csinfo_.avg_pool3d_grad), in MklLayoutRewritePass()
355 mkl_op_registry::GetMklOpName(csinfo_.concat), in MklLayoutRewritePass()
358 mkl_op_registry::GetMklOpName(csinfo_.concatv2), in MklLayoutRewritePass()
361 mkl_op_registry::GetMklOpName(csinfo_.conv2d), in MklLayoutRewritePass()
366 mkl_op_registry::GetMklOpName(csinfo_.conv2d_grad_filter), in MklLayoutRewritePass()
[all …]
|