Searched refs:transpose_to_nhwc (Results 1 – 1 of 1) sorted by relevance
737 std::vector<int> transpose_to_nhwc = {NCHW::dim::N, NCHW::dim::H, in MklLayoutRewritePass() local742 CheckForTranspose, std::placeholders::_1, transpose_to_nhwc); in MklLayoutRewritePass()3777 Node* transpose_to_nhwc = nodes[0]; in FuseTransposeMklOpTranspose() local3781 const int transpose_nhwc_num_inputs = transpose_to_nhwc->num_inputs(); in FuseTransposeMklOpTranspose()3785 FillInputs(transpose_to_nhwc, &transpose_nhwc_control_edges, in FuseTransposeMklOpTranspose()3806 if (mklop_in[i].first == transpose_to_nhwc) { in FuseTransposeMklOpTranspose()3844 (*g)->RemoveNode(transpose_to_nhwc); in FuseTransposeMklOpTranspose()