Searched refs:wsinfo_ (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | mkl_layout_pass.cc | 719 wsinfo_.push_back({csinfo_.lrn, csinfo_.lrn_grad, 0, 2, 1, 3}); in MklLayoutRewritePass()
720 wsinfo_.push_back({csinfo_.max_pool, csinfo_.max_pool_grad, 0, 1, 1, 3}); in MklLayoutRewritePass()
721 wsinfo_.push_back( in MklLayoutRewritePass()
973 std::vector<WorkSpaceInfo> wsinfo_; member in tensorflow::MklLayoutRewritePass
2456 for (auto ws : wsinfo_) { in AddWorkSpaceEdgeIfNeeded()
|