Searched refs:TryGetSubshape (Results 1 – 5 of 5) sorted by relevance
72 ShapeUtil::TryGetSubshape(on_host_shape(), index)); in SubShapedBuffer()74 ShapeUtil::TryGetSubshape(on_device_shape(), index)); in SubShapedBuffer()
367 xla::ShapeUtil::TryGetSubshape(parent->on_host_shape(), subshape)); in MakeSubBuffer()370 xla::ShapeUtil::TryGetSubshape(parent->on_device_shape(), subshape)); in MakeSubBuffer()
506 static StatusOr<const Shape*> TryGetSubshape(const Shape& shape,
674 ShapeUtil::TryGetSubshape(shape(), index)); in SubShapeTree()
792 /* static */ StatusOr<const Shape*> ShapeUtil::TryGetSubshape( in TryGetSubshape() function in xla::ShapeUtil