Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/grappler/costs/
Dop_level_cost_estimator.cc157 MaybeGetMinimumShape(input_shape_1, rank, &found_unknown_shapes); in CwiseOutputElementCount()
161 MaybeGetMinimumShape(input_shape_1, rank, &found_unknown_shapes); in CwiseOutputElementCount()
163 MaybeGetMinimumShape(input_shape_2, rank, &found_unknown_shapes); in CwiseOutputElementCount()
183 TensorShapeProto MaybeGetMinimumShape(const TensorShapeProto& original_shape, in MaybeGetMinimumShape() function
580 MaybeGetMinimumShape(original_image_shape, 4, found_unknown_shapes); in ConvolutionDimensionsFromInputs()
582 MaybeGetMinimumShape(original_filter_shape, 4, found_unknown_shapes); in ConvolutionDimensionsFromInputs()
717 MaybeGetMinimumShape(a_matrix.shape(), 2, found_unknown_shapes); in CountMatMulOperations()
719 MaybeGetMinimumShape(b_matrix.shape(), 2, found_unknown_shapes); in CountMatMulOperations()
790 TensorShapeProto a_input_shape = MaybeGetMinimumShape( in CountBatchMatMulOperations()
793 TensorShapeProto b_input_shape = MaybeGetMinimumShape( in CountBatchMatMulOperations()
[all …]
Dop_level_cost_estimator_test.cc1253 TEST_F(OpLevelCostEstimatorTest, MaybeGetMinimumShape) { in TEST_F() argument
1258 TensorShapeProto y = MaybeGetMinimumShape(x, 4, &unknown_shapes); in TEST_F()
1267 TensorShapeProto y = MaybeGetMinimumShape(x, 1, &unknown_shapes); in TEST_F()
1276 TensorShapeProto y = MaybeGetMinimumShape(x, 2, &unknown_shapes); in TEST_F()
1287 TensorShapeProto y = MaybeGetMinimumShape(x, 2, &unknown_shapes); in TEST_F()
1292 TensorShapeProto z = MaybeGetMinimumShape(x, 4, &unknown_shapes); in TEST_F()
1306 TensorShapeProto y = MaybeGetMinimumShape(x, 4, &unknown_shapes); in TEST_F()
1319 TensorShapeProto y = MaybeGetMinimumShape(x, 2, &unknown_shapes); in TEST_F()
Dop_level_cost_estimator.h29 TensorShapeProto MaybeGetMinimumShape(const TensorShapeProto& original_shape,