Home
last modified time | relevance | path

Searched refs:num_nodes_with_unknown_shapes (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/costs/
H A Dop_level_cost_estimator.cc685 node_costs.num_nodes_with_unknown_shapes; in PredictCosts()
693 costs.num_ops_with_unknown_shapes = node_costs.num_nodes_with_unknown_shapes; in PredictCosts()
1672 node_costs->num_nodes_with_unknown_shapes = 1; in PredictDefaultNodeCosts()
1696 node_costs->num_nodes_with_unknown_shapes = 1; in PredictConv2D()
1711 node_costs->num_nodes_with_unknown_shapes = 1; in PredictConv2DBackpropInput()
1727 node_costs->num_nodes_with_unknown_shapes = 1; in PredictConv2DBackpropFilter()
1820 node_costs->num_nodes_with_unknown_shapes = 1; in PredictFusedConv2DBiasActivation()
1851 node_costs->num_nodes_with_unknown_shapes = 1; in PredictEinsum()
1897 node_costs->num_nodes_with_unknown_shapes = 1; in PredictSparseTensorDenseMatMul()
1934 node_costs->num_nodes_with_unknown_shapes = 1; in PredictIdentity()
[all …]
H A Dop_level_cost_estimator.h79 int64_t num_nodes_with_unknown_shapes = 0; member