Method releaseTexture

tfjs-backend-webgpu/src/texture_manager.ts:66–96

(texture: GPUTexture)

  }

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

    const width = texture.width;
    const height = texture.height;
    const format = texture.format;
    const usage = texture.usage;

    const key = getTextureKey(width, height, format, usage);
    if (!this.freeTextures.has(key)) {
      this.freeTextures.set(key, []);
    }

    this.freeTextures.get(key).push(texture);
    this.numFreeTextures++;
    this.numUsedTextures--;

    const textureList = this.usedTextures.get(key);
    const textureIndex = textureList.indexOf(texture);
    if (textureIndex < 0) {
      throw new Error(
          'Cannot release a texture that was never provided by this ' +
          'texture manager');
    }
    textureList.splice(textureIndex, 1);
    const bytesPerElement = getBytesPerElement(format);
    const byteSize = width * height * bytesPerElement;
    this.numBytesUsed -= byteSize;
  }

  getNumUsedTextures(): number {
    return this.numUsedTextures;