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.cc935 TEST_F(OpLevelCostEstimatorTest, GetTensorShapeProtoFromTensorProto) { in TEST_F() argument
943 GetTensorShapeProtoFromTensorProto(tensor_proto, &tensor_shape_proto)); in TEST_F()
950 GetTensorShapeProtoFromTensorProto(tensor_proto, &tensor_shape_proto)); in TEST_F()
955 GetTensorShapeProtoFromTensorProto(tensor_proto, &tensor_shape_proto)); in TEST_F()
963 GetTensorShapeProtoFromTensorProto(tensor_proto, &tensor_shape_proto)); in TEST_F()
972 GetTensorShapeProtoFromTensorProto(tensor_proto, &tensor_shape_proto)); in TEST_F()
981 GetTensorShapeProtoFromTensorProto(tensor_proto, &tensor_shape_proto)); in TEST_F()
990 GetTensorShapeProtoFromTensorProto(tensor_proto, &tensor_shape_proto)); in TEST_F()
Dop_level_cost_estimator.h27 bool GetTensorShapeProtoFromTensorProto(const TensorProto& tensor_proto,
Dop_level_cost_estimator.cc865 bool GetTensorShapeProtoFromTensorProto(const TensorProto& tensor_proto, in GetTensorShapeProtoFromTensorProto() function
936 shape_found = GetTensorShapeProtoFromTensorProto(value, &input_shape); in CountConv2DBackpropInputOperations()
988 shape_found = GetTensorShapeProtoFromTensorProto(value, &filter_shape); in CountConv2DBackpropFilterOperations()
1606 shape_found = GetTensorShapeProtoFromTensorProto(value, &x_shape); in PredictAvgPoolGrad()