Searched refs:PredictIdentity (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | op_level_cost_estimator.cc | 260 {kPlaceholder, wrap(&OpLevelCostEstimator::PredictIdentity)}, in OpLevelCostEstimator()
261 {kIdentity, wrap(&OpLevelCostEstimator::PredictIdentity)}, in OpLevelCostEstimator()
262 {kIdentityN, wrap(&OpLevelCostEstimator::PredictIdentity)}, in OpLevelCostEstimator()
263 {kRefIdentity, wrap(&OpLevelCostEstimator::PredictIdentity)}, in OpLevelCostEstimator()
264 {kStopGradient, wrap(&OpLevelCostEstimator::PredictIdentity)}, in OpLevelCostEstimator()
265 {kPreventGradient, wrap(&OpLevelCostEstimator::PredictIdentity)}, in OpLevelCostEstimator()
266 {kReshape, wrap(&OpLevelCostEstimator::PredictIdentity)}, in OpLevelCostEstimator()
267 {kSqueeze, wrap(&OpLevelCostEstimator::PredictIdentity)}, in OpLevelCostEstimator()
268 {kRecv, wrap(&OpLevelCostEstimator::PredictIdentity)}, in OpLevelCostEstimator()
269 {kSend, wrap(&OpLevelCostEstimator::PredictIdentity)}, in OpLevelCostEstimator()
[all …]
|
| D | op_level_cost_estimator.h | 137 Costs PredictIdentity(const OpContext& op_context) const;
|