Searched refs:rinfo_ (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | mkl_layout_pass.cc | 386 rinfo_.push_back({csinfo_.addn, mkl_op_registry::GetMklOpName(csinfo_.addn), in MklLayoutRewritePass()
388 rinfo_.push_back({csinfo_.add, mkl_op_registry::GetMklOpName(csinfo_.add), in MklLayoutRewritePass()
391 rinfo_.push_back( in MklLayoutRewritePass()
394 rinfo_.push_back({csinfo_.avg_pool, in MklLayoutRewritePass()
397 rinfo_.push_back({csinfo_.avg_pool_grad, in MklLayoutRewritePass()
400 rinfo_.push_back({csinfo_.avg_pool3d, in MklLayoutRewritePass()
403 rinfo_.push_back({csinfo_.avg_pool3d_grad, in MklLayoutRewritePass()
406 rinfo_.push_back({csinfo_.batch_matmul, in MklLayoutRewritePass()
409 rinfo_.push_back({csinfo_.einsum, in MklLayoutRewritePass()
412 rinfo_.push_back({csinfo_.batch_matmul_v2, in MklLayoutRewritePass()
[all …]
|