Searched refs:TryGetSubshape (Results 1 – 3 of 3) sorted by relevance
300 xla::ShapeUtil::TryGetSubshape(parent->on_host_shape(), subshape)); in MakeSubBuffer()303 xla::ShapeUtil::TryGetSubshape(parent->on_device_shape(), subshape)); in MakeSubBuffer()
493 static StatusOr<const Shape*> TryGetSubshape(const Shape& shape,
792 /* static */ StatusOr<const Shape*> ShapeUtil::TryGetSubshape( in TryGetSubshape() function in xla::ShapeUtil