Searched refs:num_compute_ops (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | op_level_cost_estimator.cc | 671 node_costs.num_compute_ops, node_costs.num_total_read_bytes(), in PredictCosts()
1629 const int64 num_compute_ops, const OpContext& op_context, in PredictDefaultNodeCosts() argument
1632 node_costs->num_compute_ops = num_compute_ops; in PredictDefaultNodeCosts()
1669 int64 num_compute_ops = CountConv2DOperations(op_info, &found_unknown_shapes); in PredictConv2D() local
1670 return PredictDefaultNodeCosts(num_compute_ops, op_context, in PredictConv2D()
1684 int64 num_compute_ops = CountConv2DBackpropInputOperations( in PredictConv2DBackpropInput() local
1686 return PredictDefaultNodeCosts(num_compute_ops, op_context, in PredictConv2DBackpropInput()
1700 int64 num_compute_ops = CountConv2DBackpropFilterOperations( in PredictConv2DBackpropFilter() local
1702 return PredictDefaultNodeCosts(num_compute_ops, op_context, in PredictConv2DBackpropFilter()
1796 int64 num_compute_ops = CountMatMulOperations(op_info, &found_unknown_shapes); in PredictMatMul() local
[all …]
|
| D | op_level_cost_estimator.h | 44 int64 num_compute_ops = 0; member
309 static Status PredictDefaultNodeCosts(const int64 num_compute_ops,
|