Searched refs:PredictDefaultNodeCosts (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | op_level_cost_estimator.cc | 800 return PredictDefaultNodeCosts(op_count * op_cost, op_context, in PredictCwiseOp()
810 return PredictDefaultNodeCosts(0, op_context, &found_unknown_shapes, in PredictCostOfAnUnknownOp()
1628 Status OpLevelCostEstimator::PredictDefaultNodeCosts( in PredictDefaultNodeCosts() function in tensorflow::grappler::OpLevelCostEstimator
1670 return PredictDefaultNodeCosts(num_compute_ops, op_context, in PredictConv2D()
1686 return PredictDefaultNodeCosts(num_compute_ops, op_context, in PredictConv2DBackpropInput()
1702 return PredictDefaultNodeCosts(num_compute_ops, op_context, in PredictConv2DBackpropFilter()
1797 return PredictDefaultNodeCosts(num_compute_ops, op_context, in PredictMatMul()
1882 return PredictDefaultNodeCosts(0, op_context, &found_unknown_shapes, in PredictPureMemoryOp()
1930 return PredictDefaultNodeCosts(num_compute_ops, op_context, in PredictBatchMatMul()
2062 PredictDefaultNodeCosts(0, op_context, &found_unknown_shapes, node_costs); in PredictFusedOp()
[all …]
|
| D | op_level_cost_estimator.h | 309 static Status PredictDefaultNodeCosts(const int64 num_compute_ops,
|