mkl_layout_pass.cc | 338 rinfo_.push_back({csinfo_.addn, mkl_op_registry::GetMklOpName(csinfo_.addn), 340 rinfo_.push_back({csinfo_.add, mkl_op_registry::GetMklOpName(csinfo_.add), 342 rinfo_.push_back({csinfo_.avg_pool, 345 rinfo_.push_back({csinfo_.avg_pool_grad, 348 rinfo_.push_back({csinfo_.avg_pool3d, 351 rinfo_.push_back({csinfo_.avg_pool3d_grad, 354 rinfo_.push_back({csinfo_.concat, 357 rinfo_.push_back({csinfo_.concatv2, 360 rinfo_.push_back({csinfo_.conv2d, 363 rinfo_.push_back({csinfo_.conv2d_with_bias, csinfo_.mkl_conv2d_with_bias 757 std::vector<RewriteInfo> rinfo_; member in class:tensorflow::MklLayoutRewritePass [all...] |