Searched refs:MakeShapeFromShapeProto (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/c/ |
| D | python_api.cc | 165 ic->MakeShapeFromShapeProto(shape_and_type_proto.shape(), &shape); in SetHandleShapeAndType()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | shape_inference.cc | 46 construction_status_.Update(MakeShapeFromShapeProto(p, &shape)); in InferenceContext()
57 construction_status_.Update(MakeShapeFromShapeProto(p, &shape)); in InferenceContext()
76 MakeShapeFromShapeProto(p.first, &new_v[j].shape)); in InferenceContext()
940 Status InferenceContext::MakeShapeFromShapeProto(const TensorShapeProto& proto, in MakeShapeFromShapeProto() function in tensorflow::shape_inference::InferenceContext
|
| D | shape_inference_test.cc | 1159 TEST_F(ShapeInferenceTest, MakeShapeFromShapeProto) { in TEST_F() argument
1168 EXPECT_TRUE(c.MakeShapeFromShapeProto(proto, &out).ok()); in TEST_F()
1172 c.MakeShapeFromShapeProto(proto, &out).error_message(), in TEST_F()
1178 EXPECT_TRUE(c.MakeShapeFromShapeProto(proto, &out).ok()); in TEST_F()
1182 EXPECT_TRUE(c.MakeShapeFromShapeProto(proto, &out).ok()); in TEST_F()
1188 c.MakeShapeFromShapeProto(proto, &out).error_message(), in TEST_F()
|
| D | shape_inference.h | 488 Status MakeShapeFromShapeProto(const TensorShapeProto& proto,
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | tpu_embedding_ops.cc | 348 c->MakeShapeFromShapeProto(output_shapes[i], &output_shape)); in __anon355f22aa0902()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | shape_refiner.cc | 110 TF_RETURN_IF_ERROR(outer_context->MakeShapeFromShapeProto(proto, &handle)); in InferShapesForFunctionSubNode()
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | graph_properties.cc | 724 TF_RETURN_IF_ERROR(ic->MakeShapeFromShapeProto(shape, &out)); in UpdateFunction()
1353 TF_RETURN_IF_ERROR(ic->MakeShapeFromShapeProto(shape, &output_shape)); in UpdateOutputShapesUsingAnnotatedInformation()
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | graph_constructor.cc | 617 Status s = ic->MakeShapeFromShapeProto(p, &h); in ValidateShape()
|