Searched refs:index_to_buffer (Results 1 – 2 of 2) sorted by relevance
57 for (auto& index_to_buffer : shaped_buffer.buffers()) { in AllocateShapedBuffer() local59 xla::ShapeUtil::GetSubshape(on_device_shape, index_to_buffer.first); in AllocateShapedBuffer()66 index_to_buffer.second = buffer.Forget(); in AllocateShapedBuffer()
111 for (auto& index_to_buffer : (*buffer)->buffers()) { in AllocateScopedShapedBuffer()113 xla::ShapeUtil::GetSubshape(on_device_shape, index_to_buffer.first); in AllocateScopedShapedBuffer()119 index_to_buffer.second = buffer.Forget(); in AllocateScopedShapedBuffer()120 VLOG(2) << "Allocated buffer at " << index_to_buffer.second.opaque() in AllocateScopedShapedBuffer()121 << " index " << index_to_buffer.first.ToString(); in AllocateScopedShapedBuffer()