Searched defs:shape_handle (Results 1 – 14 of 14) sorted by relevance
45 shape_inference::ShapeHandle shape_handle; in __anon748b2ccf0102() local79 shape_inference::ShapeHandle shape_handle; in __anon748b2ccf0102() local
126 ShapeHandle shape_handle; in __anon2a9f7d410302() local241 ShapeHandle shape_handle; in __anon2a9f7d410802() local
579 shape_inference::ShapeHandle shape_handle; in __anon3692361b1002() local645 shape_inference::ShapeHandle shape_handle; in __anon3692361b1102() local
242 ShapeHandle shape_handle; in RaggedTensorToTensorShapeFn() local
228 TF_ShapeHandle* shape_handle, int64_t i, in TF_ShapeInferenceContextDim()256 TF_ShapeHandle* shape_handle, in TF_ShapeInferenceContextSubshape()
695 ShapeHandle shape_handle = c.output(i); in TFE_InferShapes() local
264 tensorflow::shape_inference::ShapeHandle shape_handle = in TF_GraphSetOutputHandleShapesAndTypes() local
862 const auto& shape_handle = shape_and_type.shape; in UpdateFunction() local1529 const ShapeHandle& shape_handle = ic->output(i); in ShouldUpdateOutputShapesAndValues() local1553 const ShapeHandle& shape_handle = ic->input_tensors_as_shapes()[i]; in CreateInputTensors() local1796 const ShapeHandle& shape_handle = in MaybeUpdateNodeContextOutput() local2451 ShapeHandle shape_handle; in UpdateQueue() 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 __anon5fb8c9c80102() local
358 ShapeHandle shape_handle = c.output(output); in InferReturnTypeComponentsForTFOp() local
736 shape_inference::ShapeHandle shape_handle; in PrepareGraphForMlir() local
459 const ShapeHandle shape_handle, in CheckFormatConstraintsOnShape()