Searched refs:PredictAssignVariableOps (Results 1 – 2 of 2) sorted by relevance
177 Status PredictAssignVariableOps(const OpContext& op_context,
530 kAssignVariableOp, wrap(&OpLevelCostEstimator::PredictAssignVariableOps)); in OpLevelCostEstimator()533 wrap(&OpLevelCostEstimator::PredictAssignVariableOps)); in OpLevelCostEstimator()536 wrap(&OpLevelCostEstimator::PredictAssignVariableOps)); in OpLevelCostEstimator()2301 Status OpLevelCostEstimator::PredictAssignVariableOps( in PredictAssignVariableOps() function in tensorflow::grappler::OpLevelCostEstimator