Home
last modified time | relevance | path

Searched refs:num_ops_with_unknown_shapes (Results 1 – 9 of 9) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/costs/
H A Dop_level_cost_estimator_test.cc680 EXPECT_EQ(cost.num_ops_with_unknown_shapes, 0); in TEST_F()
705 EXPECT_EQ(cost.num_ops_with_unknown_shapes, 0); in TEST_F()
726 EXPECT_EQ(cost.num_ops_with_unknown_shapes, 0); in TEST_F()
748 EXPECT_EQ(cost.num_ops_with_unknown_shapes, 0); in TEST_F()
771 EXPECT_EQ(cost.num_ops_with_unknown_shapes, 0); in TEST_F()
800 EXPECT_EQ(cost.num_ops_with_unknown_shapes, 0); in TEST_F()
824 EXPECT_EQ(0, cost.num_ops_with_unknown_shapes); in TEST_F()
836 EXPECT_EQ(cost.num_ops_with_unknown_shapes, 0); in TEST_F()
848 EXPECT_EQ(cost.num_ops_with_unknown_shapes, 0); in TEST_F()
881 EXPECT_EQ(1, cost.num_ops_with_unknown_shapes); in TEST_F()
[all …]
H A Dcost_estimator_test.cc37 c.num_ops_with_unknown_shapes = 0; in TEST()
52 EXPECT_EQ(sum.num_ops_with_unknown_shapes, 0); in TEST()
68 c.num_ops_with_unknown_shapes = 0; in TEST()
83 EXPECT_EQ(product.num_ops_with_unknown_shapes, 0); in TEST()
H A Dcost_estimator.cc47 result.num_ops_with_unknown_shapes += right.num_ops_with_unknown_shapes; in CombineCosts()
H A Danalytical_cost_estimator.cc205 if (node_costs.num_ops_with_unknown_shapes > 0) in PredictCosts()
207 << node_costs.num_ops_with_unknown_shapes << " unknown shapes"; in PredictCosts()
H A Dvirtual_scheduler_test.cc2424 EXPECT_EQ(0, c.num_ops_with_unknown_shapes); in TEST_F()
2440 EXPECT_EQ(0, c.num_ops_with_unknown_shapes); in TEST_F()
2862 EXPECT_EQ(0, c.num_ops_with_unknown_shapes); in TEST_F()
2880 EXPECT_EQ(0, c.num_ops_with_unknown_shapes); in TEST_F()
2933 EXPECT_EQ(0, c.num_ops_with_unknown_shapes); in TEST_F()
2995 EXPECT_EQ(0, c.num_ops_with_unknown_shapes); in TEST_F()
H A Danalytical_cost_estimator_test.cc115 EXPECT_EQ(0, summary.num_ops_with_unknown_shapes); in TEST_F()
H A Dcost_estimator.h171 int64_t num_ops_with_unknown_shapes = 0; member
H A Dvirtual_scheduler.cc1056 << graph_costs_.num_ops_with_unknown_shapes in Summary()
1135 << state.device_costs.num_ops_with_unknown_shapes in Summary()
H A Dop_level_cost_estimator.cc684 costs.num_ops_with_unknown_shapes = in PredictCosts()
693 costs.num_ops_with_unknown_shapes = node_costs.num_nodes_with_unknown_shapes; in PredictCosts()
834 costs.num_ops_with_unknown_shapes = unknown_shapes; in PredictOpCountBasedCost()