Searched refs:batch_matmul_v2 (Results 1 – 1 of 1) sorted by relevance
249 csinfo_.batch_matmul_v2 = "BatchMatMulV2"; in MklLayoutRewritePass()403 rinfo_.push_back({csinfo_.batch_matmul_v2, in MklLayoutRewritePass()404 mkl_op_registry::GetMklOpName(csinfo_.batch_matmul_v2), in MklLayoutRewritePass()910 string batch_matmul_v2; member