Searched refs:PredictAssignVariableOps (Results 1 – 2 of 2) sorted by relevance
177 Status PredictAssignVariableOps(const OpContext& op_context,
529 kAssignVariableOp, wrap(&OpLevelCostEstimator::PredictAssignVariableOps)); in OpLevelCostEstimator()532 wrap(&OpLevelCostEstimator::PredictAssignVariableOps)); in OpLevelCostEstimator()535 wrap(&OpLevelCostEstimator::PredictAssignVariableOps)); in OpLevelCostEstimator()2259 Status OpLevelCostEstimator::PredictAssignVariableOps( in PredictAssignVariableOps() function in tensorflow::grappler::OpLevelCostEstimator