Searched refs:CheckForTranspose (Results 1 – 1 of 1) sorted by relevance
572 std::bind(CheckForTranspose, std::placeholders::_1, transpose_to_nhwc); in MklLayoutRewritePass()576 std::bind(CheckForTranspose, std::placeholders::_1, transpose_to_nchw); in MklLayoutRewritePass()1085 static bool CheckForTranspose(const Node* node, std::vector<int> perm) { in CheckForTranspose() function in tensorflow::MklLayoutRewritePass