mkl_layout_pass.cc | 277 csinfo_.max_pool = "MaxPool"; 351 rinfo_.push_back({csinfo_.max_pool, 352 mkl_op_registry::GetMklOpName(csinfo_.max_pool), 378 wsinfo_.push_back({csinfo_.max_pool, csinfo_.max_pool_grad, 0, 1, 1, 3}); 474 string max_pool; member in struct:tensorflow::MklLayoutRewritePass::__anon39629 [all...] |