Searched refs:transpose_to_nchw (Results 1 – 1 of 1) sorted by relevance
739 std::vector<int> transpose_to_nchw = {NHWC::dim::N, NHWC::dim::C, in MklLayoutRewritePass() local746 CheckForTranspose, std::placeholders::_1, transpose_to_nchw); in MklLayoutRewritePass()3779 Node* transpose_to_nchw = nodes[2]; in FuseTransposeMklOpTranspose() local3793 const int transpose_nchw_num_inputs = transpose_to_nchw->num_inputs(); in FuseTransposeMklOpTranspose()3797 FillInputs(transpose_to_nchw, &transpose_nchw_control_edges, in FuseTransposeMklOpTranspose()3828 for (const Edge* e : transpose_to_nchw->out_edges()) { in FuseTransposeMklOpTranspose()3846 (*g)->RemoveNode(transpose_to_nchw); in FuseTransposeMklOpTranspose()