Searched refs:num_ops_with_unknown_shapes (Results 1 – 9 of 9) sorted by relevance
/external/tensorflow/tensorflow/core/grappler/costs/ |
D | op_level_cost_estimator_test.cc | 534 EXPECT_EQ(0, cost.num_ops_with_unknown_shapes); in TEST_F() 554 EXPECT_EQ(0, cost.num_ops_with_unknown_shapes); in TEST_F() 572 EXPECT_EQ(0, cost.num_ops_with_unknown_shapes); in TEST_F() 592 EXPECT_EQ(0, cost.num_ops_with_unknown_shapes); in TEST_F() 602 EXPECT_EQ(0, cost.num_ops_with_unknown_shapes); in TEST_F() 612 EXPECT_EQ(0, cost.num_ops_with_unknown_shapes); in TEST_F() 623 EXPECT_EQ(0, cost.num_ops_with_unknown_shapes); in TEST_F() 633 EXPECT_EQ(0, cost.num_ops_with_unknown_shapes); in TEST_F() 644 EXPECT_EQ(0, cost.num_ops_with_unknown_shapes); in TEST_F() 658 EXPECT_EQ(0, cost.num_ops_with_unknown_shapes); in TEST_F() [all …]
|
D | cost_estimator_test.cc | 37 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()
|
D | op_level_cost_estimator.cc | 484 costs.num_ops_with_unknown_shapes = found_unknown_shapes; in PredictCwiseOp() 505 costs.num_ops_with_unknown_shapes = unknown_shapes; in PredictOpCountBasedCost() 1106 costs.num_ops_with_unknown_shapes = found_unknown_shapes; in PredictConv2D() 1119 costs.num_ops_with_unknown_shapes = found_unknown_shapes; in PredictConv2DBackpropInput() 1132 costs.num_ops_with_unknown_shapes = found_unknown_shapes; in PredictConv2DBackpropFilter() 1220 costs.num_ops_with_unknown_shapes = costs.inaccurate; in PredictFusedConv2DBiasActivation() 1230 costs.num_ops_with_unknown_shapes = found_unknown_shapes; in PredictMatMul() 1271 costs.num_ops_with_unknown_shapes = found_unknown_shapes; in PredictSparseTensorDenseMatMul() 1288 result.num_ops_with_unknown_shapes = result.inaccurate; in PredictIdentity() 1301 result.num_ops_with_unknown_shapes = result.inaccurate; in PredictVariable() [all …]
|
D | cost_estimator.cc | 47 result.num_ops_with_unknown_shapes += right.num_ops_with_unknown_shapes; in CombineCosts()
|
D | analytical_cost_estimator.cc | 171 if (node_costs.num_ops_with_unknown_shapes > 0) in PredictCosts() 173 << node_costs.num_ops_with_unknown_shapes << " unknown shapes"; in PredictCosts()
|
D | analytical_cost_estimator_test.cc | 116 EXPECT_EQ(0, summary.num_ops_with_unknown_shapes); in TEST_F()
|
D | cost_estimator.h | 158 int64 num_ops_with_unknown_shapes = 0; member
|
D | virtual_scheduler_test.cc | 2294 EXPECT_EQ(0, c.num_ops_with_unknown_shapes); in TEST_F() 2310 EXPECT_EQ(0, c.num_ops_with_unknown_shapes); in TEST_F() 2631 EXPECT_EQ(0, c.num_ops_with_unknown_shapes); in TEST_F() 2649 EXPECT_EQ(0, c.num_ops_with_unknown_shapes); in TEST_F() 2702 EXPECT_EQ(0, c.num_ops_with_unknown_shapes); in TEST_F() 2764 EXPECT_EQ(0, c.num_ops_with_unknown_shapes); in TEST_F()
|
D | virtual_scheduler.cc | 859 << graph_costs_.num_ops_with_unknown_shapes in Summary() 937 << state.device_costs.num_ops_with_unknown_shapes in Summary()
|