Home
last modified time | relevance | path

Searched refs:transpose_to_nchw (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/core/graph/
Dmkl_layout_pass.cc569 std::vector<int> transpose_to_nchw = {NHWC::dim::N, NHWC::dim::C, in MklLayoutRewritePass() local
576 std::bind(CheckForTranspose, std::placeholders::_1, transpose_to_nchw); in MklLayoutRewritePass()
3348 Node* transpose_to_nchw = nodes[2]; in FuseTransposeMklOpTranspose() local
3362 const int transpose_nchw_num_inputs = transpose_to_nchw->num_inputs(); in FuseTransposeMklOpTranspose()
3366 FillInputs(transpose_to_nchw, &transpose_nchw_control_edges, in FuseTransposeMklOpTranspose()
3396 for (const Edge* e : transpose_to_nchw->out_edges()) { in FuseTransposeMklOpTranspose()
3414 (*g)->RemoveNode(transpose_to_nchw); in FuseTransposeMklOpTranspose()