Searched refs:PredictNaryOp (Results 1 – 2 of 2) sorted by relevance
134 Status PredictNaryOp(const OpContext& op_context,
536 device_cost_impl_.emplace(kAddN, wrap(&OpLevelCostEstimator::PredictNaryOp)); in OpLevelCostEstimator()2454 Status OpLevelCostEstimator::PredictNaryOp(const OpContext& op_context, in PredictNaryOp() function in tensorflow::grappler::OpLevelCostEstimator