Searched refs:ShapeHandleToProto (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/core/ops/ |
| D | ragged_conversion_ops.cc | 73 c->ShapeHandleToProto(c->input(3 + i), &partition_shape); in ValidateRowPartitionTypesAndShapes()
230 c->ShapeHandleToProto(shape_handle, &shape); in RaggedTensorToTensorShapeFn()
240 c->ShapeHandleToProto(c->input(1), &value_shape); in RaggedTensorToTensorShapeFn()
243 c->ShapeHandleToProto(c->input(2), &default_value_shape); in RaggedTensorToTensorShapeFn()
|
| /external/tensorflow/tensorflow/c/ |
| D | python_api.cc | 137 ic->ShapeHandleToProto(p.shape, out_shape_and_type->mutable_shape()); in GetHandleShapeAndType()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | shape_inference.cc | 116 context->ShapeHandleToProto(handle, &shape_proto); in PropagateShapes()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | shape_inference.h | 338 void ShapeHandleToProto(ShapeHandle handle, TensorShapeProto* proto);
|
| D | shape_inference_test.cc | 1000 c.ShapeHandleToProto(s, &proto); in TEST_F()
1014 c.ShapeHandleToProto(u0, &proto); in TEST_F()
|
| D | shape_inference.cc | 226 void InferenceContext::ShapeHandleToProto(ShapeHandle handle, in ShapeHandleToProto() function in tensorflow::shape_inference::InferenceContext
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | shape_refiner.cc | 117 node_context->ShapeHandleToProto(node_context->input(0), &proto); in InferShapesForFunctionSubNode()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | functionalize_cond.cc | 798 shape_ctx->ShapeHandleToProto(shape_ctx->output(0), &shape); in BuildIfNode()
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | graph_properties.cc | 659 input_ic->ShapeHandleToProto(handle, &proto); in UpdateFunction()
|