Method acquireTexture

tfjs-backend-webgpu/src/texture_manager.ts:29–64

(
      width: number, height: number, format: GPUTextureFormat,
      usage: GPUTextureUsageFlags)

  constructor(private device: GPUDevice) {}

  acquireTexture(
      width: number, height: number, format: GPUTextureFormat,
      usage: GPUTextureUsageFlags) {
    const bytesPerElement = getBytesPerElement(format);
    const byteSize = width * height * bytesPerElement;
    const key = getTextureKey(width, height, format, usage);
    if (!this.freeTextures.has(key)) {
      this.freeTextures.set(key, []);
    }

    if (!this.usedTextures.has(key)) {
      this.usedTextures.set(key, []);
    }

    this.numBytesUsed += byteSize;
    this.numUsedTextures++;

    if (this.freeTextures.get(key).length > 0) {
      this.numFreeTextures--;

      const newTexture = this.freeTextures.get(key).shift();
      this.usedTextures.get(key).push(newTexture);
      return newTexture;
    }

    this.numBytesAllocated += byteSize;

    const newTexture = this.device.createTexture({
      size: [width, height],
      format,
      usage,
    });
    this.usedTextures.get(key).push(newTexture);

    return newTexture;
  }

  releaseTexture(texture: GPUTexture) {
    if (this.freeTextures.size === 0) {