Searched refs:CheckForMklOp (Results 1 – 1 of 1) sorted by relevance
744 std::bind(CheckForMklOp, std::placeholders::_1, csinfo_.conv2d); in MklLayoutRewritePass()768 std::bind(CheckForMklOp, std::placeholders::_1, csinfo_.conv3d); in MklLayoutRewritePass()783 std::bind(CheckForMklOp, std::placeholders::_1, csinfo_.max_pool3d); in MklLayoutRewritePass()1449 static bool CheckForMklOp(const Node* node, string name = "") { in CheckForMklOp() function in tensorflow::MklLayoutRewritePass