Searched refs:PredictOpCountBasedCost (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | op_level_cost_estimator.cc | 480 Costs costs = PredictOpCountBasedCost(op_count * op_cost, op_info); in PredictCwiseOp()
492 auto costs = PredictOpCountBasedCost(0, op_context.op_info); in PredictCostOfAnUnknownOp()
497 Costs OpLevelCostEstimator::PredictOpCountBasedCost( in PredictOpCountBasedCost() function in tensorflow::grappler::OpLevelCostEstimator
503 PredictOpCountBasedCost(operations, input_size, output_size, op_info); in PredictOpCountBasedCost()
510 Costs OpLevelCostEstimator::PredictOpCountBasedCost( in PredictOpCountBasedCost() function in tensorflow::grappler::OpLevelCostEstimator
1103 auto costs = PredictOpCountBasedCost( in PredictConv2D()
1115 PredictOpCountBasedCost(CountConv2DBackpropInputOperations( in PredictConv2DBackpropInput()
1128 PredictOpCountBasedCost(CountConv2DBackpropFilterOperations( in PredictConv2DBackpropFilter()
1227 auto costs = PredictOpCountBasedCost( in PredictMatMul()
1269 PredictOpCountBasedCost(op_count, input_size, output_size, op_info); in PredictSparseTensorDenseMatMul()
[all …]
|
| D | op_level_cost_estimator.h | 48 Costs PredictOpCountBasedCost(double operations, const OpInfo& op_info) const;
53 Costs PredictOpCountBasedCost(double operations, double input_io_bytes,
|