Searched refs:PredictMatMul (Results 1 – 2 of 2) sorted by relevance
134 Costs PredictMatMul(const OpContext& op_context) const;
245 {kMatMul, wrap(&OpLevelCostEstimator::PredictMatMul)}, in OpLevelCostEstimator()246 {kSparseMatMul, wrap(&OpLevelCostEstimator::PredictMatMul)}, in OpLevelCostEstimator()250 {kQuantizedMatMul, wrap(&OpLevelCostEstimator::PredictMatMul)}, in OpLevelCostEstimator()251 {kQuantizedMatMulV2, wrap(&OpLevelCostEstimator::PredictMatMul)}, in OpLevelCostEstimator()1224 Costs OpLevelCostEstimator::PredictMatMul(const OpContext& op_context) const { in PredictMatMul() function in tensorflow::grappler::OpLevelCostEstimator