Home
last modified time | relevance | path

Searched refs:unknown_shapes (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/costs/
Dop_level_cost_estimator_test.cc1646 bool unknown_shapes = false; in TEST_F() local
1647 TensorShapeProto y = MaybeGetMinimumShape(x, 4, &unknown_shapes); in TEST_F()
1648 EXPECT_TRUE(unknown_shapes); in TEST_F()
1655 bool unknown_shapes = false; in TEST_F() local
1656 TensorShapeProto y = MaybeGetMinimumShape(x, 1, &unknown_shapes); in TEST_F()
1657 EXPECT_FALSE(unknown_shapes); in TEST_F()
1664 bool unknown_shapes = false; in TEST_F() local
1665 TensorShapeProto y = MaybeGetMinimumShape(x, 2, &unknown_shapes); in TEST_F()
1666 EXPECT_FALSE(unknown_shapes); in TEST_F()
1675 bool unknown_shapes = false; in TEST_F() local
[all …]
Dop_level_cost_estimator.cc816 bool unknown_shapes = false; in PredictOpCountBasedCost() local
817 const double input_size = CalculateInputSize(op_info, &unknown_shapes); in PredictOpCountBasedCost()
818 const double output_size = CalculateOutputSize(op_info, &unknown_shapes); in PredictOpCountBasedCost()
821 costs.inaccurate = unknown_shapes; in PredictOpCountBasedCost()
822 costs.num_ops_with_unknown_shapes = unknown_shapes; in PredictOpCountBasedCost()
1963 bool unknown_shapes = false; in PredictGatherOrSlice() local
1968 CalculateTensorElementCount(op_info.outputs(0), &unknown_shapes); in PredictGatherOrSlice()
1971 const int64 output_size = CalculateOutputSize(op_info, &unknown_shapes); in PredictGatherOrSlice()
1994 CalculateTensorElementCount(op_info.inputs(i), &unknown_shapes)); in PredictGatherOrSlice()
1996 if (unknown_shapes) { in PredictGatherOrSlice()
/external/tensorflow/tensorflow/python/keras/distribute/
Ddistributed_training_utils_v1.py434 unknown_shapes = [s for s in shapes if not s.is_fully_defined()]
435 return not unknown_shapes