Searched refs:transpose_to_nhwc (Results 1 – 1 of 1) sorted by relevance
567 std::vector<int> transpose_to_nhwc = {NCHW::dim::N, NCHW::dim::H, in MklLayoutRewritePass() local572 std::bind(CheckForTranspose, std::placeholders::_1, transpose_to_nhwc); in MklLayoutRewritePass()3346 Node* transpose_to_nhwc = nodes[0]; in FuseTransposeMklOpTranspose() local3350 const int transpose_nhwc_num_inputs = transpose_to_nhwc->num_inputs(); in FuseTransposeMklOpTranspose()3354 FillInputs(transpose_to_nhwc, &transpose_nhwc_control_edges, in FuseTransposeMklOpTranspose()3375 if (mklop_in[i].first == transpose_to_nhwc) { in FuseTransposeMklOpTranspose()3412 (*g)->RemoveNode(transpose_to_nhwc); in FuseTransposeMklOpTranspose()