82,120c82,120 < │ │ │ │ │ └─ Transpose{axes=[1, 0]} < │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=True, check_finite=True, b_ndim=2} < │ │ │ │ │ ├─ Switch < │ │ │ │ │ │ ├─ ExpandDims{axes=[0, 1]} < │ │ │ │ │ │ │ └─ All{axes=None} < │ │ │ │ │ │ │ └─ Gt < │ │ │ │ │ │ │ ├─ ExtractDiag{offset=0, axis1=0, axis2=1, view=False} < │ │ │ │ │ │ │ │ └─ Cholesky{lower=True, destructive=False, on_error='nan'} < │ │ │ │ │ │ │ │ └─ AdvancedSetSubtensor < │ │ │ │ │ │ │ │ ├─ Alloc < │ │ │ │ │ │ │ │ │ ├─ 0.0 < │ │ │ │ │ │ │ │ │ ├─ 2 < │ │ │ │ │ │ │ │ │ └─ 2 < │ │ │ │ │ │ │ │ ├─ Subtensor{i} < │ │ │ │ │ │ │ │ │ ├─ Composite{...}.0 < │ │ │ │ │ │ │ │ │ │ └─ ··· < < │ │ │ │ │ │ │ │ │ └─ 0 < │ │ │ │ │ │ │ │ ├─ [0 1] < │ │ │ │ │ │ │ │ └─ [0 1] < │ │ │ │ │ │ │ └─ [0] < │ │ │ │ │ │ ├─ Cholesky{lower=True, destructive=False, on_error='nan'} < │ │ │ │ │ │ │ └─ ··· < < │ │ │ │ │ │ └─ [[1]] < │ │ │ │ │ └─ Sub < │ │ │ │ │ ├─ [[ 1.08827 ... 02443723]] < │ │ │ │ │ └─ ExpandDims{axis=1} < │ │ │ │ │ └─ Subtensor{i} < │ │ │ │ │ ├─ Transpose{axes=[1, 0]} < │ │ │ │ │ │ └─ Join < │ │ │ │ │ │ ├─ 0 < │ │ │ │ │ │ ├─ ExpandDims{axis=0} < │ │ │ │ │ │ │ └─ CumOp{-1, add} < │ │ │ │ │ │ │ └─ ··· < < │ │ │ │ │ │ └─ ExpandDims{axis=0} < │ │ │ │ │ │ └─ y_coord < │ │ │ │ │ └─ 0 --- > │ │ │ │ │ └─ Blockwise{SolveTriangular{trans=0, unit_diagonal=False, lower=True, check_finite=True, b_ndim=1}, (m,m),(m)->(m)} > │ │ │ │ │ ├─ Switch > │ │ │ │ │ │ ├─ ExpandDims{axes=[0, 1, 2]} > │ │ │ │ │ │ │ └─ All{axis=0} > │ │ │ │ │ │ │ └─ Gt > │ │ │ │ │ │ │ ├─ ExtractDiag{offset=0, axis1=0, axis2=1, view=False} > │ │ │ │ │ │ │ │ └─ Cholesky{lower=True, destructive=False, on_error='nan'} > │ │ │ │ │ │ │ │ └─ AdvancedSetSubtensor > │ │ │ │ │ │ │ │ ├─ Alloc > │ │ │ │ │ │ │ │ │ ├─ 0.0 > │ │ │ │ │ │ │ │ │ ├─ 2 > │ │ │ │ │ │ │ │ │ └─ 2 > │ │ │ │ │ │ │ │ ├─ Subtensor{i} > │ │ │ │ │ │ │ │ │ ├─ Composite{...}.0 > │ │ │ │ │ │ │ │ │ │ └─ ··· > > │ │ │ │ │ │ │ │ │ └─ 0 > │ │ │ │ │ │ │ │ ├─ [0 1] > │ │ │ │ │ │ │ │ └─ [0 1] > │ │ │ │ │ │ │ └─ [0] > │ │ │ │ │ │ ├─ ExpandDims{axis=0} > │ │ │ │ │ │ │ └─ Cholesky{lower=True, destructive=False, on_error='nan'} > │ │ │ │ │ │ │ └─ ··· > > │ │ │ │ │ │ └─ [[[1]]] > │ │ │ │ │ └─ Sub > │ │ │ │ │ ├─ y_hat{[[ 1.08827 ... 02443723]]} > │ │ │ │ │ └─ ExpandDims{axis=0} > │ │ │ │ │ └─ Subtensor{i} > │ │ │ │ │ ├─ Transpose{axes=[1, 0]} > │ │ │ │ │ │ └─ Join > │ │ │ │ │ │ ├─ 0 > │ │ │ │ │ │ ├─ ExpandDims{axis=0} > │ │ │ │ │ │ │ └─ CumOp{-1, add} > │ │ │ │ │ │ │ └─ ··· > > │ │ │ │ │ │ └─ ExpandDims{axis=0} > │ │ │ │ │ │ └─ y_coord > │ │ │ │ │ └─ 0 127c127,128 < │ │ │ │ └─ ··· --- > │ │ │ │ └─ All{axis=0} > │ │ │ │ └─ ··· 132,161c133,162 < │ │ │ │ │ └─ Transpose{axes=[1, 0]} < │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=True, check_finite=True, b_ndim=2} < │ │ │ │ │ ├─ Switch < │ │ │ │ │ │ ├─ ExpandDims{axes=[0, 1]} < │ │ │ │ │ │ │ └─ All{axes=None} < │ │ │ │ │ │ │ └─ Gt < │ │ │ │ │ │ │ ├─ ExtractDiag{offset=0, axis1=0, axis2=1, view=False} < │ │ │ │ │ │ │ │ └─ Cholesky{lower=True, destructive=False, on_error='nan'} < │ │ │ │ │ │ │ │ └─ AdvancedSetSubtensor < │ │ │ │ │ │ │ │ ├─ Alloc < │ │ │ │ │ │ │ │ │ └─ ··· < < │ │ │ │ │ │ │ │ ├─ Subtensor{i} < │ │ │ │ │ │ │ │ │ ├─ Composite{...}.0 < │ │ │ │ │ │ │ │ │ │ └─ ··· < < │ │ │ │ │ │ │ │ │ └─ 1 < │ │ │ │ │ │ │ │ ├─ [0 1] < │ │ │ │ │ │ │ │ └─ [0 1] < │ │ │ │ │ │ │ └─ [0] < │ │ │ │ │ │ ├─ Cholesky{lower=True, destructive=False, on_error='nan'} < │ │ │ │ │ │ │ └─ ··· < < │ │ │ │ │ │ └─ [[1]] < │ │ │ │ │ └─ Sub < │ │ │ │ │ ├─ [[ 1.08827 ... 02443723]] < │ │ │ │ │ └─ ExpandDims{axis=1} < │ │ │ │ │ └─ Subtensor{i} < │ │ │ │ │ ├─ Transpose{axes=[1, 0]} < │ │ │ │ │ │ └─ ··· --- > │ │ │ │ │ └─ Blockwise{SolveTriangular{trans=0, unit_diagonal=False, lower=True, check_finite=True, b_ndim=1}, (m,m),(m)->(m)} > │ │ │ │ │ ├─ Switch > │ │ │ │ │ │ ├─ ExpandDims{axes=[0, 1, 2]} > │ │ │ │ │ │ │ └─ All{axis=0} > │ │ │ │ │ │ │ └─ Gt > │ │ │ │ │ │ │ ├─ ExtractDiag{offset=0, axis1=0, axis2=1, view=False} > │ │ │ │ │ │ │ │ └─ Cholesky{lower=True, destructive=False, on_error='nan'} > │ │ │ │ │ │ │ │ └─ AdvancedSetSubtensor > │ │ │ │ │ │ │ │ ├─ Alloc > │ │ │ │ │ │ │ │ │ └─ ··· > > │ │ │ │ │ │ │ │ ├─ Subtensor{i} > │ │ │ │ │ │ │ │ │ ├─ Composite{...}.0 > │ │ │ │ │ │ │ │ │ │ └─ ··· > > │ │ │ │ │ │ │ │ │ └─ 1 > │ │ │ │ │ │ │ │ ├─ [0 1] > │ │ │ │ │ │ │ │ └─ [0 1] > │ │ │ │ │ │ │ └─ [0] > │ │ │ │ │ │ ├─ ExpandDims{axis=0} > │ │ │ │ │ │ │ └─ Cholesky{lower=True, destructive=False, on_error='nan'} > │ │ │ │ │ │ │ └─ ··· > > │ │ │ │ │ │ └─ [[[1]]] > │ │ │ │ │ └─ Sub > │ │ │ │ │ ├─ y_hat{[[ 1.08827 ... 02443723]]} > │ │ │ │ │ └─ ExpandDims{axis=0} > │ │ │ │ │ └─ Subtensor{i} > │ │ │ │ │ ├─ Transpose{axes=[1, 0]} > │ │ │ │ │ │ └─ ··· 163c164 < │ │ │ │ │ └─ 1 --- > │ │ │ │ │ └─ 1 170c171,172 < │ │ │ │ └─ ··· --- > │ │ │ │ └─ All{axis=0} > │ │ │ │ └─ ··· 175,204c177,206 < │ │ │ │ │ └─ Transpose{axes=[1, 0]} < │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=True, check_finite=True, b_ndim=2} < │ │ │ │ │ ├─ Switch < │ │ │ │ │ │ ├─ ExpandDims{axes=[0, 1]} < │ │ │ │ │ │ │ └─ All{axes=None} < │ │ │ │ │ │ │ └─ Gt < │ │ │ │ │ │ │ ├─ ExtractDiag{offset=0, axis1=0, axis2=1, view=False} < │ │ │ │ │ │ │ │ └─ Cholesky{lower=True, destructive=False, on_error='nan'} < │ │ │ │ │ │ │ │ └─ AdvancedSetSubtensor < │ │ │ │ │ │ │ │ ├─ Alloc < │ │ │ │ │ │ │ │ │ └─ ··· < < │ │ │ │ │ │ │ │ ├─ Subtensor{i} < │ │ │ │ │ │ │ │ │ ├─ Composite{...}.0 < │ │ │ │ │ │ │ │ │ │ └─ ··· < < │ │ │ │ │ │ │ │ │ └─ 2 < │ │ │ │ │ │ │ │ ├─ [0 1] < │ │ │ │ │ │ │ │ └─ [0 1] < │ │ │ │ │ │ │ └─ [0] < │ │ │ │ │ │ ├─ Cholesky{lower=True, destructive=False, on_error='nan'} < │ │ │ │ │ │ │ └─ ··· < < │ │ │ │ │ │ └─ [[1]] < │ │ │ │ │ └─ Sub < │ │ │ │ │ ├─ [[ 1.08827 ... 02443723]] < │ │ │ │ │ └─ ExpandDims{axis=1} < │ │ │ │ │ └─ Subtensor{i} < │ │ │ │ │ ├─ Transpose{axes=[1, 0]} < │ │ │ │ │ │ └─ ··· --- > │ │ │ │ │ └─ Blockwise{SolveTriangular{trans=0, unit_diagonal=False, lower=True, check_finite=True, b_ndim=1}, (m,m),(m)->(m)} > │ │ │ │ │ ├─ Switch > │ │ │ │ │ │ ├─ ExpandDims{axes=[0, 1, 2]} > │ │ │ │ │ │ │ └─ All{axis=0} > │ │ │ │ │ │ │ └─ Gt > │ │ │ │ │ │ │ ├─ ExtractDiag{offset=0, axis1=0, axis2=1, view=False} > │ │ │ │ │ │ │ │ └─ Cholesky{lower=True, destructive=False, on_error='nan'} > │ │ │ │ │ │ │ │ └─ AdvancedSetSubtensor > │ │ │ │ │ │ │ │ ├─ Alloc > │ │ │ │ │ │ │ │ │ └─ ··· > > │ │ │ │ │ │ │ │ ├─ Subtensor{i} > │ │ │ │ │ │ │ │ │ ├─ Composite{...}.0 > │ │ │ │ │ │ │ │ │ │ └─ ··· > > │ │ │ │ │ │ │ │ │ └─ 2 > │ │ │ │ │ │ │ │ ├─ [0 1] > │ │ │ │ │ │ │ │ └─ [0 1] > │ │ │ │ │ │ │ └─ [0] > │ │ │ │ │ │ ├─ ExpandDims{axis=0} > │ │ │ │ │ │ │ └─ Cholesky{lower=True, destructive=False, on_error='nan'} > │ │ │ │ │ │ │ └─ ··· > > │ │ │ │ │ │ └─ [[[1]]] > │ │ │ │ │ └─ Sub > │ │ │ │ │ ├─ y_hat{[[ 1.08827 ... 02443723]]} > │ │ │ │ │ └─ ExpandDims{axis=0} > │ │ │ │ │ └─ Subtensor{i} > │ │ │ │ │ ├─ Transpose{axes=[1, 0]} > │ │ │ │ │ │ └─ ··· 206c208 < │ │ │ │ │ └─ 2 --- > │ │ │ │ │ └─ 2 213c215,216 < │ │ │ │ └─ ··· --- > │ │ │ │ └─ All{axis=0} > │ │ │ │ └─ ··· 218,247c221,250 < │ │ │ │ │ └─ Transpose{axes=[1, 0]} < │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=True, check_finite=True, b_ndim=2} < │ │ │ │ │ ├─ Switch < │ │ │ │ │ │ ├─ ExpandDims{axes=[0, 1]} < │ │ │ │ │ │ │ └─ All{axes=None} < │ │ │ │ │ │ │ └─ Gt < │ │ │ │ │ │ │ ├─ ExtractDiag{offset=0, axis1=0, axis2=1, view=False} < │ │ │ │ │ │ │ │ └─ Cholesky{lower=True, destructive=False, on_error='nan'} < │ │ │ │ │ │ │ │ └─ AdvancedSetSubtensor < │ │ │ │ │ │ │ │ ├─ Alloc < │ │ │ │ │ │ │ │ │ └─ ··· < < │ │ │ │ │ │ │ │ ├─ Subtensor{i} < │ │ │ │ │ │ │ │ │ ├─ Composite{...}.0 < │ │ │ │ │ │ │ │ │ │ └─ ··· < < │ │ │ │ │ │ │ │ │ └─ 3 < │ │ │ │ │ │ │ │ ├─ [0 1] < │ │ │ │ │ │ │ │ └─ [0 1] < │ │ │ │ │ │ │ └─ [0] < │ │ │ │ │ │ ├─ Cholesky{lower=True, destructive=False, on_error='nan'} < │ │ │ │ │ │ │ └─ ··· < < │ │ │ │ │ │ └─ [[1]] < │ │ │ │ │ └─ Sub < │ │ │ │ │ ├─ [[ 1.08827 ... 02443723]] < │ │ │ │ │ └─ ExpandDims{axis=1} < │ │ │ │ │ └─ Subtensor{i} < │ │ │ │ │ ├─ Transpose{axes=[1, 0]} < │ │ │ │ │ │ └─ ··· --- > │ │ │ │ │ └─ Blockwise{SolveTriangular{trans=0, unit_diagonal=False, lower=True, check_finite=True, b_ndim=1}, (m,m),(m)->(m)} > │ │ │ │ │ ├─ Switch > │ │ │ │ │ │ ├─ ExpandDims{axes=[0, 1, 2]} > │ │ │ │ │ │ │ └─ All{axis=0} > │ │ │ │ │ │ │ └─ Gt > │ │ │ │ │ │ │ ├─ ExtractDiag{offset=0, axis1=0, axis2=1, view=False} > │ │ │ │ │ │ │ │ └─ Cholesky{lower=True, destructive=False, on_error='nan'} > │ │ │ │ │ │ │ │ └─ AdvancedSetSubtensor > │ │ │ │ │ │ │ │ ├─ Alloc > │ │ │ │ │ │ │ │ │ └─ ··· > > │ │ │ │ │ │ │ │ ├─ Subtensor{i} > │ │ │ │ │ │ │ │ │ ├─ Composite{...}.0 > │ │ │ │ │ │ │ │ │ │ └─ ··· > > │ │ │ │ │ │ │ │ │ └─ 3 > │ │ │ │ │ │ │ │ ├─ [0 1] > │ │ │ │ │ │ │ │ └─ [0 1] > │ │ │ │ │ │ │ └─ [0] > │ │ │ │ │ │ ├─ ExpandDims{axis=0} > │ │ │ │ │ │ │ └─ Cholesky{lower=True, destructive=False, on_error='nan'} > │ │ │ │ │ │ │ └─ ··· > > │ │ │ │ │ │ └─ [[[1]]] > │ │ │ │ │ └─ Sub > │ │ │ │ │ ├─ y_hat{[[ 1.08827 ... 02443723]]} > │ │ │ │ │ └─ ExpandDims{axis=0} > │ │ │ │ │ └─ Subtensor{i} > │ │ │ │ │ ├─ Transpose{axes=[1, 0]} > │ │ │ │ │ │ └─ ··· 249c252 < │ │ │ │ │ └─ 3 --- > │ │ │ │ │ └─ 3 256c259,260 < │ │ │ │ └─ ··· --- > │ │ │ │ └─ All{axis=0} > │ │ │ │ └─ ··· 261,290c265,294 < │ │ │ │ └─ Transpose{axes=[1, 0]} < │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=True, check_finite=True, b_ndim=2} < │ │ │ │ ├─ Switch < │ │ │ │ │ ├─ ExpandDims{axes=[0, 1]} < │ │ │ │ │ │ └─ All{axes=None} < │ │ │ │ │ │ └─ Gt < │ │ │ │ │ │ ├─ ExtractDiag{offset=0, axis1=0, axis2=1, view=False} < │ │ │ │ │ │ │ └─ Cholesky{lower=True, destructive=False, on_error='nan'} < │ │ │ │ │ │ │ └─ AdvancedSetSubtensor < │ │ │ │ │ │ │ ├─ Alloc < │ │ │ │ │ │ │ │ └─ ··· < < │ │ │ │ │ │ │ ├─ Subtensor{i} < │ │ │ │ │ │ │ │ ├─ Composite{...}.0 < │ │ │ │ │ │ │ │ │ └─ ··· < < │ │ │ │ │ │ │ │ └─ 4 < │ │ │ │ │ │ │ ├─ [0 1] < │ │ │ │ │ │ │ └─ [0 1] < │ │ │ │ │ │ └─ [0] < │ │ │ │ │ ├─ Cholesky{lower=True, destructive=False, on_error='nan'} < │ │ │ │ │ │ └─ ··· < < │ │ │ │ │ └─ [[1]] < │ │ │ │ └─ Sub < │ │ │ │ ├─ [[ 1.08827 ... 02443723]] < │ │ │ │ └─ ExpandDims{axis=1} < │ │ │ │ └─ Subtensor{i} < │ │ │ │ ├─ Transpose{axes=[1, 0]} < │ │ │ │ │ └─ ··· --- > │ │ │ │ └─ Blockwise{SolveTriangular{trans=0, unit_diagonal=False, lower=True, check_finite=True, b_ndim=1}, (m,m),(m)->(m)} > │ │ │ │ ├─ Switch > │ │ │ │ │ ├─ ExpandDims{axes=[0, 1, 2]} > │ │ │ │ │ │ └─ All{axis=0} > │ │ │ │ │ │ └─ Gt > │ │ │ │ │ │ ├─ ExtractDiag{offset=0, axis1=0, axis2=1, view=False} > │ │ │ │ │ │ │ └─ Cholesky{lower=True, destructive=False, on_error='nan'} > │ │ │ │ │ │ │ └─ AdvancedSetSubtensor > │ │ │ │ │ │ │ ├─ Alloc > │ │ │ │ │ │ │ │ └─ ··· > > │ │ │ │ │ │ │ ├─ Subtensor{i} > │ │ │ │ │ │ │ │ ├─ Composite{...}.0 > │ │ │ │ │ │ │ │ │ └─ ··· > > │ │ │ │ │ │ │ │ └─ 4 > │ │ │ │ │ │ │ ├─ [0 1] > │ │ │ │ │ │ │ └─ [0 1] > │ │ │ │ │ │ └─ [0] > │ │ │ │ │ ├─ ExpandDims{axis=0} > │ │ │ │ │ │ └─ Cholesky{lower=True, destructive=False, on_error='nan'} > │ │ │ │ │ │ └─ ··· > > │ │ │ │ │ └─ [[[1]]] > │ │ │ │ └─ Sub > │ │ │ │ ├─ y_hat{[[ 1.08827 ... 02443723]]} > │ │ │ │ └─ ExpandDims{axis=0} > │ │ │ │ └─ Subtensor{i} > │ │ │ │ ├─ Transpose{axes=[1, 0]} > │ │ │ │ │ └─ ··· 292c296 < │ │ │ │ └─ 4 --- > │ │ │ │ └─ 4 299c303,304 < │ │ │ └─ ··· --- > │ │ │ └─ All{axis=0} > │ │ │ └─ ··· 532,534c537,539 < │ │ │ │ │ │ │ │ │ │ │ │ │ │ └─ Sum{axis=1} < │ │ │ │ │ │ │ │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} < │ │ │ │ │ │ │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} --- > │ │ │ │ │ │ │ │ │ │ │ │ │ │ └─ Sum{axis=0} > │ │ │ │ │ │ │ │ │ │ │ │ │ │ └─ Blockwise{SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=1}, (m,m),(m)->(m)} > │ │ │ │ │ │ │ │ │ │ │ │ │ │ ├─ Transpose{axes=[0, 2, 1]} 540,550c545,554 < │ │ │ │ │ │ │ │ │ │ │ │ │ │ ├─ DimShuffle{order=[x,0]} < │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ └─ SpecifyShape < │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ ├─ Split{5}.0 < │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ ├─ Composite{((eq(i4, i5) * i3) + switch(i2, 0, (i0 / i1)))} < │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ └─ ··· < < │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ ├─ 1 < │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ └─ [1 1 1 1 1] < │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ ├─ NoneConst{None} < │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ └─ 1 < │ │ │ │ │ │ │ │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=True, check_finite=True, b_ndim=2} --- > │ │ │ │ │ │ │ │ │ │ │ │ │ │ ├─ SpecifyShape > │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ ├─ Split{5}.0 > │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ ├─ Composite{((eq(i4, i5) * i3) + switch(i2, 0, (i0 / i1)))} > │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ └─ ··· > > │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ ├─ 1 > │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ └─ [1 1 1 1 1] > │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ ├─ NoneConst{None} > │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ └─ 1 > │ │ │ │ │ │ │ │ │ │ │ │ │ │ └─ Blockwise{SolveTriangular{trans=0, unit_diagonal=False, lower=True, check_finite=True, b_ndim=1}, (m,m),(m)->(m)} 555,557c559,561 < │ │ │ │ │ │ │ │ │ │ │ │ │ └─ Sum{axis=1} < │ │ │ │ │ │ │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} < │ │ │ │ │ │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} --- > │ │ │ │ │ │ │ │ │ │ │ │ │ └─ Sum{axis=0} > │ │ │ │ │ │ │ │ │ │ │ │ │ └─ Blockwise{SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=1}, (m,m),(m)->(m)} > │ │ │ │ │ │ │ │ │ │ │ │ │ ├─ Transpose{axes=[0, 2, 1]} 563,570c567,573 < │ │ │ │ │ │ │ │ │ │ │ │ │ ├─ DimShuffle{order=[x,0]} < │ │ │ │ │ │ │ │ │ │ │ │ │ │ └─ SpecifyShape < │ │ │ │ │ │ │ │ │ │ │ │ │ │ ├─ Split{5}.1 < │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ └─ ··· < < │ │ │ │ │ │ │ │ │ │ │ │ │ │ ├─ NoneConst{None} < │ │ │ │ │ │ │ │ │ │ │ │ │ │ └─ 1 < │ │ │ │ │ │ │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=True, check_finite=True, b_ndim=2} --- > │ │ │ │ │ │ │ │ │ │ │ │ │ ├─ SpecifyShape > │ │ │ │ │ │ │ │ │ │ │ │ │ │ ├─ Split{5}.1 > │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ └─ ··· > > │ │ │ │ │ │ │ │ │ │ │ │ │ │ ├─ NoneConst{None} > │ │ │ │ │ │ │ │ │ │ │ │ │ │ └─ 1 > │ │ │ │ │ │ │ │ │ │ │ │ │ └─ Blockwise{SolveTriangular{trans=0, unit_diagonal=False, lower=True, check_finite=True, b_ndim=1}, (m,m),(m)->(m)} 575,577c578,580 < │ │ │ │ │ │ │ │ │ │ │ │ └─ Sum{axis=1} < │ │ │ │ │ │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} < │ │ │ │ │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} --- > │ │ │ │ │ │ │ │ │ │ │ │ └─ Sum{axis=0} > │ │ │ │ │ │ │ │ │ │ │ │ └─ Blockwise{SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=1}, (m,m),(m)->(m)} > │ │ │ │ │ │ │ │ │ │ │ │ ├─ Transpose{axes=[0, 2, 1]} 583,590c586,592 < │ │ │ │ │ │ │ │ │ │ │ │ ├─ DimShuffle{order=[x,0]} < │ │ │ │ │ │ │ │ │ │ │ │ │ └─ SpecifyShape < │ │ │ │ │ │ │ │ │ │ │ │ │ ├─ Split{5}.2 < │ │ │ │ │ │ │ │ │ │ │ │ │ │ └─ ··· < < │ │ │ │ │ │ │ │ │ │ │ │ │ ├─ NoneConst{None} < │ │ │ │ │ │ │ │ │ │ │ │ │ └─ 1 < │ │ │ │ │ │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=True, check_finite=True, b_ndim=2} --- > │ │ │ │ │ │ │ │ │ │ │ │ ├─ SpecifyShape > │ │ │ │ │ │ │ │ │ │ │ │ │ ├─ Split{5}.2 > │ │ │ │ │ │ │ │ │ │ │ │ │ │ └─ ··· > > │ │ │ │ │ │ │ │ │ │ │ │ │ ├─ NoneConst{None} > │ │ │ │ │ │ │ │ │ │ │ │ │ └─ 1 > │ │ │ │ │ │ │ │ │ │ │ │ └─ Blockwise{SolveTriangular{trans=0, unit_diagonal=False, lower=True, check_finite=True, b_ndim=1}, (m,m),(m)->(m)} 595,597c597,599 < │ │ │ │ │ │ │ │ │ │ │ └─ Sum{axis=1} < │ │ │ │ │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} < │ │ │ │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} --- > │ │ │ │ │ │ │ │ │ │ │ └─ Sum{axis=0} > │ │ │ │ │ │ │ │ │ │ │ └─ Blockwise{SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=1}, (m,m),(m)->(m)} > │ │ │ │ │ │ │ │ │ │ │ ├─ Transpose{axes=[0, 2, 1]} 603,610c605,611 < │ │ │ │ │ │ │ │ │ │ │ ├─ DimShuffle{order=[x,0]} < │ │ │ │ │ │ │ │ │ │ │ │ └─ SpecifyShape < │ │ │ │ │ │ │ │ │ │ │ │ ├─ Split{5}.3 < │ │ │ │ │ │ │ │ │ │ │ │ │ └─ ··· < < │ │ │ │ │ │ │ │ │ │ │ │ ├─ NoneConst{None} < │ │ │ │ │ │ │ │ │ │ │ │ └─ 1 < │ │ │ │ │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=True, check_finite=True, b_ndim=2} --- > │ │ │ │ │ │ │ │ │ │ │ ├─ SpecifyShape > │ │ │ │ │ │ │ │ │ │ │ │ ├─ Split{5}.3 > │ │ │ │ │ │ │ │ │ │ │ │ │ └─ ··· > > │ │ │ │ │ │ │ │ │ │ │ │ ├─ NoneConst{None} > │ │ │ │ │ │ │ │ │ │ │ │ └─ 1 > │ │ │ │ │ │ │ │ │ │ │ └─ Blockwise{SolveTriangular{trans=0, unit_diagonal=False, lower=True, check_finite=True, b_ndim=1}, (m,m),(m)->(m)} 615,617c616,618 < │ │ │ │ │ │ │ │ │ │ └─ Sum{axis=1} < │ │ │ │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} < │ │ │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} --- > │ │ │ │ │ │ │ │ │ │ └─ Sum{axis=0} > │ │ │ │ │ │ │ │ │ │ └─ Blockwise{SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=1}, (m,m),(m)->(m)} > │ │ │ │ │ │ │ │ │ │ ├─ Transpose{axes=[0, 2, 1]} 623,630c624,630 < │ │ │ │ │ │ │ │ │ │ ├─ DimShuffle{order=[x,0]} < │ │ │ │ │ │ │ │ │ │ │ └─ SpecifyShape < │ │ │ │ │ │ │ │ │ │ │ ├─ Split{5}.4 < │ │ │ │ │ │ │ │ │ │ │ │ └─ ··· < < │ │ │ │ │ │ │ │ │ │ │ ├─ NoneConst{None} < │ │ │ │ │ │ │ │ │ │ │ └─ 1 < │ │ │ │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=True, check_finite=True, b_ndim=2} --- > │ │ │ │ │ │ │ │ │ │ ├─ SpecifyShape > │ │ │ │ │ │ │ │ │ │ │ ├─ Split{5}.4 > │ │ │ │ │ │ │ │ │ │ │ │ └─ ··· > > │ │ │ │ │ │ │ │ │ │ │ ├─ NoneConst{None} > │ │ │ │ │ │ │ │ │ │ │ └─ 1 > │ │ │ │ │ │ │ │ │ │ └─ Blockwise{SolveTriangular{trans=0, unit_diagonal=False, lower=True, check_finite=True, b_ndim=1}, (m,m),(m)->(m)} 709,731c709,733 < │ │ │ │ │ │ │ │ │ │ └─ Composite{switch(i4, (i3 + switch(i2, (-1.0 * i0 * i1), 0.0)), 1)} < │ │ │ │ │ │ │ │ │ │ ├─ dot < │ │ │ │ │ │ │ │ │ │ │ ├─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} < │ │ │ │ │ │ │ │ │ │ │ │ └─ ··· < < │ │ │ │ │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} < │ │ │ │ │ │ │ │ │ │ │ └─ ··· < < │ │ │ │ │ │ │ │ │ │ ├─ Tri{dtype='float64'} < │ │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} < │ │ │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{start:} < │ │ │ │ │ │ │ │ │ │ │ │ │ ├─ Shape < │ │ │ │ │ │ │ │ │ │ │ │ │ │ └─ Neg < │ │ │ │ │ │ │ │ │ │ │ │ │ │ └─ dot < │ │ │ │ │ │ │ │ │ │ │ │ │ │ └─ ··· < < │ │ │ │ │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor < │ │ │ │ │ │ │ │ │ │ │ │ │ └─ -2 < │ │ │ │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor < │ │ │ │ │ │ │ │ │ │ │ │ └─ 0 < │ │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} < │ │ │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{start:} < │ │ │ │ │ │ │ │ │ │ │ │ │ └─ ··· --- > │ │ │ │ │ │ │ │ │ │ └─ Composite{switch(i3, (i2 + switch(i1, (-i0), 0.0)), 1)} > │ │ │ │ │ │ │ │ │ │ ├─ Sum{axis=0} > │ │ │ │ │ │ │ │ │ │ │ └─ Mul > │ │ │ │ │ │ │ │ │ │ │ ├─ Blockwise{dot, (i00,i01),(i10,i11)->(o00,o01)} > │ │ │ │ │ │ │ │ │ │ │ │ ├─ ExpandDims{axis=2} > │ │ │ │ │ │ │ │ │ │ │ │ │ └─ Blockwise{SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=1}, (m,m),(m)->(m)} > │ │ │ │ │ │ │ │ │ │ │ │ │ └─ ··· > > │ │ │ │ │ │ │ │ │ │ │ │ └─ ExpandDims{axis=1} > │ │ │ │ │ │ │ │ │ │ │ │ └─ Blockwise{SolveTriangular{trans=0, unit_diagonal=False, lower=True, check_finite=True, b_ndim=1}, (m,m),(m)->(m)} > │ │ │ │ │ │ │ │ │ │ │ │ └─ ··· > > │ │ │ │ │ │ │ │ │ │ │ └─ Blockwise{Tri{dtype='float64'}, (),(),()->(o00,o01)} > │ │ │ │ │ │ │ │ │ │ │ ├─ Blockwise{Subtensor{i}, (i00),()->()} > │ │ │ │ │ │ │ │ │ │ │ │ ├─ Blockwise{Subtensor{start:}, (i00),()->(o00)} > │ │ │ │ │ │ │ │ │ │ │ │ │ ├─ Blockwise{Shape, (i00,i01)->(o00)} > │ │ │ │ │ │ │ │ │ │ │ │ │ │ └─ Neg > │ │ │ │ │ │ │ │ │ │ │ │ │ │ └─ Blockwise{dot, (i00,i01),(i10,i11)->(o00,o01)} > │ │ │ │ │ │ │ │ │ │ │ │ │ │ └─ ··· > > │ │ │ │ │ │ │ │ │ │ │ │ │ └─ [-2] > │ │ │ │ │ │ │ │ │ │ │ │ └─ [0] > │ │ │ │ │ │ │ │ │ │ │ ├─ Blockwise{Subtensor{i}, (i00),()->()} > │ │ │ │ │ │ │ │ │ │ │ │ ├─ Blockwise{Subtensor{start:}, (i00),()->(o00)} > │ │ │ │ │ │ │ │ │ │ │ │ │ └─ ··· 733,735c735,736 < │ │ │ │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor < │ │ │ │ │ │ │ │ │ │ │ │ └─ 1 < │ │ │ │ │ │ │ │ │ │ │ └─ 0 --- > │ │ │ │ │ │ │ │ │ │ │ │ └─ [1] > │ │ │ │ │ │ │ │ │ │ │ └─ [0] 776,777c777 < │ │ │ │ │ │ │ │ │ │ │ │ └─ ··· < --- > │ │ │ │ │ │ │ │ │ │ │ │ └─ -2 779,780c779 < │ │ │ │ │ │ │ │ │ │ │ └─ ··· < --- > │ │ │ │ │ │ │ │ │ │ │ └─ 0 786,787c785 < │ │ │ │ │ │ │ │ │ │ │ └─ ··· < --- > │ │ │ │ │ │ │ │ │ │ │ └─ 1 904,918c902,926 < │ │ │ │ │ │ │ │ └─ Composite{switch(i4, (i3 + switch(i2, (-1.0 * i0 * i1), 0.0)), 1)} < │ │ │ │ │ │ │ │ ├─ dot < │ │ │ │ │ │ │ │ │ ├─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} < │ │ │ │ │ │ │ │ │ │ └─ ··· < < │ │ │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} < │ │ │ │ │ │ │ │ │ └─ ··· < < │ │ │ │ │ │ │ │ ├─ Tri{dtype='float64'} < │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} < │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{start:} < │ │ │ │ │ │ │ │ │ │ │ ├─ Shape < │ │ │ │ │ │ │ │ │ │ │ │ └─ Neg < │ │ │ │ │ │ │ │ │ │ │ │ └─ dot < │ │ │ │ │ │ │ │ │ │ │ │ └─ ··· --- > │ │ │ │ │ │ │ │ └─ Composite{switch(i3, (i2 + switch(i1, (-i0), 0.0)), 1)} > │ │ │ │ │ │ │ │ ├─ Sum{axis=0} > │ │ │ │ │ │ │ │ │ └─ Mul > │ │ │ │ │ │ │ │ │ ├─ Blockwise{dot, (i00,i01),(i10,i11)->(o00,o01)} > │ │ │ │ │ │ │ │ │ │ ├─ ExpandDims{axis=2} > │ │ │ │ │ │ │ │ │ │ │ └─ Blockwise{SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=1}, (m,m),(m)->(m)} > │ │ │ │ │ │ │ │ │ │ │ └─ ··· > > │ │ │ │ │ │ │ │ │ │ └─ ExpandDims{axis=1} > │ │ │ │ │ │ │ │ │ │ └─ Blockwise{SolveTriangular{trans=0, unit_diagonal=False, lower=True, check_finite=True, b_ndim=1}, (m,m),(m)->(m)} > │ │ │ │ │ │ │ │ │ │ └─ ··· > > │ │ │ │ │ │ │ │ │ └─ Blockwise{Tri{dtype='float64'}, (),(),()->(o00,o01)} > │ │ │ │ │ │ │ │ │ ├─ Blockwise{Subtensor{i}, (i00),()->()} > │ │ │ │ │ │ │ │ │ │ ├─ Blockwise{Subtensor{start:}, (i00),()->(o00)} > │ │ │ │ │ │ │ │ │ │ │ ├─ Blockwise{Shape, (i00,i01)->(o00)} > │ │ │ │ │ │ │ │ │ │ │ │ └─ Neg > │ │ │ │ │ │ │ │ │ │ │ │ └─ Blockwise{dot, (i00,i01),(i10,i11)->(o00,o01)} > │ │ │ │ │ │ │ │ │ │ │ │ └─ ··· > > │ │ │ │ │ │ │ │ │ │ │ └─ [-2] > │ │ │ │ │ │ │ │ │ │ └─ [0] > │ │ │ │ │ │ │ │ │ ├─ Blockwise{Subtensor{i}, (i00),()->()} > │ │ │ │ │ │ │ │ │ │ ├─ Blockwise{Subtensor{start:}, (i00),()->(o00)} > │ │ │ │ │ │ │ │ │ │ │ └─ ··· 920,933c928,929 < │ │ │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor < │ │ │ │ │ │ │ │ │ │ │ └─ ··· < < │ │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor < │ │ │ │ │ │ │ │ │ │ └─ ··· < < │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} < │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{start:} < │ │ │ │ │ │ │ │ │ │ │ └─ ··· < < │ │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor < │ │ │ │ │ │ │ │ │ │ └─ ··· < < │ │ │ │ │ │ │ │ │ └─ 0 --- > │ │ │ │ │ │ │ │ │ │ └─ [1] > │ │ │ │ │ │ │ │ │ └─ [0] 1364,1391c1360,1384 < │ │ │ │ │ │ │ └─ Composite{switch(i4, (i3 + switch(i2, (-1.0 * i0 * i1), 0.0)), 1)} < │ │ │ │ │ │ │ ├─ dot < │ │ │ │ │ │ │ │ ├─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} < │ │ │ │ │ │ │ │ │ └─ ··· < < │ │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} < │ │ │ │ │ │ │ │ └─ ··· < < │ │ │ │ │ │ │ ├─ Tri{dtype='float64'} < │ │ │ │ │ │ │ │ ├─ Subtensor{i} < │ │ │ │ │ │ │ │ │ ├─ Subtensor{start:} < │ │ │ │ │ │ │ │ │ │ ├─ Shape < │ │ │ │ │ │ │ │ │ │ │ └─ Neg < │ │ │ │ │ │ │ │ │ │ │ └─ dot < │ │ │ │ │ │ │ │ │ │ │ └─ ··· < < │ │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor < │ │ │ │ │ │ │ │ │ │ └─ ··· < < │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor < │ │ │ │ │ │ │ │ │ └─ ··· < < │ │ │ │ │ │ │ │ ├─ Subtensor{i} < │ │ │ │ │ │ │ │ │ ├─ Subtensor{start:} < │ │ │ │ │ │ │ │ │ │ └─ ··· < < │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor < │ │ │ │ │ │ │ │ │ └─ ··· --- > │ │ │ │ │ │ │ └─ Composite{switch(i3, (i2 + switch(i1, (-i0), 0.0)), 1)} > │ │ │ │ │ │ │ ├─ Sum{axis=0} > │ │ │ │ │ │ │ │ └─ Mul > │ │ │ │ │ │ │ │ ├─ Blockwise{dot, (i00,i01),(i10,i11)->(o00,o01)} > │ │ │ │ │ │ │ │ │ ├─ ExpandDims{axis=2} > │ │ │ │ │ │ │ │ │ │ └─ Blockwise{SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=1}, (m,m),(m)->(m)} > │ │ │ │ │ │ │ │ │ │ └─ ··· > > │ │ │ │ │ │ │ │ │ └─ ExpandDims{axis=1} > │ │ │ │ │ │ │ │ │ └─ Blockwise{SolveTriangular{trans=0, unit_diagonal=False, lower=True, check_finite=True, b_ndim=1}, (m,m),(m)->(m)} > │ │ │ │ │ │ │ │ │ └─ ··· > > │ │ │ │ │ │ │ │ └─ Blockwise{Tri{dtype='float64'}, (),(),()->(o00,o01)} > │ │ │ │ │ │ │ │ ├─ Blockwise{Subtensor{i}, (i00),()->()} > │ │ │ │ │ │ │ │ │ ├─ Blockwise{Subtensor{start:}, (i00),()->(o00)} > │ │ │ │ │ │ │ │ │ │ ├─ Blockwise{Shape, (i00,i01)->(o00)} > │ │ │ │ │ │ │ │ │ │ │ └─ Neg > │ │ │ │ │ │ │ │ │ │ │ └─ Blockwise{dot, (i00,i01),(i10,i11)->(o00,o01)} > │ │ │ │ │ │ │ │ │ │ │ └─ ··· > > │ │ │ │ │ │ │ │ │ │ └─ [-2] > │ │ │ │ │ │ │ │ │ └─ [0] > │ │ │ │ │ │ │ │ ├─ Blockwise{Subtensor{i}, (i00),()->()} > │ │ │ │ │ │ │ │ │ ├─ Blockwise{Subtensor{start:}, (i00),()->(o00)} > │ │ │ │ │ │ │ │ │ │ └─ ··· 1393c1386,1387 < │ │ │ │ │ │ │ │ └─ 0 --- > │ │ │ │ │ │ │ │ │ └─ [1] > │ │ │ │ │ │ │ │ └─ [0] 1824,1844c1818,1842 < │ │ │ │ │ │ └─ Composite{switch(i4, (i3 + switch(i2, (-1.0 * i0 * i1), 0.0)), 1)} < │ │ │ │ │ │ ├─ dot < │ │ │ │ │ │ │ ├─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} < │ │ │ │ │ │ │ │ └─ ··· < < │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} < │ │ │ │ │ │ │ └─ ··· < < │ │ │ │ │ │ ├─ Tri{dtype='float64'} < │ │ │ │ │ │ │ ├─ Subtensor{i} < │ │ │ │ │ │ │ │ ├─ Subtensor{start:} < │ │ │ │ │ │ │ │ │ ├─ Shape < │ │ │ │ │ │ │ │ │ │ └─ Neg < │ │ │ │ │ │ │ │ │ │ └─ dot < │ │ │ │ │ │ │ │ │ │ └─ ··· < < │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor < │ │ │ │ │ │ │ │ │ └─ ··· < < │ │ │ │ │ │ │ │ └─ ScalarFromTensor < │ │ │ │ │ │ │ │ └─ ··· --- > │ │ │ │ │ │ └─ Composite{switch(i3, (i2 + switch(i1, (-i0), 0.0)), 1)} > │ │ │ │ │ │ ├─ Sum{axis=0} > │ │ │ │ │ │ │ └─ Mul > │ │ │ │ │ │ │ ├─ Blockwise{dot, (i00,i01),(i10,i11)->(o00,o01)} > │ │ │ │ │ │ │ │ ├─ ExpandDims{axis=2} > │ │ │ │ │ │ │ │ │ └─ Blockwise{SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=1}, (m,m),(m)->(m)} > │ │ │ │ │ │ │ │ │ └─ ··· > > │ │ │ │ │ │ │ │ └─ ExpandDims{axis=1} > │ │ │ │ │ │ │ │ └─ Blockwise{SolveTriangular{trans=0, unit_diagonal=False, lower=True, check_finite=True, b_ndim=1}, (m,m),(m)->(m)} > │ │ │ │ │ │ │ │ └─ ··· > > │ │ │ │ │ │ │ └─ Blockwise{Tri{dtype='float64'}, (),(),()->(o00,o01)} > │ │ │ │ │ │ │ ├─ Blockwise{Subtensor{i}, (i00),()->()} > │ │ │ │ │ │ │ │ ├─ Blockwise{Subtensor{start:}, (i00),()->(o00)} > │ │ │ │ │ │ │ │ │ ├─ Blockwise{Shape, (i00,i01)->(o00)} > │ │ │ │ │ │ │ │ │ │ └─ Neg > │ │ │ │ │ │ │ │ │ │ └─ Blockwise{dot, (i00,i01),(i10,i11)->(o00,o01)} > │ │ │ │ │ │ │ │ │ │ └─ ··· > > │ │ │ │ │ │ │ │ │ └─ [-2] > │ │ │ │ │ │ │ │ └─ [0] > │ │ │ │ │ │ │ ├─ Blockwise{Subtensor{i}, (i00),()->()} > │ │ │ │ │ │ │ │ ├─ Blockwise{Subtensor{start:}, (i00),()->(o00)} > │ │ │ │ │ │ │ │ │ └─ ··· 1846,1853c1844,1845 < │ │ │ │ │ │ │ ├─ Subtensor{i} < │ │ │ │ │ │ │ │ ├─ Subtensor{start:} < │ │ │ │ │ │ │ │ │ └─ ··· < < │ │ │ │ │ │ │ │ └─ ScalarFromTensor < │ │ │ │ │ │ │ │ └─ ··· < < │ │ │ │ │ │ │ └─ 0 --- > │ │ │ │ │ │ │ │ └─ [1] > │ │ │ │ │ │ │ └─ [0] 2284,2308c2276,2300 < │ │ │ │ │ └─ Composite{switch(i4, (i3 + switch(i2, (-1.0 * i0 * i1), 0.0)), 1)} < │ │ │ │ │ ├─ dot < │ │ │ │ │ │ ├─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} < │ │ │ │ │ │ │ └─ ··· < < │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} < │ │ │ │ │ │ └─ ··· < < │ │ │ │ │ ├─ Tri{dtype='float64'} < │ │ │ │ │ │ ├─ Subtensor{i} < │ │ │ │ │ │ │ ├─ Subtensor{start:} < │ │ │ │ │ │ │ │ ├─ Shape < │ │ │ │ │ │ │ │ │ └─ Neg < │ │ │ │ │ │ │ │ │ └─ dot < │ │ │ │ │ │ │ │ │ └─ ··· < < │ │ │ │ │ │ │ │ └─ ScalarFromTensor < │ │ │ │ │ │ │ │ └─ ··· < < │ │ │ │ │ │ │ └─ ScalarFromTensor < │ │ │ │ │ │ │ └─ ··· < < │ │ │ │ │ │ ├─ Subtensor{i} < │ │ │ │ │ │ │ ├─ Subtensor{start:} < │ │ │ │ │ │ │ │ └─ ··· --- > │ │ │ │ │ └─ Composite{switch(i3, (i2 + switch(i1, (-i0), 0.0)), 1)} > │ │ │ │ │ ├─ Sum{axis=0} > │ │ │ │ │ │ └─ Mul > │ │ │ │ │ │ ├─ Blockwise{dot, (i00,i01),(i10,i11)->(o00,o01)} > │ │ │ │ │ │ │ ├─ ExpandDims{axis=2} > │ │ │ │ │ │ │ │ └─ Blockwise{SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=1}, (m,m),(m)->(m)} > │ │ │ │ │ │ │ │ └─ ··· > > │ │ │ │ │ │ │ └─ ExpandDims{axis=1} > │ │ │ │ │ │ │ └─ Blockwise{SolveTriangular{trans=0, unit_diagonal=False, lower=True, check_finite=True, b_ndim=1}, (m,m),(m)->(m)} > │ │ │ │ │ │ │ └─ ··· > > │ │ │ │ │ │ └─ Blockwise{Tri{dtype='float64'}, (),(),()->(o00,o01)} > │ │ │ │ │ │ ├─ Blockwise{Subtensor{i}, (i00),()->()} > │ │ │ │ │ │ │ ├─ Blockwise{Subtensor{start:}, (i00),()->(o00)} > │ │ │ │ │ │ │ │ ├─ Blockwise{Shape, (i00,i01)->(o00)} > │ │ │ │ │ │ │ │ │ └─ Neg > │ │ │ │ │ │ │ │ │ └─ Blockwise{dot, (i00,i01),(i10,i11)->(o00,o01)} > │ │ │ │ │ │ │ │ │ └─ ··· > > │ │ │ │ │ │ │ │ └─ [-2] > │ │ │ │ │ │ │ └─ [0] > │ │ │ │ │ │ ├─ Blockwise{Subtensor{i}, (i00),()->()} > │ │ │ │ │ │ │ ├─ Blockwise{Subtensor{start:}, (i00),()->(o00)} > │ │ │ │ │ │ │ │ └─ ··· 2310,2313c2302,2303 < │ │ │ │ │ │ │ └─ ScalarFromTensor < │ │ │ │ │ │ │ └─ ··· < < │ │ │ │ │ │ └─ 0 --- > │ │ │ │ │ │ │ └─ [1] > │ │ │ │ │ │ └─ [0] 3152c3142 < Composite{switch(i4, (i3 + switch(i2, (-1.0 * i0 * i1), 0.0)), 1)} --- > Composite{switch(i3, (i2 + switch(i1, (-i0), 0.0)), 1)} 3154c3144 < ├─ i4 --- > ├─ i3 3156c3146 < │ ├─ i3 --- > │ ├─ i2 3158,3162c3148,3150 < │ ├─ i2 < │ ├─ mul < │ │ ├─ -1.0 < │ │ ├─ i0 < │ │ └─ i1 --- > │ ├─ i1 > │ ├─ neg > │ │ └─ i0 3188c3176 < Composite{switch(i4, (i3 + switch(i2, (-1.0 * i0 * i1), 0.0)), 1)} --- > Composite{switch(i3, (i2 + switch(i1, (-i0), 0.0)), 1)} 3190c3178 < ├─ i4 --- > ├─ i3 3192c3180 < │ ├─ i3 --- > │ ├─ i2 3194,3198c3182,3184 < │ ├─ i2 < │ ├─ mul < │ │ ├─ -1.0 < │ │ ├─ i0 < │ │ └─ i1 --- > │ ├─ i1 > │ ├─ neg > │ │ └─ i0 3296c3282 < Composite{switch(i4, (i3 + switch(i2, (-1.0 * i0 * i1), 0.0)), 1)} --- > Composite{switch(i3, (i2 + switch(i1, (-i0), 0.0)), 1)} 3298c3284 < ├─ i4 --- > ├─ i3 3300c3286 < │ ├─ i3 --- > │ ├─ i2 3302,3306c3288,3290 < │ ├─ i2 < │ ├─ mul < │ │ ├─ -1.0 < │ │ ├─ i0 < │ │ └─ i1 --- > │ ├─ i1 > │ ├─ neg > │ │ └─ i0 3404c3388 < Composite{switch(i4, (i3 + switch(i2, (-1.0 * i0 * i1), 0.0)), 1)} --- > Composite{switch(i3, (i2 + switch(i1, (-i0), 0.0)), 1)} 3406c3390 < ├─ i4 --- > ├─ i3 3408c3392 < │ ├─ i3 --- > │ ├─ i2 3410,3414c3394,3396 < │ ├─ i2 < │ ├─ mul < │ │ ├─ -1.0 < │ │ ├─ i0 < │ │ └─ i1 --- > │ ├─ i1 > │ ├─ neg > │ │ └─ i0 3512c3494 < Composite{switch(i4, (i3 + switch(i2, (-1.0 * i0 * i1), 0.0)), 1)} --- > Composite{switch(i3, (i2 + switch(i1, (-i0), 0.0)), 1)} 3514c3496 < ├─ i4 --- > ├─ i3 3516c3498 < │ ├─ i3 --- > │ ├─ i2 3518,3522c3500,3502 < │ ├─ i2 < │ ├─ mul < │ │ ├─ -1.0 < │ │ ├─ i0 < │ │ └─ i1 --- > │ ├─ i1 > │ ├─ neg > │ │ └─ i0