Searched refs:wsinfo_ (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/graph/ |
| D | mkl_layout_pass.cc | 533 wsinfo_.push_back({csinfo_.lrn, csinfo_.lrn_grad, 0, 2, 1, 3}); in MklLayoutRewritePass()
534 wsinfo_.push_back({csinfo_.max_pool, csinfo_.max_pool_grad, 0, 1, 1, 3}); in MklLayoutRewritePass()
535 wsinfo_.push_back( in MklLayoutRewritePass()
760 std::vector<WorkSpaceInfo> wsinfo_; member in tensorflow::MklLayoutRewritePass
1936 for (auto ws : wsinfo_) { in AddWorkSpaceEdgeIfNeeded()
|