Searched refs:ShapeHandleToProto (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/core/ops/ |
| D | ragged_conversion_ops.cc | 73 c->ShapeHandleToProto(c->input(3 + i), &partition_shape); in ValidateRowPartitionTypesAndShapes()
248 c->ShapeHandleToProto(shape_handle, &shape); in RaggedTensorToTensorShapeFn()
258 c->ShapeHandleToProto(c->input(1), &value_shape); in RaggedTensorToTensorShapeFn()
261 c->ShapeHandleToProto(c->input(2), &default_value_shape); in RaggedTensorToTensorShapeFn()
|
| /external/tensorflow/tensorflow/c/ |
| D | python_api.cc | 101 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/tpu/kernels/xla/ |
| D | host_compute_ops.cc | 354 shape_ctx->ShapeHandleToProto(handle, &shape_proto); in InferOutputShapes()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | shape_inference.h | 359 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 | 228 void InferenceContext::ShapeHandleToProto(ShapeHandle handle, in ShapeHandleToProto() function in tensorflow::shape_inference::InferenceContext
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | shape_refiner.cc | 119 node_context->ShapeHandleToProto(node_context->input(0), &proto); in InferShapesForFunctionSubNode()
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | graph_properties.cc | 815 input_ic->ShapeHandleToProto(handle, &proto); in UpdateFunction()
835 input_ic->ShapeHandleToProto( in UpdateFunction()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | functionalize_cond.cc | 806 shape_ctx->ShapeHandleToProto(shape_ctx->output(0), &shape); in BuildIfNode()
|