Searched refs:PredictBatchMatMul (Results 1 – 2 of 2) sorted by relevance
155 Status PredictBatchMatMul(const OpContext& op_context,
399 wrap(&OpLevelCostEstimator::PredictBatchMatMul)); in OpLevelCostEstimator()401 wrap(&OpLevelCostEstimator::PredictBatchMatMul)); in OpLevelCostEstimator()1924 Status OpLevelCostEstimator::PredictBatchMatMul(const OpContext& op_context, in PredictBatchMatMul() function in tensorflow::grappler::OpLevelCostEstimator