Searched refs:GetTensorShapeProtoFromTensorProto (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/grappler/costs/ |
| H A D | op_level_cost_estimator_test.cc | 1346 TEST_F(OpLevelCostEstimatorTest, GetTensorShapeProtoFromTensorProto) { in TEST_F() argument
1354 GetTensorShapeProtoFromTensorProto(tensor_proto, &tensor_shape_proto)); in TEST_F()
1361 GetTensorShapeProtoFromTensorProto(tensor_proto, &tensor_shape_proto)); in TEST_F()
1366 GetTensorShapeProtoFromTensorProto(tensor_proto, &tensor_shape_proto)); in TEST_F()
1374 GetTensorShapeProtoFromTensorProto(tensor_proto, &tensor_shape_proto)); in TEST_F()
1383 GetTensorShapeProtoFromTensorProto(tensor_proto, &tensor_shape_proto)); in TEST_F()
1392 GetTensorShapeProtoFromTensorProto(tensor_proto, &tensor_shape_proto)); in TEST_F()
1401 GetTensorShapeProtoFromTensorProto(tensor_proto, &tensor_shape_proto)); in TEST_F()
|
| H A D | op_level_cost_estimator.h | 30 bool GetTensorShapeProtoFromTensorProto(const TensorProto& tensor_proto,
|
| H A D | op_level_cost_estimator.cc | 1374 bool GetTensorShapeProtoFromTensorProto(const TensorProto& tensor_proto, in GetTensorShapeProtoFromTensorProto() function
1445 shape_found = GetTensorShapeProtoFromTensorProto(value, &input_shape); in CountConv2DBackpropInputOperations()
1497 shape_found = GetTensorShapeProtoFromTensorProto(value, &filter_shape); in CountConv2DBackpropFilterOperations()
2378 shape_found = GetTensorShapeProtoFromTensorProto(value, &x_shape); in PredictAvgPoolGrad()
|