Searched refs:tensor_list_shape (Results 1 – 2 of 2) sorted by relevance
227 bool leading_dim_is_dynamic, xla::Shape* tensor_list_shape) { in GetTensorListShapeFromElementTensorListShape() argument243 *tensor_list_shape = xla::ShapeUtil::MakeTupleShape(shapes); in GetTensorListShapeFromElementTensorListShape()250 xla::Shape* tensor_list_shape) { in GetTensorListShapeFromElementShape() argument265 *tensor_list_shape = xla::ShapeUtil::MakeTupleShape(shapes); in GetTensorListShapeFromElementShape()
79 xla::Shape* tensor_list_shape);