type Node

bendtorch-core@0.1.0.0 · src/autograd/tensor_backward.bend:82

import Base
import bendtorch-core@0.1.0.0/src/shape/core.bend as Shape
import bendtorch-core@0.1.0.0/src/tensor/slice.bend as Slice
import bendtorch-core@0.1.0.0/src/tensor/gather.bend as Gather

82type Node is Data:
83  Leaf{}
84  Add{a: Nat, b: Nat, left: Shape.Shape, right: Shape.Shape}
85  Sub{a: Nat, b: Nat, left: Shape.Shape, right: Shape.Shape}
86  Mul{a: Nat, b: Nat, left: Shape.Shape, right: Shape.Shape}
87  Sum{a: Nat, input: Shape.Shape}
88  SumDims{a: Nat, input: Shape.Shape, kept: Shape.Shape}
89  Matmul{a: Nat, b: Nat, left: Shape.Shape, right: Shape.Shape}
90  MatmulNT{a: Nat, b: Nat, left: Shape.Shape, right: Shape.Shape}
91  Relu{a: Nat, input: Shape.Shape}
92  MseLoss{a: Nat, b: Nat, left: Shape.Shape, right: Shape.Shape}
93  CrossEntropy{a: Nat, input: Shape.Shape}
94  BatchedMatmul{a: Nat, b: Nat, left: Shape.Shape, right: Shape.Shape}
95  Slice{a: Nat, input: Shape.Shape, kept: Shape.Shape, plan: Slice.Plan}
96  Gathered{a: Nat, input: Shape.Shape, plan: Gather.Plan, entries: List<&2, U32>}
97  Constant{}
98  Frozen{id: Nat, node: Node}