Searched refs:index_to_buffer (Results 1 – 2 of 2) sorted by relevance
53 for (auto& index_to_buffer : shaped_buffer.buffers()) { in AllocateShapedBuffer() local55 xla::ShapeUtil::GetSubshape(on_device_shape, index_to_buffer.first); in AllocateShapedBuffer()63 index_to_buffer.second = buffer.Release(); in AllocateShapedBuffer()
97 for (auto& index_to_buffer : (*buffer)->buffers()) { in AllocateScopedShapedBuffer()99 xla::ShapeUtil::GetSubshape(on_device_shape, index_to_buffer.first); in AllocateScopedShapedBuffer()105 index_to_buffer.second = buffer.Release(); in AllocateScopedShapedBuffer()106 VLOG(2) << "Allocated buffer at " << index_to_buffer.second.opaque() in AllocateScopedShapedBuffer()107 << " index " << index_to_buffer.first.ToString() << " (" << size in AllocateScopedShapedBuffer()