Searched refs:kBatchMatMul (Results 1 – 5 of 5) sorted by relevance
68 {{OperatorType::kBatchMatMul, 1}, kPendingReleaseOpVersion}, in GetMinimumRuntimeVersionForModel()
138 if (batch_op_it->get()->type != OperatorType::kBatchMatMul) { in Run()
2364 case OperatorType::kBatchMatMul: in Run()
44 kBatchMatMul, enumerator1032 BatchMatMulOperator() : Operator(OperatorType::kBatchMatMul) {}
68 constexpr char kBatchMatMul[] = "BatchMatMul"; variable398 device_cost_impl_.emplace(kBatchMatMul, in OpLevelCostEstimator()1242 if (op_info.op() != kBatchMatMul && op_info.op() != kBatchMatMulV2) { in CountBatchMatMulOperations()