Function createGPUBufferFromData

tfjs-backend-webgpu/src/backend_webgpu_test.ts:510–528

(
    device: GPUDevice, data: number[], dtype: tf.DataType,
    bufferUsage = GPUBufferUsage.COPY_DST | GPUBufferUsage.STORAGE |
        GPUBufferUsage.COPY_SRC)

}

function createGPUBufferFromData(
    device: GPUDevice, data: number[], dtype: tf.DataType,
    bufferUsage = GPUBufferUsage.COPY_DST | GPUBufferUsage.STORAGE |
        GPUBufferUsage.COPY_SRC) {
  const bytesPerElement = 4;
  const sizeInBytes = data.length * bytesPerElement;

  const gpuWriteBuffer = createStagingGPUBufferFromData(device, data, dtype);
  const gpuReadBuffer = device.createBuffer(
      {mappedAtCreation: false, size: sizeInBytes, usage: bufferUsage});

  const copyEncoder = device.createCommandEncoder();
  copyEncoder.copyBufferToBuffer(
      gpuWriteBuffer, 0, gpuReadBuffer, 0, sizeInBytes);
  const copyCommands = copyEncoder.finish();
  device.queue.submit([copyCommands]);
  gpuWriteBuffer.destroy();
  return gpuReadBuffer;
}

async function testCreateTensorFromGPUBuffer(
    dtype: tf.DataType, useDefaultShapeAndType = false, zeroCopy = false) {