Searched refs:transpose_to_nchw (Results 1 – 1 of 1) sorted by relevance
569 std::vector<int> transpose_to_nchw = {NHWC::dim::N, NHWC::dim::C, in MklLayoutRewritePass() local576 std::bind(CheckForTranspose, std::placeholders::_1, transpose_to_nchw); in MklLayoutRewritePass()3348 Node* transpose_to_nchw = nodes[2]; in FuseTransposeMklOpTranspose() local3362 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()