tensor/tensor_cpu_gpu library

Functions

encapsulateGPUGraph<T>(List<Tensor> cpuDynamicInputs, List<GPUTensor> gpuDynamicInputs, List<Tensor> cpuStaticParams, List<GPUTensor> gpuStaticParams, GPUTensor gpuOutput, T initialOutputValue, Uint8List forwardTape, Uint8List backwardTape) → Tensor<T>

executeGPUGraph<T>(List<Tensor> cpuInputs, List<GPUTensor> gpuInputs, GPUTensor gpuOutput, CommandBuffer forwardTape, CommandBuffer backwardTape, {String opName = 'gpu_subgraph'}) → Tensor<T>

Wraps an entire compiled GPU tape into a single CPU Tensor Autograd node.