Searched refs:kBatchMatMul (Results 1 – 4 of 4) sorted by relevance
218 if (batch_op_it->get()->type != OperatorType::kBatchMatMul) { in Run()
2281 case OperatorType::kBatchMatMul: in Run()
43 kBatchMatMul, enumerator982 BatchMatMulOperator() : Operator(OperatorType::kBatchMatMul) {}
52 constexpr char kBatchMatMul[] = "BatchMatMul"; variable249 {kBatchMatMul, wrap(&OpLevelCostEstimator::PredictBatchMatMul)}, in OpLevelCostEstimator()759 if (op_info.op() != kBatchMatMul) { in CountBatchMatMulOperations()