Searched refs:PredictNodeCosts (Results 1 – 2 of 2) sorted by relevance
117 Status PredictNodeCosts(const OpContext& op_context,
651 if (PredictNodeCosts(op_context, &node_costs).ok()) { in PredictCosts()694 Status OpLevelCostEstimator::PredictNodeCosts(const OpContext& op_context, in PredictNodeCosts() function in tensorflow::grappler::OpLevelCostEstimator1822 return PredictNodeCosts(batch_matmul_op_context, node_costs); in PredictEinsum()2066 s.Update(PredictNodeCosts(fused_op, &fused_node_costs)); in PredictFusedOp()