Searched refs:CheckForTranspose (Results 1 – 1 of 1) sorted by relevance
742 CheckForTranspose, std::placeholders::_1, transpose_to_nhwc); in MklLayoutRewritePass()746 CheckForTranspose, std::placeholders::_1, transpose_to_nchw); in MklLayoutRewritePass()766 CheckForTranspose, std::placeholders::_1, transpose_to_ndhwc); in MklLayoutRewritePass()770 CheckForTranspose, std::placeholders::_1, transpose_to_ncdhw); in MklLayoutRewritePass()1378 static bool CheckForTranspose(const Node* node, std::vector<int> perm) { in CheckForTranspose() function in tensorflow::MklLayoutRewritePass