GPUNode constructor

GPUNode(
  1. List<GPUTensor> inputs,
  2. void backwardFn(
    1. CommandBuffer
    ), {
  3. String opName = 'op',
  4. int cost = 0,
  5. Map<String, dynamic> extraParams = const {},
})

Implementation

GPUNode(
    this.inputs,
    this.backwardFn, {
      this.opName = 'op',
      this.cost = 0,
      this.extraParams = const {},
    });