BackwardBuffers (DeepJ 0.13.0-alpha API)

java.lang.Object

io.github.kirstenali.deepj.tensor.BackwardBuffers

All Implemented Interfaces:: AutoCloseable

public final class BackwardBuffers extends Object implements AutoCloseable

Internal lifetime tracking for non-escaping buffers produced by library backward rules.

Method Summary

Modifier and Type

Method

Description

static BackwardBuffers

begin()

<T> T

capture(Supplier<T> operation)

void

close()

void

expose(Tensor tensor)

void

releaseTemporaryResources()

Releases this private computation's scratch without invalidating unrelated caller tensors.

Methods inherited from class java.lang.Object
clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Method Details
- begin
  
  public static BackwardBuffers begin()
- capture
  
  public <T> T capture(Supplier<T> operation)
- expose
  
  public void expose(Tensor tensor)
- releaseTemporaryResources
  
  public void releaseTemporaryResources()
  
  Releases this private computation's scratch without invalidating unrelated caller tensors.
- close
  
  public void close()
  
  Specified by:
  
  close in interface AutoCloseable