Searched refs:element_tensor_list_shape (Results 1 – 1 of 1) sorted by relevance
226 const xla::Shape& element_tensor_list_shape, int64 leading_dim, in GetTensorListShapeFromElementTensorListShape() argument229 int tuple_size = xla::ShapeUtil::TupleElementCount(element_tensor_list_shape); in GetTensorListShapeFromElementTensorListShape()232 xla::ShapeUtil::GetTupleElementShape(element_tensor_list_shape, i); in GetTensorListShapeFromElementTensorListShape()