MCPcopy Index your code
hub / github.com/tensorflow/tfjs / releaseResource

Method releaseResource

tfjs-backend-webgpu/src/backend_webgpu.ts:224–241  ·  view source on GitHub ↗
(dataId: DataId)

Source from the content-addressed store, hash-verified

222 }
223
224 private releaseResource(dataId: DataId) {
225 const tensorData = this.tensorMap.get(dataId);
226 if (!tensorData || !tensorData.resource) {
227 return;
228 }
229
230 // If tensor's resource is from external, do not release.
231 if (tensorData.external) {
232 tensorData.resource = null;
233 return;
234 }
235 if (tensorData.resource instanceof GPUBuffer) {
236 this.bufferManager.releaseBuffer(tensorData.resource);
237 } else if (tensorData.resource instanceof GPUTexture) {
238 this.textureManager.releaseTexture(tensorData.resource);
239 }
240 tensorData.resource = null;
241 }
242
243 /** Return refCount of a `TensorData`. */
244 override refCount(dataId: DataId): number {

Callers 2

disposeDataMethod · 0.95
submitQueueMethod · 0.95

Calls 3

releaseBufferMethod · 0.80
getMethod · 0.45
releaseTextureMethod · 0.45

Tested by

no test coverage detected