Searched refs:PredictBatchMatMul (Results 1 – 2 of 2) sorted by relevance
139 Costs PredictBatchMatMul(const OpContext& op_context) const;
249 {kBatchMatMul, wrap(&OpLevelCostEstimator::PredictBatchMatMul)}, in OpLevelCostEstimator()1308 Costs OpLevelCostEstimator::PredictBatchMatMul( in PredictBatchMatMul() function in tensorflow::grappler::OpLevelCostEstimator