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.cc290 TensorShapeProto a_input_shape = MaybeGetMinimumShape( in IsEinsumCorrectlyFormed()
293 TensorShapeProto b_input_shape = MaybeGetMinimumShape( in IsEinsumCorrectlyFormed()
322 TensorShapeProto MaybeGetMinimumShape(const TensorShapeProto& original_shape, in MaybeGetMinimumShape() function
938 auto image_shape = MaybeGetMinimumShape(original_image_shape, in ConvolutionDimensionsFromInputs()
941 auto filter_shape = MaybeGetMinimumShape(original_filter_shape, in ConvolutionDimensionsFromInputs()
1070 MaybeGetMinimumShape(a_matrix.shape(), 2, found_unknown_shapes); in CountMatMulOperations()
1072 MaybeGetMinimumShape(b_matrix.shape(), 2, found_unknown_shapes); in CountMatMulOperations()
1155 TensorShapeProto a_input_shape = MaybeGetMinimumShape( in GenerateBatchMatmulContextFromEinsum()
1158 TensorShapeProto b_input_shape = MaybeGetMinimumShape( in GenerateBatchMatmulContextFromEinsum()
1273 TensorShapeProto a_input_shape = MaybeGetMinimumShape( in CountBatchMatMulOperations()
[all …]
Dop_level_cost_estimator_test.cc1642 TEST_F(OpLevelCostEstimatorTest, MaybeGetMinimumShape) { in TEST_F() argument
1647 TensorShapeProto y = MaybeGetMinimumShape(x, 4, &unknown_shapes); in TEST_F()
1656 TensorShapeProto y = MaybeGetMinimumShape(x, 1, &unknown_shapes); in TEST_F()
1665 TensorShapeProto y = MaybeGetMinimumShape(x, 2, &unknown_shapes); in TEST_F()
1676 TensorShapeProto y = MaybeGetMinimumShape(x, 2, &unknown_shapes); in TEST_F()
1681 TensorShapeProto z = MaybeGetMinimumShape(x, 4, &unknown_shapes); in TEST_F()
1695 TensorShapeProto y = MaybeGetMinimumShape(x, 4, &unknown_shapes); in TEST_F()
1708 TensorShapeProto y = MaybeGetMinimumShape(x, 2, &unknown_shapes); in TEST_F()
Dop_level_cost_estimator.h32 TensorShapeProto MaybeGetMinimumShape(const TensorShapeProto& original_shape,