Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/grappler/costs/
Dop_level_cost_estimator_test.cc1312 TEST_F(OpLevelCostEstimatorTest, GetTensorShapeProtoFromTensorProto) { in TEST_F() argument
1320 GetTensorShapeProtoFromTensorProto(tensor_proto, &tensor_shape_proto)); in TEST_F()
1327 GetTensorShapeProtoFromTensorProto(tensor_proto, &tensor_shape_proto)); in TEST_F()
1332 GetTensorShapeProtoFromTensorProto(tensor_proto, &tensor_shape_proto)); in TEST_F()
1340 GetTensorShapeProtoFromTensorProto(tensor_proto, &tensor_shape_proto)); in TEST_F()
1349 GetTensorShapeProtoFromTensorProto(tensor_proto, &tensor_shape_proto)); in TEST_F()
1358 GetTensorShapeProtoFromTensorProto(tensor_proto, &tensor_shape_proto)); in TEST_F()
1367 GetTensorShapeProtoFromTensorProto(tensor_proto, &tensor_shape_proto)); in TEST_F()
Dop_level_cost_estimator.h30 bool GetTensorShapeProtoFromTensorProto(const TensorProto& tensor_proto,
Dop_level_cost_estimator.cc1362 bool GetTensorShapeProtoFromTensorProto(const TensorProto& tensor_proto, in GetTensorShapeProtoFromTensorProto() function
1433 shape_found = GetTensorShapeProtoFromTensorProto(value, &input_shape); in CountConv2DBackpropInputOperations()
1485 shape_found = GetTensorShapeProtoFromTensorProto(value, &filter_shape); in CountConv2DBackpropFilterOperations()
2334 shape_found = GetTensorShapeProtoFromTensorProto(value, &x_shape); in PredictAvgPoolGrad()