Searched defs:shape_handle (Results 1 – 16 of 16) sorted by relevance
45 shape_inference::ShapeHandle shape_handle; in __anon5bd79b600102() local79 shape_inference::ShapeHandle shape_handle; in __anon5bd79b600102() local
67 shape_inference::ShapeHandle shape_handle; in __anon6270b1470302() local
121 ShapeHandle shape_handle; in __anon52e493120302() local235 ShapeHandle shape_handle; in __anon52e493120802() local
245 ShapeHandle shape_handle; in RaggedTensorToTensorShapeFn() local
579 shape_inference::ShapeHandle shape_handle; in __anon770f674c1002() local645 shape_inference::ShapeHandle shape_handle; in __anon770f674c1102() local
51 shape_inference::ShapeHandle shape_handle = ic.output(i); in RunShapeInference() local
546 shape_inference::ShapeHandle* shape_handle) { in InferenceShape()576 const shape_inference::ShapeHandle& shape_handle) { in SetInferenceShape()
69 ShapeHandle shape_handle = shapes_and_types->at(0).shape; in __anon2b8a3a590102() local
228 TF_ShapeHandle* shape_handle, int64_t i, in TF_ShapeInferenceContextDim()256 TF_ShapeHandle* shape_handle, in TF_ShapeInferenceContextSubshape()
688 ShapeHandle shape_handle = c.output(i); in TFE_InferShapes() local
299 tensorflow::shape_inference::ShapeHandle shape_handle = in TF_GraphSetOutputHandleShapesAndTypes() local
472 shape_inference::ShapeHandle shape_handle = context->output(output_index); in RegisterConstantNode() local617 shape_inference::ShapeHandle shape_handle = context->output(0); in IsNodeFlattenReshape() local964 const shape_inference::ShapeHandle& shape_handle, in BuildShapeArray()
306 ShapeHandle shape_handle = c.output(output); in InferReturnTypeComponentsForTFOp() local
833 const auto& shape_handle = shape_and_type.shape; in UpdateFunction() local1490 const ShapeHandle& shape_handle = ic->output(i); in ShouldUpdateOutputShapesAndValues() local1514 const ShapeHandle& shape_handle = ic->input_tensors_as_shapes()[i]; in CreateInputTensors() local1757 const ShapeHandle& shape_handle = in MaybeUpdateNodeContextOutput() local2407 ShapeHandle shape_handle; in UpdateQueue() local
501 shape_inference::ShapeHandle shape_handle = context->output(output_index); in BuildTensorShapeMapFromGraph() local
459 const ShapeHandle shape_handle, in CheckFormatConstraintsOnShape()