Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/contrib/data/python/kernel_tests/
Dassert_element_shape_test.py86 unknown_shapes = (tensor_shape.TensorShape(None),
88 self.assertEqual(unknown_shapes,
118 unknown_shapes = (tensor_shape.TensorShape(None),
120 self.assertEqual(unknown_shapes,
185 unknown_shapes = (tensor_shape.TensorShape(None),
187 self.assertEqual(unknown_shapes,
217 unknown_shapes = (tensor_shape.TensorShape(None),
219 self.assertEqual(unknown_shapes,
/external/tensorflow/tensorflow/core/grappler/costs/
Dop_level_cost_estimator_test.cc1257 bool unknown_shapes = false; in TEST_F() local
1258 TensorShapeProto y = MaybeGetMinimumShape(x, 4, &unknown_shapes); in TEST_F()
1259 EXPECT_TRUE(unknown_shapes); in TEST_F()
1266 bool unknown_shapes = false; in TEST_F() local
1267 TensorShapeProto y = MaybeGetMinimumShape(x, 1, &unknown_shapes); in TEST_F()
1268 EXPECT_FALSE(unknown_shapes); in TEST_F()
1275 bool unknown_shapes = false; in TEST_F() local
1276 TensorShapeProto y = MaybeGetMinimumShape(x, 2, &unknown_shapes); in TEST_F()
1277 EXPECT_FALSE(unknown_shapes); in TEST_F()
1286 bool unknown_shapes = false; in TEST_F() local
[all …]
Dop_level_cost_estimator.cc499 bool unknown_shapes = false; in PredictOpCountBasedCost() local
500 const double input_size = CalculateInputSize(op_info, &unknown_shapes); in PredictOpCountBasedCost()
501 const double output_size = CalculateOutputSize(op_info, &unknown_shapes); in PredictOpCountBasedCost()
504 costs.inaccurate = unknown_shapes; in PredictOpCountBasedCost()
505 costs.num_ops_with_unknown_shapes = unknown_shapes; in PredictOpCountBasedCost()
1345 bool unknown_shapes = false; in PredictGatherOrSlice() local
1350 CalculateTensorElementCount(op_info.outputs(0), &unknown_shapes); in PredictGatherOrSlice()
1352 const double output_size = CalculateOutputSize(op_info, &unknown_shapes); in PredictGatherOrSlice()
1357 CalculateTensorElementCount(op_info.inputs(1), &unknown_shapes) + in PredictGatherOrSlice()
1358 CalculateTensorElementCount(op_info.inputs(2), &unknown_shapes); in PredictGatherOrSlice()
[all …]
/external/tensorflow/tensorflow/python/keras/engine/
Ddistributed_training_utils.py402 unknown_shapes = [s for s in shapes if not s.is_fully_defined()]
403 return not unknown_shapes