Home
last modified time | relevance | path

Searched refs:output_shape_handle (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/ops/
Dfunctional_ops.cc105 shape_inference::ShapeHandle output_shape_handle; in IfShapeInferenceFn() local
107 output_shapes[i], &output_shape_handle)); in IfShapeInferenceFn()
108 c->set_output(static_cast<int>(i), output_shape_handle); in IfShapeInferenceFn()
150 shape_inference::ShapeHandle output_shape_handle; in CaseShapeInferenceFn() local
152 output_shapes[i], &output_shape_handle)); in CaseShapeInferenceFn()
153 c->set_output(static_cast<int>(i), output_shape_handle); in CaseShapeInferenceFn()
222 shape_inference::ShapeHandle output_shape_handle; in WhileShapeInferenceFn() local
224 output_shapes[i], &output_shape_handle)); in WhileShapeInferenceFn()
225 c->set_output(static_cast<int>(i), output_shape_handle); in WhileShapeInferenceFn()
Dragged_conversion_ops.cc278 ShapeHandle output_shape_handle; in RaggedTensorToTensorShapeFn() local
280 c->MakeShapeFromShapeProto(output_shape, &output_shape_handle)); in RaggedTensorToTensorShapeFn()
281 c->set_output(0, output_shape_handle); in RaggedTensorToTensorShapeFn()
Ddataset_ops.cc966 shape_inference::ShapeHandle output_shape_handle; in __anon42b7b0c02002() local
969 c->MakeShapeFromPartialTensorShape(s, &output_shape_handle)); in __anon42b7b0c02002()
970 c->set_output(static_cast<int>(i), output_shape_handle); in __anon42b7b0c02002()
/external/tensorflow/tensorflow/core/framework/
Dcommon_shape_fns.cc486 shape_inference::ShapeHandle output_shape_handle; in DatasetIteratorShape() local
488 output_shapes[i], &output_shape_handle)); in DatasetIteratorShape()
489 c->set_output(static_cast<int>(i), output_shape_handle); in DatasetIteratorShape()