Searched refs:MakeShapeFromShapeProto (Results 1 – 12 of 12) sorted by relevance
| /external/tensorflow/tensorflow/core/tpu/ops/ |
| D | host_compute_ops.cc | 77 c->MakeShapeFromShapeProto(shapes->list().shape(i), &handle)); in __anonc8f95cfc0202()
112 c->MakeShapeFromShapeProto(shape_attr->shape(), &handle)); in __anonc8f95cfc0402()
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | function_ops.cc | 47 context->MakeShapeFromShapeProto(shape_proto, &shape_handle)); in __anon5bd79b600102()
81 context->MakeShapeFromShapeProto(shape_proto, &shape_handle)); in __anon5bd79b600102()
|
| D | tpu_embedding_ops.cc | 81 c->MakeShapeFromShapeProto(output_shapes[i], &output_shape)); in __anon222eb4110102()
|
| D | ragged_conversion_ops.cc | 280 c->MakeShapeFromShapeProto(output_shape, &output_shape_handle)); in RaggedTensorToTensorShapeFn()
|
| /external/tensorflow/tensorflow/c/ |
| D | python_api.cc | 129 ic->MakeShapeFromShapeProto(shape_and_type_proto.shape(), &shape); in SetHandleShapeAndType()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | shape_inference_test.cc | 1169 TEST_F(ShapeInferenceTest, MakeShapeFromShapeProto) { in TEST_F() argument
1178 EXPECT_TRUE(c.MakeShapeFromShapeProto(proto, &out).ok()); in TEST_F()
1181 EXPECT_CONTAINS(c.MakeShapeFromShapeProto(proto, &out).error_message(), in TEST_F()
1187 EXPECT_TRUE(c.MakeShapeFromShapeProto(proto, &out).ok()); in TEST_F()
1191 EXPECT_TRUE(c.MakeShapeFromShapeProto(proto, &out).ok()); in TEST_F()
1196 EXPECT_CONTAINS(c.MakeShapeFromShapeProto(proto, &out).error_message(), in TEST_F()
|
| D | shape_inference.h | 474 Status MakeShapeFromShapeProto(const TensorShapeProto& proto,
|
| D | shape_inference.cc | 884 Status InferenceContext::MakeShapeFromShapeProto(const TensorShapeProto& proto, in MakeShapeFromShapeProto() function in tensorflow::shape_inference::InferenceContext
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | shape_refiner.cc | 120 TF_RETURN_IF_ERROR(outer_context->MakeShapeFromShapeProto(proto, &handle)); in InferShapesForFunctionSubNode()
|
| D | graph_constructor.cc | 817 Status s = ic->MakeShapeFromShapeProto(p, &h); in ValidateShape()
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | graph_properties.cc | 943 TF_RETURN_IF_ERROR(ic->MakeShapeFromShapeProto(shape, &out)); in UpdateFunction()
1626 TF_RETURN_IF_ERROR(ic->MakeShapeFromShapeProto(shape, &output_shape)); in UpdateOutputShapesUsingAnnotatedInformation()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/ |
| D | import_model.cc | 795 node_context->MakeShapeFromShapeProto(shape.value(), &handle); in AddNodesToShapeRefiner()
832 node_context->MakeShapeFromShapeProto(attr->shape(), &handle); in AddNodesToShapeRefiner()
|