1,3c1,3 < Composite{...}.0 [id A] 258 < ├─ Sum{axes=None} [id B] 186 < │ └─ Composite{...}.0 [id C] 163 --- > Composite{...}.0 [id A] 248 > ├─ Sum{axes=None} [id B] 175 > │ └─ Composite{...}.0 [id C] 153 12,13c12,13 < │ ├─ ExpandDims{axis=0} [id K] 121 < │ │ └─ Max{axis=0} [id L] 'max' 102 --- > │ ├─ ExpandDims{axis=0} [id K] 116 > │ │ └─ Max{axis=0} [id L] 'max' 97 16,17c16,17 < │ ├─ Isinf [id M] 140 < │ │ └─ ExpandDims{axis=0} [id K] 121 --- > │ ├─ Isinf [id M] 135 > │ │ └─ ExpandDims{axis=0} [id K] 116 19,20c19,20 < │ └─ Exp [id N] 139 < │ └─ ExpandDims{axis=0} [id K] 121 --- > │ └─ Exp [id N] 134 > │ └─ ExpandDims{axis=0} [id K] 116 22c22 < ├─ Max{axis=0} [id L] 'max' 102 --- > ├─ Max{axis=0} [id L] 'max' 97 26,27c26,27 < ├─ Sum{axes=None} [id Q] 119 < │ └─ Composite{((-0.5 * sqr(i0)) - 0.9189385332046727)} [id R] 100 --- > ├─ Sum{axes=None} [id Q] 114 > │ └─ Composite{((-0.5 * sqr(i0)) - 0.9189385332046727)} [id R] 95 52,59c52,59 < ├─ Sum{axes=None} [id BL] 255 < │ └─ Composite{switch(i2, (i1 + log(i0)), -inf)} [id BM] '0 <= weights <= 1, sum(weights) == 1' 252 < │ ├─ Sum{axis=1} [id BN] 250 < │ │ └─ Composite{...}.0 [id BO] 249 < │ │ ├─ Add [id BP] 240 < │ │ │ ├─ ExpandDims{axis=0} [id BQ] 179 < │ │ │ │ └─ LogSoftmax{axis=0} [id BR] 161 < │ │ │ │ └─ Sub [id BS] 138 --- > ├─ Sum{axes=None} [id BL] 245 > │ └─ Composite{switch(i2, (i1 + log(i0)), -inf)} [id BM] '0 <= weights <= 1, sum(weights) == 1' 242 > │ ├─ Sum{axis=1} [id BN] 240 > │ │ └─ Composite{...}.0 [id BO] 239 > │ │ ├─ Add [id BP] 231 > │ │ │ ├─ ExpandDims{axis=0} [id BQ] 168 > │ │ │ │ └─ LogSoftmax{axis=0} [id BR] 151 > │ │ │ │ └─ Sub [id BS] 133 66,67c66,67 < │ │ │ │ └─ ExpandDims{axis=0} [id BV] 120 < │ │ │ │ └─ Max{axis=0} [id BW] 'max' 101 --- > │ │ │ │ └─ ExpandDims{axis=0} [id BV] 115 > │ │ │ │ └─ Max{axis=0} [id BW] 'max' 96 70c70 < │ │ │ └─ Join [id BX] 238 --- > │ │ │ └─ Join [id BX] 229 72,113c72,113 < │ │ │ ├─ Composite{switch(i2, ((-1.8378770664093453 - (0.5 * i0)) - i1), -inf)} [id BZ] 236 < │ │ │ │ ├─ ExpandDims{axis=1} [id CA] 228 < │ │ │ │ │ └─ CAReduce{Composite{(i0 + sqr(i1))}, axis=1} [id CB] 219 < │ │ │ │ │ └─ Blockwise{SolveTriangular{trans=0, unit_diagonal=False, lower=True, check_finite=True, b_ndim=1}, (m,m),(m)->(m)} [id CC] 207 < │ │ │ │ │ ├─ Switch [id CD] 166 < │ │ │ │ │ │ ├─ ExpandDims{axes=[0, 1, 2]} [id CE] 145 < │ │ │ │ │ │ │ └─ All{axis=0} [id CF] 125 < │ │ │ │ │ │ │ └─ Gt [id CG] 106 < │ │ │ │ │ │ │ ├─ ExtractDiag{offset=0, axis1=0, axis2=1, view=False} [id CH] 82 < │ │ │ │ │ │ │ │ └─ Cholesky{lower=True, destructive=False, on_error='nan'} [id CI] 71 < │ │ │ │ │ │ │ │ └─ AdvancedSetSubtensor [id CJ] 63 < │ │ │ │ │ │ │ │ ├─ Alloc [id CK] 1 < │ │ │ │ │ │ │ │ │ ├─ 0.0 [id W] < │ │ │ │ │ │ │ │ │ ├─ 2 [id CL] < │ │ │ │ │ │ │ │ │ └─ 2 [id CL] < │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id CM] 52 < │ │ │ │ │ │ │ │ │ ├─ Composite{...}.0 [id BJ] 0 < │ │ │ │ │ │ │ │ │ │ └─ ··· < │ │ │ │ │ │ │ │ │ └─ 0 [id BA] < │ │ │ │ │ │ │ │ ├─ [0 1] [id CN] < │ │ │ │ │ │ │ │ └─ [0 1] [id CN] < │ │ │ │ │ │ │ └─ [0] [id CO] < │ │ │ │ │ │ ├─ ExpandDims{axis=0} [id CP] 83 < │ │ │ │ │ │ │ └─ Cholesky{lower=True, destructive=False, on_error='nan'} [id CI] 71 < │ │ │ │ │ │ │ └─ ··· < │ │ │ │ │ │ └─ [[[1]]] [id CQ] < │ │ │ │ │ └─ Sub [id CR] 196 < │ │ │ │ │ ├─ y_hat{[[ 1.08827 ... 02443723]]} [id CS] < │ │ │ │ │ └─ ExpandDims{axis=0} [id CT] 178 < │ │ │ │ │ └─ Subtensor{i} [id CU] 160 < │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id CV] 137 < │ │ │ │ │ │ └─ Join [id CW] 118 < │ │ │ │ │ │ ├─ 0 [id E] < │ │ │ │ │ │ ├─ ExpandDims{axis=0} [id CX] 98 < │ │ │ │ │ │ │ └─ CumOp{-1, add} [id S] 75 < │ │ │ │ │ │ │ └─ ··· < │ │ │ │ │ │ └─ ExpandDims{axis=0} [id CY] 3 < │ │ │ │ │ │ └─ y_coord [id BH] < │ │ │ │ │ └─ 0 [id BA] < │ │ │ │ ├─ ExpandDims{axes=[0, 1]} [id CZ] 126 < │ │ │ │ │ └─ CAReduce{Composite{(i0 + log(i1))}, axes=None} [id DA] 107 < │ │ │ │ │ └─ ExtractDiag{offset=0, axis1=0, axis2=1, view=False} [id CH] 82 --- > │ │ │ ├─ Composite{switch(i2, ((-1.8378770664093453 - (0.5 * i0)) - i1), -inf)} [id BZ] 227 > │ │ │ │ ├─ ExpandDims{axis=1} [id CA] 221 > │ │ │ │ │ └─ CAReduce{Composite{(i0 + sqr(i1))}, axis=1} [id CB] 213 > │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id CC] 205 > │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=True, check_finite=True, b_ndim=2} [id CD] 197 > │ │ │ │ │ ├─ Switch [id CE] 156 > │ │ │ │ │ │ ├─ ExpandDims{axes=[0, 1]} [id CF] 138 > │ │ │ │ │ │ │ └─ All{axes=None} [id CG] 120 > │ │ │ │ │ │ │ └─ Gt [id CH] 101 > │ │ │ │ │ │ │ ├─ ExtractDiag{offset=0, axis1=0, axis2=1, view=False} [id CI] 81 > │ │ │ │ │ │ │ │ └─ Cholesky{lower=True, destructive=False, on_error='nan'} [id CJ] 71 > │ │ │ │ │ │ │ │ └─ AdvancedSetSubtensor [id CK] 63 > │ │ │ │ │ │ │ │ ├─ Alloc [id CL] 1 > │ │ │ │ │ │ │ │ │ ├─ 0.0 [id W] > │ │ │ │ │ │ │ │ │ ├─ 2 [id CM] > │ │ │ │ │ │ │ │ │ └─ 2 [id CM] > │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id CN] 52 > │ │ │ │ │ │ │ │ │ ├─ Composite{...}.0 [id BJ] 0 > │ │ │ │ │ │ │ │ │ │ └─ ··· > │ │ │ │ │ │ │ │ │ └─ 0 [id BA] > │ │ │ │ │ │ │ │ ├─ [0 1] [id CO] > │ │ │ │ │ │ │ │ └─ [0 1] [id CO] > │ │ │ │ │ │ │ └─ [0] [id CP] > │ │ │ │ │ │ ├─ Cholesky{lower=True, destructive=False, on_error='nan'} [id CJ] 71 > │ │ │ │ │ │ │ └─ ··· > │ │ │ │ │ │ └─ [[1]] [id CQ] > │ │ │ │ │ └─ Sub [id CR] 186 > │ │ │ │ │ ├─ [[ 1.08827 ... 02443723]] [id CS] > │ │ │ │ │ └─ ExpandDims{axis=1} [id CT] 167 > │ │ │ │ │ └─ Subtensor{i} [id CU] 150 > │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id CV] 132 > │ │ │ │ │ │ └─ Join [id CW] 113 > │ │ │ │ │ │ ├─ 0 [id E] > │ │ │ │ │ │ ├─ ExpandDims{axis=0} [id CX] 93 > │ │ │ │ │ │ │ └─ CumOp{-1, add} [id S] 75 > │ │ │ │ │ │ │ └─ ··· > │ │ │ │ │ │ └─ ExpandDims{axis=0} [id CY] 3 > │ │ │ │ │ │ └─ y_coord [id BH] > │ │ │ │ │ └─ 0 [id BA] > │ │ │ │ ├─ ExpandDims{axes=[0, 1]} [id CZ] 121 > │ │ │ │ │ └─ CAReduce{Composite{(i0 + log(i1))}, axes=None} [id DA] 102 > │ │ │ │ │ └─ ExtractDiag{offset=0, axis1=0, axis2=1, view=False} [id CI] 81 115,151c115,150 < │ │ │ │ └─ ExpandDims{axes=[0, 1]} [id DB] 144 < │ │ │ │ └─ All{axis=0} [id CF] 125 < │ │ │ │ └─ ··· < │ │ │ ├─ Composite{switch(i2, ((-1.8378770664093453 - (0.5 * i0)) - i1), -inf)} [id DC] 235 < │ │ │ │ ├─ ExpandDims{axis=1} [id DD] 227 < │ │ │ │ │ └─ CAReduce{Composite{(i0 + sqr(i1))}, axis=1} [id DE] 217 < │ │ │ │ │ └─ Blockwise{SolveTriangular{trans=0, unit_diagonal=False, lower=True, check_finite=True, b_ndim=1}, (m,m),(m)->(m)} [id DF] 206 < │ │ │ │ │ ├─ Switch [id DG] 172 < │ │ │ │ │ │ ├─ ExpandDims{axes=[0, 1, 2]} [id DH] 153 < │ │ │ │ │ │ │ └─ All{axis=0} [id DI] 134 < │ │ │ │ │ │ │ └─ Gt [id DJ] 115 < │ │ │ │ │ │ │ ├─ ExtractDiag{offset=0, axis1=0, axis2=1, view=False} [id DK] 95 < │ │ │ │ │ │ │ │ └─ Cholesky{lower=True, destructive=False, on_error='nan'} [id DL] 74 < │ │ │ │ │ │ │ │ └─ AdvancedSetSubtensor [id DM] 66 < │ │ │ │ │ │ │ │ ├─ Alloc [id CK] 1 < │ │ │ │ │ │ │ │ │ └─ ··· < │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id DN] 55 < │ │ │ │ │ │ │ │ │ ├─ Composite{...}.0 [id BJ] 0 < │ │ │ │ │ │ │ │ │ │ └─ ··· < │ │ │ │ │ │ │ │ │ └─ 1 [id DO] < │ │ │ │ │ │ │ │ ├─ [0 1] [id CN] < │ │ │ │ │ │ │ │ └─ [0 1] [id CN] < │ │ │ │ │ │ │ └─ [0] [id CO] < │ │ │ │ │ │ ├─ ExpandDims{axis=0} [id DP] 94 < │ │ │ │ │ │ │ └─ Cholesky{lower=True, destructive=False, on_error='nan'} [id DL] 74 < │ │ │ │ │ │ │ └─ ··· < │ │ │ │ │ │ └─ [[[1]]] [id CQ] < │ │ │ │ │ └─ Sub [id DQ] 195 < │ │ │ │ │ ├─ y_hat{[[ 1.08827 ... 02443723]]} [id CS] < │ │ │ │ │ └─ ExpandDims{axis=0} [id DR] 177 < │ │ │ │ │ └─ Subtensor{i} [id DS] 159 < │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id CV] 137 < │ │ │ │ │ │ └─ ··· < │ │ │ │ │ └─ 1 [id DO] < │ │ │ │ ├─ ExpandDims{axes=[0, 1]} [id DT] 135 < │ │ │ │ │ └─ CAReduce{Composite{(i0 + log(i1))}, axes=None} [id DU] 116 < │ │ │ │ │ └─ ExtractDiag{offset=0, axis1=0, axis2=1, view=False} [id DK] 95 --- > │ │ │ │ └─ ExpandDims{axes=[0, 1]} [id CF] 138 > │ │ │ │ └─ ··· > │ │ │ ├─ Composite{switch(i2, ((-1.8378770664093453 - (0.5 * i0)) - i1), -inf)} [id DB] 226 > │ │ │ │ ├─ ExpandDims{axis=1} [id DC] 220 > │ │ │ │ │ └─ CAReduce{Composite{(i0 + sqr(i1))}, axis=1} [id DD] 212 > │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id DE] 204 > │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=True, check_finite=True, b_ndim=2} [id DF] 196 > │ │ │ │ │ ├─ Switch [id DG] 161 > │ │ │ │ │ │ ├─ ExpandDims{axes=[0, 1]} [id DH] 144 > │ │ │ │ │ │ │ └─ All{axes=None} [id DI] 129 > │ │ │ │ │ │ │ └─ Gt [id DJ] 110 > │ │ │ │ │ │ │ ├─ ExtractDiag{offset=0, axis1=0, axis2=1, view=False} [id DK] 90 > │ │ │ │ │ │ │ │ └─ Cholesky{lower=True, destructive=False, on_error='nan'} [id DL] 74 > │ │ │ │ │ │ │ │ └─ AdvancedSetSubtensor [id DM] 66 > │ │ │ │ │ │ │ │ ├─ Alloc [id CL] 1 > │ │ │ │ │ │ │ │ │ └─ ··· > │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id DN] 55 > │ │ │ │ │ │ │ │ │ ├─ Composite{...}.0 [id BJ] 0 > │ │ │ │ │ │ │ │ │ │ └─ ··· > │ │ │ │ │ │ │ │ │ └─ 1 [id DO] > │ │ │ │ │ │ │ │ ├─ [0 1] [id CO] > │ │ │ │ │ │ │ │ └─ [0 1] [id CO] > │ │ │ │ │ │ │ └─ [0] [id CP] > │ │ │ │ │ │ ├─ Cholesky{lower=True, destructive=False, on_error='nan'} [id DL] 74 > │ │ │ │ │ │ │ └─ ··· > │ │ │ │ │ │ └─ [[1]] [id CQ] > │ │ │ │ │ └─ Sub [id DP] 185 > │ │ │ │ │ ├─ [[ 1.08827 ... 02443723]] [id CS] > │ │ │ │ │ └─ ExpandDims{axis=1} [id DQ] 166 > │ │ │ │ │ └─ Subtensor{i} [id DR] 149 > │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id CV] 132 > │ │ │ │ │ │ └─ ··· > │ │ │ │ │ └─ 1 [id DO] > │ │ │ │ ├─ ExpandDims{axes=[0, 1]} [id DS] 130 > │ │ │ │ │ └─ CAReduce{Composite{(i0 + log(i1))}, axes=None} [id DT] 111 > │ │ │ │ │ └─ ExtractDiag{offset=0, axis1=0, axis2=1, view=False} [id DK] 90 153,189c152,187 < │ │ │ │ └─ ExpandDims{axes=[0, 1]} [id DV] 154 < │ │ │ │ └─ All{axis=0} [id DI] 134 < │ │ │ │ └─ ··· < │ │ │ ├─ Composite{switch(i2, ((-1.8378770664093453 - (0.5 * i0)) - i1), -inf)} [id DW] 234 < │ │ │ │ ├─ ExpandDims{axis=1} [id DX] 226 < │ │ │ │ │ └─ CAReduce{Composite{(i0 + sqr(i1))}, axis=1} [id DY] 215 < │ │ │ │ │ └─ Blockwise{SolveTriangular{trans=0, unit_diagonal=False, lower=True, check_finite=True, b_ndim=1}, (m,m),(m)->(m)} [id DZ] 205 < │ │ │ │ │ ├─ Switch [id EA] 170 < │ │ │ │ │ │ ├─ ExpandDims{axes=[0, 1, 2]} [id EB] 151 < │ │ │ │ │ │ │ └─ All{axis=0} [id EC] 131 < │ │ │ │ │ │ │ └─ Gt [id ED] 112 < │ │ │ │ │ │ │ ├─ ExtractDiag{offset=0, axis1=0, axis2=1, view=False} [id EE] 90 < │ │ │ │ │ │ │ │ └─ Cholesky{lower=True, destructive=False, on_error='nan'} [id EF] 73 < │ │ │ │ │ │ │ │ └─ AdvancedSetSubtensor [id EG] 65 < │ │ │ │ │ │ │ │ ├─ Alloc [id CK] 1 < │ │ │ │ │ │ │ │ │ └─ ··· < │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id EH] 54 < │ │ │ │ │ │ │ │ │ ├─ Composite{...}.0 [id BJ] 0 < │ │ │ │ │ │ │ │ │ │ └─ ··· < │ │ │ │ │ │ │ │ │ └─ 2 [id EI] < │ │ │ │ │ │ │ │ ├─ [0 1] [id CN] < │ │ │ │ │ │ │ │ └─ [0 1] [id CN] < │ │ │ │ │ │ │ └─ [0] [id CO] < │ │ │ │ │ │ ├─ ExpandDims{axis=0} [id EJ] 91 < │ │ │ │ │ │ │ └─ Cholesky{lower=True, destructive=False, on_error='nan'} [id EF] 73 < │ │ │ │ │ │ │ └─ ··· < │ │ │ │ │ │ └─ [[[1]]] [id CQ] < │ │ │ │ │ └─ Sub [id EK] 194 < │ │ │ │ │ ├─ y_hat{[[ 1.08827 ... 02443723]]} [id CS] < │ │ │ │ │ └─ ExpandDims{axis=0} [id EL] 176 < │ │ │ │ │ └─ Subtensor{i} [id EM] 158 < │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id CV] 137 < │ │ │ │ │ │ └─ ··· < │ │ │ │ │ └─ 2 [id EI] < │ │ │ │ ├─ ExpandDims{axes=[0, 1]} [id EN] 132 < │ │ │ │ │ └─ CAReduce{Composite{(i0 + log(i1))}, axes=None} [id EO] 113 < │ │ │ │ │ └─ ExtractDiag{offset=0, axis1=0, axis2=1, view=False} [id EE] 90 --- > │ │ │ │ └─ ExpandDims{axes=[0, 1]} [id DH] 144 > │ │ │ │ └─ ··· > │ │ │ ├─ Composite{switch(i2, ((-1.8378770664093453 - (0.5 * i0)) - i1), -inf)} [id DU] 225 > │ │ │ │ ├─ ExpandDims{axis=1} [id DV] 219 > │ │ │ │ │ └─ CAReduce{Composite{(i0 + sqr(i1))}, axis=1} [id DW] 211 > │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id DX] 203 > │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=True, check_finite=True, b_ndim=2} [id DY] 195 > │ │ │ │ │ ├─ Switch [id DZ] 159 > │ │ │ │ │ │ ├─ ExpandDims{axes=[0, 1]} [id EA] 142 > │ │ │ │ │ │ │ └─ All{axes=None} [id EB] 126 > │ │ │ │ │ │ │ └─ Gt [id EC] 107 > │ │ │ │ │ │ │ ├─ ExtractDiag{offset=0, axis1=0, axis2=1, view=False} [id ED] 87 > │ │ │ │ │ │ │ │ └─ Cholesky{lower=True, destructive=False, on_error='nan'} [id EE] 73 > │ │ │ │ │ │ │ │ └─ AdvancedSetSubtensor [id EF] 65 > │ │ │ │ │ │ │ │ ├─ Alloc [id CL] 1 > │ │ │ │ │ │ │ │ │ └─ ··· > │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id EG] 54 > │ │ │ │ │ │ │ │ │ ├─ Composite{...}.0 [id BJ] 0 > │ │ │ │ │ │ │ │ │ │ └─ ··· > │ │ │ │ │ │ │ │ │ └─ 2 [id EH] > │ │ │ │ │ │ │ │ ├─ [0 1] [id CO] > │ │ │ │ │ │ │ │ └─ [0 1] [id CO] > │ │ │ │ │ │ │ └─ [0] [id CP] > │ │ │ │ │ │ ├─ Cholesky{lower=True, destructive=False, on_error='nan'} [id EE] 73 > │ │ │ │ │ │ │ └─ ··· > │ │ │ │ │ │ └─ [[1]] [id CQ] > │ │ │ │ │ └─ Sub [id EI] 184 > │ │ │ │ │ ├─ [[ 1.08827 ... 02443723]] [id CS] > │ │ │ │ │ └─ ExpandDims{axis=1} [id EJ] 165 > │ │ │ │ │ └─ Subtensor{i} [id EK] 148 > │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id CV] 132 > │ │ │ │ │ │ └─ ··· > │ │ │ │ │ └─ 2 [id EH] > │ │ │ │ ├─ ExpandDims{axes=[0, 1]} [id EL] 127 > │ │ │ │ │ └─ CAReduce{Composite{(i0 + log(i1))}, axes=None} [id EM] 108 > │ │ │ │ │ └─ ExtractDiag{offset=0, axis1=0, axis2=1, view=False} [id ED] 87 191,227c189,224 < │ │ │ │ └─ ExpandDims{axes=[0, 1]} [id EP] 150 < │ │ │ │ └─ All{axis=0} [id EC] 131 < │ │ │ │ └─ ··· < │ │ │ ├─ Composite{switch(i2, ((-1.8378770664093453 - (0.5 * i0)) - i1), -inf)} [id EQ] 233 < │ │ │ │ ├─ ExpandDims{axis=1} [id ER] 225 < │ │ │ │ │ └─ CAReduce{Composite{(i0 + sqr(i1))}, axis=1} [id ES] 213 < │ │ │ │ │ └─ Blockwise{SolveTriangular{trans=0, unit_diagonal=False, lower=True, check_finite=True, b_ndim=1}, (m,m),(m)->(m)} [id ET] 204 < │ │ │ │ │ ├─ Switch [id EU] 168 < │ │ │ │ │ │ ├─ ExpandDims{axes=[0, 1, 2]} [id EV] 148 < │ │ │ │ │ │ │ └─ All{axis=0} [id EW] 128 < │ │ │ │ │ │ │ └─ Gt [id EX] 109 < │ │ │ │ │ │ │ ├─ ExtractDiag{offset=0, axis1=0, axis2=1, view=False} [id EY] 86 < │ │ │ │ │ │ │ │ └─ Cholesky{lower=True, destructive=False, on_error='nan'} [id EZ] 72 < │ │ │ │ │ │ │ │ └─ AdvancedSetSubtensor [id FA] 64 < │ │ │ │ │ │ │ │ ├─ Alloc [id CK] 1 < │ │ │ │ │ │ │ │ │ └─ ··· < │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id FB] 53 < │ │ │ │ │ │ │ │ │ ├─ Composite{...}.0 [id BJ] 0 < │ │ │ │ │ │ │ │ │ │ └─ ··· < │ │ │ │ │ │ │ │ │ └─ 3 [id FC] < │ │ │ │ │ │ │ │ ├─ [0 1] [id CN] < │ │ │ │ │ │ │ │ └─ [0 1] [id CN] < │ │ │ │ │ │ │ └─ [0] [id CO] < │ │ │ │ │ │ ├─ ExpandDims{axis=0} [id FD] 87 < │ │ │ │ │ │ │ └─ Cholesky{lower=True, destructive=False, on_error='nan'} [id EZ] 72 < │ │ │ │ │ │ │ └─ ··· < │ │ │ │ │ │ └─ [[[1]]] [id CQ] < │ │ │ │ │ └─ Sub [id FE] 193 < │ │ │ │ │ ├─ y_hat{[[ 1.08827 ... 02443723]]} [id CS] < │ │ │ │ │ └─ ExpandDims{axis=0} [id FF] 175 < │ │ │ │ │ └─ Subtensor{i} [id FG] 157 < │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id CV] 137 < │ │ │ │ │ │ └─ ··· < │ │ │ │ │ └─ 3 [id FC] < │ │ │ │ ├─ ExpandDims{axes=[0, 1]} [id FH] 129 < │ │ │ │ │ └─ CAReduce{Composite{(i0 + log(i1))}, axes=None} [id FI] 110 < │ │ │ │ │ └─ ExtractDiag{offset=0, axis1=0, axis2=1, view=False} [id EY] 86 --- > │ │ │ │ └─ ExpandDims{axes=[0, 1]} [id EA] 142 > │ │ │ │ └─ ··· > │ │ │ ├─ Composite{switch(i2, ((-1.8378770664093453 - (0.5 * i0)) - i1), -inf)} [id EN] 224 > │ │ │ │ ├─ ExpandDims{axis=1} [id EO] 218 > │ │ │ │ │ └─ CAReduce{Composite{(i0 + sqr(i1))}, axis=1} [id EP] 210 > │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id EQ] 202 > │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=True, check_finite=True, b_ndim=2} [id ER] 194 > │ │ │ │ │ ├─ Switch [id ES] 157 > │ │ │ │ │ │ ├─ ExpandDims{axes=[0, 1]} [id ET] 140 > │ │ │ │ │ │ │ └─ All{axes=None} [id EU] 123 > │ │ │ │ │ │ │ └─ Gt [id EV] 104 > │ │ │ │ │ │ │ ├─ ExtractDiag{offset=0, axis1=0, axis2=1, view=False} [id EW] 84 > │ │ │ │ │ │ │ │ └─ Cholesky{lower=True, destructive=False, on_error='nan'} [id EX] 72 > │ │ │ │ │ │ │ │ └─ AdvancedSetSubtensor [id EY] 64 > │ │ │ │ │ │ │ │ ├─ Alloc [id CL] 1 > │ │ │ │ │ │ │ │ │ └─ ··· > │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id EZ] 53 > │ │ │ │ │ │ │ │ │ ├─ Composite{...}.0 [id BJ] 0 > │ │ │ │ │ │ │ │ │ │ └─ ··· > │ │ │ │ │ │ │ │ │ └─ 3 [id FA] > │ │ │ │ │ │ │ │ ├─ [0 1] [id CO] > │ │ │ │ │ │ │ │ └─ [0 1] [id CO] > │ │ │ │ │ │ │ └─ [0] [id CP] > │ │ │ │ │ │ ├─ Cholesky{lower=True, destructive=False, on_error='nan'} [id EX] 72 > │ │ │ │ │ │ │ └─ ··· > │ │ │ │ │ │ └─ [[1]] [id CQ] > │ │ │ │ │ └─ Sub [id FB] 183 > │ │ │ │ │ ├─ [[ 1.08827 ... 02443723]] [id CS] > │ │ │ │ │ └─ ExpandDims{axis=1} [id FC] 164 > │ │ │ │ │ └─ Subtensor{i} [id FD] 147 > │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id CV] 132 > │ │ │ │ │ │ └─ ··· > │ │ │ │ │ └─ 3 [id FA] > │ │ │ │ ├─ ExpandDims{axes=[0, 1]} [id FE] 124 > │ │ │ │ │ └─ CAReduce{Composite{(i0 + log(i1))}, axes=None} [id FF] 105 > │ │ │ │ │ └─ ExtractDiag{offset=0, axis1=0, axis2=1, view=False} [id EW] 84 229,265c226,261 < │ │ │ │ └─ ExpandDims{axes=[0, 1]} [id FJ] 147 < │ │ │ │ └─ All{axis=0} [id EW] 128 < │ │ │ │ └─ ··· < │ │ │ └─ Composite{switch(i2, ((-1.8378770664093453 - (0.5 * i0)) - i1), -inf)} [id FK] 232 < │ │ │ ├─ ExpandDims{axis=1} [id FL] 224 < │ │ │ │ └─ CAReduce{Composite{(i0 + sqr(i1))}, axis=1} [id FM] 211 < │ │ │ │ └─ Blockwise{SolveTriangular{trans=0, unit_diagonal=False, lower=True, check_finite=True, b_ndim=1}, (m,m),(m)->(m)} [id FN] 203 < │ │ │ │ ├─ Switch [id FO] 165 < │ │ │ │ │ ├─ ExpandDims{axes=[0, 1, 2]} [id FP] 143 < │ │ │ │ │ │ └─ All{axis=0} [id FQ] 123 < │ │ │ │ │ │ └─ Gt [id FR] 104 < │ │ │ │ │ │ ├─ ExtractDiag{offset=0, axis1=0, axis2=1, view=False} [id FS] 79 < │ │ │ │ │ │ │ └─ Cholesky{lower=True, destructive=False, on_error='nan'} [id FT] 70 < │ │ │ │ │ │ │ └─ AdvancedSetSubtensor [id FU] 62 < │ │ │ │ │ │ │ ├─ Alloc [id CK] 1 < │ │ │ │ │ │ │ │ └─ ··· < │ │ │ │ │ │ │ ├─ Subtensor{i} [id FV] 51 < │ │ │ │ │ │ │ │ ├─ Composite{...}.0 [id BJ] 0 < │ │ │ │ │ │ │ │ │ └─ ··· < │ │ │ │ │ │ │ │ └─ 4 [id FW] < │ │ │ │ │ │ │ ├─ [0 1] [id CN] < │ │ │ │ │ │ │ └─ [0 1] [id CN] < │ │ │ │ │ │ └─ [0] [id CO] < │ │ │ │ │ ├─ ExpandDims{axis=0} [id FX] 80 < │ │ │ │ │ │ └─ Cholesky{lower=True, destructive=False, on_error='nan'} [id FT] 70 < │ │ │ │ │ │ └─ ··· < │ │ │ │ │ └─ [[[1]]] [id CQ] < │ │ │ │ └─ Sub [id FY] 192 < │ │ │ │ ├─ y_hat{[[ 1.08827 ... 02443723]]} [id CS] < │ │ │ │ └─ ExpandDims{axis=0} [id FZ] 174 < │ │ │ │ └─ Subtensor{i} [id GA] 156 < │ │ │ │ ├─ Transpose{axes=[1, 0]} [id CV] 137 < │ │ │ │ │ └─ ··· < │ │ │ │ └─ 4 [id FW] < │ │ │ ├─ ExpandDims{axes=[0, 1]} [id GB] 124 < │ │ │ │ └─ CAReduce{Composite{(i0 + log(i1))}, axes=None} [id GC] 105 < │ │ │ │ └─ ExtractDiag{offset=0, axis1=0, axis2=1, view=False} [id FS] 79 --- > │ │ │ │ └─ ExpandDims{axes=[0, 1]} [id ET] 140 > │ │ │ │ └─ ··· > │ │ │ └─ Composite{switch(i2, ((-1.8378770664093453 - (0.5 * i0)) - i1), -inf)} [id FG] 223 > │ │ │ ├─ ExpandDims{axis=1} [id FH] 217 > │ │ │ │ └─ CAReduce{Composite{(i0 + sqr(i1))}, axis=1} [id FI] 209 > │ │ │ │ └─ Transpose{axes=[1, 0]} [id FJ] 201 > │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=True, check_finite=True, b_ndim=2} [id FK] 193 > │ │ │ │ ├─ Switch [id FL] 155 > │ │ │ │ │ ├─ ExpandDims{axes=[0, 1]} [id FM] 137 > │ │ │ │ │ │ └─ All{axes=None} [id FN] 118 > │ │ │ │ │ │ └─ Gt [id FO] 99 > │ │ │ │ │ │ ├─ ExtractDiag{offset=0, axis1=0, axis2=1, view=False} [id FP] 79 > │ │ │ │ │ │ │ └─ Cholesky{lower=True, destructive=False, on_error='nan'} [id FQ] 70 > │ │ │ │ │ │ │ └─ AdvancedSetSubtensor [id FR] 62 > │ │ │ │ │ │ │ ├─ Alloc [id CL] 1 > │ │ │ │ │ │ │ │ └─ ··· > │ │ │ │ │ │ │ ├─ Subtensor{i} [id FS] 51 > │ │ │ │ │ │ │ │ ├─ Composite{...}.0 [id BJ] 0 > │ │ │ │ │ │ │ │ │ └─ ··· > │ │ │ │ │ │ │ │ └─ 4 [id FT] > │ │ │ │ │ │ │ ├─ [0 1] [id CO] > │ │ │ │ │ │ │ └─ [0 1] [id CO] > │ │ │ │ │ │ └─ [0] [id CP] > │ │ │ │ │ ├─ Cholesky{lower=True, destructive=False, on_error='nan'} [id FQ] 70 > │ │ │ │ │ │ └─ ··· > │ │ │ │ │ └─ [[1]] [id CQ] > │ │ │ │ └─ Sub [id FU] 182 > │ │ │ │ ├─ [[ 1.08827 ... 02443723]] [id CS] > │ │ │ │ └─ ExpandDims{axis=1} [id FV] 163 > │ │ │ │ └─ Subtensor{i} [id FW] 146 > │ │ │ │ ├─ Transpose{axes=[1, 0]} [id CV] 132 > │ │ │ │ │ └─ ··· > │ │ │ │ └─ 4 [id FT] > │ │ │ ├─ ExpandDims{axes=[0, 1]} [id FX] 119 > │ │ │ │ └─ CAReduce{Composite{(i0 + log(i1))}, axes=None} [id FY] 100 > │ │ │ │ └─ ExtractDiag{offset=0, axis1=0, axis2=1, view=False} [id FP] 79 267,272c263,267 < │ │ │ └─ ExpandDims{axes=[0, 1]} [id GD] 142 < │ │ │ └─ All{axis=0} [id FQ] 123 < │ │ │ └─ ··· < │ │ ├─ ExpandDims{axis=1} [id GE] 244 < │ │ │ └─ Max{axis=1} [id GF] 'max' 242 < │ │ │ └─ Add [id BP] 240 --- > │ │ │ └─ ExpandDims{axes=[0, 1]} [id FM] 137 > │ │ │ └─ ··· > │ │ ├─ ExpandDims{axis=1} [id FZ] 235 > │ │ │ └─ Max{axis=1} [id GA] 'max' 233 > │ │ │ └─ Add [id BP] 231 274,275c269,270 < │ │ ├─ Isinf [id GG] 246 < │ │ │ └─ ExpandDims{axis=1} [id GE] 244 --- > │ │ ├─ Isinf [id GB] 237 > │ │ │ └─ ExpandDims{axis=1} [id FZ] 235 277,278c272,273 < │ │ └─ Exp [id GH] 247 < │ │ └─ ExpandDims{axis=1} [id GE] 244 --- > │ │ └─ Exp [id GC] 238 > │ │ └─ ExpandDims{axis=1} [id FZ] 235 280c275 < │ ├─ Max{axis=1} [id GF] 'max' 242 --- > │ ├─ Max{axis=1} [id GA] 'max' 233 282,288c277,283 < │ └─ ExpandDims{axis=0} [id GI] 229 < │ └─ All{axes=None} [id GJ] 221 < │ └─ MakeVector{dtype='bool'} [id GK] 208 < │ ├─ All{axes=None} [id GL] 201 < │ │ └─ Ge [id GM] 185 < │ │ ├─ Softmax{axis=0} [id GN] 162 < │ │ │ └─ Sub [id BS] 138 --- > │ └─ ExpandDims{axis=0} [id GD] 214 > │ └─ All{axes=None} [id GE] 206 > │ └─ MakeVector{dtype='bool'} [id GF] 198 > │ ├─ All{axes=None} [id GG] 191 > │ │ └─ Ge [id GH] 174 > │ │ ├─ Softmax{axis=0} [id GI] 152 > │ │ │ └─ Sub [id BS] 133 290,293c285,288 < │ │ └─ [0] [id CO] < │ ├─ All{axes=None} [id GO] 200 < │ │ └─ Le [id GP] 184 < │ │ ├─ Softmax{axis=0} [id GN] 162 --- > │ │ └─ [0] [id CP] > │ ├─ All{axes=None} [id GJ] 190 > │ │ └─ Le [id GK] 173 > │ │ ├─ Softmax{axis=0} [id GI] 152 295,296c290,291 < │ │ └─ [1] [id GQ] < │ └─ True [id GR] --- > │ │ └─ [1] [id GL] > │ └─ True [id GM] 299,303c294,298 < ├─ -inf [id GS] < ├─ 144.56574394634487 [id GT] < ├─ Sum{axes=None} [id GU] 199 < │ └─ Composite{...}.0 [id GV] 183 < │ ├─ LogSoftmax{axis=0} [id BR] 161 --- > ├─ -inf [id GN] > ├─ 144.56574394634487 [id GO] > ├─ Sum{axes=None} [id GP] 189 > │ └─ Composite{...}.0 [id GQ] 172 > │ ├─ LogSoftmax{axis=0} [id BR] 151 305c300 < │ └─ Softmax{axis=0} [id GN] 162 --- > │ └─ Softmax{axis=0} [id GI] 152 307,309c302,304 < ├─ Any{axis=0} [id GW] 198 < │ └─ Lt [id GX] 182 < │ ├─ Softmax{axis=0} [id GN] 162 --- > ├─ Any{axis=0} [id GR] 188 > │ └─ Lt [id GS] 171 > │ ├─ Softmax{axis=0} [id GI] 152 311,314c306,309 < │ └─ [0] [id CO] < └─ Any{axis=0} [id GY] 197 < └─ Gt [id GZ] 181 < ├─ Softmax{axis=0} [id GN] 162 --- > │ └─ [0] [id CP] > └─ Any{axis=0} [id GT] 187 > └─ Gt [id GU] 170 > ├─ Softmax{axis=0} [id GI] 152 316,322c311,317 < └─ [1] [id GQ] < Add [id HA] 'w_simplex___grad' 390 < ├─ [5.] [id HB] < ├─ SpecifyShape [id HC] 376 < │ ├─ Split{2}.0 [id HD] 362 < │ │ ├─ Composite{(i3 + (-1.0 * eq(i0, i1) * i2))} [id HE] 350 < │ │ │ ├─ ExpandDims{axis=0} [id BV] 120 --- > └─ [1] [id GL] > Add [id GV] 'w_simplex___grad' 379 > ├─ [5.] [id GW] > ├─ SpecifyShape [id GX] 365 > │ ├─ Split{2}.0 [id GY] 351 > │ │ ├─ Composite{(i3 + (-1.0 * eq(i0, i1) * i2))} [id GZ] 339 > │ │ │ ├─ ExpandDims{axis=0} [id BV] 115 326,332c321,327 < │ │ │ ├─ ExpandDims{axis=0} [id HF] 339 < │ │ │ │ └─ Sum{axes=None} [id HG] 327 < │ │ │ │ └─ Composite{(i3 + (i2 - (i0 * i1)))} [id HH] 311 < │ │ │ │ ├─ ExpandDims{axis=0} [id HI] 295 < │ │ │ │ │ └─ Sum{axes=None} [id HJ] 284 < │ │ │ │ │ └─ Composite{(switch(i0, 0, i1) + i2)} [id HK] 273 < │ │ │ │ │ ├─ Composite{...}.1 [id GV] 183 --- > │ │ │ ├─ ExpandDims{axis=0} [id HA] 328 > │ │ │ │ └─ Sum{axes=None} [id HB] 312 > │ │ │ │ └─ Composite{(i3 + (i2 - (i0 * i1)))} [id HC] 296 > │ │ │ │ ├─ ExpandDims{axis=0} [id HD] 285 > │ │ │ │ │ └─ Sum{axes=None} [id HE] 274 > │ │ │ │ │ └─ Composite{(switch(i0, 0, i1) + i2)} [id HF] 263 > │ │ │ │ │ ├─ Composite{...}.1 [id GQ] 172 334,336c329,331 < │ │ │ │ │ ├─ Switch [id HL] 263 < │ │ │ │ │ │ ├─ ExpandDims{axis=0} [id HM] 261 < │ │ │ │ │ │ │ └─ Composite{...}.1 [id A] 258 --- > │ │ │ │ │ ├─ Switch [id HG] 253 > │ │ │ │ │ │ ├─ ExpandDims{axis=0} [id HH] 251 > │ │ │ │ │ │ │ └─ Composite{...}.1 [id A] 248 338,342c333,337 < │ │ │ │ │ │ ├─ [0] [id CO] < │ │ │ │ │ │ └─ [9.] [id HN] < │ │ │ │ │ └─ Sum{axis=0} [id HO] 267 < │ │ │ │ │ └─ Composite{((eq(i4, i5) * i3) + switch(i2, 0, (i0 / i1)))} [id HP] 265 < │ │ │ │ │ ├─ Composite{...}.1 [id BO] 249 --- > │ │ │ │ │ │ ├─ [0] [id CP] > │ │ │ │ │ │ └─ [9.] [id HI] > │ │ │ │ │ └─ Sum{axis=0} [id HJ] 257 > │ │ │ │ │ └─ Composite{((eq(i4, i5) * i3) + switch(i2, 0, (i0 / i1)))} [id HK] 255 > │ │ │ │ │ ├─ Composite{...}.1 [id BO] 239 344,345c339,340 < │ │ │ │ │ ├─ ExpandDims{axis=1} [id HQ] 251 < │ │ │ │ │ │ └─ Sum{axis=1} [id BN] 250 --- > │ │ │ │ │ ├─ ExpandDims{axis=1} [id HL] 241 > │ │ │ │ │ │ └─ Sum{axis=1} [id BN] 240 347c342 < │ │ │ │ │ ├─ Isinf [id GG] 246 --- > │ │ │ │ │ ├─ Isinf [id GB] 237 349,354c344,349 < │ │ │ │ │ ├─ Composite{(1.0 + (i0 * i1) + i2)} [id HR] 264 < │ │ │ │ │ │ ├─ ExpandDims{axis=1} [id HS] 262 < │ │ │ │ │ │ │ └─ Sum{axis=1} [id HT] 260 < │ │ │ │ │ │ │ └─ Alloc [id HU] 257 < │ │ │ │ │ │ │ ├─ Composite{switch(i1, reciprocal(i0), 0.0)} [id HV] 254 < │ │ │ │ │ │ │ │ ├─ ExpandDims{axis=1} [id HQ] 251 --- > │ │ │ │ │ ├─ Composite{(1.0 + (i0 * i1) + i2)} [id HM] 254 > │ │ │ │ │ │ ├─ ExpandDims{axis=1} [id HN] 252 > │ │ │ │ │ │ │ └─ Sum{axis=1} [id HO] 250 > │ │ │ │ │ │ │ └─ Alloc [id HP] 247 > │ │ │ │ │ │ │ ├─ Composite{switch(i1, reciprocal(i0), 0.0)} [id HQ] 244 > │ │ │ │ │ │ │ │ ├─ ExpandDims{axis=1} [id HL] 241 356c351 < │ │ │ │ │ │ │ │ └─ Isinf [id GG] 246 --- > │ │ │ │ │ │ │ │ └─ Isinf [id GB] 237 358,360c353,355 < │ │ │ │ │ │ │ ├─ 1000 [id HW] < │ │ │ │ │ │ │ └─ 5 [id HX] < │ │ │ │ │ │ ├─ Exp [id GH] 247 --- > │ │ │ │ │ │ │ ├─ 1000 [id HR] > │ │ │ │ │ │ │ └─ 5 [id HS] > │ │ │ │ │ │ ├─ Exp [id GC] 238 362,365c357,360 < │ │ │ │ │ │ └─ ExpandDims{axis=1} [id HY] 259 < │ │ │ │ │ │ └─ Sum{axis=1} [id HZ] 256 < │ │ │ │ │ │ └─ Composite{switch(i2, 0, (i0 / i1))} [id IA] 253 < │ │ │ │ │ │ ├─ Composite{...}.2 [id BO] 249 --- > │ │ │ │ │ │ └─ ExpandDims{axis=1} [id HT] 249 > │ │ │ │ │ │ └─ Sum{axis=1} [id HU] 246 > │ │ │ │ │ │ └─ Composite{switch(i2, 0, (i0 / i1))} [id HV] 243 > │ │ │ │ │ │ ├─ Composite{...}.2 [id BO] 239 367c362 < │ │ │ │ │ │ ├─ ExpandDims{axis=1} [id HQ] 251 --- > │ │ │ │ │ │ ├─ ExpandDims{axis=1} [id HL] 241 369c364 < │ │ │ │ │ │ └─ Isinf [id GG] 246 --- > │ │ │ │ │ │ └─ Isinf [id GB] 237 371c366 < │ │ │ │ │ ├─ ExpandDims{axis=1} [id GE] 244 --- > │ │ │ │ │ ├─ ExpandDims{axis=1} [id FZ] 235 373c368 < │ │ │ │ │ └─ Add [id BP] 240 --- > │ │ │ │ │ └─ Add [id BP] 231 375c370 < │ │ │ │ ├─ Softmax{axis=0} [id GN] 162 --- > │ │ │ │ ├─ Softmax{axis=0} [id GI] 152 377c372 < │ │ │ │ ├─ Composite{(switch(i0, 0, i1) + i2)} [id HK] 273 --- > │ │ │ │ ├─ Composite{(switch(i0, 0, i1) + i2)} [id HF] 263 379,381c374,376 < │ │ │ │ └─ SoftmaxGrad{axis=0} [id IB] 180 < │ │ │ │ ├─ [0. 0. 0. 0. 0.] [id IC] < │ │ │ │ └─ Softmax{axis=0} [id GN] 162 --- > │ │ │ │ └─ SoftmaxGrad{axis=0} [id HW] 169 > │ │ │ │ ├─ [0. 0. 0. 0. 0.] [id HX] > │ │ │ │ └─ Softmax{axis=0} [id GI] 152 383c378 < │ │ │ └─ Composite{(i3 + (i2 - (i0 * i1)))} [id HH] 311 --- > │ │ │ └─ Composite{(i3 + (i2 - (i0 * i1)))} [id HC] 296 386,390c381,385 < │ │ └─ [4 1] [id ID] < │ └─ 4 [id IE] < ├─ Neg [id IF] 383 < │ └─ SpecifyShape [id IG] 375 < │ ├─ Split{2}.1 [id HD] 362 --- > │ │ └─ [4 1] [id HY] > │ └─ 4 [id HZ] > ├─ Neg [id IA] 372 > │ └─ SpecifyShape [id IB] 364 > │ ├─ Split{2}.1 [id GY] 351 393,397c388,392 < ├─ ExpandDims{axis=0} [id IH] 248 < │ └─ Sum{axes=None} [id II] 245 < │ └─ Split{2}.0 [id IJ] 243 < │ ├─ Composite{((eq(i4, i5) * i3) + switch(i2, 0, (i0 / i1)))} [id IK] 241 < │ │ ├─ Composite{...}.1 [id C] 163 --- > ├─ ExpandDims{axis=0} [id IC] 236 > │ └─ Sum{axes=None} [id ID] 234 > │ └─ Split{2}.0 [id IE] 232 > │ ├─ Composite{((eq(i4, i5) * i3) + switch(i2, 0, (i0 / i1)))} [id IF] 230 > │ │ ├─ Composite{...}.1 [id C] 153 399,400c394,395 < │ │ ├─ ExpandDims{axis=0} [id IL] 202 < │ │ │ └─ Sum{axes=None} [id B] 186 --- > │ │ ├─ ExpandDims{axis=0} [id IG] 192 > │ │ │ └─ Sum{axes=None} [id B] 175 402c397 < │ │ ├─ Isinf [id M] 140 --- > │ │ ├─ Isinf [id M] 135 404,409c399,404 < │ │ ├─ Composite{(-5.0 + (i0 * i1) + i2)} [id IM] 239 < │ │ │ ├─ ExpandDims{axis=0} [id IN] 237 < │ │ │ │ └─ Sum{axes=None} [id IO] 231 < │ │ │ │ └─ Alloc [id IP] 223 < │ │ │ │ ├─ Composite{switch(i1, (-5.0 / i0), 0.0)} [id IQ] 210 < │ │ │ │ │ ├─ ExpandDims{axis=0} [id IL] 202 --- > │ │ ├─ Composite{(-5.0 + (i0 * i1) + i2)} [id IH] 228 > │ │ │ ├─ ExpandDims{axis=0} [id II] 222 > │ │ │ │ └─ Sum{axes=None} [id IJ] 216 > │ │ │ │ └─ Alloc [id IK] 208 > │ │ │ │ ├─ Composite{switch(i1, (-5.0 / i0), 0.0)} [id IL] 200 > │ │ │ │ │ ├─ ExpandDims{axis=0} [id IG] 192 411c406 < │ │ │ │ │ └─ Isinf [id M] 140 --- > │ │ │ │ │ └─ Isinf [id M] 135 413,414c408,409 < │ │ │ │ └─ 5 [id HX] < │ │ │ ├─ Exp [id N] 139 --- > │ │ │ │ └─ 5 [id HS] > │ │ │ ├─ Exp [id N] 134 416,419c411,414 < │ │ │ └─ ExpandDims{axis=0} [id IR] 230 < │ │ │ └─ Sum{axes=None} [id IS] 222 < │ │ │ └─ Composite{switch(i2, 0, (i0 / i1))} [id IT] 209 < │ │ │ ├─ Composite{...}.2 [id C] 163 --- > │ │ │ └─ ExpandDims{axis=0} [id IM] 215 > │ │ │ └─ Sum{axes=None} [id IN] 207 > │ │ │ └─ Composite{switch(i2, 0, (i0 / i1))} [id IO] 199 > │ │ │ ├─ Composite{...}.2 [id C] 153 421c416 < │ │ │ ├─ ExpandDims{axis=0} [id IL] 202 --- > │ │ │ ├─ ExpandDims{axis=0} [id IG] 192 423c418 < │ │ │ └─ Isinf [id M] 140 --- > │ │ │ └─ Isinf [id M] 135 425c420 < │ │ ├─ ExpandDims{axis=0} [id K] 121 --- > │ │ ├─ ExpandDims{axis=0} [id K] 116 430,431c425,426 < │ └─ [4 1] [id ID] < └─ Split{2}.0 [id IJ] 243 --- > │ └─ [4 1] [id HY] > └─ Split{2}.0 [id IE] 232 433,435c428,430 < IncSubtensor{start:} [id IU] 'x_coord_ordered___grad' 687 < ├─ IncSubtensor{i} [id IV] 645 < │ ├─ Alloc [id IW] 50 --- > IncSubtensor{start:} [id IP] 'x_coord_ordered___grad' 782 > ├─ IncSubtensor{i} [id IQ] 747 > │ ├─ Alloc [id IR] 50 437,461c432,456 < │ │ └─ 5 [id HX] < │ ├─ Subtensor{i} [id IX] 603 < │ │ ├─ SetSubtensor{start:} [id IY] 597 < │ │ │ ├─ Subtensor{::step} [id IZ] 494 < │ │ │ │ ├─ CumOp{-1, add} [id JA] 442 < │ │ │ │ │ └─ Sub [id JB] 430 < │ │ │ │ │ ├─ Subtensor{::step} [id JC] 415 < │ │ │ │ │ │ ├─ DropDims{axis=0} [id JD] 406 < │ │ │ │ │ │ │ └─ SpecifyShape [id JE] 399 < │ │ │ │ │ │ │ ├─ Split{2}.0 [id JF] 392 < │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id JG] 385 < │ │ │ │ │ │ │ │ │ └─ IncSubtensor{i} [id JH] 378 < │ │ │ │ │ │ │ │ │ ├─ IncSubtensor{i} [id JI] 366 < │ │ │ │ │ │ │ │ │ │ ├─ IncSubtensor{i} [id JJ] 354 < │ │ │ │ │ │ │ │ │ │ │ ├─ IncSubtensor{i} [id JK] 342 < │ │ │ │ │ │ │ │ │ │ │ │ ├─ IncSubtensor{i} [id JL] 330 < │ │ │ │ │ │ │ │ │ │ │ │ │ ├─ Alloc [id JM] 10 < │ │ │ │ │ │ │ │ │ │ │ │ │ │ ├─ [[0.]] [id JN] < │ │ │ │ │ │ │ │ │ │ │ │ │ │ ├─ 5 [id HX] < │ │ │ │ │ │ │ │ │ │ │ │ │ │ └─ 2 [id CL] < │ │ │ │ │ │ │ │ │ │ │ │ │ ├─ Neg [id JO] 314 < │ │ │ │ │ │ │ │ │ │ │ │ │ │ └─ Sum{axis=0} [id JP] 298 < │ │ │ │ │ │ │ │ │ │ │ │ │ │ └─ Blockwise{SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=1}, (m,m),(m)->(m)} [id JQ] 286 < │ │ │ │ │ │ │ │ │ │ │ │ │ │ ├─ Transpose{axes=[0, 2, 1]} [id JR] 188 < │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ └─ Switch [id CD] 166 --- > │ │ └─ 5 [id HS] > │ ├─ Subtensor{i} [id IS] 705 > │ │ ├─ SetSubtensor{start:} [id IT] 663 > │ │ │ ├─ Subtensor{::step} [id IU] 575 > │ │ │ │ ├─ CumOp{-1, add} [id IV] 568 > │ │ │ │ │ └─ Sub [id IW] 518 > │ │ │ │ │ ├─ Subtensor{::step} [id IX] 470 > │ │ │ │ │ │ ├─ DropDims{axis=0} [id IY] 418 > │ │ │ │ │ │ │ └─ SpecifyShape [id IZ] 405 > │ │ │ │ │ │ │ ├─ Split{2}.0 [id JA] 389 > │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id JB] 381 > │ │ │ │ │ │ │ │ │ └─ IncSubtensor{i} [id JC] 374 > │ │ │ │ │ │ │ │ │ ├─ IncSubtensor{i} [id JD] 367 > │ │ │ │ │ │ │ │ │ │ ├─ IncSubtensor{i} [id JE] 355 > │ │ │ │ │ │ │ │ │ │ │ ├─ IncSubtensor{i} [id JF] 343 > │ │ │ │ │ │ │ │ │ │ │ │ ├─ IncSubtensor{i} [id JG] 331 > │ │ │ │ │ │ │ │ │ │ │ │ │ ├─ Alloc [id JH] 10 > │ │ │ │ │ │ │ │ │ │ │ │ │ │ ├─ [[0.]] [id JI] > │ │ │ │ │ │ │ │ │ │ │ │ │ │ ├─ 5 [id HS] > │ │ │ │ │ │ │ │ │ │ │ │ │ │ └─ 2 [id CM] > │ │ │ │ │ │ │ │ │ │ │ │ │ ├─ Neg [id JJ] 315 > │ │ │ │ │ │ │ │ │ │ │ │ │ │ └─ Sum{axis=1} [id JK] 299 > │ │ │ │ │ │ │ │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id JL] 287 > │ │ │ │ │ │ │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id JM] 178 > │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ └─ Switch [id CE] 156 463,473c458,469 < │ │ │ │ │ │ │ │ │ │ │ │ │ │ └─ Mul [id JS] 275 < │ │ │ │ │ │ │ │ │ │ │ │ │ │ ├─ [[-1.]] [id JT] < │ │ │ │ │ │ │ │ │ │ │ │ │ │ ├─ SpecifyShape [id JU] 268 < │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ ├─ Split{5}.0 [id JV] 266 < │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ ├─ Composite{((eq(i4, i5) * i3) + switch(i2, 0, (i0 / i1)))} [id HP] 265 < │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ └─ ··· < │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ ├─ 1 [id BY] < │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ └─ [1 1 1 1 1] [id JW] < │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ ├─ NoneConst{None} [id JX] < │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ └─ 1 [id BY] < │ │ │ │ │ │ │ │ │ │ │ │ │ │ └─ Blockwise{SolveTriangular{trans=0, unit_diagonal=False, lower=True, check_finite=True, b_ndim=1}, (m,m),(m)->(m)} [id CC] 207 --- > │ │ │ │ │ │ │ │ │ │ │ │ │ │ └─ Mul [id JN] 276 > │ │ │ │ │ │ │ │ │ │ │ │ │ │ ├─ [[-1.]] [id JO] > │ │ │ │ │ │ │ │ │ │ │ │ │ │ ├─ DimShuffle{order=[x,0]} [id JP] 265 > │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ └─ SpecifyShape [id JQ] 258 > │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ ├─ Split{5}.0 [id JR] 256 > │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ ├─ Composite{((eq(i4, i5) * i3) + switch(i2, 0, (i0 / i1)))} [id HK] 255 > │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ └─ ··· > │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ ├─ 1 [id BY] > │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ └─ [1 1 1 1 1] [id JS] > │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ ├─ NoneConst{None} [id JT] > │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ └─ 1 [id BY] > │ │ │ │ │ │ │ │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=True, check_finite=True, b_ndim=2} [id CD] 197 476,480c472,476 < │ │ │ │ │ │ │ │ │ │ │ │ ├─ Neg [id JY] 317 < │ │ │ │ │ │ │ │ │ │ │ │ │ └─ Sum{axis=0} [id JZ] 301 < │ │ │ │ │ │ │ │ │ │ │ │ │ └─ Blockwise{SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=1}, (m,m),(m)->(m)} [id KA] 288 < │ │ │ │ │ │ │ │ │ │ │ │ │ ├─ Transpose{axes=[0, 2, 1]} [id KB] 191 < │ │ │ │ │ │ │ │ │ │ │ │ │ │ └─ Switch [id DG] 172 --- > │ │ │ │ │ │ │ │ │ │ │ │ ├─ Neg [id JU] 318 > │ │ │ │ │ │ │ │ │ │ │ │ │ └─ Sum{axis=1} [id JV] 302 > │ │ │ │ │ │ │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id JW] 289 > │ │ │ │ │ │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id JX] 181 > │ │ │ │ │ │ │ │ │ │ │ │ │ │ └─ Switch [id DG] 161 482,489c478,486 < │ │ │ │ │ │ │ │ │ │ │ │ │ └─ Mul [id KC] 277 < │ │ │ │ │ │ │ │ │ │ │ │ │ ├─ [[-1.]] [id JT] < │ │ │ │ │ │ │ │ │ │ │ │ │ ├─ SpecifyShape [id KD] 269 < │ │ │ │ │ │ │ │ │ │ │ │ │ │ ├─ Split{5}.1 [id JV] 266 < │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ └─ ··· < │ │ │ │ │ │ │ │ │ │ │ │ │ │ ├─ NoneConst{None} [id JX] < │ │ │ │ │ │ │ │ │ │ │ │ │ │ └─ 1 [id BY] < │ │ │ │ │ │ │ │ │ │ │ │ │ └─ Blockwise{SolveTriangular{trans=0, unit_diagonal=False, lower=True, check_finite=True, b_ndim=1}, (m,m),(m)->(m)} [id DF] 206 --- > │ │ │ │ │ │ │ │ │ │ │ │ │ └─ Mul [id JY] 278 > │ │ │ │ │ │ │ │ │ │ │ │ │ ├─ [[-1.]] [id JO] > │ │ │ │ │ │ │ │ │ │ │ │ │ ├─ DimShuffle{order=[x,0]} [id JZ] 267 > │ │ │ │ │ │ │ │ │ │ │ │ │ │ └─ SpecifyShape [id KA] 259 > │ │ │ │ │ │ │ │ │ │ │ │ │ │ ├─ Split{5}.1 [id JR] 256 > │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ └─ ··· > │ │ │ │ │ │ │ │ │ │ │ │ │ │ ├─ NoneConst{None} [id JT] > │ │ │ │ │ │ │ │ │ │ │ │ │ │ └─ 1 [id BY] > │ │ │ │ │ │ │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=True, check_finite=True, b_ndim=2} [id DF] 196 492,496c489,493 < │ │ │ │ │ │ │ │ │ │ │ ├─ Neg [id KE] 320 < │ │ │ │ │ │ │ │ │ │ │ │ └─ Sum{axis=0} [id KF] 304 < │ │ │ │ │ │ │ │ │ │ │ │ └─ Blockwise{SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=1}, (m,m),(m)->(m)} [id KG] 290 < │ │ │ │ │ │ │ │ │ │ │ │ ├─ Transpose{axes=[0, 2, 1]} [id KH] 190 < │ │ │ │ │ │ │ │ │ │ │ │ │ └─ Switch [id EA] 170 --- > │ │ │ │ │ │ │ │ │ │ │ ├─ Neg [id KB] 321 > │ │ │ │ │ │ │ │ │ │ │ │ └─ Sum{axis=1} [id KC] 305 > │ │ │ │ │ │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id KD] 291 > │ │ │ │ │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id KE] 180 > │ │ │ │ │ │ │ │ │ │ │ │ │ └─ Switch [id DZ] 159 498,505c495,503 < │ │ │ │ │ │ │ │ │ │ │ │ └─ Mul [id KI] 279 < │ │ │ │ │ │ │ │ │ │ │ │ ├─ [[-1.]] [id JT] < │ │ │ │ │ │ │ │ │ │ │ │ ├─ SpecifyShape [id KJ] 270 < │ │ │ │ │ │ │ │ │ │ │ │ │ ├─ Split{5}.2 [id JV] 266 < │ │ │ │ │ │ │ │ │ │ │ │ │ │ └─ ··· < │ │ │ │ │ │ │ │ │ │ │ │ │ ├─ NoneConst{None} [id JX] < │ │ │ │ │ │ │ │ │ │ │ │ │ └─ 1 [id BY] < │ │ │ │ │ │ │ │ │ │ │ │ └─ Blockwise{SolveTriangular{trans=0, unit_diagonal=False, lower=True, check_finite=True, b_ndim=1}, (m,m),(m)->(m)} [id DZ] 205 --- > │ │ │ │ │ │ │ │ │ │ │ │ └─ Mul [id KF] 280 > │ │ │ │ │ │ │ │ │ │ │ │ ├─ [[-1.]] [id JO] > │ │ │ │ │ │ │ │ │ │ │ │ ├─ DimShuffle{order=[x,0]} [id KG] 269 > │ │ │ │ │ │ │ │ │ │ │ │ │ └─ SpecifyShape [id KH] 260 > │ │ │ │ │ │ │ │ │ │ │ │ │ ├─ Split{5}.2 [id JR] 256 > │ │ │ │ │ │ │ │ │ │ │ │ │ │ └─ ··· > │ │ │ │ │ │ │ │ │ │ │ │ │ ├─ NoneConst{None} [id JT] > │ │ │ │ │ │ │ │ │ │ │ │ │ └─ 1 [id BY] > │ │ │ │ │ │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=True, check_finite=True, b_ndim=2} [id DY] 195 507,512c505,510 < │ │ │ │ │ │ │ │ │ │ │ └─ 2 [id EI] < │ │ │ │ │ │ │ │ │ │ ├─ Neg [id KK] 323 < │ │ │ │ │ │ │ │ │ │ │ └─ Sum{axis=0} [id KL] 307 < │ │ │ │ │ │ │ │ │ │ │ └─ Blockwise{SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=1}, (m,m),(m)->(m)} [id KM] 292 < │ │ │ │ │ │ │ │ │ │ │ ├─ Transpose{axes=[0, 2, 1]} [id KN] 189 < │ │ │ │ │ │ │ │ │ │ │ │ └─ Switch [id EU] 168 --- > │ │ │ │ │ │ │ │ │ │ │ └─ 2 [id EH] > │ │ │ │ │ │ │ │ │ │ ├─ Neg [id KI] 324 > │ │ │ │ │ │ │ │ │ │ │ └─ Sum{axis=1} [id KJ] 308 > │ │ │ │ │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id KK] 293 > │ │ │ │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id KL] 179 > │ │ │ │ │ │ │ │ │ │ │ │ └─ Switch [id ES] 157 514,521c512,520 < │ │ │ │ │ │ │ │ │ │ │ └─ Mul [id KO] 281 < │ │ │ │ │ │ │ │ │ │ │ ├─ [[-1.]] [id JT] < │ │ │ │ │ │ │ │ │ │ │ ├─ SpecifyShape [id KP] 271 < │ │ │ │ │ │ │ │ │ │ │ │ ├─ Split{5}.3 [id JV] 266 < │ │ │ │ │ │ │ │ │ │ │ │ │ └─ ··· < │ │ │ │ │ │ │ │ │ │ │ │ ├─ NoneConst{None} [id JX] < │ │ │ │ │ │ │ │ │ │ │ │ └─ 1 [id BY] < │ │ │ │ │ │ │ │ │ │ │ └─ Blockwise{SolveTriangular{trans=0, unit_diagonal=False, lower=True, check_finite=True, b_ndim=1}, (m,m),(m)->(m)} [id ET] 204 --- > │ │ │ │ │ │ │ │ │ │ │ └─ Mul [id KM] 282 > │ │ │ │ │ │ │ │ │ │ │ ├─ [[-1.]] [id JO] > │ │ │ │ │ │ │ │ │ │ │ ├─ DimShuffle{order=[x,0]} [id KN] 271 > │ │ │ │ │ │ │ │ │ │ │ │ └─ SpecifyShape [id KO] 261 > │ │ │ │ │ │ │ │ │ │ │ │ ├─ Split{5}.3 [id JR] 256 > │ │ │ │ │ │ │ │ │ │ │ │ │ └─ ··· > │ │ │ │ │ │ │ │ │ │ │ │ ├─ NoneConst{None} [id JT] > │ │ │ │ │ │ │ │ │ │ │ │ └─ 1 [id BY] > │ │ │ │ │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=True, check_finite=True, b_ndim=2} [id ER] 194 523,528c522,527 < │ │ │ │ │ │ │ │ │ │ └─ 3 [id FC] < │ │ │ │ │ │ │ │ │ ├─ Neg [id KQ] 326 < │ │ │ │ │ │ │ │ │ │ └─ Sum{axis=0} [id KR] 310 < │ │ │ │ │ │ │ │ │ │ └─ Blockwise{SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=1}, (m,m),(m)->(m)} [id KS] 294 < │ │ │ │ │ │ │ │ │ │ ├─ Transpose{axes=[0, 2, 1]} [id KT] 187 < │ │ │ │ │ │ │ │ │ │ │ └─ Switch [id FO] 165 --- > │ │ │ │ │ │ │ │ │ │ └─ 3 [id FA] > │ │ │ │ │ │ │ │ │ ├─ Neg [id KP] 327 > │ │ │ │ │ │ │ │ │ │ └─ Sum{axis=1} [id KQ] 311 > │ │ │ │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id KR] 295 > │ │ │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id KS] 177 > │ │ │ │ │ │ │ │ │ │ │ └─ Switch [id FL] 155 530,537c529,537 < │ │ │ │ │ │ │ │ │ │ └─ Mul [id KU] 283 < │ │ │ │ │ │ │ │ │ │ ├─ [[-1.]] [id JT] < │ │ │ │ │ │ │ │ │ │ ├─ SpecifyShape [id KV] 272 < │ │ │ │ │ │ │ │ │ │ │ ├─ Split{5}.4 [id JV] 266 < │ │ │ │ │ │ │ │ │ │ │ │ └─ ··· < │ │ │ │ │ │ │ │ │ │ │ ├─ NoneConst{None} [id JX] < │ │ │ │ │ │ │ │ │ │ │ └─ 1 [id BY] < │ │ │ │ │ │ │ │ │ │ └─ Blockwise{SolveTriangular{trans=0, unit_diagonal=False, lower=True, check_finite=True, b_ndim=1}, (m,m),(m)->(m)} [id FN] 203 --- > │ │ │ │ │ │ │ │ │ │ └─ Mul [id KT] 284 > │ │ │ │ │ │ │ │ │ │ ├─ [[-1.]] [id JO] > │ │ │ │ │ │ │ │ │ │ ├─ DimShuffle{order=[x,0]} [id KU] 273 > │ │ │ │ │ │ │ │ │ │ │ └─ SpecifyShape [id KV] 262 > │ │ │ │ │ │ │ │ │ │ │ ├─ Split{5}.4 [id JR] 256 > │ │ │ │ │ │ │ │ │ │ │ │ └─ ··· > │ │ │ │ │ │ │ │ │ │ │ ├─ NoneConst{None} [id JT] > │ │ │ │ │ │ │ │ │ │ │ └─ 1 [id BY] > │ │ │ │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=True, check_finite=True, b_ndim=2} [id FK] 193 539c539 < │ │ │ │ │ │ │ │ │ └─ 4 [id FW] --- > │ │ │ │ │ │ │ │ │ └─ 4 [id FT] 543c543 < │ │ │ │ │ │ │ └─ NoneConst{None} [id JX] --- > │ │ │ │ │ │ │ └─ NoneConst{None} [id JT] 545c545 < │ │ │ │ │ └─ Subtensor{::step} [id KY] 99 --- > │ │ │ │ │ └─ Subtensor{::step} [id KY] 94 554,557c554,557 < ├─ Composite{(1.0 + (i0 * i1))} [id LA] 592 < │ ├─ SpecifyShape [id LB] 542 < │ │ ├─ Subtensor{start:stop:step} [id LC] 493 < │ │ │ ├─ CumOp{-1, add} [id JA] 442 --- > ├─ Composite{(1.0 + (i0 * i1))} [id LA] 623 > │ ├─ SpecifyShape [id LB] 581 > │ │ ├─ Subtensor{start:stop:step} [id LC] 574 > │ │ │ ├─ CumOp{-1, add} [id IV] 568 562c562 < │ │ └─ 4 [id IE] --- > │ │ └─ 4 [id HZ] 566,569c566,569 < Sub [id LF] 'y_coord_grad' 414 < ├─ DropDims{axis=0} [id LG] 405 < │ └─ SpecifyShape [id LH] 398 < │ ├─ Split{2}.1 [id JF] 392 --- > Sub [id LF] 'y_coord_grad' 469 > ├─ DropDims{axis=0} [id LG] 417 > │ └─ SpecifyShape [id LH] 404 > │ ├─ Split{2}.1 [id JA] 389 572c572 < │ └─ NoneConst{None} [id JX] --- > │ └─ NoneConst{None} [id JT] 574,579c574,579 < Composite{(1.0 + (i0 * i1))} [id LI] 'sigma_log___grad' 914 < ├─ IncSubtensor{i} [id LJ] 913 < │ ├─ IncSubtensor{i} [id LK] 912 < │ │ ├─ IncSubtensor{i} [id LL] 911 < │ │ │ ├─ IncSubtensor{i} [id LM] 910 < │ │ │ │ ├─ IncSubtensor{i} [id LN] 909 --- > Composite{(1.0 + (i0 * i1))} [id LI] 'sigma_log___grad' 894 > ├─ IncSubtensor{i} [id LJ] 893 > │ ├─ IncSubtensor{i} [id LK] 892 > │ │ ├─ IncSubtensor{i} [id LL] 891 > │ │ │ ├─ IncSubtensor{i} [id LM] 890 > │ │ │ │ ├─ IncSubtensor{i} [id LN] 889 582,591c582,591 < │ │ │ │ │ ├─ AdvancedSubtensor [id LO] 904 < │ │ │ │ │ │ ├─ Composite{switch(i3, ((i0 * i1) - i2), i4)} [id LP] 899 < │ │ │ │ │ │ │ ├─ Add [id LQ] 805 < │ │ │ │ │ │ │ │ ├─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id LR] 729 < │ │ │ │ │ │ │ │ │ ├─ Switch [id LS] 167 < │ │ │ │ │ │ │ │ │ │ ├─ Invert [id LT] 146 < │ │ │ │ │ │ │ │ │ │ │ └─ ExpandDims{axes=[0, 1]} [id LU] 127 < │ │ │ │ │ │ │ │ │ │ │ └─ Any{axes=None} [id LV] 108 < │ │ │ │ │ │ │ │ │ │ │ └─ Isnan [id LW] 84 < │ │ │ │ │ │ │ │ │ │ │ └─ Cholesky{lower=True, destructive=False, on_error='nan'} [id CI] 71 --- > │ │ │ │ │ ├─ AdvancedSubtensor [id LO] 884 > │ │ │ │ │ │ ├─ Composite{switch(i3, ((i0 * i1) - i2), i4)} [id LP] 879 > │ │ │ │ │ │ │ ├─ Add [id LQ] 784 > │ │ │ │ │ │ │ │ ├─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id LR] 707 > │ │ │ │ │ │ │ │ │ ├─ Switch [id LS] 176 > │ │ │ │ │ │ │ │ │ │ ├─ Invert [id LT] 139 > │ │ │ │ │ │ │ │ │ │ │ └─ ExpandDims{axes=[0, 1]} [id LU] 122 > │ │ │ │ │ │ │ │ │ │ │ └─ Any{axes=None} [id LV] 103 > │ │ │ │ │ │ │ │ │ │ │ └─ Isnan [id LW] 82 > │ │ │ │ │ │ │ │ │ │ │ └─ Cholesky{lower=True, destructive=False, on_error='nan'} [id CJ] 71 593,594c593,594 < │ │ │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id LX] 85 < │ │ │ │ │ │ │ │ │ │ │ └─ Cholesky{lower=True, destructive=False, on_error='nan'} [id CI] 71 --- > │ │ │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id LX] 83 > │ │ │ │ │ │ │ │ │ │ │ └─ Cholesky{lower=True, destructive=False, on_error='nan'} [id CJ] 71 596,599c596,599 < │ │ │ │ │ │ │ │ │ │ └─ [[1]] [id LY] < │ │ │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id LZ] 688 < │ │ │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id MA] 646 < │ │ │ │ │ │ │ │ │ ├─ Switch [id LS] 167 --- > │ │ │ │ │ │ │ │ │ │ └─ [[1]] [id CQ] > │ │ │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id LY] 665 > │ │ │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id LZ] 622 > │ │ │ │ │ │ │ │ │ ├─ Switch [id LS] 176 601,604c601,604 < │ │ │ │ │ │ │ │ │ └─ Composite{((i0 * i1) - i2)} [id MB] 604 < │ │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id MC] 413 < │ │ │ │ │ │ │ │ │ │ └─ dot [id MD] 404 < │ │ │ │ │ │ │ │ │ │ ├─ Switch [id LS] 167 --- > │ │ │ │ │ │ │ │ │ └─ Composite{((i0 * i1) - i2)} [id MA] 580 > │ │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id MB] 388 > │ │ │ │ │ │ │ │ │ │ └─ dot [id MC] 380 > │ │ │ │ │ │ │ │ │ │ ├─ Switch [id LS] 176 606,630c606,629 < │ │ │ │ │ │ │ │ │ │ └─ Composite{switch(i3, (i2 + switch(i1, (-i0), 0.0)), 1)} [id ME] 397 < │ │ │ │ │ │ │ │ │ │ ├─ Sum{axis=0} [id MF] 391 < │ │ │ │ │ │ │ │ │ │ │ └─ Mul [id MG] 384 < │ │ │ │ │ │ │ │ │ │ │ ├─ Blockwise{dot, (i00,i01),(i10,i11)->(o00,o01)} [id MH] 313 < │ │ │ │ │ │ │ │ │ │ │ │ ├─ ExpandDims{axis=2} [id MI] 297 < │ │ │ │ │ │ │ │ │ │ │ │ │ └─ Blockwise{SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=1}, (m,m),(m)->(m)} [id JQ] 286 < │ │ │ │ │ │ │ │ │ │ │ │ │ └─ ··· < │ │ │ │ │ │ │ │ │ │ │ │ └─ ExpandDims{axis=1} [id MJ] 220 < │ │ │ │ │ │ │ │ │ │ │ │ └─ Blockwise{SolveTriangular{trans=0, unit_diagonal=False, lower=True, check_finite=True, b_ndim=1}, (m,m),(m)->(m)} [id CC] 207 < │ │ │ │ │ │ │ │ │ │ │ │ └─ ··· < │ │ │ │ │ │ │ │ │ │ │ └─ Blockwise{Tri{dtype='float64'}, (),(),()->(o00,o01)} [id MK] 377 < │ │ │ │ │ │ │ │ │ │ │ ├─ Blockwise{Subtensor{i}, (i00),()->()} [id ML] 365 < │ │ │ │ │ │ │ │ │ │ │ │ ├─ Blockwise{Subtensor{start:}, (i00),()->(o00)} [id MM] 353 < │ │ │ │ │ │ │ │ │ │ │ │ │ ├─ Blockwise{Shape, (i00,i01)->(o00)} [id MN] 341 < │ │ │ │ │ │ │ │ │ │ │ │ │ │ └─ Neg [id MO] 329 < │ │ │ │ │ │ │ │ │ │ │ │ │ │ └─ Blockwise{dot, (i00,i01),(i10,i11)->(o00,o01)} [id MH] 313 < │ │ │ │ │ │ │ │ │ │ │ │ │ │ └─ ··· < │ │ │ │ │ │ │ │ │ │ │ │ │ └─ [-2] [id MP] < │ │ │ │ │ │ │ │ │ │ │ │ └─ [0] [id MQ] < │ │ │ │ │ │ │ │ │ │ │ ├─ Blockwise{Subtensor{i}, (i00),()->()} [id MR] 364 < │ │ │ │ │ │ │ │ │ │ │ │ ├─ Blockwise{Subtensor{start:}, (i00),()->(o00)} [id MM] 353 < │ │ │ │ │ │ │ │ │ │ │ │ │ └─ ··· < │ │ │ │ │ │ │ │ │ │ │ │ └─ [1] [id MS] < │ │ │ │ │ │ │ │ │ │ │ └─ [0] [id CO] < │ │ │ │ │ │ │ │ │ │ ├─ ExpandDims{axes=[0, 1]} [id DB] 144 --- > │ │ │ │ │ │ │ │ │ │ └─ Composite{switch(i4, (i3 + switch(i2, (-1.0 * i0 * i1), 0.0)), 1)} [id MD] 373 > │ │ │ │ │ │ │ │ │ │ ├─ dot [id ME] 298 > │ │ │ │ │ │ │ │ │ │ │ ├─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id JL] 287 > │ │ │ │ │ │ │ │ │ │ │ │ └─ ··· > │ │ │ │ │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id CC] 205 > │ │ │ │ │ │ │ │ │ │ │ └─ ··· > │ │ │ │ │ │ │ │ │ │ ├─ Tri{dtype='float64'} [id MF] 366 > │ │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id MG] 354 > │ │ │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{start:} [id MH] 342 > │ │ │ │ │ │ │ │ │ │ │ │ │ ├─ Shape [id MI] 330 > │ │ │ │ │ │ │ │ │ │ │ │ │ │ └─ Neg [id MJ] 314 > │ │ │ │ │ │ │ │ │ │ │ │ │ │ └─ dot [id ME] 298 > │ │ │ │ │ │ │ │ │ │ │ │ │ │ └─ ··· > │ │ │ │ │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id MK] 12 > │ │ │ │ │ │ │ │ │ │ │ │ │ └─ -2 [id ML] > │ │ │ │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id MM] 13 > │ │ │ │ │ │ │ │ │ │ │ │ └─ 0 [id MN] > │ │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id MO] 353 > │ │ │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{start:} [id MH] 342 > │ │ │ │ │ │ │ │ │ │ │ │ │ └─ ··· > │ │ │ │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id MP] 11 > │ │ │ │ │ │ │ │ │ │ │ │ └─ 1 [id MQ] > │ │ │ │ │ │ │ │ │ │ │ └─ 0 [id E] > │ │ │ │ │ │ │ │ │ │ ├─ ExpandDims{axes=[0, 1]} [id CF] 138 632,638c631,637 < │ │ │ │ │ │ │ │ │ │ ├─ SetSubtensor{:stop, start:stop} [id MT] 352 < │ │ │ │ │ │ │ │ │ │ │ ├─ Alloc [id MU] 9 < │ │ │ │ │ │ │ │ │ │ │ │ ├─ [[0.]] [id JN] < │ │ │ │ │ │ │ │ │ │ │ │ ├─ 2 [id CL] < │ │ │ │ │ │ │ │ │ │ │ │ └─ 2 [id CL] < │ │ │ │ │ │ │ │ │ │ │ ├─ AdvancedSetSubtensor [id MV] 340 < │ │ │ │ │ │ │ │ │ │ │ │ ├─ Alloc [id CK] 1 --- > │ │ │ │ │ │ │ │ │ │ ├─ SetSubtensor{:stop, start:stop} [id MR] 341 > │ │ │ │ │ │ │ │ │ │ │ ├─ Alloc [id MS] 9 > │ │ │ │ │ │ │ │ │ │ │ │ ├─ [[0.]] [id JI] > │ │ │ │ │ │ │ │ │ │ │ │ ├─ 2 [id CM] > │ │ │ │ │ │ │ │ │ │ │ │ └─ 2 [id CM] > │ │ │ │ │ │ │ │ │ │ │ ├─ AdvancedSetSubtensor [id MT] 329 > │ │ │ │ │ │ │ │ │ │ │ │ ├─ Alloc [id CL] 1 640,645c639,644 < │ │ │ │ │ │ │ │ │ │ │ │ ├─ True_div [id MW] 328 < │ │ │ │ │ │ │ │ │ │ │ │ │ ├─ Neg [id MX] 312 < │ │ │ │ │ │ │ │ │ │ │ │ │ │ └─ ExpandDims{axis=0} [id MY] 296 < │ │ │ │ │ │ │ │ │ │ │ │ │ │ └─ Sum{axes=None} [id MZ] 285 < │ │ │ │ │ │ │ │ │ │ │ │ │ │ └─ DropDims{axis=1} [id NA] 274 < │ │ │ │ │ │ │ │ │ │ │ │ │ │ └─ SpecifyShape [id JU] 268 --- > │ │ │ │ │ │ │ │ │ │ │ │ ├─ True_div [id MU] 313 > │ │ │ │ │ │ │ │ │ │ │ │ │ ├─ Neg [id MV] 297 > │ │ │ │ │ │ │ │ │ │ │ │ │ │ └─ ExpandDims{axis=0} [id MW] 286 > │ │ │ │ │ │ │ │ │ │ │ │ │ │ └─ Sum{axes=None} [id MX] 275 > │ │ │ │ │ │ │ │ │ │ │ │ │ │ └─ DropDims{axis=1} [id MY] 264 > │ │ │ │ │ │ │ │ │ │ │ │ │ │ └─ SpecifyShape [id JQ] 258 647c646 < │ │ │ │ │ │ │ │ │ │ │ │ │ └─ ExtractDiag{offset=0, axis1=0, axis2=1, view=False} [id CH] 82 --- > │ │ │ │ │ │ │ │ │ │ │ │ │ └─ ExtractDiag{offset=0, axis1=0, axis2=1, view=False} [id CI] 81 649,654c648,653 < │ │ │ │ │ │ │ │ │ │ │ │ ├─ [0 1] [id CN] < │ │ │ │ │ │ │ │ │ │ │ │ └─ [0 1] [id CN] < │ │ │ │ │ │ │ │ │ │ │ ├─ 2 [id NB] < │ │ │ │ │ │ │ │ │ │ │ ├─ 0 [id NC] < │ │ │ │ │ │ │ │ │ │ │ └─ 2 [id NB] < │ │ │ │ │ │ │ │ │ │ └─ Invert [id LT] 146 --- > │ │ │ │ │ │ │ │ │ │ │ │ ├─ [0 1] [id CO] > │ │ │ │ │ │ │ │ │ │ │ │ └─ [0 1] [id CO] > │ │ │ │ │ │ │ │ │ │ │ ├─ 2 [id MZ] > │ │ │ │ │ │ │ │ │ │ │ ├─ 0 [id NA] > │ │ │ │ │ │ │ │ │ │ │ └─ 2 [id MZ] > │ │ │ │ │ │ │ │ │ │ └─ Invert [id LT] 139 656,661c655,660 < │ │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id ND] 541 < │ │ │ │ │ │ │ │ │ │ └─ Tri{dtype='float64'} [id NE] 492 < │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id NF] 441 < │ │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{start:} [id NG] 429 < │ │ │ │ │ │ │ │ │ │ │ │ ├─ Shape [id NH] 412 < │ │ │ │ │ │ │ │ │ │ │ │ │ └─ dot [id MD] 404 --- > │ │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id NB] 517 > │ │ │ │ │ │ │ │ │ │ └─ Tri{dtype='float64'} [id NC] 468 > │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id ND] 416 > │ │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{start:} [id NE] 403 > │ │ │ │ │ │ │ │ │ │ │ │ ├─ Shape [id NF] 387 > │ │ │ │ │ │ │ │ │ │ │ │ │ └─ dot [id MC] 380 663,668c662,667 < │ │ │ │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id NI] 12 < │ │ │ │ │ │ │ │ │ │ │ │ └─ -2 [id NJ] < │ │ │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id NK] 13 < │ │ │ │ │ │ │ │ │ │ │ └─ 0 [id NL] < │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id NM] 440 < │ │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{start:} [id NG] 429 --- > │ │ │ │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id MK] 12 > │ │ │ │ │ │ │ │ │ │ │ │ └─ ··· > │ │ │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id MM] 13 > │ │ │ │ │ │ │ │ │ │ │ └─ ··· > │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id NG] 415 > │ │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{start:} [id NE] 403 670,671c669,670 < │ │ │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id NN] 11 < │ │ │ │ │ │ │ │ │ │ │ └─ 1 [id NO] --- > │ │ │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id MP] 11 > │ │ │ │ │ │ │ │ │ │ │ └─ ··· 673,675c672,674 < │ │ │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id NP] 598 < │ │ │ │ │ │ │ │ │ └─ AdvancedSetSubtensor [id NQ] 591 < │ │ │ │ │ │ │ │ │ ├─ Alloc [id NR] 540 --- > │ │ │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id NH] 573 > │ │ │ │ │ │ │ │ │ └─ AdvancedSetSubtensor [id NI] 567 > │ │ │ │ │ │ │ │ │ ├─ Alloc [id NJ] 516 677,682c676,681 < │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id NS] 491 < │ │ │ │ │ │ │ │ │ │ │ ├─ Shape [id NT] 439 < │ │ │ │ │ │ │ │ │ │ │ │ └─ Mul [id NU] 428 < │ │ │ │ │ │ │ │ │ │ │ │ ├─ [0.5] [id NV] < │ │ │ │ │ │ │ │ │ │ │ │ └─ ExtractDiag{offset=0, axis1=0, axis2=1, view=False} [id NW] 411 < │ │ │ │ │ │ │ │ │ │ │ │ └─ dot [id MD] 404 --- > │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id NK] 467 > │ │ │ │ │ │ │ │ │ │ │ ├─ Shape [id NL] 414 > │ │ │ │ │ │ │ │ │ │ │ │ └─ Mul [id NM] 402 > │ │ │ │ │ │ │ │ │ │ │ │ ├─ [0.5] [id NN] > │ │ │ │ │ │ │ │ │ │ │ │ └─ ExtractDiag{offset=0, axis1=0, axis2=1, view=False} [id NO] 386 > │ │ │ │ │ │ │ │ │ │ │ │ └─ dot [id MC] 380 684,686c683,685 < │ │ │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id NX] 2 < │ │ │ │ │ │ │ │ │ │ │ └─ -1 [id NY] < │ │ │ │ │ │ │ │ │ │ └─ Subtensor{i} [id NS] 491 --- > │ │ │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id NP] 2 > │ │ │ │ │ │ │ │ │ │ │ └─ -1 [id NQ] > │ │ │ │ │ │ │ │ │ │ └─ Subtensor{i} [id NK] 467 688c687 < │ │ │ │ │ │ │ │ │ ├─ Mul [id NU] 428 --- > │ │ │ │ │ │ │ │ │ ├─ Mul [id NM] 402 690c689 < │ │ │ │ │ │ │ │ │ ├─ ARange{dtype='int64'} [id NZ] 539 --- > │ │ │ │ │ │ │ │ │ ├─ ARange{dtype='int64'} [id NR] 515 692c691 < │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id NS] 491 --- > │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id NK] 467 695c694 < │ │ │ │ │ │ │ │ │ └─ ARange{dtype='int64'} [id NZ] 539 --- > │ │ │ │ │ │ │ │ │ └─ ARange{dtype='int64'} [id NR] 515 697,698c696,697 < │ │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id OA] 771 < │ │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id LR] 729 --- > │ │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id NS] 749 > │ │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id LR] 707 700,704c699,703 < │ │ │ │ │ │ │ ├─ Tri{dtype='float64'} [id OB] 894 < │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id OC] 881 < │ │ │ │ │ │ │ │ │ ├─ Subtensor{start:} [id OD] 858 < │ │ │ │ │ │ │ │ │ │ ├─ Shape [id OE] 831 < │ │ │ │ │ │ │ │ │ │ │ └─ Add [id LQ] 805 --- > │ │ │ │ │ │ │ ├─ Tri{dtype='float64'} [id NT] 874 > │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id NU] 860 > │ │ │ │ │ │ │ │ │ ├─ Subtensor{start:} [id NV] 836 > │ │ │ │ │ │ │ │ │ │ ├─ Shape [id NW] 810 > │ │ │ │ │ │ │ │ │ │ │ └─ Add [id LQ] 784 706c705 < │ │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id NI] 12 --- > │ │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id MK] 12 708c707 < │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id NK] 13 --- > │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id MM] 13 710,711c709,710 < │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id OF] 880 < │ │ │ │ │ │ │ │ │ ├─ Subtensor{start:} [id OD] 858 --- > │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id NX] 859 > │ │ │ │ │ │ │ │ │ ├─ Subtensor{start:} [id NV] 836 713c712 < │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id NN] 11 --- > │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id MP] 11 716,717c715,716 < │ │ │ │ │ │ │ ├─ AdvancedSetSubtensor [id OG] 879 < │ │ │ │ │ │ │ │ ├─ Alloc [id OH] 857 --- > │ │ │ │ │ │ │ ├─ AdvancedSetSubtensor [id NY] 861 > │ │ │ │ │ │ │ │ ├─ Alloc [id NZ] 838 719,722c718,721 < │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id OI] 830 < │ │ │ │ │ │ │ │ │ │ ├─ Shape [id OJ] 804 < │ │ │ │ │ │ │ │ │ │ │ └─ ExtractDiag{offset=0, axis1=0, axis2=1, view=False} [id OK] 770 < │ │ │ │ │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id LR] 729 --- > │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id OA] 811 > │ │ │ │ │ │ │ │ │ │ ├─ Shape [id OB] 785 > │ │ │ │ │ │ │ │ │ │ │ └─ ExtractDiag{offset=0, axis1=0, axis2=1, view=False} [id OC] 750 > │ │ │ │ │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id LR] 707 724c723 < │ │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id NX] 2 --- > │ │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id NP] 2 726c725 < │ │ │ │ │ │ │ │ │ └─ Subtensor{i} [id OI] 830 --- > │ │ │ │ │ │ │ │ │ └─ Subtensor{i} [id OA] 811 728c727 < │ │ │ │ │ │ │ │ ├─ ExtractDiag{offset=0, axis1=0, axis2=1, view=False} [id OK] 770 --- > │ │ │ │ │ │ │ │ ├─ ExtractDiag{offset=0, axis1=0, axis2=1, view=False} [id OC] 750 730c729 < │ │ │ │ │ │ │ │ ├─ ARange{dtype='int64'} [id OL] 856 --- > │ │ │ │ │ │ │ │ ├─ ARange{dtype='int64'} [id OD] 837 732c731 < │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id OI] 830 --- > │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id OA] 811 735c734 < │ │ │ │ │ │ │ │ └─ ARange{dtype='int64'} [id OL] 856 --- > │ │ │ │ │ │ │ │ └─ ARange{dtype='int64'} [id OD] 837 737c736 < │ │ │ │ │ │ │ ├─ Invert [id LT] 146 --- > │ │ │ │ │ │ │ ├─ Invert [id LT] 139 739,741c738,740 < │ │ │ │ │ │ │ └─ [[nan]] [id OM] < │ │ │ │ │ │ ├─ [0 1] [id CN] < │ │ │ │ │ │ └─ [0 1] [id CN] --- > │ │ │ │ │ │ │ └─ [[nan]] [id OE] > │ │ │ │ │ │ ├─ [0 1] [id CO] > │ │ │ │ │ │ └─ [0 1] [id CO] 743,750c742,749 < │ │ │ │ ├─ AdvancedSubtensor [id ON] 905 < │ │ │ │ │ ├─ Composite{switch(i4, (((i0 + i1) * i2) - i3), i5)} [id OO] 900 < │ │ │ │ │ │ ├─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id OP] 730 < │ │ │ │ │ │ │ ├─ Switch [id OQ] 173 < │ │ │ │ │ │ │ │ ├─ Invert [id OR] 155 < │ │ │ │ │ │ │ │ │ └─ ExpandDims{axes=[0, 1]} [id OS] 136 < │ │ │ │ │ │ │ │ │ └─ Any{axes=None} [id OT] 117 < │ │ │ │ │ │ │ │ │ └─ Isnan [id OU] 96 --- > │ │ │ │ ├─ AdvancedSubtensor [id OF] 885 > │ │ │ │ │ ├─ Composite{switch(i4, (((i0 + i1) * i2) - i3), i5)} [id OG] 880 > │ │ │ │ │ │ ├─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id OH] 708 > │ │ │ │ │ │ │ ├─ Switch [id OI] 162 > │ │ │ │ │ │ │ │ ├─ Invert [id OJ] 145 > │ │ │ │ │ │ │ │ │ └─ ExpandDims{axes=[0, 1]} [id OK] 131 > │ │ │ │ │ │ │ │ │ └─ Any{axes=None} [id OL] 112 > │ │ │ │ │ │ │ │ │ └─ Isnan [id OM] 91 753c752 < │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id OV] 97 --- > │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id ON] 92 756,759c755,758 < │ │ │ │ │ │ │ │ └─ [[1]] [id LY] < │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id OW] 689 < │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id OX] 647 < │ │ │ │ │ │ │ ├─ Switch [id OQ] 173 --- > │ │ │ │ │ │ │ │ └─ [[1]] [id CQ] > │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id OO] 666 > │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id OP] 624 > │ │ │ │ │ │ │ ├─ Switch [id OI] 162 761,764c760,763 < │ │ │ │ │ │ │ └─ Composite{((i0 * i1) - i2)} [id OY] 605 < │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id OZ] 418 < │ │ │ │ │ │ │ │ └─ dot [id PA] 407 < │ │ │ │ │ │ │ │ ├─ Switch [id OQ] 173 --- > │ │ │ │ │ │ │ └─ Composite{((i0 * i1) - i2)} [id OQ] 582 > │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id OR] 392 > │ │ │ │ │ │ │ │ └─ dot [id OS] 382 > │ │ │ │ │ │ │ │ ├─ Switch [id OI] 162 766,790c765,788 < │ │ │ │ │ │ │ │ └─ Composite{switch(i3, (i2 + switch(i1, (-i0), 0.0)), 1)} [id PB] 400 < │ │ │ │ │ │ │ │ ├─ Sum{axis=0} [id PC] 393 < │ │ │ │ │ │ │ │ │ └─ Mul [id PD] 386 < │ │ │ │ │ │ │ │ │ ├─ Blockwise{dot, (i00,i01),(i10,i11)->(o00,o01)} [id PE] 316 < │ │ │ │ │ │ │ │ │ │ ├─ ExpandDims{axis=2} [id PF] 300 < │ │ │ │ │ │ │ │ │ │ │ └─ Blockwise{SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=1}, (m,m),(m)->(m)} [id KA] 288 < │ │ │ │ │ │ │ │ │ │ │ └─ ··· < │ │ │ │ │ │ │ │ │ │ └─ ExpandDims{axis=1} [id PG] 218 < │ │ │ │ │ │ │ │ │ │ └─ Blockwise{SolveTriangular{trans=0, unit_diagonal=False, lower=True, check_finite=True, b_ndim=1}, (m,m),(m)->(m)} [id DF] 206 < │ │ │ │ │ │ │ │ │ │ └─ ··· < │ │ │ │ │ │ │ │ │ └─ Blockwise{Tri{dtype='float64'}, (),(),()->(o00,o01)} [id PH] 379 < │ │ │ │ │ │ │ │ │ ├─ Blockwise{Subtensor{i}, (i00),()->()} [id PI] 368 < │ │ │ │ │ │ │ │ │ │ ├─ Blockwise{Subtensor{start:}, (i00),()->(o00)} [id PJ] 356 < │ │ │ │ │ │ │ │ │ │ │ ├─ Blockwise{Shape, (i00,i01)->(o00)} [id PK] 344 < │ │ │ │ │ │ │ │ │ │ │ │ └─ Neg [id PL] 332 < │ │ │ │ │ │ │ │ │ │ │ │ └─ Blockwise{dot, (i00,i01),(i10,i11)->(o00,o01)} [id PE] 316 < │ │ │ │ │ │ │ │ │ │ │ │ └─ ··· < │ │ │ │ │ │ │ │ │ │ │ └─ [-2] [id MP] < │ │ │ │ │ │ │ │ │ │ └─ [0] [id MQ] < │ │ │ │ │ │ │ │ │ ├─ Blockwise{Subtensor{i}, (i00),()->()} [id PM] 367 < │ │ │ │ │ │ │ │ │ │ ├─ Blockwise{Subtensor{start:}, (i00),()->(o00)} [id PJ] 356 < │ │ │ │ │ │ │ │ │ │ │ └─ ··· < │ │ │ │ │ │ │ │ │ │ └─ [1] [id MS] < │ │ │ │ │ │ │ │ │ └─ [0] [id CO] < │ │ │ │ │ │ │ │ ├─ ExpandDims{axes=[0, 1]} [id DV] 154 --- > │ │ │ │ │ │ │ │ └─ Composite{switch(i4, (i3 + switch(i2, (-1.0 * i0 * i1), 0.0)), 1)} [id OT] 375 > │ │ │ │ │ │ │ │ ├─ dot [id OU] 301 > │ │ │ │ │ │ │ │ │ ├─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id JW] 289 > │ │ │ │ │ │ │ │ │ │ └─ ··· > │ │ │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id DE] 204 > │ │ │ │ │ │ │ │ │ └─ ··· > │ │ │ │ │ │ │ │ ├─ Tri{dtype='float64'} [id OV] 368 > │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id OW] 357 > │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{start:} [id OX] 345 > │ │ │ │ │ │ │ │ │ │ │ ├─ Shape [id OY] 333 > │ │ │ │ │ │ │ │ │ │ │ │ └─ Neg [id OZ] 317 > │ │ │ │ │ │ │ │ │ │ │ │ └─ dot [id OU] 301 > │ │ │ │ │ │ │ │ │ │ │ │ └─ ··· > │ │ │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id MK] 12 > │ │ │ │ │ │ │ │ │ │ │ └─ ··· > │ │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id MM] 13 > │ │ │ │ │ │ │ │ │ │ └─ ··· > │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id PA] 356 > │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{start:} [id OX] 345 > │ │ │ │ │ │ │ │ │ │ │ └─ ··· > │ │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id MP] 11 > │ │ │ │ │ │ │ │ │ │ └─ ··· > │ │ │ │ │ │ │ │ │ └─ 0 [id E] > │ │ │ │ │ │ │ │ ├─ ExpandDims{axes=[0, 1]} [id DH] 144 792,793c790,791 < │ │ │ │ │ │ │ │ ├─ SetSubtensor{:stop, start:stop} [id PN] 355 < │ │ │ │ │ │ │ │ │ ├─ Alloc [id MU] 9 --- > │ │ │ │ │ │ │ │ ├─ SetSubtensor{:stop, start:stop} [id PB] 344 > │ │ │ │ │ │ │ │ │ ├─ Alloc [id MS] 9 795,796c793,794 < │ │ │ │ │ │ │ │ │ ├─ AdvancedSetSubtensor [id PO] 343 < │ │ │ │ │ │ │ │ │ │ ├─ Alloc [id CK] 1 --- > │ │ │ │ │ │ │ │ │ ├─ AdvancedSetSubtensor [id PC] 332 > │ │ │ │ │ │ │ │ │ │ ├─ Alloc [id CL] 1 798,803c796,801 < │ │ │ │ │ │ │ │ │ │ ├─ True_div [id PP] 331 < │ │ │ │ │ │ │ │ │ │ │ ├─ Neg [id PQ] 315 < │ │ │ │ │ │ │ │ │ │ │ │ └─ ExpandDims{axis=0} [id PR] 299 < │ │ │ │ │ │ │ │ │ │ │ │ └─ Sum{axes=None} [id PS] 287 < │ │ │ │ │ │ │ │ │ │ │ │ └─ DropDims{axis=1} [id PT] 276 < │ │ │ │ │ │ │ │ │ │ │ │ └─ SpecifyShape [id KD] 269 --- > │ │ │ │ │ │ │ │ │ │ ├─ True_div [id PD] 316 > │ │ │ │ │ │ │ │ │ │ │ ├─ Neg [id PE] 300 > │ │ │ │ │ │ │ │ │ │ │ │ └─ ExpandDims{axis=0} [id PF] 288 > │ │ │ │ │ │ │ │ │ │ │ │ └─ Sum{axes=None} [id PG] 277 > │ │ │ │ │ │ │ │ │ │ │ │ └─ DropDims{axis=1} [id PH] 266 > │ │ │ │ │ │ │ │ │ │ │ │ └─ SpecifyShape [id KA] 259 805c803 < │ │ │ │ │ │ │ │ │ │ │ └─ ExtractDiag{offset=0, axis1=0, axis2=1, view=False} [id DK] 95 --- > │ │ │ │ │ │ │ │ │ │ │ └─ ExtractDiag{offset=0, axis1=0, axis2=1, view=False} [id DK] 90 807,812c805,810 < │ │ │ │ │ │ │ │ │ │ ├─ [0 1] [id CN] < │ │ │ │ │ │ │ │ │ │ └─ [0 1] [id CN] < │ │ │ │ │ │ │ │ │ ├─ 2 [id NB] < │ │ │ │ │ │ │ │ │ ├─ 0 [id NC] < │ │ │ │ │ │ │ │ │ └─ 2 [id NB] < │ │ │ │ │ │ │ │ └─ Invert [id OR] 155 --- > │ │ │ │ │ │ │ │ │ │ ├─ [0 1] [id CO] > │ │ │ │ │ │ │ │ │ │ └─ [0 1] [id CO] > │ │ │ │ │ │ │ │ │ ├─ 2 [id MZ] > │ │ │ │ │ │ │ │ │ ├─ 0 [id NA] > │ │ │ │ │ │ │ │ │ └─ 2 [id MZ] > │ │ │ │ │ │ │ │ └─ Invert [id OJ] 145 814,819c812,817 < │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id PU] 545 < │ │ │ │ │ │ │ │ └─ Tri{dtype='float64'} [id PV] 496 < │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id PW] 445 < │ │ │ │ │ │ │ │ │ ├─ Subtensor{start:} [id PX] 432 < │ │ │ │ │ │ │ │ │ │ ├─ Shape [id PY] 417 < │ │ │ │ │ │ │ │ │ │ │ └─ dot [id PA] 407 --- > │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id PI] 521 > │ │ │ │ │ │ │ │ └─ Tri{dtype='float64'} [id PJ] 472 > │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id PK] 421 > │ │ │ │ │ │ │ │ │ ├─ Subtensor{start:} [id PL] 407 > │ │ │ │ │ │ │ │ │ │ ├─ Shape [id PM] 391 > │ │ │ │ │ │ │ │ │ │ │ └─ dot [id OS] 382 821c819 < │ │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id NI] 12 --- > │ │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id MK] 12 823c821 < │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id NK] 13 --- > │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id MM] 13 825,826c823,824 < │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id PZ] 444 < │ │ │ │ │ │ │ │ │ ├─ Subtensor{start:} [id PX] 432 --- > │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id PN] 420 > │ │ │ │ │ │ │ │ │ ├─ Subtensor{start:} [id PL] 407 828c826 < │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id NN] 11 --- > │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id MP] 11 831,833c829,831 < │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id QA] 599 < │ │ │ │ │ │ │ └─ AdvancedSetSubtensor [id QB] 593 < │ │ │ │ │ │ │ ├─ Alloc [id QC] 544 --- > │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id PO] 576 > │ │ │ │ │ │ │ └─ AdvancedSetSubtensor [id PP] 569 > │ │ │ │ │ │ │ ├─ Alloc [id PQ] 520 835,840c833,838 < │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id QD] 495 < │ │ │ │ │ │ │ │ │ ├─ Shape [id QE] 443 < │ │ │ │ │ │ │ │ │ │ └─ Mul [id QF] 431 < │ │ │ │ │ │ │ │ │ │ ├─ [0.5] [id NV] < │ │ │ │ │ │ │ │ │ │ └─ ExtractDiag{offset=0, axis1=0, axis2=1, view=False} [id QG] 416 < │ │ │ │ │ │ │ │ │ │ └─ dot [id PA] 407 --- > │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id PR] 471 > │ │ │ │ │ │ │ │ │ ├─ Shape [id PS] 419 > │ │ │ │ │ │ │ │ │ │ └─ Mul [id PT] 406 > │ │ │ │ │ │ │ │ │ │ ├─ [0.5] [id NN] > │ │ │ │ │ │ │ │ │ │ └─ ExtractDiag{offset=0, axis1=0, axis2=1, view=False} [id PU] 390 > │ │ │ │ │ │ │ │ │ │ └─ dot [id OS] 382 842c840 < │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id NX] 2 --- > │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id NP] 2 844c842 < │ │ │ │ │ │ │ │ └─ Subtensor{i} [id QD] 495 --- > │ │ │ │ │ │ │ │ └─ Subtensor{i} [id PR] 471 846c844 < │ │ │ │ │ │ │ ├─ Mul [id QF] 431 --- > │ │ │ │ │ │ │ ├─ Mul [id PT] 406 848c846 < │ │ │ │ │ │ │ ├─ ARange{dtype='int64'} [id QH] 543 --- > │ │ │ │ │ │ │ ├─ ARange{dtype='int64'} [id PV] 519 850c848 < │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id QD] 495 --- > │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id PR] 471 853c851 < │ │ │ │ │ │ │ └─ ARange{dtype='int64'} [id QH] 543 --- > │ │ │ │ │ │ │ └─ ARange{dtype='int64'} [id PV] 519 855,857c853,855 < │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id QI] 779 < │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id QJ] 739 < │ │ │ │ │ │ │ ├─ Switch [id OQ] 173 --- > │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id PW] 758 > │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id PX] 717 > │ │ │ │ │ │ │ ├─ Switch [id OI] 162 859,861c857,859 < │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id QK] 698 < │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id QL] 656 < │ │ │ │ │ │ │ ├─ Switch [id OQ] 173 --- > │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id PY] 675 > │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id PZ] 633 > │ │ │ │ │ │ │ ├─ Switch [id OI] 162 863,864c861,862 < │ │ │ │ │ │ │ └─ Composite{((i0 * i1) - i2)} [id QM] 614 < │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id OZ] 418 --- > │ │ │ │ │ │ │ └─ Composite{((i0 * i1) - i2)} [id QA] 591 > │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id OR] 392 866,868c864,866 < │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id QN] 554 < │ │ │ │ │ │ │ │ └─ Tri{dtype='float64'} [id QO] 505 < │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id PW] 445 --- > │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id QB] 530 > │ │ │ │ │ │ │ │ └─ Tri{dtype='float64'} [id QC] 481 > │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id PK] 421 870,871c868,869 < │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id QP] 454 < │ │ │ │ │ │ │ │ │ ├─ Subtensor{start:} [id PX] 432 --- > │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id QD] 430 > │ │ │ │ │ │ │ │ │ ├─ Subtensor{start:} [id PL] 407 873,874c871,872 < │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id QQ] 49 < │ │ │ │ │ │ │ │ │ └─ 1 [id QR] --- > │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id QE] 49 > │ │ │ │ │ │ │ │ │ └─ 1 [id QF] 876c874 < │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id QA] 599 --- > │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id PO] 576 878,884c876,882 < │ │ │ │ │ │ ├─ Tri{dtype='float64'} [id QS] 895 < │ │ │ │ │ │ │ ├─ Subtensor{i} [id QT] 884 < │ │ │ │ │ │ │ │ ├─ Subtensor{start:} [id QU] 863 < │ │ │ │ │ │ │ │ │ ├─ Shape [id QV] 837 < │ │ │ │ │ │ │ │ │ │ └─ Add [id QW] 811 < │ │ │ │ │ │ │ │ │ │ ├─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id QX] 738 < │ │ │ │ │ │ │ │ │ │ │ ├─ Switch [id OQ] 173 --- > │ │ │ │ │ │ ├─ Tri{dtype='float64'} [id QG] 875 > │ │ │ │ │ │ │ ├─ Subtensor{i} [id QH] 864 > │ │ │ │ │ │ │ │ ├─ Subtensor{start:} [id QI] 843 > │ │ │ │ │ │ │ │ │ ├─ Shape [id QJ] 817 > │ │ │ │ │ │ │ │ │ │ └─ Add [id QK] 791 > │ │ │ │ │ │ │ │ │ │ ├─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id QL] 716 > │ │ │ │ │ │ │ │ │ │ │ ├─ Switch [id OI] 162 886,888c884,886 < │ │ │ │ │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id QY] 697 < │ │ │ │ │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id QZ] 655 < │ │ │ │ │ │ │ │ │ │ │ ├─ Switch [id OQ] 173 --- > │ │ │ │ │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id QM] 674 > │ │ │ │ │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id QN] 632 > │ │ │ │ │ │ │ │ │ │ │ ├─ Switch [id OI] 162 890,891c888,889 < │ │ │ │ │ │ │ │ │ │ │ └─ Composite{((i0 * i1) - i2)} [id RA] 613 < │ │ │ │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id OZ] 418 --- > │ │ │ │ │ │ │ │ │ │ │ └─ Composite{((i0 * i1) - i2)} [id QO] 590 > │ │ │ │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id OR] 392 893,895c891,893 < │ │ │ │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id RB] 553 < │ │ │ │ │ │ │ │ │ │ │ │ └─ Tri{dtype='float64'} [id RC] 504 < │ │ │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id PW] 445 --- > │ │ │ │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id QP] 529 > │ │ │ │ │ │ │ │ │ │ │ │ └─ Tri{dtype='float64'} [id QQ] 480 > │ │ │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id PK] 421 897,898c895,896 < │ │ │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id RD] 453 < │ │ │ │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{start:} [id PX] 432 --- > │ │ │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id QR] 429 > │ │ │ │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{start:} [id PL] 407 900,901c898,899 < │ │ │ │ │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id RE] 48 < │ │ │ │ │ │ │ │ │ │ │ │ │ └─ 1 [id RF] --- > │ │ │ │ │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id QS] 48 > │ │ │ │ │ │ │ │ │ │ │ │ │ └─ 1 [id QT] 903c901 < │ │ │ │ │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id QA] 599 --- > │ │ │ │ │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id PO] 576 905,907c903,905 < │ │ │ │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id RG] 778 < │ │ │ │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id RH] 737 < │ │ │ │ │ │ │ │ │ │ ├─ Switch [id OQ] 173 --- > │ │ │ │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id QU] 757 > │ │ │ │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id QV] 715 > │ │ │ │ │ │ │ │ │ │ ├─ Switch [id OI] 162 909,911c907,909 < │ │ │ │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id RI] 696 < │ │ │ │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id RJ] 654 < │ │ │ │ │ │ │ │ │ │ ├─ Switch [id OQ] 173 --- > │ │ │ │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id QW] 673 > │ │ │ │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id QX] 631 > │ │ │ │ │ │ │ │ │ │ ├─ Switch [id OI] 162 913,914c911,912 < │ │ │ │ │ │ │ │ │ │ └─ Composite{((i0 * i1) - i2)} [id RK] 612 < │ │ │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id OZ] 418 --- > │ │ │ │ │ │ │ │ │ │ └─ Composite{((i0 * i1) - i2)} [id QY] 589 > │ │ │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id OR] 392 916,918c914,916 < │ │ │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id RL] 552 < │ │ │ │ │ │ │ │ │ │ │ └─ Tri{dtype='float64'} [id RM] 503 < │ │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id PW] 445 --- > │ │ │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id QZ] 528 > │ │ │ │ │ │ │ │ │ │ │ └─ Tri{dtype='float64'} [id RA] 479 > │ │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id PK] 421 920,921c918,919 < │ │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id RN] 452 < │ │ │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{start:} [id PX] 432 --- > │ │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id RB] 428 > │ │ │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{start:} [id PL] 407 923,924c921,922 < │ │ │ │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id RO] 47 < │ │ │ │ │ │ │ │ │ │ │ │ └─ 1 [id RP] --- > │ │ │ │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id RC] 47 > │ │ │ │ │ │ │ │ │ │ │ │ └─ 1 [id RD] 926c924 < │ │ │ │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id QA] 599 --- > │ │ │ │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id PO] 576 928c926 < │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id NI] 12 --- > │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id MK] 12 930c928 < │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id NK] 13 --- > │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id MM] 13 932,937c930,935 < │ │ │ │ │ │ │ ├─ Subtensor{i} [id RQ] 883 < │ │ │ │ │ │ │ │ ├─ Subtensor{start:} [id RR] 862 < │ │ │ │ │ │ │ │ │ ├─ Shape [id RS] 836 < │ │ │ │ │ │ │ │ │ │ └─ Add [id RT] 810 < │ │ │ │ │ │ │ │ │ │ ├─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id RU] 736 < │ │ │ │ │ │ │ │ │ │ │ ├─ Switch [id OQ] 173 --- > │ │ │ │ │ │ │ ├─ Subtensor{i} [id RE] 863 > │ │ │ │ │ │ │ │ ├─ Subtensor{start:} [id RF] 842 > │ │ │ │ │ │ │ │ │ ├─ Shape [id RG] 816 > │ │ │ │ │ │ │ │ │ │ └─ Add [id RH] 790 > │ │ │ │ │ │ │ │ │ │ ├─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id RI] 714 > │ │ │ │ │ │ │ │ │ │ │ ├─ Switch [id OI] 162 939,941c937,939 < │ │ │ │ │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id RV] 695 < │ │ │ │ │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id RW] 653 < │ │ │ │ │ │ │ │ │ │ │ ├─ Switch [id OQ] 173 --- > │ │ │ │ │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id RJ] 672 > │ │ │ │ │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id RK] 630 > │ │ │ │ │ │ │ │ │ │ │ ├─ Switch [id OI] 162 943,944c941,942 < │ │ │ │ │ │ │ │ │ │ │ └─ Composite{((i0 * i1) - i2)} [id RX] 611 < │ │ │ │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id OZ] 418 --- > │ │ │ │ │ │ │ │ │ │ │ └─ Composite{((i0 * i1) - i2)} [id RL] 588 > │ │ │ │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id OR] 392 946,948c944,946 < │ │ │ │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id RY] 551 < │ │ │ │ │ │ │ │ │ │ │ │ └─ Tri{dtype='float64'} [id RZ] 502 < │ │ │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id PW] 445 --- > │ │ │ │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id RM] 527 > │ │ │ │ │ │ │ │ │ │ │ │ └─ Tri{dtype='float64'} [id RN] 478 > │ │ │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id PK] 421 950,951c948,949 < │ │ │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id SA] 451 < │ │ │ │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{start:} [id PX] 432 --- > │ │ │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id RO] 427 > │ │ │ │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{start:} [id PL] 407 953,954c951,952 < │ │ │ │ │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id SB] 46 < │ │ │ │ │ │ │ │ │ │ │ │ │ └─ 1 [id SC] --- > │ │ │ │ │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id RP] 46 > │ │ │ │ │ │ │ │ │ │ │ │ │ └─ 1 [id RQ] 956c954 < │ │ │ │ │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id QA] 599 --- > │ │ │ │ │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id PO] 576 958,960c956,958 < │ │ │ │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id SD] 777 < │ │ │ │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id SE] 735 < │ │ │ │ │ │ │ │ │ │ ├─ Switch [id OQ] 173 --- > │ │ │ │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id RR] 756 > │ │ │ │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id RS] 713 > │ │ │ │ │ │ │ │ │ │ ├─ Switch [id OI] 162 962,964c960,962 < │ │ │ │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id SF] 694 < │ │ │ │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id SG] 652 < │ │ │ │ │ │ │ │ │ │ ├─ Switch [id OQ] 173 --- > │ │ │ │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id RT] 671 > │ │ │ │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id RU] 629 > │ │ │ │ │ │ │ │ │ │ ├─ Switch [id OI] 162 966,967c964,965 < │ │ │ │ │ │ │ │ │ │ └─ Composite{((i0 * i1) - i2)} [id SH] 610 < │ │ │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id OZ] 418 --- > │ │ │ │ │ │ │ │ │ │ └─ Composite{((i0 * i1) - i2)} [id RV] 587 > │ │ │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id OR] 392 969,971c967,969 < │ │ │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id SI] 550 < │ │ │ │ │ │ │ │ │ │ │ └─ Tri{dtype='float64'} [id SJ] 501 < │ │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id PW] 445 --- > │ │ │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id RW] 526 > │ │ │ │ │ │ │ │ │ │ │ └─ Tri{dtype='float64'} [id RX] 477 > │ │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id PK] 421 973,974c971,972 < │ │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id SK] 450 < │ │ │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{start:} [id PX] 432 --- > │ │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id RY] 426 > │ │ │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{start:} [id PL] 407 976,977c974,975 < │ │ │ │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id SL] 45 < │ │ │ │ │ │ │ │ │ │ │ │ └─ 1 [id SM] --- > │ │ │ │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id RZ] 45 > │ │ │ │ │ │ │ │ │ │ │ │ └─ 1 [id SA] 979c977 < │ │ │ │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id QA] 599 --- > │ │ │ │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id PO] 576 981c979 < │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id NI] 12 --- > │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id MK] 12 983c981 < │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id NN] 11 --- > │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id MP] 11 986,987c984,985 < │ │ │ │ │ │ ├─ AdvancedSetSubtensor [id SN] 882 < │ │ │ │ │ │ │ ├─ Alloc [id SO] 861 --- > │ │ │ │ │ │ ├─ AdvancedSetSubtensor [id SB] 862 > │ │ │ │ │ │ │ ├─ Alloc [id SC] 841 989,993c987,991 < │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id SP] 835 < │ │ │ │ │ │ │ │ │ ├─ Shape [id SQ] 809 < │ │ │ │ │ │ │ │ │ │ └─ ExtractDiag{offset=0, axis1=0, axis2=1, view=False} [id SR] 776 < │ │ │ │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id SS] 734 < │ │ │ │ │ │ │ │ │ │ ├─ Switch [id OQ] 173 --- > │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id SD] 815 > │ │ │ │ │ │ │ │ │ ├─ Shape [id SE] 789 > │ │ │ │ │ │ │ │ │ │ └─ ExtractDiag{offset=0, axis1=0, axis2=1, view=False} [id SF] 755 > │ │ │ │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id SG] 712 > │ │ │ │ │ │ │ │ │ │ ├─ Switch [id OI] 162 995,997c993,995 < │ │ │ │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id ST] 693 < │ │ │ │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id SU] 651 < │ │ │ │ │ │ │ │ │ │ ├─ Switch [id OQ] 173 --- > │ │ │ │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id SH] 670 > │ │ │ │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id SI] 628 > │ │ │ │ │ │ │ │ │ │ ├─ Switch [id OI] 162 999,1000c997,998 < │ │ │ │ │ │ │ │ │ │ └─ Composite{((i0 * i1) - i2)} [id SV] 609 < │ │ │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id OZ] 418 --- > │ │ │ │ │ │ │ │ │ │ └─ Composite{((i0 * i1) - i2)} [id SJ] 586 > │ │ │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id OR] 392 1002,1004c1000,1002 < │ │ │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id SW] 549 < │ │ │ │ │ │ │ │ │ │ │ └─ Tri{dtype='float64'} [id SX] 500 < │ │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id PW] 445 --- > │ │ │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id SK] 525 > │ │ │ │ │ │ │ │ │ │ │ └─ Tri{dtype='float64'} [id SL] 476 > │ │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id PK] 421 1006,1007c1004,1005 < │ │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id SY] 449 < │ │ │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{start:} [id PX] 432 --- > │ │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id SM] 425 > │ │ │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{start:} [id PL] 407 1009,1010c1007,1008 < │ │ │ │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id SZ] 44 < │ │ │ │ │ │ │ │ │ │ │ │ └─ 1 [id TA] --- > │ │ │ │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id SN] 44 > │ │ │ │ │ │ │ │ │ │ │ │ └─ 1 [id SO] 1012c1010 < │ │ │ │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id QA] 599 --- > │ │ │ │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id PO] 576 1014c1012 < │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id NX] 2 --- > │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id NP] 2 1016,1020c1014,1018 < │ │ │ │ │ │ │ │ └─ Subtensor{i} [id TB] 834 < │ │ │ │ │ │ │ │ ├─ Shape [id TC] 808 < │ │ │ │ │ │ │ │ │ └─ ExtractDiag{offset=0, axis1=0, axis2=1, view=False} [id TD] 775 < │ │ │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id TE] 733 < │ │ │ │ │ │ │ │ │ ├─ Switch [id OQ] 173 --- > │ │ │ │ │ │ │ │ └─ Subtensor{i} [id SP] 814 > │ │ │ │ │ │ │ │ ├─ Shape [id SQ] 788 > │ │ │ │ │ │ │ │ │ └─ ExtractDiag{offset=0, axis1=0, axis2=1, view=False} [id SR] 754 > │ │ │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id SS] 711 > │ │ │ │ │ │ │ │ │ ├─ Switch [id OI] 162 1022,1024c1020,1022 < │ │ │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id TF] 692 < │ │ │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id TG] 650 < │ │ │ │ │ │ │ │ │ ├─ Switch [id OQ] 173 --- > │ │ │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id ST] 669 > │ │ │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id SU] 627 > │ │ │ │ │ │ │ │ │ ├─ Switch [id OI] 162 1026,1027c1024,1025 < │ │ │ │ │ │ │ │ │ └─ Composite{((i0 * i1) - i2)} [id TH] 608 < │ │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id OZ] 418 --- > │ │ │ │ │ │ │ │ │ └─ Composite{((i0 * i1) - i2)} [id SV] 585 > │ │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id OR] 392 1029,1031c1027,1029 < │ │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id TI] 548 < │ │ │ │ │ │ │ │ │ │ └─ Tri{dtype='float64'} [id TJ] 499 < │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id PW] 445 --- > │ │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id SW] 524 > │ │ │ │ │ │ │ │ │ │ └─ Tri{dtype='float64'} [id SX] 475 > │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id PK] 421 1033,1034c1031,1032 < │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id TK] 448 < │ │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{start:} [id PX] 432 --- > │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id SY] 424 > │ │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{start:} [id PL] 407 1036,1037c1034,1035 < │ │ │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id TL] 43 < │ │ │ │ │ │ │ │ │ │ │ └─ 1 [id TM] --- > │ │ │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id SZ] 43 > │ │ │ │ │ │ │ │ │ │ │ └─ 1 [id TA] 1039c1037 < │ │ │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id QA] 599 --- > │ │ │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id PO] 576 1041c1039 < │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id NX] 2 --- > │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id NP] 2 1043,1045c1041,1043 < │ │ │ │ │ │ │ ├─ ExtractDiag{offset=0, axis1=0, axis2=1, view=False} [id TN] 774 < │ │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id TO] 732 < │ │ │ │ │ │ │ │ ├─ Switch [id OQ] 173 --- > │ │ │ │ │ │ │ ├─ ExtractDiag{offset=0, axis1=0, axis2=1, view=False} [id TB] 753 > │ │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id TC] 710 > │ │ │ │ │ │ │ │ ├─ Switch [id OI] 162 1047,1049c1045,1047 < │ │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id TP] 691 < │ │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id TQ] 649 < │ │ │ │ │ │ │ │ ├─ Switch [id OQ] 173 --- > │ │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id TD] 668 > │ │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id TE] 626 > │ │ │ │ │ │ │ │ ├─ Switch [id OI] 162 1051,1052c1049,1050 < │ │ │ │ │ │ │ │ └─ Composite{((i0 * i1) - i2)} [id TR] 607 < │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id OZ] 418 --- > │ │ │ │ │ │ │ │ └─ Composite{((i0 * i1) - i2)} [id TF] 584 > │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id OR] 392 1054,1056c1052,1054 < │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id TS] 547 < │ │ │ │ │ │ │ │ │ └─ Tri{dtype='float64'} [id TT] 498 < │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id PW] 445 --- > │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id TG] 523 > │ │ │ │ │ │ │ │ │ └─ Tri{dtype='float64'} [id TH] 474 > │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id PK] 421 1058,1059c1056,1057 < │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id TU] 447 < │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{start:} [id PX] 432 --- > │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id TI] 423 > │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{start:} [id PL] 407 1061,1062c1059,1060 < │ │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id TV] 42 < │ │ │ │ │ │ │ │ │ │ └─ 1 [id TW] --- > │ │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id TJ] 42 > │ │ │ │ │ │ │ │ │ │ └─ 1 [id TK] 1064c1062 < │ │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id QA] 599 --- > │ │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id PO] 576 1066c1064 < │ │ │ │ │ │ │ ├─ ARange{dtype='int64'} [id TX] 860 --- > │ │ │ │ │ │ │ ├─ ARange{dtype='int64'} [id TL] 840 1068,1072c1066,1070 < │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id TY] 833 < │ │ │ │ │ │ │ │ │ ├─ Shape [id TZ] 807 < │ │ │ │ │ │ │ │ │ │ └─ ExtractDiag{offset=0, axis1=0, axis2=1, view=False} [id UA] 773 < │ │ │ │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id UB] 731 < │ │ │ │ │ │ │ │ │ │ ├─ Switch [id OQ] 173 --- > │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id TM] 813 > │ │ │ │ │ │ │ │ │ ├─ Shape [id TN] 787 > │ │ │ │ │ │ │ │ │ │ └─ ExtractDiag{offset=0, axis1=0, axis2=1, view=False} [id TO] 752 > │ │ │ │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id TP] 709 > │ │ │ │ │ │ │ │ │ │ ├─ Switch [id OI] 162 1074,1076c1072,1074 < │ │ │ │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id UC] 690 < │ │ │ │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id UD] 648 < │ │ │ │ │ │ │ │ │ │ ├─ Switch [id OQ] 173 --- > │ │ │ │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id TQ] 667 > │ │ │ │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id TR] 625 > │ │ │ │ │ │ │ │ │ │ ├─ Switch [id OI] 162 1078,1079c1076,1077 < │ │ │ │ │ │ │ │ │ │ └─ Composite{((i0 * i1) - i2)} [id UE] 606 < │ │ │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id OZ] 418 --- > │ │ │ │ │ │ │ │ │ │ └─ Composite{((i0 * i1) - i2)} [id TS] 583 > │ │ │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id OR] 392 1081,1083c1079,1081 < │ │ │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id UF] 546 < │ │ │ │ │ │ │ │ │ │ │ └─ Tri{dtype='float64'} [id UG] 497 < │ │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id PW] 445 --- > │ │ │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id TT] 522 > │ │ │ │ │ │ │ │ │ │ │ └─ Tri{dtype='float64'} [id TU] 473 > │ │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id PK] 421 1085,1086c1083,1084 < │ │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id UH] 446 < │ │ │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{start:} [id PX] 432 --- > │ │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id TV] 422 > │ │ │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{start:} [id PL] 407 1088,1089c1086,1087 < │ │ │ │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id UI] 41 < │ │ │ │ │ │ │ │ │ │ │ │ └─ 1 [id UJ] --- > │ │ │ │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id TW] 41 > │ │ │ │ │ │ │ │ │ │ │ │ └─ 1 [id TX] 1091c1089 < │ │ │ │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id QA] 599 --- > │ │ │ │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id PO] 576 1093c1091 < │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id NX] 2 --- > │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id NP] 2 1096c1094 < │ │ │ │ │ │ │ └─ ARange{dtype='int64'} [id UK] 859 --- > │ │ │ │ │ │ │ └─ ARange{dtype='int64'} [id TY] 839 1098,1101c1096,1099 < │ │ │ │ │ │ │ ├─ Subtensor{i} [id UL] 832 < │ │ │ │ │ │ │ │ ├─ Shape [id UM] 806 < │ │ │ │ │ │ │ │ │ └─ ExtractDiag{offset=0, axis1=0, axis2=1, view=False} [id UN] 772 < │ │ │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id OP] 730 --- > │ │ │ │ │ │ │ ├─ Subtensor{i} [id TZ] 812 > │ │ │ │ │ │ │ │ ├─ Shape [id UA] 786 > │ │ │ │ │ │ │ │ │ └─ ExtractDiag{offset=0, axis1=0, axis2=1, view=False} [id UB] 751 > │ │ │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id OH] 708 1103c1101 < │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id NX] 2 --- > │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id NP] 2 1106c1104 < │ │ │ │ │ │ ├─ Invert [id OR] 155 --- > │ │ │ │ │ │ ├─ Invert [id OJ] 145 1108,1110c1106,1108 < │ │ │ │ │ │ └─ [[nan]] [id OM] < │ │ │ │ │ ├─ [0 1] [id CN] < │ │ │ │ │ └─ [0 1] [id CN] --- > │ │ │ │ │ │ └─ [[nan]] [id OE] > │ │ │ │ │ ├─ [0 1] [id CO] > │ │ │ │ │ └─ [0 1] [id CO] 1112,1120c1110,1118 < │ │ │ ├─ AdvancedSubtensor [id UO] 906 < │ │ │ │ ├─ Composite{switch(i4, (((i0 + i1) * i2) - i3), i5)} [id UP] 901 < │ │ │ │ │ ├─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id UQ] 740 < │ │ │ │ │ │ ├─ Switch [id UR] 171 < │ │ │ │ │ │ │ ├─ Invert [id US] 152 < │ │ │ │ │ │ │ │ └─ ExpandDims{axes=[0, 1]} [id UT] 133 < │ │ │ │ │ │ │ │ └─ Any{axes=None} [id UU] 114 < │ │ │ │ │ │ │ │ └─ Isnan [id UV] 92 < │ │ │ │ │ │ │ │ └─ Cholesky{lower=True, destructive=False, on_error='nan'} [id EF] 73 --- > │ │ │ ├─ AdvancedSubtensor [id UC] 886 > │ │ │ │ ├─ Composite{switch(i4, (((i0 + i1) * i2) - i3), i5)} [id UD] 881 > │ │ │ │ │ ├─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id UE] 718 > │ │ │ │ │ │ ├─ Switch [id UF] 160 > │ │ │ │ │ │ │ ├─ Invert [id UG] 143 > │ │ │ │ │ │ │ │ └─ ExpandDims{axes=[0, 1]} [id UH] 128 > │ │ │ │ │ │ │ │ └─ Any{axes=None} [id UI] 109 > │ │ │ │ │ │ │ │ └─ Isnan [id UJ] 88 > │ │ │ │ │ │ │ │ └─ Cholesky{lower=True, destructive=False, on_error='nan'} [id EE] 73 1122,1123c1120,1121 < │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id UW] 93 < │ │ │ │ │ │ │ │ └─ Cholesky{lower=True, destructive=False, on_error='nan'} [id EF] 73 --- > │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id UK] 89 > │ │ │ │ │ │ │ │ └─ Cholesky{lower=True, destructive=False, on_error='nan'} [id EE] 73 1125,1128c1123,1126 < │ │ │ │ │ │ │ └─ [[1]] [id LY] < │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id UX] 699 < │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id UY] 657 < │ │ │ │ │ │ ├─ Switch [id UR] 171 --- > │ │ │ │ │ │ │ └─ [[1]] [id CQ] > │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id UL] 676 > │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id UM] 634 > │ │ │ │ │ │ ├─ Switch [id UF] 160 1130,1133c1128,1131 < │ │ │ │ │ │ └─ Composite{((i0 * i1) - i2)} [id UZ] 615 < │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id VA] 421 < │ │ │ │ │ │ │ └─ dot [id VB] 408 < │ │ │ │ │ │ │ ├─ Switch [id UR] 171 --- > │ │ │ │ │ │ └─ Composite{((i0 * i1) - i2)} [id UN] 592 > │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id UO] 395 > │ │ │ │ │ │ │ └─ dot [id UP] 383 > │ │ │ │ │ │ │ ├─ Switch [id UF] 160 1135,1159c1133,1156 < │ │ │ │ │ │ │ └─ Composite{switch(i3, (i2 + switch(i1, (-i0), 0.0)), 1)} [id VC] 401 < │ │ │ │ │ │ │ ├─ Sum{axis=0} [id VD] 394 < │ │ │ │ │ │ │ │ └─ Mul [id VE] 387 < │ │ │ │ │ │ │ │ ├─ Blockwise{dot, (i00,i01),(i10,i11)->(o00,o01)} [id VF] 319 < │ │ │ │ │ │ │ │ │ ├─ ExpandDims{axis=2} [id VG] 303 < │ │ │ │ │ │ │ │ │ │ └─ Blockwise{SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=1}, (m,m),(m)->(m)} [id KG] 290 < │ │ │ │ │ │ │ │ │ │ └─ ··· < │ │ │ │ │ │ │ │ │ └─ ExpandDims{axis=1} [id VH] 216 < │ │ │ │ │ │ │ │ │ └─ Blockwise{SolveTriangular{trans=0, unit_diagonal=False, lower=True, check_finite=True, b_ndim=1}, (m,m),(m)->(m)} [id DZ] 205 < │ │ │ │ │ │ │ │ │ └─ ··· < │ │ │ │ │ │ │ │ └─ Blockwise{Tri{dtype='float64'}, (),(),()->(o00,o01)} [id VI] 380 < │ │ │ │ │ │ │ │ ├─ Blockwise{Subtensor{i}, (i00),()->()} [id VJ] 370 < │ │ │ │ │ │ │ │ │ ├─ Blockwise{Subtensor{start:}, (i00),()->(o00)} [id VK] 358 < │ │ │ │ │ │ │ │ │ │ ├─ Blockwise{Shape, (i00,i01)->(o00)} [id VL] 346 < │ │ │ │ │ │ │ │ │ │ │ └─ Neg [id VM] 334 < │ │ │ │ │ │ │ │ │ │ │ └─ Blockwise{dot, (i00,i01),(i10,i11)->(o00,o01)} [id VF] 319 < │ │ │ │ │ │ │ │ │ │ │ └─ ··· < │ │ │ │ │ │ │ │ │ │ └─ [-2] [id MP] < │ │ │ │ │ │ │ │ │ └─ [0] [id MQ] < │ │ │ │ │ │ │ │ ├─ Blockwise{Subtensor{i}, (i00),()->()} [id VN] 369 < │ │ │ │ │ │ │ │ │ ├─ Blockwise{Subtensor{start:}, (i00),()->(o00)} [id VK] 358 < │ │ │ │ │ │ │ │ │ │ └─ ··· < │ │ │ │ │ │ │ │ │ └─ [1] [id MS] < │ │ │ │ │ │ │ │ └─ [0] [id CO] < │ │ │ │ │ │ │ ├─ ExpandDims{axes=[0, 1]} [id EP] 150 --- > │ │ │ │ │ │ │ └─ Composite{switch(i4, (i3 + switch(i2, (-1.0 * i0 * i1), 0.0)), 1)} [id UQ] 376 > │ │ │ │ │ │ │ ├─ dot [id UR] 304 > │ │ │ │ │ │ │ │ ├─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id KD] 291 > │ │ │ │ │ │ │ │ │ └─ ··· > │ │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id DX] 203 > │ │ │ │ │ │ │ │ └─ ··· > │ │ │ │ │ │ │ ├─ Tri{dtype='float64'} [id US] 369 > │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id UT] 359 > │ │ │ │ │ │ │ │ │ ├─ Subtensor{start:} [id UU] 347 > │ │ │ │ │ │ │ │ │ │ ├─ Shape [id UV] 335 > │ │ │ │ │ │ │ │ │ │ │ └─ Neg [id UW] 320 > │ │ │ │ │ │ │ │ │ │ │ └─ dot [id UR] 304 > │ │ │ │ │ │ │ │ │ │ │ └─ ··· > │ │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id MK] 12 > │ │ │ │ │ │ │ │ │ │ └─ ··· > │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id MM] 13 > │ │ │ │ │ │ │ │ │ └─ ··· > │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id UX] 358 > │ │ │ │ │ │ │ │ │ ├─ Subtensor{start:} [id UU] 347 > │ │ │ │ │ │ │ │ │ │ └─ ··· > │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id MP] 11 > │ │ │ │ │ │ │ │ │ └─ ··· > │ │ │ │ │ │ │ │ └─ 0 [id E] > │ │ │ │ │ │ │ ├─ ExpandDims{axes=[0, 1]} [id EA] 142 1161,1162c1158,1159 < │ │ │ │ │ │ │ ├─ SetSubtensor{:stop, start:stop} [id VO] 357 < │ │ │ │ │ │ │ │ ├─ Alloc [id MU] 9 --- > │ │ │ │ │ │ │ ├─ SetSubtensor{:stop, start:stop} [id UY] 346 > │ │ │ │ │ │ │ │ ├─ Alloc [id MS] 9 1164,1165c1161,1162 < │ │ │ │ │ │ │ │ ├─ AdvancedSetSubtensor [id VP] 345 < │ │ │ │ │ │ │ │ │ ├─ Alloc [id CK] 1 --- > │ │ │ │ │ │ │ │ ├─ AdvancedSetSubtensor [id UZ] 334 > │ │ │ │ │ │ │ │ │ ├─ Alloc [id CL] 1 1167,1172c1164,1169 < │ │ │ │ │ │ │ │ │ ├─ True_div [id VQ] 333 < │ │ │ │ │ │ │ │ │ │ ├─ Neg [id VR] 318 < │ │ │ │ │ │ │ │ │ │ │ └─ ExpandDims{axis=0} [id VS] 302 < │ │ │ │ │ │ │ │ │ │ │ └─ Sum{axes=None} [id VT] 289 < │ │ │ │ │ │ │ │ │ │ │ └─ DropDims{axis=1} [id VU] 278 < │ │ │ │ │ │ │ │ │ │ │ └─ SpecifyShape [id KJ] 270 --- > │ │ │ │ │ │ │ │ │ ├─ True_div [id VA] 319 > │ │ │ │ │ │ │ │ │ │ ├─ Neg [id VB] 303 > │ │ │ │ │ │ │ │ │ │ │ └─ ExpandDims{axis=0} [id VC] 290 > │ │ │ │ │ │ │ │ │ │ │ └─ Sum{axes=None} [id VD] 279 > │ │ │ │ │ │ │ │ │ │ │ └─ DropDims{axis=1} [id VE] 268 > │ │ │ │ │ │ │ │ │ │ │ └─ SpecifyShape [id KH] 260 1174c1171 < │ │ │ │ │ │ │ │ │ │ └─ ExtractDiag{offset=0, axis1=0, axis2=1, view=False} [id EE] 90 --- > │ │ │ │ │ │ │ │ │ │ └─ ExtractDiag{offset=0, axis1=0, axis2=1, view=False} [id ED] 87 1176,1181c1173,1178 < │ │ │ │ │ │ │ │ │ ├─ [0 1] [id CN] < │ │ │ │ │ │ │ │ │ └─ [0 1] [id CN] < │ │ │ │ │ │ │ │ ├─ 2 [id NB] < │ │ │ │ │ │ │ │ ├─ 0 [id NC] < │ │ │ │ │ │ │ │ └─ 2 [id NB] < │ │ │ │ │ │ │ └─ Invert [id US] 152 --- > │ │ │ │ │ │ │ │ │ ├─ [0 1] [id CO] > │ │ │ │ │ │ │ │ │ └─ [0 1] [id CO] > │ │ │ │ │ │ │ │ ├─ 2 [id MZ] > │ │ │ │ │ │ │ │ ├─ 0 [id NA] > │ │ │ │ │ │ │ │ └─ 2 [id MZ] > │ │ │ │ │ │ │ └─ Invert [id UG] 143 1183,1188c1180,1185 < │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id VV] 557 < │ │ │ │ │ │ │ └─ Tri{dtype='float64'} [id VW] 507 < │ │ │ │ │ │ │ ├─ Subtensor{i} [id VX] 457 < │ │ │ │ │ │ │ │ ├─ Subtensor{start:} [id VY] 434 < │ │ │ │ │ │ │ │ │ ├─ Shape [id VZ] 420 < │ │ │ │ │ │ │ │ │ │ └─ dot [id VB] 408 --- > │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id VF] 533 > │ │ │ │ │ │ │ └─ Tri{dtype='float64'} [id VG] 483 > │ │ │ │ │ │ │ ├─ Subtensor{i} [id VH] 433 > │ │ │ │ │ │ │ │ ├─ Subtensor{start:} [id VI] 409 > │ │ │ │ │ │ │ │ │ ├─ Shape [id VJ] 394 > │ │ │ │ │ │ │ │ │ │ └─ dot [id UP] 383 1190c1187 < │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id NI] 12 --- > │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id MK] 12 1192c1189 < │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id NK] 13 --- > │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id MM] 13 1194,1195c1191,1192 < │ │ │ │ │ │ │ ├─ Subtensor{i} [id WA] 456 < │ │ │ │ │ │ │ │ ├─ Subtensor{start:} [id VY] 434 --- > │ │ │ │ │ │ │ ├─ Subtensor{i} [id VK] 432 > │ │ │ │ │ │ │ │ ├─ Subtensor{start:} [id VI] 409 1197c1194 < │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id NN] 11 --- > │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id MP] 11 1200,1202c1197,1199 < │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id WB] 600 < │ │ │ │ │ │ └─ AdvancedSetSubtensor [id WC] 594 < │ │ │ │ │ │ ├─ Alloc [id WD] 556 --- > │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id VL] 577 > │ │ │ │ │ │ └─ AdvancedSetSubtensor [id VM] 570 > │ │ │ │ │ │ ├─ Alloc [id VN] 532 1204,1209c1201,1206 < │ │ │ │ │ │ │ ├─ Subtensor{i} [id WE] 506 < │ │ │ │ │ │ │ │ ├─ Shape [id WF] 455 < │ │ │ │ │ │ │ │ │ └─ Mul [id WG] 433 < │ │ │ │ │ │ │ │ │ ├─ [0.5] [id NV] < │ │ │ │ │ │ │ │ │ └─ ExtractDiag{offset=0, axis1=0, axis2=1, view=False} [id WH] 419 < │ │ │ │ │ │ │ │ │ └─ dot [id VB] 408 --- > │ │ │ │ │ │ │ ├─ Subtensor{i} [id VO] 482 > │ │ │ │ │ │ │ │ ├─ Shape [id VP] 431 > │ │ │ │ │ │ │ │ │ └─ Mul [id VQ] 408 > │ │ │ │ │ │ │ │ │ ├─ [0.5] [id NN] > │ │ │ │ │ │ │ │ │ └─ ExtractDiag{offset=0, axis1=0, axis2=1, view=False} [id VR] 393 > │ │ │ │ │ │ │ │ │ └─ dot [id UP] 383 1211c1208 < │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id NX] 2 --- > │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id NP] 2 1213c1210 < │ │ │ │ │ │ │ └─ Subtensor{i} [id WE] 506 --- > │ │ │ │ │ │ │ └─ Subtensor{i} [id VO] 482 1215c1212 < │ │ │ │ │ │ ├─ Mul [id WG] 433 --- > │ │ │ │ │ │ ├─ Mul [id VQ] 408 1217c1214 < │ │ │ │ │ │ ├─ ARange{dtype='int64'} [id WI] 555 --- > │ │ │ │ │ │ ├─ ARange{dtype='int64'} [id VS] 531 1219c1216 < │ │ │ │ │ │ │ ├─ Subtensor{i} [id WE] 506 --- > │ │ │ │ │ │ │ ├─ Subtensor{i} [id VO] 482 1222c1219 < │ │ │ │ │ │ └─ ARange{dtype='int64'} [id WI] 555 --- > │ │ │ │ │ │ └─ ARange{dtype='int64'} [id VS] 531 1224,1226c1221,1223 < │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id WJ] 787 < │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id WK] 749 < │ │ │ │ │ │ ├─ Switch [id UR] 171 --- > │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id VT] 760 > │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id VU] 719 > │ │ │ │ │ │ ├─ Switch [id UF] 160 1228,1230c1225,1227 < │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id WL] 708 < │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id WM] 666 < │ │ │ │ │ │ ├─ Switch [id UR] 171 --- > │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id VV] 677 > │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id VW] 635 > │ │ │ │ │ │ ├─ Switch [id UF] 160 1232,1233c1229,1230 < │ │ │ │ │ │ └─ Composite{((i0 * i1) - i2)} [id WN] 624 < │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id VA] 421 --- > │ │ │ │ │ │ └─ Composite{((i0 * i1) - i2)} [id VX] 593 > │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id UO] 395 1235,1237c1232,1234 < │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id WO] 566 < │ │ │ │ │ │ │ └─ Tri{dtype='float64'} [id WP] 516 < │ │ │ │ │ │ │ ├─ Subtensor{i} [id VX] 457 --- > │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id VY] 534 > │ │ │ │ │ │ │ └─ Tri{dtype='float64'} [id VZ] 484 > │ │ │ │ │ │ │ ├─ Subtensor{i} [id VH] 433 1239,1240c1236,1237 < │ │ │ │ │ │ │ ├─ Subtensor{i} [id WQ] 466 < │ │ │ │ │ │ │ │ ├─ Subtensor{start:} [id VY] 434 --- > │ │ │ │ │ │ │ ├─ Subtensor{i} [id WA] 434 > │ │ │ │ │ │ │ │ ├─ Subtensor{start:} [id VI] 409 1242,1243c1239,1240 < │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id WR] 40 < │ │ │ │ │ │ │ │ └─ 1 [id WS] --- > │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id WB] 32 > │ │ │ │ │ │ │ │ └─ 1 [id WC] 1245c1242 < │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id WB] 600 --- > │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id VL] 577 1247,1253c1244,1250 < │ │ │ │ │ ├─ Tri{dtype='float64'} [id WT] 896 < │ │ │ │ │ │ ├─ Subtensor{i} [id WU] 887 < │ │ │ │ │ │ │ ├─ Subtensor{start:} [id WV] 868 < │ │ │ │ │ │ │ │ ├─ Shape [id WW] 843 < │ │ │ │ │ │ │ │ │ └─ Add [id WX] 817 < │ │ │ │ │ │ │ │ │ ├─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id WY] 748 < │ │ │ │ │ │ │ │ │ │ ├─ Switch [id UR] 171 --- > │ │ │ │ │ ├─ Tri{dtype='float64'} [id WD] 876 > │ │ │ │ │ │ ├─ Subtensor{i} [id WE] 865 > │ │ │ │ │ │ │ ├─ Subtensor{start:} [id WF] 845 > │ │ │ │ │ │ │ │ ├─ Shape [id WG] 819 > │ │ │ │ │ │ │ │ │ └─ Add [id WH] 793 > │ │ │ │ │ │ │ │ │ ├─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id WI] 720 > │ │ │ │ │ │ │ │ │ │ ├─ Switch [id UF] 160 1255,1257c1252,1254 < │ │ │ │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id WZ] 707 < │ │ │ │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id XA] 665 < │ │ │ │ │ │ │ │ │ │ ├─ Switch [id UR] 171 --- > │ │ │ │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id WJ] 678 > │ │ │ │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id WK] 636 > │ │ │ │ │ │ │ │ │ │ ├─ Switch [id UF] 160 1259,1260c1256,1257 < │ │ │ │ │ │ │ │ │ │ └─ Composite{((i0 * i1) - i2)} [id XB] 623 < │ │ │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id VA] 421 --- > │ │ │ │ │ │ │ │ │ │ └─ Composite{((i0 * i1) - i2)} [id WL] 594 > │ │ │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id UO] 395 1262,1264c1259,1261 < │ │ │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id XC] 565 < │ │ │ │ │ │ │ │ │ │ │ └─ Tri{dtype='float64'} [id XD] 515 < │ │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id VX] 457 --- > │ │ │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id WM] 535 > │ │ │ │ │ │ │ │ │ │ │ └─ Tri{dtype='float64'} [id WN] 485 > │ │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id VH] 433 1266,1267c1263,1264 < │ │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id XE] 465 < │ │ │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{start:} [id VY] 434 --- > │ │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id WO] 435 > │ │ │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{start:} [id VI] 409 1269,1270c1266,1267 < │ │ │ │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id XF] 39 < │ │ │ │ │ │ │ │ │ │ │ │ └─ 1 [id XG] --- > │ │ │ │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id WP] 33 > │ │ │ │ │ │ │ │ │ │ │ │ └─ 1 [id WQ] 1272c1269 < │ │ │ │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id WB] 600 --- > │ │ │ │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id VL] 577 1274,1276c1271,1273 < │ │ │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id XH] 786 < │ │ │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id XI] 747 < │ │ │ │ │ │ │ │ │ ├─ Switch [id UR] 171 --- > │ │ │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id WR] 761 > │ │ │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id WS] 721 > │ │ │ │ │ │ │ │ │ ├─ Switch [id UF] 160 1278,1280c1275,1277 < │ │ │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id XJ] 706 < │ │ │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id XK] 664 < │ │ │ │ │ │ │ │ │ ├─ Switch [id UR] 171 --- > │ │ │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id WT] 679 > │ │ │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id WU] 637 > │ │ │ │ │ │ │ │ │ ├─ Switch [id UF] 160 1282,1283c1279,1280 < │ │ │ │ │ │ │ │ │ └─ Composite{((i0 * i1) - i2)} [id XL] 622 < │ │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id VA] 421 --- > │ │ │ │ │ │ │ │ │ └─ Composite{((i0 * i1) - i2)} [id WV] 595 > │ │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id UO] 395 1285,1287c1282,1284 < │ │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id XM] 564 < │ │ │ │ │ │ │ │ │ │ └─ Tri{dtype='float64'} [id XN] 514 < │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id VX] 457 --- > │ │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id WW] 536 > │ │ │ │ │ │ │ │ │ │ └─ Tri{dtype='float64'} [id WX] 486 > │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id VH] 433 1289,1290c1286,1287 < │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id XO] 464 < │ │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{start:} [id VY] 434 --- > │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id WY] 436 > │ │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{start:} [id VI] 409 1292,1293c1289,1290 < │ │ │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id XP] 38 < │ │ │ │ │ │ │ │ │ │ │ └─ 1 [id XQ] --- > │ │ │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id WZ] 34 > │ │ │ │ │ │ │ │ │ │ │ └─ 1 [id XA] 1295c1292 < │ │ │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id WB] 600 --- > │ │ │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id VL] 577 1297c1294 < │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id NI] 12 --- > │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id MK] 12 1299c1296 < │ │ │ │ │ │ │ └─ ScalarFromTensor [id NK] 13 --- > │ │ │ │ │ │ │ └─ ScalarFromTensor [id MM] 13 1301,1306c1298,1303 < │ │ │ │ │ │ ├─ Subtensor{i} [id XR] 886 < │ │ │ │ │ │ │ ├─ Subtensor{start:} [id XS] 867 < │ │ │ │ │ │ │ │ ├─ Shape [id XT] 842 < │ │ │ │ │ │ │ │ │ └─ Add [id XU] 816 < │ │ │ │ │ │ │ │ │ ├─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id XV] 746 < │ │ │ │ │ │ │ │ │ │ ├─ Switch [id UR] 171 --- > │ │ │ │ │ │ ├─ Subtensor{i} [id XB] 866 > │ │ │ │ │ │ │ ├─ Subtensor{start:} [id XC] 846 > │ │ │ │ │ │ │ │ ├─ Shape [id XD] 820 > │ │ │ │ │ │ │ │ │ └─ Add [id XE] 794 > │ │ │ │ │ │ │ │ │ ├─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id XF] 722 > │ │ │ │ │ │ │ │ │ │ ├─ Switch [id UF] 160 1308,1310c1305,1307 < │ │ │ │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id XW] 705 < │ │ │ │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id XX] 663 < │ │ │ │ │ │ │ │ │ │ ├─ Switch [id UR] 171 --- > │ │ │ │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id XG] 680 > │ │ │ │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id XH] 638 > │ │ │ │ │ │ │ │ │ │ ├─ Switch [id UF] 160 1312,1313c1309,1310 < │ │ │ │ │ │ │ │ │ │ └─ Composite{((i0 * i1) - i2)} [id XY] 621 < │ │ │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id VA] 421 --- > │ │ │ │ │ │ │ │ │ │ └─ Composite{((i0 * i1) - i2)} [id XI] 596 > │ │ │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id UO] 395 1315,1317c1312,1314 < │ │ │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id XZ] 563 < │ │ │ │ │ │ │ │ │ │ │ └─ Tri{dtype='float64'} [id YA] 513 < │ │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id VX] 457 --- > │ │ │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id XJ] 537 > │ │ │ │ │ │ │ │ │ │ │ └─ Tri{dtype='float64'} [id XK] 487 > │ │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id VH] 433 1319,1320c1316,1317 < │ │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id YB] 463 < │ │ │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{start:} [id VY] 434 --- > │ │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id XL] 437 > │ │ │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{start:} [id VI] 409 1322,1323c1319,1320 < │ │ │ │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id YC] 37 < │ │ │ │ │ │ │ │ │ │ │ │ └─ 1 [id YD] --- > │ │ │ │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id XM] 35 > │ │ │ │ │ │ │ │ │ │ │ │ └─ 1 [id XN] 1325c1322 < │ │ │ │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id WB] 600 --- > │ │ │ │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id VL] 577 1327,1329c1324,1326 < │ │ │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id YE] 785 < │ │ │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id YF] 745 < │ │ │ │ │ │ │ │ │ ├─ Switch [id UR] 171 --- > │ │ │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id XO] 762 > │ │ │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id XP] 723 > │ │ │ │ │ │ │ │ │ ├─ Switch [id UF] 160 1331,1333c1328,1330 < │ │ │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id YG] 704 < │ │ │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id YH] 662 < │ │ │ │ │ │ │ │ │ ├─ Switch [id UR] 171 --- > │ │ │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id XQ] 681 > │ │ │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id XR] 639 > │ │ │ │ │ │ │ │ │ ├─ Switch [id UF] 160 1335,1336c1332,1333 < │ │ │ │ │ │ │ │ │ └─ Composite{((i0 * i1) - i2)} [id YI] 620 < │ │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id VA] 421 --- > │ │ │ │ │ │ │ │ │ └─ Composite{((i0 * i1) - i2)} [id XS] 597 > │ │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id UO] 395 1338,1340c1335,1337 < │ │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id YJ] 562 < │ │ │ │ │ │ │ │ │ │ └─ Tri{dtype='float64'} [id YK] 512 < │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id VX] 457 --- > │ │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id XT] 538 > │ │ │ │ │ │ │ │ │ │ └─ Tri{dtype='float64'} [id XU] 488 > │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id VH] 433 1342,1343c1339,1340 < │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id YL] 462 < │ │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{start:} [id VY] 434 --- > │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id XV] 438 > │ │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{start:} [id VI] 409 1345,1346c1342,1343 < │ │ │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id YM] 36 < │ │ │ │ │ │ │ │ │ │ │ └─ 1 [id YN] --- > │ │ │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id XW] 36 > │ │ │ │ │ │ │ │ │ │ │ └─ 1 [id XX] 1348c1345 < │ │ │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id WB] 600 --- > │ │ │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id VL] 577 1350c1347 < │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id NI] 12 --- > │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id MK] 12 1352c1349 < │ │ │ │ │ │ │ └─ ScalarFromTensor [id NN] 11 --- > │ │ │ │ │ │ │ └─ ScalarFromTensor [id MP] 11 1355,1356c1352,1353 < │ │ │ │ │ ├─ AdvancedSetSubtensor [id YO] 885 < │ │ │ │ │ │ ├─ Alloc [id YP] 866 --- > │ │ │ │ │ ├─ AdvancedSetSubtensor [id XY] 873 > │ │ │ │ │ │ ├─ Alloc [id XZ] 847 1358,1362c1355,1359 < │ │ │ │ │ │ │ ├─ Subtensor{i} [id YQ] 841 < │ │ │ │ │ │ │ │ ├─ Shape [id YR] 815 < │ │ │ │ │ │ │ │ │ └─ ExtractDiag{offset=0, axis1=0, axis2=1, view=False} [id YS] 784 < │ │ │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id YT] 744 < │ │ │ │ │ │ │ │ │ ├─ Switch [id UR] 171 --- > │ │ │ │ │ │ │ ├─ Subtensor{i} [id YA] 821 > │ │ │ │ │ │ │ │ ├─ Shape [id YB] 795 > │ │ │ │ │ │ │ │ │ └─ ExtractDiag{offset=0, axis1=0, axis2=1, view=False} [id YC] 763 > │ │ │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id YD] 724 > │ │ │ │ │ │ │ │ │ ├─ Switch [id UF] 160 1364,1366c1361,1363 < │ │ │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id YU] 703 < │ │ │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id YV] 661 < │ │ │ │ │ │ │ │ │ ├─ Switch [id UR] 171 --- > │ │ │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id YE] 682 > │ │ │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id YF] 640 > │ │ │ │ │ │ │ │ │ ├─ Switch [id UF] 160 1368,1369c1365,1366 < │ │ │ │ │ │ │ │ │ └─ Composite{((i0 * i1) - i2)} [id YW] 619 < │ │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id VA] 421 --- > │ │ │ │ │ │ │ │ │ └─ Composite{((i0 * i1) - i2)} [id YG] 598 > │ │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id UO] 395 1371,1373c1368,1370 < │ │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id YX] 561 < │ │ │ │ │ │ │ │ │ │ └─ Tri{dtype='float64'} [id YY] 511 < │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id VX] 457 --- > │ │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id YH] 539 > │ │ │ │ │ │ │ │ │ │ └─ Tri{dtype='float64'} [id YI] 489 > │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id VH] 433 1375,1376c1372,1373 < │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id YZ] 461 < │ │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{start:} [id VY] 434 --- > │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id YJ] 439 > │ │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{start:} [id VI] 409 1378,1379c1375,1376 < │ │ │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id ZA] 35 < │ │ │ │ │ │ │ │ │ │ │ └─ 1 [id ZB] --- > │ │ │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id YK] 37 > │ │ │ │ │ │ │ │ │ │ │ └─ 1 [id YL] 1381c1378 < │ │ │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id WB] 600 --- > │ │ │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id VL] 577 1383c1380 < │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id NX] 2 --- > │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id NP] 2 1385,1389c1382,1386 < │ │ │ │ │ │ │ └─ Subtensor{i} [id ZC] 840 < │ │ │ │ │ │ │ ├─ Shape [id ZD] 814 < │ │ │ │ │ │ │ │ └─ ExtractDiag{offset=0, axis1=0, axis2=1, view=False} [id ZE] 783 < │ │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id ZF] 743 < │ │ │ │ │ │ │ │ ├─ Switch [id UR] 171 --- > │ │ │ │ │ │ │ └─ Subtensor{i} [id YM] 822 > │ │ │ │ │ │ │ ├─ Shape [id YN] 796 > │ │ │ │ │ │ │ │ └─ ExtractDiag{offset=0, axis1=0, axis2=1, view=False} [id YO] 764 > │ │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id YP] 725 > │ │ │ │ │ │ │ │ ├─ Switch [id UF] 160 1391,1393c1388,1390 < │ │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id ZG] 702 < │ │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id ZH] 660 < │ │ │ │ │ │ │ │ ├─ Switch [id UR] 171 --- > │ │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id YQ] 683 > │ │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id YR] 641 > │ │ │ │ │ │ │ │ ├─ Switch [id UF] 160 1395,1396c1392,1393 < │ │ │ │ │ │ │ │ └─ Composite{((i0 * i1) - i2)} [id ZI] 618 < │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id VA] 421 --- > │ │ │ │ │ │ │ │ └─ Composite{((i0 * i1) - i2)} [id YS] 599 > │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id UO] 395 1398,1400c1395,1397 < │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id ZJ] 560 < │ │ │ │ │ │ │ │ │ └─ Tri{dtype='float64'} [id ZK] 510 < │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id VX] 457 --- > │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id YT] 540 > │ │ │ │ │ │ │ │ │ └─ Tri{dtype='float64'} [id YU] 490 > │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id VH] 433 1402,1403c1399,1400 < │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id ZL] 460 < │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{start:} [id VY] 434 --- > │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id YV] 440 > │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{start:} [id VI] 409 1405,1406c1402,1403 < │ │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id ZM] 34 < │ │ │ │ │ │ │ │ │ │ └─ 1 [id ZN] --- > │ │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id YW] 38 > │ │ │ │ │ │ │ │ │ │ └─ 1 [id YX] 1408c1405 < │ │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id WB] 600 --- > │ │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id VL] 577 1410c1407 < │ │ │ │ │ │ │ └─ ScalarFromTensor [id NX] 2 --- > │ │ │ │ │ │ │ └─ ScalarFromTensor [id NP] 2 1412,1414c1409,1411 < │ │ │ │ │ │ ├─ ExtractDiag{offset=0, axis1=0, axis2=1, view=False} [id ZO] 782 < │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id ZP] 742 < │ │ │ │ │ │ │ ├─ Switch [id UR] 171 --- > │ │ │ │ │ │ ├─ ExtractDiag{offset=0, axis1=0, axis2=1, view=False} [id YY] 765 > │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id YZ] 726 > │ │ │ │ │ │ │ ├─ Switch [id UF] 160 1416,1418c1413,1415 < │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id ZQ] 701 < │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id ZR] 659 < │ │ │ │ │ │ │ ├─ Switch [id UR] 171 --- > │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id ZA] 684 > │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id ZB] 642 > │ │ │ │ │ │ │ ├─ Switch [id UF] 160 1420,1421c1417,1418 < │ │ │ │ │ │ │ └─ Composite{((i0 * i1) - i2)} [id ZS] 617 < │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id VA] 421 --- > │ │ │ │ │ │ │ └─ Composite{((i0 * i1) - i2)} [id ZC] 600 > │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id UO] 395 1423,1425c1420,1422 < │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id ZT] 559 < │ │ │ │ │ │ │ │ └─ Tri{dtype='float64'} [id ZU] 509 < │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id VX] 457 --- > │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id ZD] 541 > │ │ │ │ │ │ │ │ └─ Tri{dtype='float64'} [id ZE] 491 > │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id VH] 433 1427,1428c1424,1425 < │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id ZV] 459 < │ │ │ │ │ │ │ │ │ ├─ Subtensor{start:} [id VY] 434 --- > │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id ZF] 441 > │ │ │ │ │ │ │ │ │ ├─ Subtensor{start:} [id VI] 409 1430,1431c1427,1428 < │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id ZW] 33 < │ │ │ │ │ │ │ │ │ └─ 1 [id ZX] --- > │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id ZG] 39 > │ │ │ │ │ │ │ │ │ └─ 1 [id ZH] 1433c1430 < │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id WB] 600 --- > │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id VL] 577 1435c1432 < │ │ │ │ │ │ ├─ ARange{dtype='int64'} [id ZY] 865 --- > │ │ │ │ │ │ ├─ ARange{dtype='int64'} [id ZI] 858 1437,1441c1434,1438 < │ │ │ │ │ │ │ ├─ Subtensor{i} [id ZZ] 839 < │ │ │ │ │ │ │ │ ├─ Shape [id BAA] 813 < │ │ │ │ │ │ │ │ │ └─ ExtractDiag{offset=0, axis1=0, axis2=1, view=False} [id BAB] 781 < │ │ │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id BAC] 741 < │ │ │ │ │ │ │ │ │ ├─ Switch [id UR] 171 --- > │ │ │ │ │ │ │ ├─ Subtensor{i} [id ZJ] 835 > │ │ │ │ │ │ │ │ ├─ Shape [id ZK] 809 > │ │ │ │ │ │ │ │ │ └─ ExtractDiag{offset=0, axis1=0, axis2=1, view=False} [id ZL] 783 > │ │ │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id ZM] 748 > │ │ │ │ │ │ │ │ │ ├─ Switch [id UF] 160 1443,1445c1440,1442 < │ │ │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id BAD] 700 < │ │ │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id BAE] 658 < │ │ │ │ │ │ │ │ │ ├─ Switch [id UR] 171 --- > │ │ │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id ZN] 706 > │ │ │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id ZO] 664 > │ │ │ │ │ │ │ │ │ ├─ Switch [id UF] 160 1447,1448c1444,1445 < │ │ │ │ │ │ │ │ │ └─ Composite{((i0 * i1) - i2)} [id BAF] 616 < │ │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id VA] 421 --- > │ │ │ │ │ │ │ │ │ └─ Composite{((i0 * i1) - i2)} [id ZP] 621 > │ │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id UO] 395 1450,1452c1447,1449 < │ │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id BAG] 558 < │ │ │ │ │ │ │ │ │ │ └─ Tri{dtype='float64'} [id BAH] 508 < │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id VX] 457 --- > │ │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id ZQ] 542 > │ │ │ │ │ │ │ │ │ │ └─ Tri{dtype='float64'} [id ZR] 492 > │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id VH] 433 1454,1455c1451,1452 < │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id BAI] 458 < │ │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{start:} [id VY] 434 --- > │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id ZS] 442 > │ │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{start:} [id VI] 409 1457,1458c1454,1455 < │ │ │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id BAJ] 32 < │ │ │ │ │ │ │ │ │ │ │ └─ 1 [id BAK] --- > │ │ │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id ZT] 40 > │ │ │ │ │ │ │ │ │ │ │ └─ 1 [id ZU] 1460c1457 < │ │ │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id WB] 600 --- > │ │ │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id VL] 577 1462c1459 < │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id NX] 2 --- > │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id NP] 2 1465c1462 < │ │ │ │ │ │ └─ ARange{dtype='int64'} [id BAL] 864 --- > │ │ │ │ │ │ └─ ARange{dtype='int64'} [id ZV] 844 1467,1470c1464,1467 < │ │ │ │ │ │ ├─ Subtensor{i} [id BAM] 838 < │ │ │ │ │ │ │ ├─ Shape [id BAN] 812 < │ │ │ │ │ │ │ │ └─ ExtractDiag{offset=0, axis1=0, axis2=1, view=False} [id BAO] 780 < │ │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id UQ] 740 --- > │ │ │ │ │ │ ├─ Subtensor{i} [id ZW] 818 > │ │ │ │ │ │ │ ├─ Shape [id ZX] 792 > │ │ │ │ │ │ │ │ └─ ExtractDiag{offset=0, axis1=0, axis2=1, view=False} [id ZY] 759 > │ │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id UE] 718 1472c1469 < │ │ │ │ │ │ │ └─ ScalarFromTensor [id NX] 2 --- > │ │ │ │ │ │ │ └─ ScalarFromTensor [id NP] 2 1475c1472 < │ │ │ │ │ ├─ Invert [id US] 152 --- > │ │ │ │ │ ├─ Invert [id UG] 143 1477,1489c1474,1486 < │ │ │ │ │ └─ [[nan]] [id OM] < │ │ │ │ ├─ [0 1] [id CN] < │ │ │ │ └─ [0 1] [id CN] < │ │ │ └─ 2 [id EI] < │ │ ├─ AdvancedSubtensor [id BAP] 907 < │ │ │ ├─ Composite{switch(i4, (((i0 + i1) * i2) - i3), i5)} [id BAQ] 902 < │ │ │ │ ├─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id BAR] 750 < │ │ │ │ │ ├─ Switch [id BAS] 169 < │ │ │ │ │ │ ├─ Invert [id BAT] 149 < │ │ │ │ │ │ │ └─ ExpandDims{axes=[0, 1]} [id BAU] 130 < │ │ │ │ │ │ │ └─ Any{axes=None} [id BAV] 111 < │ │ │ │ │ │ │ └─ Isnan [id BAW] 88 < │ │ │ │ │ │ │ └─ Cholesky{lower=True, destructive=False, on_error='nan'} [id EZ] 72 --- > │ │ │ │ │ └─ [[nan]] [id OE] > │ │ │ │ ├─ [0 1] [id CO] > │ │ │ │ └─ [0 1] [id CO] > │ │ │ └─ 2 [id EH] > │ │ ├─ AdvancedSubtensor [id ZZ] 887 > │ │ │ ├─ Composite{switch(i4, (((i0 + i1) * i2) - i3), i5)} [id BAA] 882 > │ │ │ │ ├─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id BAB] 727 > │ │ │ │ │ ├─ Switch [id BAC] 158 > │ │ │ │ │ │ ├─ Invert [id BAD] 141 > │ │ │ │ │ │ │ └─ ExpandDims{axes=[0, 1]} [id BAE] 125 > │ │ │ │ │ │ │ └─ Any{axes=None} [id BAF] 106 > │ │ │ │ │ │ │ └─ Isnan [id BAG] 85 > │ │ │ │ │ │ │ └─ Cholesky{lower=True, destructive=False, on_error='nan'} [id EX] 72 1491,1492c1488,1489 < │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id BAX] 89 < │ │ │ │ │ │ │ └─ Cholesky{lower=True, destructive=False, on_error='nan'} [id EZ] 72 --- > │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id BAH] 86 > │ │ │ │ │ │ │ └─ Cholesky{lower=True, destructive=False, on_error='nan'} [id EX] 72 1494,1497c1491,1494 < │ │ │ │ │ │ └─ [[1]] [id LY] < │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id BAY] 709 < │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id BAZ] 667 < │ │ │ │ │ ├─ Switch [id BAS] 169 --- > │ │ │ │ │ │ └─ [[1]] [id CQ] > │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id BAI] 685 > │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id BAJ] 643 > │ │ │ │ │ ├─ Switch [id BAC] 158 1499,1502c1496,1499 < │ │ │ │ │ └─ Composite{((i0 * i1) - i2)} [id BBA] 625 < │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id BBB] 424 < │ │ │ │ │ │ └─ dot [id BBC] 409 < │ │ │ │ │ │ ├─ Switch [id BAS] 169 --- > │ │ │ │ │ └─ Composite{((i0 * i1) - i2)} [id BAK] 601 > │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id BAL] 398 > │ │ │ │ │ │ └─ dot [id BAM] 384 > │ │ │ │ │ │ ├─ Switch [id BAC] 158 1504,1528c1501,1524 < │ │ │ │ │ │ └─ Composite{switch(i3, (i2 + switch(i1, (-i0), 0.0)), 1)} [id BBD] 402 < │ │ │ │ │ │ ├─ Sum{axis=0} [id BBE] 395 < │ │ │ │ │ │ │ └─ Mul [id BBF] 388 < │ │ │ │ │ │ │ ├─ Blockwise{dot, (i00,i01),(i10,i11)->(o00,o01)} [id BBG] 322 < │ │ │ │ │ │ │ │ ├─ ExpandDims{axis=2} [id BBH] 306 < │ │ │ │ │ │ │ │ │ └─ Blockwise{SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=1}, (m,m),(m)->(m)} [id KM] 292 < │ │ │ │ │ │ │ │ │ └─ ··· < │ │ │ │ │ │ │ │ └─ ExpandDims{axis=1} [id BBI] 214 < │ │ │ │ │ │ │ │ └─ Blockwise{SolveTriangular{trans=0, unit_diagonal=False, lower=True, check_finite=True, b_ndim=1}, (m,m),(m)->(m)} [id ET] 204 < │ │ │ │ │ │ │ │ └─ ··· < │ │ │ │ │ │ │ └─ Blockwise{Tri{dtype='float64'}, (),(),()->(o00,o01)} [id BBJ] 381 < │ │ │ │ │ │ │ ├─ Blockwise{Subtensor{i}, (i00),()->()} [id BBK] 372 < │ │ │ │ │ │ │ │ ├─ Blockwise{Subtensor{start:}, (i00),()->(o00)} [id BBL] 360 < │ │ │ │ │ │ │ │ │ ├─ Blockwise{Shape, (i00,i01)->(o00)} [id BBM] 348 < │ │ │ │ │ │ │ │ │ │ └─ Neg [id BBN] 336 < │ │ │ │ │ │ │ │ │ │ └─ Blockwise{dot, (i00,i01),(i10,i11)->(o00,o01)} [id BBG] 322 < │ │ │ │ │ │ │ │ │ │ └─ ··· < │ │ │ │ │ │ │ │ │ └─ [-2] [id MP] < │ │ │ │ │ │ │ │ └─ [0] [id MQ] < │ │ │ │ │ │ │ ├─ Blockwise{Subtensor{i}, (i00),()->()} [id BBO] 371 < │ │ │ │ │ │ │ │ ├─ Blockwise{Subtensor{start:}, (i00),()->(o00)} [id BBL] 360 < │ │ │ │ │ │ │ │ │ └─ ··· < │ │ │ │ │ │ │ │ └─ [1] [id MS] < │ │ │ │ │ │ │ └─ [0] [id CO] < │ │ │ │ │ │ ├─ ExpandDims{axes=[0, 1]} [id FJ] 147 --- > │ │ │ │ │ │ └─ Composite{switch(i4, (i3 + switch(i2, (-1.0 * i0 * i1), 0.0)), 1)} [id BAN] 377 > │ │ │ │ │ │ ├─ dot [id BAO] 307 > │ │ │ │ │ │ │ ├─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id KK] 293 > │ │ │ │ │ │ │ │ └─ ··· > │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id EQ] 202 > │ │ │ │ │ │ │ └─ ··· > │ │ │ │ │ │ ├─ Tri{dtype='float64'} [id BAP] 370 > │ │ │ │ │ │ │ ├─ Subtensor{i} [id BAQ] 361 > │ │ │ │ │ │ │ │ ├─ Subtensor{start:} [id BAR] 349 > │ │ │ │ │ │ │ │ │ ├─ Shape [id BAS] 337 > │ │ │ │ │ │ │ │ │ │ └─ Neg [id BAT] 323 > │ │ │ │ │ │ │ │ │ │ └─ dot [id BAO] 307 > │ │ │ │ │ │ │ │ │ │ └─ ··· > │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id MK] 12 > │ │ │ │ │ │ │ │ │ └─ ··· > │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id MM] 13 > │ │ │ │ │ │ │ │ └─ ··· > │ │ │ │ │ │ │ ├─ Subtensor{i} [id BAU] 360 > │ │ │ │ │ │ │ │ ├─ Subtensor{start:} [id BAR] 349 > │ │ │ │ │ │ │ │ │ └─ ··· > │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id MP] 11 > │ │ │ │ │ │ │ │ └─ ··· > │ │ │ │ │ │ │ └─ 0 [id E] > │ │ │ │ │ │ ├─ ExpandDims{axes=[0, 1]} [id ET] 140 1530,1531c1526,1527 < │ │ │ │ │ │ ├─ SetSubtensor{:stop, start:stop} [id BBP] 359 < │ │ │ │ │ │ │ ├─ Alloc [id MU] 9 --- > │ │ │ │ │ │ ├─ SetSubtensor{:stop, start:stop} [id BAV] 348 > │ │ │ │ │ │ │ ├─ Alloc [id MS] 9 1533,1534c1529,1530 < │ │ │ │ │ │ │ ├─ AdvancedSetSubtensor [id BBQ] 347 < │ │ │ │ │ │ │ │ ├─ Alloc [id CK] 1 --- > │ │ │ │ │ │ │ ├─ AdvancedSetSubtensor [id BAW] 336 > │ │ │ │ │ │ │ │ ├─ Alloc [id CL] 1 1536,1541c1532,1537 < │ │ │ │ │ │ │ │ ├─ True_div [id BBR] 335 < │ │ │ │ │ │ │ │ │ ├─ Neg [id BBS] 321 < │ │ │ │ │ │ │ │ │ │ └─ ExpandDims{axis=0} [id BBT] 305 < │ │ │ │ │ │ │ │ │ │ └─ Sum{axes=None} [id BBU] 291 < │ │ │ │ │ │ │ │ │ │ └─ DropDims{axis=1} [id BBV] 280 < │ │ │ │ │ │ │ │ │ │ └─ SpecifyShape [id KP] 271 --- > │ │ │ │ │ │ │ │ ├─ True_div [id BAX] 322 > │ │ │ │ │ │ │ │ │ ├─ Neg [id BAY] 306 > │ │ │ │ │ │ │ │ │ │ └─ ExpandDims{axis=0} [id BAZ] 292 > │ │ │ │ │ │ │ │ │ │ └─ Sum{axes=None} [id BBA] 281 > │ │ │ │ │ │ │ │ │ │ └─ DropDims{axis=1} [id BBB] 270 > │ │ │ │ │ │ │ │ │ │ └─ SpecifyShape [id KO] 261 1543c1539 < │ │ │ │ │ │ │ │ │ └─ ExtractDiag{offset=0, axis1=0, axis2=1, view=False} [id EY] 86 --- > │ │ │ │ │ │ │ │ │ └─ ExtractDiag{offset=0, axis1=0, axis2=1, view=False} [id EW] 84 1545,1550c1541,1546 < │ │ │ │ │ │ │ │ ├─ [0 1] [id CN] < │ │ │ │ │ │ │ │ └─ [0 1] [id CN] < │ │ │ │ │ │ │ ├─ 2 [id NB] < │ │ │ │ │ │ │ ├─ 0 [id NC] < │ │ │ │ │ │ │ └─ 2 [id NB] < │ │ │ │ │ │ └─ Invert [id BAT] 149 --- > │ │ │ │ │ │ │ │ ├─ [0 1] [id CO] > │ │ │ │ │ │ │ │ └─ [0 1] [id CO] > │ │ │ │ │ │ │ ├─ 2 [id MZ] > │ │ │ │ │ │ │ ├─ 0 [id NA] > │ │ │ │ │ │ │ └─ 2 [id MZ] > │ │ │ │ │ │ └─ Invert [id BAD] 141 1552,1557c1548,1553 < │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id BBW] 569 < │ │ │ │ │ │ └─ Tri{dtype='float64'} [id BBX] 518 < │ │ │ │ │ │ ├─ Subtensor{i} [id BBY] 469 < │ │ │ │ │ │ │ ├─ Subtensor{start:} [id BBZ] 436 < │ │ │ │ │ │ │ │ ├─ Shape [id BCA] 423 < │ │ │ │ │ │ │ │ │ └─ dot [id BBC] 409 --- > │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id BBC] 545 > │ │ │ │ │ │ └─ Tri{dtype='float64'} [id BBD] 494 > │ │ │ │ │ │ ├─ Subtensor{i} [id BBE] 445 > │ │ │ │ │ │ │ ├─ Subtensor{start:} [id BBF] 411 > │ │ │ │ │ │ │ │ ├─ Shape [id BBG] 397 > │ │ │ │ │ │ │ │ │ └─ dot [id BAM] 384 1559c1555 < │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id NI] 12 --- > │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id MK] 12 1561c1557 < │ │ │ │ │ │ │ └─ ScalarFromTensor [id NK] 13 --- > │ │ │ │ │ │ │ └─ ScalarFromTensor [id MM] 13 1563,1564c1559,1560 < │ │ │ │ │ │ ├─ Subtensor{i} [id BCB] 468 < │ │ │ │ │ │ │ ├─ Subtensor{start:} [id BBZ] 436 --- > │ │ │ │ │ │ ├─ Subtensor{i} [id BBH] 444 > │ │ │ │ │ │ │ ├─ Subtensor{start:} [id BBF] 411 1566c1562 < │ │ │ │ │ │ │ └─ ScalarFromTensor [id NN] 11 --- > │ │ │ │ │ │ │ └─ ScalarFromTensor [id MP] 11 1569,1571c1565,1567 < │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id BCC] 601 < │ │ │ │ │ └─ AdvancedSetSubtensor [id BCD] 595 < │ │ │ │ │ ├─ Alloc [id BCE] 568 --- > │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id BBI] 578 > │ │ │ │ │ └─ AdvancedSetSubtensor [id BBJ] 571 > │ │ │ │ │ ├─ Alloc [id BBK] 544 1573,1578c1569,1574 < │ │ │ │ │ │ ├─ Subtensor{i} [id BCF] 517 < │ │ │ │ │ │ │ ├─ Shape [id BCG] 467 < │ │ │ │ │ │ │ │ └─ Mul [id BCH] 435 < │ │ │ │ │ │ │ │ ├─ [0.5] [id NV] < │ │ │ │ │ │ │ │ └─ ExtractDiag{offset=0, axis1=0, axis2=1, view=False} [id BCI] 422 < │ │ │ │ │ │ │ │ └─ dot [id BBC] 409 --- > │ │ │ │ │ │ ├─ Subtensor{i} [id BBL] 493 > │ │ │ │ │ │ │ ├─ Shape [id BBM] 443 > │ │ │ │ │ │ │ │ └─ Mul [id BBN] 410 > │ │ │ │ │ │ │ │ ├─ [0.5] [id NN] > │ │ │ │ │ │ │ │ └─ ExtractDiag{offset=0, axis1=0, axis2=1, view=False} [id BBO] 396 > │ │ │ │ │ │ │ │ └─ dot [id BAM] 384 1580c1576 < │ │ │ │ │ │ │ └─ ScalarFromTensor [id NX] 2 --- > │ │ │ │ │ │ │ └─ ScalarFromTensor [id NP] 2 1582c1578 < │ │ │ │ │ │ └─ Subtensor{i} [id BCF] 517 --- > │ │ │ │ │ │ └─ Subtensor{i} [id BBL] 493 1584c1580 < │ │ │ │ │ ├─ Mul [id BCH] 435 --- > │ │ │ │ │ ├─ Mul [id BBN] 410 1586c1582 < │ │ │ │ │ ├─ ARange{dtype='int64'} [id BCJ] 567 --- > │ │ │ │ │ ├─ ARange{dtype='int64'} [id BBP] 543 1588c1584 < │ │ │ │ │ │ ├─ Subtensor{i} [id BCF] 517 --- > │ │ │ │ │ │ ├─ Subtensor{i} [id BBL] 493 1591c1587 < │ │ │ │ │ └─ ARange{dtype='int64'} [id BCJ] 567 --- > │ │ │ │ │ └─ ARange{dtype='int64'} [id BBP] 543 1593,1595c1589,1591 < │ │ │ │ ├─ Transpose{axes=[1, 0]} [id BCK] 795 < │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id BCL] 759 < │ │ │ │ │ ├─ Switch [id BAS] 169 --- > │ │ │ │ ├─ Transpose{axes=[1, 0]} [id BBQ] 773 > │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id BBR] 736 > │ │ │ │ │ ├─ Switch [id BAC] 158 1597,1599c1593,1595 < │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id BCM] 718 < │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id BCN] 676 < │ │ │ │ │ ├─ Switch [id BAS] 169 --- > │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id BBS] 694 > │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id BBT] 652 > │ │ │ │ │ ├─ Switch [id BAC] 158 1601,1602c1597,1598 < │ │ │ │ │ └─ Composite{((i0 * i1) - i2)} [id BCO] 634 < │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id BBB] 424 --- > │ │ │ │ │ └─ Composite{((i0 * i1) - i2)} [id BBU] 610 > │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id BAL] 398 1604,1606c1600,1602 < │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id BCP] 578 < │ │ │ │ │ │ └─ Tri{dtype='float64'} [id BCQ] 527 < │ │ │ │ │ │ ├─ Subtensor{i} [id BBY] 469 --- > │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id BBV] 554 > │ │ │ │ │ │ └─ Tri{dtype='float64'} [id BBW] 503 > │ │ │ │ │ │ ├─ Subtensor{i} [id BBE] 445 1608,1609c1604,1605 < │ │ │ │ │ │ ├─ Subtensor{i} [id BCR] 478 < │ │ │ │ │ │ │ ├─ Subtensor{start:} [id BBZ] 436 --- > │ │ │ │ │ │ ├─ Subtensor{i} [id BBX] 454 > │ │ │ │ │ │ │ ├─ Subtensor{start:} [id BBF] 411 1611,1612c1607,1608 < │ │ │ │ │ │ │ └─ ScalarFromTensor [id BCS] 31 < │ │ │ │ │ │ │ └─ 1 [id BCT] --- > │ │ │ │ │ │ │ └─ ScalarFromTensor [id BBY] 31 > │ │ │ │ │ │ │ └─ 1 [id BBZ] 1614c1610 < │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id BCC] 601 --- > │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id BBI] 578 1616,1622c1612,1618 < │ │ │ │ ├─ Tri{dtype='float64'} [id BCU] 897 < │ │ │ │ │ ├─ Subtensor{i} [id BCV] 890 < │ │ │ │ │ │ ├─ Subtensor{start:} [id BCW] 873 < │ │ │ │ │ │ │ ├─ Shape [id BCX] 849 < │ │ │ │ │ │ │ │ └─ Add [id BCY] 823 < │ │ │ │ │ │ │ │ ├─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id BCZ] 758 < │ │ │ │ │ │ │ │ │ ├─ Switch [id BAS] 169 --- > │ │ │ │ ├─ Tri{dtype='float64'} [id BCA] 877 > │ │ │ │ │ ├─ Subtensor{i} [id BCB] 869 > │ │ │ │ │ │ ├─ Subtensor{start:} [id BCC] 852 > │ │ │ │ │ │ │ ├─ Shape [id BCD] 828 > │ │ │ │ │ │ │ │ └─ Add [id BCE] 802 > │ │ │ │ │ │ │ │ ├─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id BCF] 735 > │ │ │ │ │ │ │ │ │ ├─ Switch [id BAC] 158 1624,1626c1620,1622 < │ │ │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id BDA] 717 < │ │ │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id BDB] 675 < │ │ │ │ │ │ │ │ │ ├─ Switch [id BAS] 169 --- > │ │ │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id BCG] 693 > │ │ │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id BCH] 651 > │ │ │ │ │ │ │ │ │ ├─ Switch [id BAC] 158 1628,1629c1624,1625 < │ │ │ │ │ │ │ │ │ └─ Composite{((i0 * i1) - i2)} [id BDC] 633 < │ │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id BBB] 424 --- > │ │ │ │ │ │ │ │ │ └─ Composite{((i0 * i1) - i2)} [id BCI] 609 > │ │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id BAL] 398 1631,1633c1627,1629 < │ │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id BDD] 577 < │ │ │ │ │ │ │ │ │ │ └─ Tri{dtype='float64'} [id BDE] 526 < │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id BBY] 469 --- > │ │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id BCJ] 553 > │ │ │ │ │ │ │ │ │ │ └─ Tri{dtype='float64'} [id BCK] 502 > │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id BBE] 445 1635,1636c1631,1632 < │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id BDF] 477 < │ │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{start:} [id BBZ] 436 --- > │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id BCL] 453 > │ │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{start:} [id BBF] 411 1638,1639c1634,1635 < │ │ │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id BDG] 30 < │ │ │ │ │ │ │ │ │ │ │ └─ 1 [id BDH] --- > │ │ │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id BCM] 30 > │ │ │ │ │ │ │ │ │ │ │ └─ 1 [id BCN] 1641c1637 < │ │ │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id BCC] 601 --- > │ │ │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id BBI] 578 1643,1645c1639,1641 < │ │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id BDI] 794 < │ │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id BDJ] 757 < │ │ │ │ │ │ │ │ ├─ Switch [id BAS] 169 --- > │ │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id BCO] 772 > │ │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id BCP] 734 > │ │ │ │ │ │ │ │ ├─ Switch [id BAC] 158 1647,1649c1643,1645 < │ │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id BDK] 716 < │ │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id BDL] 674 < │ │ │ │ │ │ │ │ ├─ Switch [id BAS] 169 --- > │ │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id BCQ] 692 > │ │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id BCR] 650 > │ │ │ │ │ │ │ │ ├─ Switch [id BAC] 158 1651,1652c1647,1648 < │ │ │ │ │ │ │ │ └─ Composite{((i0 * i1) - i2)} [id BDM] 632 < │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id BBB] 424 --- > │ │ │ │ │ │ │ │ └─ Composite{((i0 * i1) - i2)} [id BCS] 608 > │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id BAL] 398 1654,1656c1650,1652 < │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id BDN] 576 < │ │ │ │ │ │ │ │ │ └─ Tri{dtype='float64'} [id BDO] 525 < │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id BBY] 469 --- > │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id BCT] 552 > │ │ │ │ │ │ │ │ │ └─ Tri{dtype='float64'} [id BCU] 501 > │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id BBE] 445 1658,1659c1654,1655 < │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id BDP] 476 < │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{start:} [id BBZ] 436 --- > │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id BCV] 452 > │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{start:} [id BBF] 411 1661,1662c1657,1658 < │ │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id BDQ] 29 < │ │ │ │ │ │ │ │ │ │ └─ 1 [id BDR] --- > │ │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id BCW] 29 > │ │ │ │ │ │ │ │ │ │ └─ 1 [id BCX] 1664c1660 < │ │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id BCC] 601 --- > │ │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id BBI] 578 1666c1662 < │ │ │ │ │ │ │ └─ ScalarFromTensor [id NI] 12 --- > │ │ │ │ │ │ │ └─ ScalarFromTensor [id MK] 12 1668c1664 < │ │ │ │ │ │ └─ ScalarFromTensor [id NK] 13 --- > │ │ │ │ │ │ └─ ScalarFromTensor [id MM] 13 1670,1675c1666,1671 < │ │ │ │ │ ├─ Subtensor{i} [id BDS] 889 < │ │ │ │ │ │ ├─ Subtensor{start:} [id BDT] 872 < │ │ │ │ │ │ │ ├─ Shape [id BDU] 848 < │ │ │ │ │ │ │ │ └─ Add [id BDV] 822 < │ │ │ │ │ │ │ │ ├─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id BDW] 756 < │ │ │ │ │ │ │ │ │ ├─ Switch [id BAS] 169 --- > │ │ │ │ │ ├─ Subtensor{i} [id BCY] 868 > │ │ │ │ │ │ ├─ Subtensor{start:} [id BCZ] 851 > │ │ │ │ │ │ │ ├─ Shape [id BDA] 827 > │ │ │ │ │ │ │ │ └─ Add [id BDB] 801 > │ │ │ │ │ │ │ │ ├─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id BDC] 733 > │ │ │ │ │ │ │ │ │ ├─ Switch [id BAC] 158 1677,1679c1673,1675 < │ │ │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id BDX] 715 < │ │ │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id BDY] 673 < │ │ │ │ │ │ │ │ │ ├─ Switch [id BAS] 169 --- > │ │ │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id BDD] 691 > │ │ │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id BDE] 649 > │ │ │ │ │ │ │ │ │ ├─ Switch [id BAC] 158 1681,1682c1677,1678 < │ │ │ │ │ │ │ │ │ └─ Composite{((i0 * i1) - i2)} [id BDZ] 631 < │ │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id BBB] 424 --- > │ │ │ │ │ │ │ │ │ └─ Composite{((i0 * i1) - i2)} [id BDF] 607 > │ │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id BAL] 398 1684,1686c1680,1682 < │ │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id BEA] 575 < │ │ │ │ │ │ │ │ │ │ └─ Tri{dtype='float64'} [id BEB] 524 < │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id BBY] 469 --- > │ │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id BDG] 551 > │ │ │ │ │ │ │ │ │ │ └─ Tri{dtype='float64'} [id BDH] 500 > │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id BBE] 445 1688,1689c1684,1685 < │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id BEC] 475 < │ │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{start:} [id BBZ] 436 --- > │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id BDI] 451 > │ │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{start:} [id BBF] 411 1691,1692c1687,1688 < │ │ │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id BED] 28 < │ │ │ │ │ │ │ │ │ │ │ └─ 1 [id BEE] --- > │ │ │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id BDJ] 28 > │ │ │ │ │ │ │ │ │ │ │ └─ 1 [id BDK] 1694c1690 < │ │ │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id BCC] 601 --- > │ │ │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id BBI] 578 1696,1698c1692,1694 < │ │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id BEF] 793 < │ │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id BEG] 755 < │ │ │ │ │ │ │ │ ├─ Switch [id BAS] 169 --- > │ │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id BDL] 771 > │ │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id BDM] 732 > │ │ │ │ │ │ │ │ ├─ Switch [id BAC] 158 1700,1702c1696,1698 < │ │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id BEH] 714 < │ │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id BEI] 672 < │ │ │ │ │ │ │ │ ├─ Switch [id BAS] 169 --- > │ │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id BDN] 690 > │ │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id BDO] 648 > │ │ │ │ │ │ │ │ ├─ Switch [id BAC] 158 1704,1705c1700,1701 < │ │ │ │ │ │ │ │ └─ Composite{((i0 * i1) - i2)} [id BEJ] 630 < │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id BBB] 424 --- > │ │ │ │ │ │ │ │ └─ Composite{((i0 * i1) - i2)} [id BDP] 606 > │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id BAL] 398 1707,1709c1703,1705 < │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id BEK] 574 < │ │ │ │ │ │ │ │ │ └─ Tri{dtype='float64'} [id BEL] 523 < │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id BBY] 469 --- > │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id BDQ] 550 > │ │ │ │ │ │ │ │ │ └─ Tri{dtype='float64'} [id BDR] 499 > │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id BBE] 445 1711,1712c1707,1708 < │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id BEM] 474 < │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{start:} [id BBZ] 436 --- > │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id BDS] 450 > │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{start:} [id BBF] 411 1714,1715c1710,1711 < │ │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id BEN] 27 < │ │ │ │ │ │ │ │ │ │ └─ 1 [id BEO] --- > │ │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id BDT] 27 > │ │ │ │ │ │ │ │ │ │ └─ 1 [id BDU] 1717c1713 < │ │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id BCC] 601 --- > │ │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id BBI] 578 1719c1715 < │ │ │ │ │ │ │ └─ ScalarFromTensor [id NI] 12 --- > │ │ │ │ │ │ │ └─ ScalarFromTensor [id MK] 12 1721c1717 < │ │ │ │ │ │ └─ ScalarFromTensor [id NN] 11 --- > │ │ │ │ │ │ └─ ScalarFromTensor [id MP] 11 1724,1725c1720,1721 < │ │ │ │ ├─ AdvancedSetSubtensor [id BEP] 888 < │ │ │ │ │ ├─ Alloc [id BEQ] 871 --- > │ │ │ │ ├─ AdvancedSetSubtensor [id BDV] 867 > │ │ │ │ │ ├─ Alloc [id BDW] 850 1727,1731c1723,1727 < │ │ │ │ │ │ ├─ Subtensor{i} [id BER] 847 < │ │ │ │ │ │ │ ├─ Shape [id BES] 821 < │ │ │ │ │ │ │ │ └─ ExtractDiag{offset=0, axis1=0, axis2=1, view=False} [id BET] 792 < │ │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id BEU] 754 < │ │ │ │ │ │ │ │ ├─ Switch [id BAS] 169 --- > │ │ │ │ │ │ ├─ Subtensor{i} [id BDX] 826 > │ │ │ │ │ │ │ ├─ Shape [id BDY] 800 > │ │ │ │ │ │ │ │ └─ ExtractDiag{offset=0, axis1=0, axis2=1, view=False} [id BDZ] 770 > │ │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id BEA] 731 > │ │ │ │ │ │ │ │ ├─ Switch [id BAC] 158 1733,1735c1729,1731 < │ │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id BEV] 713 < │ │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id BEW] 671 < │ │ │ │ │ │ │ │ ├─ Switch [id BAS] 169 --- > │ │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id BEB] 689 > │ │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id BEC] 647 > │ │ │ │ │ │ │ │ ├─ Switch [id BAC] 158 1737,1738c1733,1734 < │ │ │ │ │ │ │ │ └─ Composite{((i0 * i1) - i2)} [id BEX] 629 < │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id BBB] 424 --- > │ │ │ │ │ │ │ │ └─ Composite{((i0 * i1) - i2)} [id BED] 605 > │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id BAL] 398 1740,1742c1736,1738 < │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id BEY] 573 < │ │ │ │ │ │ │ │ │ └─ Tri{dtype='float64'} [id BEZ] 522 < │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id BBY] 469 --- > │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id BEE] 549 > │ │ │ │ │ │ │ │ │ └─ Tri{dtype='float64'} [id BEF] 498 > │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id BBE] 445 1744,1745c1740,1741 < │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id BFA] 473 < │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{start:} [id BBZ] 436 --- > │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id BEG] 449 > │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{start:} [id BBF] 411 1747,1748c1743,1744 < │ │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id BFB] 26 < │ │ │ │ │ │ │ │ │ │ └─ 1 [id BFC] --- > │ │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id BEH] 26 > │ │ │ │ │ │ │ │ │ │ └─ 1 [id BEI] 1750c1746 < │ │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id BCC] 601 --- > │ │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id BBI] 578 1752c1748 < │ │ │ │ │ │ │ └─ ScalarFromTensor [id NX] 2 --- > │ │ │ │ │ │ │ └─ ScalarFromTensor [id NP] 2 1754,1758c1750,1754 < │ │ │ │ │ │ └─ Subtensor{i} [id BFD] 846 < │ │ │ │ │ │ ├─ Shape [id BFE] 820 < │ │ │ │ │ │ │ └─ ExtractDiag{offset=0, axis1=0, axis2=1, view=False} [id BFF] 791 < │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id BFG] 753 < │ │ │ │ │ │ │ ├─ Switch [id BAS] 169 --- > │ │ │ │ │ │ └─ Subtensor{i} [id BEJ] 825 > │ │ │ │ │ │ ├─ Shape [id BEK] 799 > │ │ │ │ │ │ │ └─ ExtractDiag{offset=0, axis1=0, axis2=1, view=False} [id BEL] 769 > │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id BEM] 730 > │ │ │ │ │ │ │ ├─ Switch [id BAC] 158 1760,1762c1756,1758 < │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id BFH] 712 < │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id BFI] 670 < │ │ │ │ │ │ │ ├─ Switch [id BAS] 169 --- > │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id BEN] 688 > │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id BEO] 646 > │ │ │ │ │ │ │ ├─ Switch [id BAC] 158 1764,1765c1760,1761 < │ │ │ │ │ │ │ └─ Composite{((i0 * i1) - i2)} [id BFJ] 628 < │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id BBB] 424 --- > │ │ │ │ │ │ │ └─ Composite{((i0 * i1) - i2)} [id BEP] 604 > │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id BAL] 398 1767,1769c1763,1765 < │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id BFK] 572 < │ │ │ │ │ │ │ │ └─ Tri{dtype='float64'} [id BFL] 521 < │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id BBY] 469 --- > │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id BEQ] 548 > │ │ │ │ │ │ │ │ └─ Tri{dtype='float64'} [id BER] 497 > │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id BBE] 445 1771,1772c1767,1768 < │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id BFM] 472 < │ │ │ │ │ │ │ │ │ ├─ Subtensor{start:} [id BBZ] 436 --- > │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id BES] 448 > │ │ │ │ │ │ │ │ │ ├─ Subtensor{start:} [id BBF] 411 1774,1775c1770,1771 < │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id BFN] 25 < │ │ │ │ │ │ │ │ │ └─ 1 [id BFO] --- > │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id BET] 25 > │ │ │ │ │ │ │ │ │ └─ 1 [id BEU] 1777c1773 < │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id BCC] 601 --- > │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id BBI] 578 1779c1775 < │ │ │ │ │ │ └─ ScalarFromTensor [id NX] 2 --- > │ │ │ │ │ │ └─ ScalarFromTensor [id NP] 2 1781,1783c1777,1779 < │ │ │ │ │ ├─ ExtractDiag{offset=0, axis1=0, axis2=1, view=False} [id BFP] 790 < │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id BFQ] 752 < │ │ │ │ │ │ ├─ Switch [id BAS] 169 --- > │ │ │ │ │ ├─ ExtractDiag{offset=0, axis1=0, axis2=1, view=False} [id BEV] 768 > │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id BEW] 729 > │ │ │ │ │ │ ├─ Switch [id BAC] 158 1785,1787c1781,1783 < │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id BFR] 711 < │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id BFS] 669 < │ │ │ │ │ │ ├─ Switch [id BAS] 169 --- > │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id BEX] 687 > │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id BEY] 645 > │ │ │ │ │ │ ├─ Switch [id BAC] 158 1789,1790c1785,1786 < │ │ │ │ │ │ └─ Composite{((i0 * i1) - i2)} [id BFT] 627 < │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id BBB] 424 --- > │ │ │ │ │ │ └─ Composite{((i0 * i1) - i2)} [id BEZ] 603 > │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id BAL] 398 1792,1794c1788,1790 < │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id BFU] 571 < │ │ │ │ │ │ │ └─ Tri{dtype='float64'} [id BFV] 520 < │ │ │ │ │ │ │ ├─ Subtensor{i} [id BBY] 469 --- > │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id BFA] 547 > │ │ │ │ │ │ │ └─ Tri{dtype='float64'} [id BFB] 496 > │ │ │ │ │ │ │ ├─ Subtensor{i} [id BBE] 445 1796,1797c1792,1793 < │ │ │ │ │ │ │ ├─ Subtensor{i} [id BFW] 471 < │ │ │ │ │ │ │ │ ├─ Subtensor{start:} [id BBZ] 436 --- > │ │ │ │ │ │ │ ├─ Subtensor{i} [id BFC] 447 > │ │ │ │ │ │ │ │ ├─ Subtensor{start:} [id BBF] 411 1799,1800c1795,1796 < │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id BFX] 24 < │ │ │ │ │ │ │ │ └─ 1 [id BFY] --- > │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id BFD] 24 > │ │ │ │ │ │ │ │ └─ 1 [id BFE] 1802c1798 < │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id BCC] 601 --- > │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id BBI] 578 1804c1800 < │ │ │ │ │ ├─ ARange{dtype='int64'} [id BFZ] 870 --- > │ │ │ │ │ ├─ ARange{dtype='int64'} [id BFF] 849 1806,1810c1802,1806 < │ │ │ │ │ │ ├─ Subtensor{i} [id BGA] 845 < │ │ │ │ │ │ │ ├─ Shape [id BGB] 819 < │ │ │ │ │ │ │ │ └─ ExtractDiag{offset=0, axis1=0, axis2=1, view=False} [id BGC] 789 < │ │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id BGD] 751 < │ │ │ │ │ │ │ │ ├─ Switch [id BAS] 169 --- > │ │ │ │ │ │ ├─ Subtensor{i} [id BFG] 824 > │ │ │ │ │ │ │ ├─ Shape [id BFH] 798 > │ │ │ │ │ │ │ │ └─ ExtractDiag{offset=0, axis1=0, axis2=1, view=False} [id BFI] 767 > │ │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id BFJ] 728 > │ │ │ │ │ │ │ │ ├─ Switch [id BAC] 158 1812,1814c1808,1810 < │ │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id BGE] 710 < │ │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id BGF] 668 < │ │ │ │ │ │ │ │ ├─ Switch [id BAS] 169 --- > │ │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id BFK] 686 > │ │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id BFL] 644 > │ │ │ │ │ │ │ │ ├─ Switch [id BAC] 158 1816,1817c1812,1813 < │ │ │ │ │ │ │ │ └─ Composite{((i0 * i1) - i2)} [id BGG] 626 < │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id BBB] 424 --- > │ │ │ │ │ │ │ │ └─ Composite{((i0 * i1) - i2)} [id BFM] 602 > │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id BAL] 398 1819,1821c1815,1817 < │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id BGH] 570 < │ │ │ │ │ │ │ │ │ └─ Tri{dtype='float64'} [id BGI] 519 < │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id BBY] 469 --- > │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id BFN] 546 > │ │ │ │ │ │ │ │ │ └─ Tri{dtype='float64'} [id BFO] 495 > │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id BBE] 445 1823,1824c1819,1820 < │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id BGJ] 470 < │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{start:} [id BBZ] 436 --- > │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id BFP] 446 > │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{start:} [id BBF] 411 1826,1827c1822,1823 < │ │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id BGK] 23 < │ │ │ │ │ │ │ │ │ │ └─ 1 [id BGL] --- > │ │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id BFQ] 23 > │ │ │ │ │ │ │ │ │ │ └─ 1 [id BFR] 1829c1825 < │ │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id BCC] 601 --- > │ │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id BBI] 578 1831c1827 < │ │ │ │ │ │ │ └─ ScalarFromTensor [id NX] 2 --- > │ │ │ │ │ │ │ └─ ScalarFromTensor [id NP] 2 1834c1830 < │ │ │ │ │ └─ ARange{dtype='int64'} [id BGM] 869 --- > │ │ │ │ │ └─ ARange{dtype='int64'} [id BFS] 848 1836,1839c1832,1835 < │ │ │ │ │ ├─ Subtensor{i} [id BGN] 844 < │ │ │ │ │ │ ├─ Shape [id BGO] 818 < │ │ │ │ │ │ │ └─ ExtractDiag{offset=0, axis1=0, axis2=1, view=False} [id BGP] 788 < │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id BAR] 750 --- > │ │ │ │ │ ├─ Subtensor{i} [id BFT] 823 > │ │ │ │ │ │ ├─ Shape [id BFU] 797 > │ │ │ │ │ │ │ └─ ExtractDiag{offset=0, axis1=0, axis2=1, view=False} [id BFV] 766 > │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id BAB] 727 1841c1837 < │ │ │ │ │ │ └─ ScalarFromTensor [id NX] 2 --- > │ │ │ │ │ │ └─ ScalarFromTensor [id NP] 2 1844c1840 < │ │ │ │ ├─ Invert [id BAT] 149 --- > │ │ │ │ ├─ Invert [id BAD] 141 1846,1858c1842,1854 < │ │ │ │ └─ [[nan]] [id OM] < │ │ │ ├─ [0 1] [id CN] < │ │ │ └─ [0 1] [id CN] < │ │ └─ 3 [id FC] < │ ├─ AdvancedSubtensor [id BGQ] 908 < │ │ ├─ Composite{switch(i4, (((i0 + i1) * i2) - i3), i5)} [id BGR] 903 < │ │ │ ├─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id BGS] 760 < │ │ │ │ ├─ Switch [id BGT] 164 < │ │ │ │ │ ├─ Invert [id BGU] 141 < │ │ │ │ │ │ └─ ExpandDims{axes=[0, 1]} [id BGV] 122 < │ │ │ │ │ │ └─ Any{axes=None} [id BGW] 103 < │ │ │ │ │ │ └─ Isnan [id BGX] 78 < │ │ │ │ │ │ └─ Cholesky{lower=True, destructive=False, on_error='nan'} [id FT] 70 --- > │ │ │ │ └─ [[nan]] [id OE] > │ │ │ ├─ [0 1] [id CO] > │ │ │ └─ [0 1] [id CO] > │ │ └─ 3 [id FA] > │ ├─ AdvancedSubtensor [id BFW] 888 > │ │ ├─ Composite{switch(i4, (((i0 + i1) * i2) - i3), i5)} [id BFX] 883 > │ │ │ ├─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id BFY] 737 > │ │ │ │ ├─ Switch [id BFZ] 154 > │ │ │ │ │ ├─ Invert [id BGA] 136 > │ │ │ │ │ │ └─ ExpandDims{axes=[0, 1]} [id BGB] 117 > │ │ │ │ │ │ └─ Any{axes=None} [id BGC] 98 > │ │ │ │ │ │ └─ Isnan [id BGD] 78 > │ │ │ │ │ │ └─ Cholesky{lower=True, destructive=False, on_error='nan'} [id FQ] 70 1860,1861c1856,1857 < │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id BGY] 81 < │ │ │ │ │ │ └─ Cholesky{lower=True, destructive=False, on_error='nan'} [id FT] 70 --- > │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id BGE] 80 > │ │ │ │ │ │ └─ Cholesky{lower=True, destructive=False, on_error='nan'} [id FQ] 70 1863,1866c1859,1862 < │ │ │ │ │ └─ [[1]] [id LY] < │ │ │ │ └─ Transpose{axes=[1, 0]} [id BGZ] 719 < │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id BHA] 677 < │ │ │ │ ├─ Switch [id BGT] 164 --- > │ │ │ │ │ └─ [[1]] [id CQ] > │ │ │ │ └─ Transpose{axes=[1, 0]} [id BGF] 695 > │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id BGG] 653 > │ │ │ │ ├─ Switch [id BFZ] 154 1868,1871c1864,1867 < │ │ │ │ └─ Composite{((i0 * i1) - i2)} [id BHB] 635 < │ │ │ │ ├─ Transpose{axes=[1, 0]} [id BHC] 427 < │ │ │ │ │ └─ dot [id BHD] 410 < │ │ │ │ │ ├─ Switch [id BGT] 164 --- > │ │ │ │ └─ Composite{((i0 * i1) - i2)} [id BGH] 611 > │ │ │ │ ├─ Transpose{axes=[1, 0]} [id BGI] 401 > │ │ │ │ │ └─ dot [id BGJ] 385 > │ │ │ │ │ ├─ Switch [id BFZ] 154 1873,1897c1869,1892 < │ │ │ │ │ └─ Composite{switch(i3, (i2 + switch(i1, (-i0), 0.0)), 1)} [id BHE] 403 < │ │ │ │ │ ├─ Sum{axis=0} [id BHF] 396 < │ │ │ │ │ │ └─ Mul [id BHG] 389 < │ │ │ │ │ │ ├─ Blockwise{dot, (i00,i01),(i10,i11)->(o00,o01)} [id BHH] 325 < │ │ │ │ │ │ │ ├─ ExpandDims{axis=2} [id BHI] 309 < │ │ │ │ │ │ │ │ └─ Blockwise{SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=1}, (m,m),(m)->(m)} [id KS] 294 < │ │ │ │ │ │ │ │ └─ ··· < │ │ │ │ │ │ │ └─ ExpandDims{axis=1} [id BHJ] 212 < │ │ │ │ │ │ │ └─ Blockwise{SolveTriangular{trans=0, unit_diagonal=False, lower=True, check_finite=True, b_ndim=1}, (m,m),(m)->(m)} [id FN] 203 < │ │ │ │ │ │ │ └─ ··· < │ │ │ │ │ │ └─ Blockwise{Tri{dtype='float64'}, (),(),()->(o00,o01)} [id BHK] 382 < │ │ │ │ │ │ ├─ Blockwise{Subtensor{i}, (i00),()->()} [id BHL] 374 < │ │ │ │ │ │ │ ├─ Blockwise{Subtensor{start:}, (i00),()->(o00)} [id BHM] 361 < │ │ │ │ │ │ │ │ ├─ Blockwise{Shape, (i00,i01)->(o00)} [id BHN] 349 < │ │ │ │ │ │ │ │ │ └─ Neg [id BHO] 338 < │ │ │ │ │ │ │ │ │ └─ Blockwise{dot, (i00,i01),(i10,i11)->(o00,o01)} [id BHH] 325 < │ │ │ │ │ │ │ │ │ └─ ··· < │ │ │ │ │ │ │ │ └─ [-2] [id MP] < │ │ │ │ │ │ │ └─ [0] [id MQ] < │ │ │ │ │ │ ├─ Blockwise{Subtensor{i}, (i00),()->()} [id BHP] 373 < │ │ │ │ │ │ │ ├─ Blockwise{Subtensor{start:}, (i00),()->(o00)} [id BHM] 361 < │ │ │ │ │ │ │ │ └─ ··· < │ │ │ │ │ │ │ └─ [1] [id MS] < │ │ │ │ │ │ └─ [0] [id CO] < │ │ │ │ │ ├─ ExpandDims{axes=[0, 1]} [id GD] 142 --- > │ │ │ │ │ └─ Composite{switch(i4, (i3 + switch(i2, (-1.0 * i0 * i1), 0.0)), 1)} [id BGK] 378 > │ │ │ │ │ ├─ dot [id BGL] 310 > │ │ │ │ │ │ ├─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id KR] 295 > │ │ │ │ │ │ │ └─ ··· > │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id FJ] 201 > │ │ │ │ │ │ └─ ··· > │ │ │ │ │ ├─ Tri{dtype='float64'} [id BGM] 371 > │ │ │ │ │ │ ├─ Subtensor{i} [id BGN] 363 > │ │ │ │ │ │ │ ├─ Subtensor{start:} [id BGO] 350 > │ │ │ │ │ │ │ │ ├─ Shape [id BGP] 338 > │ │ │ │ │ │ │ │ │ └─ Neg [id BGQ] 326 > │ │ │ │ │ │ │ │ │ └─ dot [id BGL] 310 > │ │ │ │ │ │ │ │ │ └─ ··· > │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id MK] 12 > │ │ │ │ │ │ │ │ └─ ··· > │ │ │ │ │ │ │ └─ ScalarFromTensor [id MM] 13 > │ │ │ │ │ │ │ └─ ··· > │ │ │ │ │ │ ├─ Subtensor{i} [id BGR] 362 > │ │ │ │ │ │ │ ├─ Subtensor{start:} [id BGO] 350 > │ │ │ │ │ │ │ │ └─ ··· > │ │ │ │ │ │ │ └─ ScalarFromTensor [id MP] 11 > │ │ │ │ │ │ │ └─ ··· > │ │ │ │ │ │ └─ 0 [id E] > │ │ │ │ │ ├─ ExpandDims{axes=[0, 1]} [id FM] 137 1899,1900c1894,1895 < │ │ │ │ │ ├─ SetSubtensor{:stop, start:stop} [id BHQ] 363 < │ │ │ │ │ │ ├─ Alloc [id MU] 9 --- > │ │ │ │ │ ├─ SetSubtensor{:stop, start:stop} [id BGS] 352 > │ │ │ │ │ │ ├─ Alloc [id MS] 9 1902,1903c1897,1898 < │ │ │ │ │ │ ├─ AdvancedSetSubtensor [id BHR] 351 < │ │ │ │ │ │ │ ├─ Alloc [id CK] 1 --- > │ │ │ │ │ │ ├─ AdvancedSetSubtensor [id BGT] 340 > │ │ │ │ │ │ │ ├─ Alloc [id CL] 1 1905,1910c1900,1905 < │ │ │ │ │ │ │ ├─ True_div [id BHS] 337 < │ │ │ │ │ │ │ │ ├─ Neg [id BHT] 324 < │ │ │ │ │ │ │ │ │ └─ ExpandDims{axis=0} [id BHU] 308 < │ │ │ │ │ │ │ │ │ └─ Sum{axes=None} [id BHV] 293 < │ │ │ │ │ │ │ │ │ └─ DropDims{axis=1} [id BHW] 282 < │ │ │ │ │ │ │ │ │ └─ SpecifyShape [id KV] 272 --- > │ │ │ │ │ │ │ ├─ True_div [id BGU] 325 > │ │ │ │ │ │ │ │ ├─ Neg [id BGV] 309 > │ │ │ │ │ │ │ │ │ └─ ExpandDims{axis=0} [id BGW] 294 > │ │ │ │ │ │ │ │ │ └─ Sum{axes=None} [id BGX] 283 > │ │ │ │ │ │ │ │ │ └─ DropDims{axis=1} [id BGY] 272 > │ │ │ │ │ │ │ │ │ └─ SpecifyShape [id KV] 262 1912c1907 < │ │ │ │ │ │ │ │ └─ ExtractDiag{offset=0, axis1=0, axis2=1, view=False} [id FS] 79 --- > │ │ │ │ │ │ │ │ └─ ExtractDiag{offset=0, axis1=0, axis2=1, view=False} [id FP] 79 1914,1919c1909,1914 < │ │ │ │ │ │ │ ├─ [0 1] [id CN] < │ │ │ │ │ │ │ └─ [0 1] [id CN] < │ │ │ │ │ │ ├─ 2 [id NB] < │ │ │ │ │ │ ├─ 0 [id NC] < │ │ │ │ │ │ └─ 2 [id NB] < │ │ │ │ │ └─ Invert [id BGU] 141 --- > │ │ │ │ │ │ │ ├─ [0 1] [id CO] > │ │ │ │ │ │ │ └─ [0 1] [id CO] > │ │ │ │ │ │ ├─ 2 [id MZ] > │ │ │ │ │ │ ├─ 0 [id NA] > │ │ │ │ │ │ └─ 2 [id MZ] > │ │ │ │ │ └─ Invert [id BGA] 136 1921,1926c1916,1921 < │ │ │ │ ├─ Transpose{axes=[1, 0]} [id BHX] 581 < │ │ │ │ │ └─ Tri{dtype='float64'} [id BHY] 529 < │ │ │ │ │ ├─ Subtensor{i} [id BHZ] 481 < │ │ │ │ │ │ ├─ Subtensor{start:} [id BIA] 438 < │ │ │ │ │ │ │ ├─ Shape [id BIB] 426 < │ │ │ │ │ │ │ │ └─ dot [id BHD] 410 --- > │ │ │ │ ├─ Transpose{axes=[1, 0]} [id BGZ] 557 > │ │ │ │ │ └─ Tri{dtype='float64'} [id BHA] 505 > │ │ │ │ │ ├─ Subtensor{i} [id BHB] 457 > │ │ │ │ │ │ ├─ Subtensor{start:} [id BHC] 413 > │ │ │ │ │ │ │ ├─ Shape [id BHD] 400 > │ │ │ │ │ │ │ │ └─ dot [id BGJ] 385 1928c1923 < │ │ │ │ │ │ │ └─ ScalarFromTensor [id NI] 12 --- > │ │ │ │ │ │ │ └─ ScalarFromTensor [id MK] 12 1930c1925 < │ │ │ │ │ │ └─ ScalarFromTensor [id NK] 13 --- > │ │ │ │ │ │ └─ ScalarFromTensor [id MM] 13 1932,1933c1927,1928 < │ │ │ │ │ ├─ Subtensor{i} [id BIC] 480 < │ │ │ │ │ │ ├─ Subtensor{start:} [id BIA] 438 --- > │ │ │ │ │ ├─ Subtensor{i} [id BHE] 456 > │ │ │ │ │ │ ├─ Subtensor{start:} [id BHC] 413 1935c1930 < │ │ │ │ │ │ └─ ScalarFromTensor [id NN] 11 --- > │ │ │ │ │ │ └─ ScalarFromTensor [id MP] 11 1938,1940c1933,1935 < │ │ │ │ └─ Transpose{axes=[1, 0]} [id BID] 602 < │ │ │ │ └─ AdvancedSetSubtensor [id BIE] 596 < │ │ │ │ ├─ Alloc [id BIF] 580 --- > │ │ │ │ └─ Transpose{axes=[1, 0]} [id BHF] 579 > │ │ │ │ └─ AdvancedSetSubtensor [id BHG] 572 > │ │ │ │ ├─ Alloc [id BHH] 556 1942,1947c1937,1942 < │ │ │ │ │ ├─ Subtensor{i} [id BIG] 528 < │ │ │ │ │ │ ├─ Shape [id BIH] 479 < │ │ │ │ │ │ │ └─ Mul [id BII] 437 < │ │ │ │ │ │ │ ├─ [0.5] [id NV] < │ │ │ │ │ │ │ └─ ExtractDiag{offset=0, axis1=0, axis2=1, view=False} [id BIJ] 425 < │ │ │ │ │ │ │ └─ dot [id BHD] 410 --- > │ │ │ │ │ ├─ Subtensor{i} [id BHI] 504 > │ │ │ │ │ │ ├─ Shape [id BHJ] 455 > │ │ │ │ │ │ │ └─ Mul [id BHK] 412 > │ │ │ │ │ │ │ ├─ [0.5] [id NN] > │ │ │ │ │ │ │ └─ ExtractDiag{offset=0, axis1=0, axis2=1, view=False} [id BHL] 399 > │ │ │ │ │ │ │ └─ dot [id BGJ] 385 1949c1944 < │ │ │ │ │ │ └─ ScalarFromTensor [id NX] 2 --- > │ │ │ │ │ │ └─ ScalarFromTensor [id NP] 2 1951c1946 < │ │ │ │ │ └─ Subtensor{i} [id BIG] 528 --- > │ │ │ │ │ └─ Subtensor{i} [id BHI] 504 1953c1948 < │ │ │ │ ├─ Mul [id BII] 437 --- > │ │ │ │ ├─ Mul [id BHK] 412 1955c1950 < │ │ │ │ ├─ ARange{dtype='int64'} [id BIK] 579 --- > │ │ │ │ ├─ ARange{dtype='int64'} [id BHM] 555 1957c1952 < │ │ │ │ │ ├─ Subtensor{i} [id BIG] 528 --- > │ │ │ │ │ ├─ Subtensor{i} [id BHI] 504 1960c1955 < │ │ │ │ └─ ARange{dtype='int64'} [id BIK] 579 --- > │ │ │ │ └─ ARange{dtype='int64'} [id BHM] 555 1962,1964c1957,1959 < │ │ │ ├─ Transpose{axes=[1, 0]} [id BIL] 803 < │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id BIM] 769 < │ │ │ │ ├─ Switch [id BGT] 164 --- > │ │ │ ├─ Transpose{axes=[1, 0]} [id BHN] 781 > │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id BHO] 746 > │ │ │ │ ├─ Switch [id BFZ] 154 1966,1968c1961,1963 < │ │ │ │ └─ Transpose{axes=[1, 0]} [id BIN] 728 < │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id BIO] 686 < │ │ │ │ ├─ Switch [id BGT] 164 --- > │ │ │ │ └─ Transpose{axes=[1, 0]} [id BHP] 704 > │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id BHQ] 662 > │ │ │ │ ├─ Switch [id BFZ] 154 1970,1971c1965,1966 < │ │ │ │ └─ Composite{((i0 * i1) - i2)} [id BIP] 644 < │ │ │ │ ├─ Transpose{axes=[1, 0]} [id BHC] 427 --- > │ │ │ │ └─ Composite{((i0 * i1) - i2)} [id BHR] 620 > │ │ │ │ ├─ Transpose{axes=[1, 0]} [id BGI] 401 1973,1975c1968,1970 < │ │ │ │ ├─ Transpose{axes=[1, 0]} [id BIQ] 590 < │ │ │ │ │ └─ Tri{dtype='float64'} [id BIR] 538 < │ │ │ │ │ ├─ Subtensor{i} [id BHZ] 481 --- > │ │ │ │ ├─ Transpose{axes=[1, 0]} [id BHS] 566 > │ │ │ │ │ └─ Tri{dtype='float64'} [id BHT] 514 > │ │ │ │ │ ├─ Subtensor{i} [id BHB] 457 1977,1978c1972,1973 < │ │ │ │ │ ├─ Subtensor{i} [id BIS] 490 < │ │ │ │ │ │ ├─ Subtensor{start:} [id BIA] 438 --- > │ │ │ │ │ ├─ Subtensor{i} [id BHU] 466 > │ │ │ │ │ │ ├─ Subtensor{start:} [id BHC] 413 1980,1981c1975,1976 < │ │ │ │ │ │ └─ ScalarFromTensor [id BIT] 22 < │ │ │ │ │ │ └─ 1 [id BIU] --- > │ │ │ │ │ │ └─ ScalarFromTensor [id BHV] 22 > │ │ │ │ │ │ └─ 1 [id BHW] 1983c1978 < │ │ │ │ └─ Transpose{axes=[1, 0]} [id BID] 602 --- > │ │ │ │ └─ Transpose{axes=[1, 0]} [id BHF] 579 1985,1991c1980,1986 < │ │ │ ├─ Tri{dtype='float64'} [id BIV] 898 < │ │ │ │ ├─ Subtensor{i} [id BIW] 893 < │ │ │ │ │ ├─ Subtensor{start:} [id BIX] 878 < │ │ │ │ │ │ ├─ Shape [id BIY] 855 < │ │ │ │ │ │ │ └─ Add [id BIZ] 829 < │ │ │ │ │ │ │ ├─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id BJA] 768 < │ │ │ │ │ │ │ │ ├─ Switch [id BGT] 164 --- > │ │ │ ├─ Tri{dtype='float64'} [id BHX] 878 > │ │ │ │ ├─ Subtensor{i} [id BHY] 872 > │ │ │ │ │ ├─ Subtensor{start:} [id BHZ] 857 > │ │ │ │ │ │ ├─ Shape [id BIA] 834 > │ │ │ │ │ │ │ └─ Add [id BIB] 808 > │ │ │ │ │ │ │ ├─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id BIC] 745 > │ │ │ │ │ │ │ │ ├─ Switch [id BFZ] 154 1993,1995c1988,1990 < │ │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id BJB] 727 < │ │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id BJC] 685 < │ │ │ │ │ │ │ │ ├─ Switch [id BGT] 164 --- > │ │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id BID] 703 > │ │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id BIE] 661 > │ │ │ │ │ │ │ │ ├─ Switch [id BFZ] 154 1997,1998c1992,1993 < │ │ │ │ │ │ │ │ └─ Composite{((i0 * i1) - i2)} [id BJD] 643 < │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id BHC] 427 --- > │ │ │ │ │ │ │ │ └─ Composite{((i0 * i1) - i2)} [id BIF] 619 > │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id BGI] 401 2000,2002c1995,1997 < │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id BJE] 589 < │ │ │ │ │ │ │ │ │ └─ Tri{dtype='float64'} [id BJF] 537 < │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id BHZ] 481 --- > │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id BIG] 565 > │ │ │ │ │ │ │ │ │ └─ Tri{dtype='float64'} [id BIH] 513 > │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id BHB] 457 2004,2005c1999,2000 < │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id BJG] 489 < │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{start:} [id BIA] 438 --- > │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id BII] 465 > │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{start:} [id BHC] 413 2007,2008c2002,2003 < │ │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id BJH] 21 < │ │ │ │ │ │ │ │ │ │ └─ 1 [id BJI] --- > │ │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id BIJ] 21 > │ │ │ │ │ │ │ │ │ │ └─ 1 [id BIK] 2010c2005 < │ │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id BID] 602 --- > │ │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id BHF] 579 2012,2014c2007,2009 < │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id BJJ] 802 < │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id BJK] 767 < │ │ │ │ │ │ │ ├─ Switch [id BGT] 164 --- > │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id BIL] 780 > │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id BIM] 744 > │ │ │ │ │ │ │ ├─ Switch [id BFZ] 154 2016,2018c2011,2013 < │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id BJL] 726 < │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id BJM] 684 < │ │ │ │ │ │ │ ├─ Switch [id BGT] 164 --- > │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id BIN] 702 > │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id BIO] 660 > │ │ │ │ │ │ │ ├─ Switch [id BFZ] 154 2020,2021c2015,2016 < │ │ │ │ │ │ │ └─ Composite{((i0 * i1) - i2)} [id BJN] 642 < │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id BHC] 427 --- > │ │ │ │ │ │ │ └─ Composite{((i0 * i1) - i2)} [id BIP] 618 > │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id BGI] 401 2023,2025c2018,2020 < │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id BJO] 588 < │ │ │ │ │ │ │ │ └─ Tri{dtype='float64'} [id BJP] 536 < │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id BHZ] 481 --- > │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id BIQ] 564 > │ │ │ │ │ │ │ │ └─ Tri{dtype='float64'} [id BIR] 512 > │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id BHB] 457 2027,2028c2022,2023 < │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id BJQ] 488 < │ │ │ │ │ │ │ │ │ ├─ Subtensor{start:} [id BIA] 438 --- > │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id BIS] 464 > │ │ │ │ │ │ │ │ │ ├─ Subtensor{start:} [id BHC] 413 2030,2031c2025,2026 < │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id BJR] 20 < │ │ │ │ │ │ │ │ │ └─ 1 [id BJS] --- > │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id BIT] 20 > │ │ │ │ │ │ │ │ │ └─ 1 [id BIU] 2033c2028 < │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id BID] 602 --- > │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id BHF] 579 2035c2030 < │ │ │ │ │ │ └─ ScalarFromTensor [id NI] 12 --- > │ │ │ │ │ │ └─ ScalarFromTensor [id MK] 12 2037c2032 < │ │ │ │ │ └─ ScalarFromTensor [id NK] 13 --- > │ │ │ │ │ └─ ScalarFromTensor [id MM] 13 2039,2044c2034,2039 < │ │ │ │ ├─ Subtensor{i} [id BJT] 892 < │ │ │ │ │ ├─ Subtensor{start:} [id BJU] 877 < │ │ │ │ │ │ ├─ Shape [id BJV] 854 < │ │ │ │ │ │ │ └─ Add [id BJW] 828 < │ │ │ │ │ │ │ ├─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id BJX] 766 < │ │ │ │ │ │ │ │ ├─ Switch [id BGT] 164 --- > │ │ │ │ ├─ Subtensor{i} [id BIV] 871 > │ │ │ │ │ ├─ Subtensor{start:} [id BIW] 856 > │ │ │ │ │ │ ├─ Shape [id BIX] 833 > │ │ │ │ │ │ │ └─ Add [id BIY] 807 > │ │ │ │ │ │ │ ├─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id BIZ] 743 > │ │ │ │ │ │ │ │ ├─ Switch [id BFZ] 154 2046,2048c2041,2043 < │ │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id BJY] 725 < │ │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id BJZ] 683 < │ │ │ │ │ │ │ │ ├─ Switch [id BGT] 164 --- > │ │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id BJA] 701 > │ │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id BJB] 659 > │ │ │ │ │ │ │ │ ├─ Switch [id BFZ] 154 2050,2051c2045,2046 < │ │ │ │ │ │ │ │ └─ Composite{((i0 * i1) - i2)} [id BKA] 641 < │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id BHC] 427 --- > │ │ │ │ │ │ │ │ └─ Composite{((i0 * i1) - i2)} [id BJC] 617 > │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id BGI] 401 2053,2055c2048,2050 < │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id BKB] 587 < │ │ │ │ │ │ │ │ │ └─ Tri{dtype='float64'} [id BKC] 535 < │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id BHZ] 481 --- > │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id BJD] 563 > │ │ │ │ │ │ │ │ │ └─ Tri{dtype='float64'} [id BJE] 511 > │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id BHB] 457 2057,2058c2052,2053 < │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id BKD] 487 < │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{start:} [id BIA] 438 --- > │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id BJF] 463 > │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{start:} [id BHC] 413 2060,2061c2055,2056 < │ │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id BKE] 19 < │ │ │ │ │ │ │ │ │ │ └─ 1 [id BKF] --- > │ │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id BJG] 19 > │ │ │ │ │ │ │ │ │ │ └─ 1 [id BJH] 2063c2058 < │ │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id BID] 602 --- > │ │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id BHF] 579 2065,2067c2060,2062 < │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id BKG] 801 < │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id BKH] 765 < │ │ │ │ │ │ │ ├─ Switch [id BGT] 164 --- > │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id BJI] 779 > │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id BJJ] 742 > │ │ │ │ │ │ │ ├─ Switch [id BFZ] 154 2069,2071c2064,2066 < │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id BKI] 724 < │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id BKJ] 682 < │ │ │ │ │ │ │ ├─ Switch [id BGT] 164 --- > │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id BJK] 700 > │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id BJL] 658 > │ │ │ │ │ │ │ ├─ Switch [id BFZ] 154 2073,2074c2068,2069 < │ │ │ │ │ │ │ └─ Composite{((i0 * i1) - i2)} [id BKK] 640 < │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id BHC] 427 --- > │ │ │ │ │ │ │ └─ Composite{((i0 * i1) - i2)} [id BJM] 616 > │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id BGI] 401 2076,2078c2071,2073 < │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id BKL] 586 < │ │ │ │ │ │ │ │ └─ Tri{dtype='float64'} [id BKM] 534 < │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id BHZ] 481 --- > │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id BJN] 562 > │ │ │ │ │ │ │ │ └─ Tri{dtype='float64'} [id BJO] 510 > │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id BHB] 457 2080,2081c2075,2076 < │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id BKN] 486 < │ │ │ │ │ │ │ │ │ ├─ Subtensor{start:} [id BIA] 438 --- > │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id BJP] 462 > │ │ │ │ │ │ │ │ │ ├─ Subtensor{start:} [id BHC] 413 2083,2084c2078,2079 < │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id BKO] 18 < │ │ │ │ │ │ │ │ │ └─ 1 [id BKP] --- > │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id BJQ] 18 > │ │ │ │ │ │ │ │ │ └─ 1 [id BJR] 2086c2081 < │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id BID] 602 --- > │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id BHF] 579 2088c2083 < │ │ │ │ │ │ └─ ScalarFromTensor [id NI] 12 --- > │ │ │ │ │ │ └─ ScalarFromTensor [id MK] 12 2090c2085 < │ │ │ │ │ └─ ScalarFromTensor [id NN] 11 --- > │ │ │ │ │ └─ ScalarFromTensor [id MP] 11 2093,2094c2088,2089 < │ │ │ ├─ AdvancedSetSubtensor [id BKQ] 891 < │ │ │ │ ├─ Alloc [id BKR] 876 --- > │ │ │ ├─ AdvancedSetSubtensor [id BJS] 870 > │ │ │ │ ├─ Alloc [id BJT] 855 2096,2100c2091,2095 < │ │ │ │ │ ├─ Subtensor{i} [id BKS] 853 < │ │ │ │ │ │ ├─ Shape [id BKT] 827 < │ │ │ │ │ │ │ └─ ExtractDiag{offset=0, axis1=0, axis2=1, view=False} [id BKU] 800 < │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id BKV] 764 < │ │ │ │ │ │ │ ├─ Switch [id BGT] 164 --- > │ │ │ │ │ ├─ Subtensor{i} [id BJU] 832 > │ │ │ │ │ │ ├─ Shape [id BJV] 806 > │ │ │ │ │ │ │ └─ ExtractDiag{offset=0, axis1=0, axis2=1, view=False} [id BJW] 778 > │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id BJX] 741 > │ │ │ │ │ │ │ ├─ Switch [id BFZ] 154 2102,2104c2097,2099 < │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id BKW] 723 < │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id BKX] 681 < │ │ │ │ │ │ │ ├─ Switch [id BGT] 164 --- > │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id BJY] 699 > │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id BJZ] 657 > │ │ │ │ │ │ │ ├─ Switch [id BFZ] 154 2106,2107c2101,2102 < │ │ │ │ │ │ │ └─ Composite{((i0 * i1) - i2)} [id BKY] 639 < │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id BHC] 427 --- > │ │ │ │ │ │ │ └─ Composite{((i0 * i1) - i2)} [id BKA] 615 > │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id BGI] 401 2109,2111c2104,2106 < │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id BKZ] 585 < │ │ │ │ │ │ │ │ └─ Tri{dtype='float64'} [id BLA] 533 < │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id BHZ] 481 --- > │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id BKB] 561 > │ │ │ │ │ │ │ │ └─ Tri{dtype='float64'} [id BKC] 509 > │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id BHB] 457 2113,2114c2108,2109 < │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id BLB] 485 < │ │ │ │ │ │ │ │ │ ├─ Subtensor{start:} [id BIA] 438 --- > │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id BKD] 461 > │ │ │ │ │ │ │ │ │ ├─ Subtensor{start:} [id BHC] 413 2116,2117c2111,2112 < │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id BLC] 17 < │ │ │ │ │ │ │ │ │ └─ 1 [id BLD] --- > │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id BKE] 17 > │ │ │ │ │ │ │ │ │ └─ 1 [id BKF] 2119c2114 < │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id BID] 602 --- > │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id BHF] 579 2121c2116 < │ │ │ │ │ │ └─ ScalarFromTensor [id NX] 2 --- > │ │ │ │ │ │ └─ ScalarFromTensor [id NP] 2 2123,2127c2118,2122 < │ │ │ │ │ └─ Subtensor{i} [id BLE] 852 < │ │ │ │ │ ├─ Shape [id BLF] 826 < │ │ │ │ │ │ └─ ExtractDiag{offset=0, axis1=0, axis2=1, view=False} [id BLG] 799 < │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id BLH] 763 < │ │ │ │ │ │ ├─ Switch [id BGT] 164 --- > │ │ │ │ │ └─ Subtensor{i} [id BKG] 831 > │ │ │ │ │ ├─ Shape [id BKH] 805 > │ │ │ │ │ │ └─ ExtractDiag{offset=0, axis1=0, axis2=1, view=False} [id BKI] 777 > │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id BKJ] 740 > │ │ │ │ │ │ ├─ Switch [id BFZ] 154 2129,2131c2124,2126 < │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id BLI] 722 < │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id BLJ] 680 < │ │ │ │ │ │ ├─ Switch [id BGT] 164 --- > │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id BKK] 698 > │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id BKL] 656 > │ │ │ │ │ │ ├─ Switch [id BFZ] 154 2133,2134c2128,2129 < │ │ │ │ │ │ └─ Composite{((i0 * i1) - i2)} [id BLK] 638 < │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id BHC] 427 --- > │ │ │ │ │ │ └─ Composite{((i0 * i1) - i2)} [id BKM] 614 > │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id BGI] 401 2136,2138c2131,2133 < │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id BLL] 584 < │ │ │ │ │ │ │ └─ Tri{dtype='float64'} [id BLM] 532 < │ │ │ │ │ │ │ ├─ Subtensor{i} [id BHZ] 481 --- > │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id BKN] 560 > │ │ │ │ │ │ │ └─ Tri{dtype='float64'} [id BKO] 508 > │ │ │ │ │ │ │ ├─ Subtensor{i} [id BHB] 457 2140,2141c2135,2136 < │ │ │ │ │ │ │ ├─ Subtensor{i} [id BLN] 484 < │ │ │ │ │ │ │ │ ├─ Subtensor{start:} [id BIA] 438 --- > │ │ │ │ │ │ │ ├─ Subtensor{i} [id BKP] 460 > │ │ │ │ │ │ │ │ ├─ Subtensor{start:} [id BHC] 413 2143,2144c2138,2139 < │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id BLO] 16 < │ │ │ │ │ │ │ │ └─ 1 [id BLP] --- > │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id BKQ] 16 > │ │ │ │ │ │ │ │ └─ 1 [id BKR] 2146c2141 < │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id BID] 602 --- > │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id BHF] 579 2148c2143 < │ │ │ │ │ └─ ScalarFromTensor [id NX] 2 --- > │ │ │ │ │ └─ ScalarFromTensor [id NP] 2 2150,2152c2145,2147 < │ │ │ │ ├─ ExtractDiag{offset=0, axis1=0, axis2=1, view=False} [id BLQ] 798 < │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id BLR] 762 < │ │ │ │ │ ├─ Switch [id BGT] 164 --- > │ │ │ │ ├─ ExtractDiag{offset=0, axis1=0, axis2=1, view=False} [id BKS] 776 > │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id BKT] 739 > │ │ │ │ │ ├─ Switch [id BFZ] 154 2154,2156c2149,2151 < │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id BLS] 721 < │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id BLT] 679 < │ │ │ │ │ ├─ Switch [id BGT] 164 --- > │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id BKU] 697 > │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id BKV] 655 > │ │ │ │ │ ├─ Switch [id BFZ] 154 2158,2159c2153,2154 < │ │ │ │ │ └─ Composite{((i0 * i1) - i2)} [id BLU] 637 < │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id BHC] 427 --- > │ │ │ │ │ └─ Composite{((i0 * i1) - i2)} [id BKW] 613 > │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id BGI] 401 2161,2163c2156,2158 < │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id BLV] 583 < │ │ │ │ │ │ └─ Tri{dtype='float64'} [id BLW] 531 < │ │ │ │ │ │ ├─ Subtensor{i} [id BHZ] 481 --- > │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id BKX] 559 > │ │ │ │ │ │ └─ Tri{dtype='float64'} [id BKY] 507 > │ │ │ │ │ │ ├─ Subtensor{i} [id BHB] 457 2165,2166c2160,2161 < │ │ │ │ │ │ ├─ Subtensor{i} [id BLX] 483 < │ │ │ │ │ │ │ ├─ Subtensor{start:} [id BIA] 438 --- > │ │ │ │ │ │ ├─ Subtensor{i} [id BKZ] 459 > │ │ │ │ │ │ │ ├─ Subtensor{start:} [id BHC] 413 2168,2169c2163,2164 < │ │ │ │ │ │ │ └─ ScalarFromTensor [id BLY] 15 < │ │ │ │ │ │ │ └─ 1 [id BLZ] --- > │ │ │ │ │ │ │ └─ ScalarFromTensor [id BLA] 15 > │ │ │ │ │ │ │ └─ 1 [id BLB] 2171c2166 < │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id BID] 602 --- > │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id BHF] 579 2173c2168 < │ │ │ │ ├─ ARange{dtype='int64'} [id BMA] 875 --- > │ │ │ │ ├─ ARange{dtype='int64'} [id BLC] 854 2175,2179c2170,2174 < │ │ │ │ │ ├─ Subtensor{i} [id BMB] 851 < │ │ │ │ │ │ ├─ Shape [id BMC] 825 < │ │ │ │ │ │ │ └─ ExtractDiag{offset=0, axis1=0, axis2=1, view=False} [id BMD] 797 < │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id BME] 761 < │ │ │ │ │ │ │ ├─ Switch [id BGT] 164 --- > │ │ │ │ │ ├─ Subtensor{i} [id BLD] 830 > │ │ │ │ │ │ ├─ Shape [id BLE] 804 > │ │ │ │ │ │ │ └─ ExtractDiag{offset=0, axis1=0, axis2=1, view=False} [id BLF] 775 > │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id BLG] 738 > │ │ │ │ │ │ │ ├─ Switch [id BFZ] 154 2181,2183c2176,2178 < │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id BMF] 720 < │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id BMG] 678 < │ │ │ │ │ │ │ ├─ Switch [id BGT] 164 --- > │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id BLH] 696 > │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id BLI] 654 > │ │ │ │ │ │ │ ├─ Switch [id BFZ] 154 2185,2186c2180,2181 < │ │ │ │ │ │ │ └─ Composite{((i0 * i1) - i2)} [id BMH] 636 < │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id BHC] 427 --- > │ │ │ │ │ │ │ └─ Composite{((i0 * i1) - i2)} [id BLJ] 612 > │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id BGI] 401 2188,2190c2183,2185 < │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id BMI] 582 < │ │ │ │ │ │ │ │ └─ Tri{dtype='float64'} [id BMJ] 530 < │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id BHZ] 481 --- > │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id BLK] 558 > │ │ │ │ │ │ │ │ └─ Tri{dtype='float64'} [id BLL] 506 > │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id BHB] 457 2192,2193c2187,2188 < │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id BMK] 482 < │ │ │ │ │ │ │ │ │ ├─ Subtensor{start:} [id BIA] 438 --- > │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id BLM] 458 > │ │ │ │ │ │ │ │ │ ├─ Subtensor{start:} [id BHC] 413 2195,2196c2190,2191 < │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id BML] 14 < │ │ │ │ │ │ │ │ │ └─ 1 [id BMM] --- > │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id BLN] 14 > │ │ │ │ │ │ │ │ │ └─ 1 [id BLO] 2198c2193 < │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id BID] 602 --- > │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id BHF] 579 2200c2195 < │ │ │ │ │ │ └─ ScalarFromTensor [id NX] 2 --- > │ │ │ │ │ │ └─ ScalarFromTensor [id NP] 2 2203c2198 < │ │ │ │ └─ ARange{dtype='int64'} [id BMN] 874 --- > │ │ │ │ └─ ARange{dtype='int64'} [id BLP] 853 2205,2208c2200,2203 < │ │ │ │ ├─ Subtensor{i} [id BMO] 850 < │ │ │ │ │ ├─ Shape [id BMP] 824 < │ │ │ │ │ │ └─ ExtractDiag{offset=0, axis1=0, axis2=1, view=False} [id BMQ] 796 < │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id BGS] 760 --- > │ │ │ │ ├─ Subtensor{i} [id BLQ] 829 > │ │ │ │ │ ├─ Shape [id BLR] 803 > │ │ │ │ │ │ └─ ExtractDiag{offset=0, axis1=0, axis2=1, view=False} [id BLS] 774 > │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id BFY] 737 2210c2205 < │ │ │ │ │ └─ ScalarFromTensor [id NX] 2 --- > │ │ │ │ │ └─ ScalarFromTensor [id NP] 2 2213c2208 < │ │ │ ├─ Invert [id BGU] 141 --- > │ │ │ ├─ Invert [id BGA] 136 2215,2218c2210,2213 < │ │ │ └─ [[nan]] [id OM] < │ │ ├─ [0 1] [id CN] < │ │ └─ [0 1] [id CN] < │ └─ 4 [id FW] --- > │ │ │ └─ [[nan]] [id OE] > │ │ ├─ [0 1] [id CO] > │ │ └─ [0 1] [id CO] > │ └─ 4 [id FT] 2225,2250c2220,2245 < ← sub [id BMR] 'o0' < ├─ add [id BMS] < │ ├─ i3 [id BMT] < │ ├─ Switch [id BMU] < │ │ ├─ OR [id BMV] 'o1' < │ │ │ ├─ i13 [id BMW] < │ │ │ └─ i14 [id BMX] < │ │ ├─ i10 [id BMY] < │ │ └─ add [id BMZ] < │ │ ├─ i11 [id BNA] < │ │ └─ i12 [id BNB] < │ ├─ mul [id BNC] < │ │ ├─ i2 [id BND] < │ │ └─ i9 [id BNE] < │ ├─ i4 [id BNF] < │ ├─ i5 [id BNG] < │ ├─ i6 [id BNH] < │ ├─ i7 [id BNI] < │ └─ i8 [id BNJ] < └─ mul [id BNK] < ├─ i2 [id BND] < └─ add [id BNL] < ├─ i1 [id BNM] < └─ log [id BNN] < └─ i0 [id BNO] < ← OR [id BMV] 'o1' --- > ← sub [id BLT] 'o0' > ├─ add [id BLU] > │ ├─ i3 [id BLV] > │ ├─ Switch [id BLW] > │ │ ├─ OR [id BLX] 'o1' > │ │ │ ├─ i13 [id BLY] > │ │ │ └─ i14 [id BLZ] > │ │ ├─ i10 [id BMA] > │ │ └─ add [id BMB] > │ │ ├─ i11 [id BMC] > │ │ └─ i12 [id BMD] > │ ├─ mul [id BME] > │ │ ├─ i2 [id BMF] > │ │ └─ i9 [id BMG] > │ ├─ i4 [id BMH] > │ ├─ i5 [id BMI] > │ ├─ i6 [id BMJ] > │ ├─ i7 [id BMK] > │ └─ i8 [id BML] > └─ mul [id BMM] > ├─ i2 [id BMF] > └─ add [id BMN] > ├─ i1 [id BMO] > └─ log [id BMP] > └─ i0 [id BMQ] > ← OR [id BLX] 'o1' 2254,2263c2249,2258 < ← Switch [id BNP] 'o0' < ├─ i2 [id BNQ] < ├─ i3 [id BNR] < └─ exp [id BNS] 't1' < └─ sub [id BNT] < ├─ i0 [id BNU] < └─ i1 [id BNV] < ← mul [id BNW] 'o1' < ├─ -5.0 [id BNX] < └─ exp [id BNS] 't1' --- > ← Switch [id BMR] 'o0' > ├─ i2 [id BMS] > ├─ i3 [id BMT] > └─ exp [id BMU] 't9' > └─ sub [id BMV] > ├─ i0 [id BMW] > └─ i1 [id BMX] > ← mul [id BMY] 'o1' > ├─ -5.0 [id BMZ] > └─ exp [id BMU] 't9' 2265,2267c2260,2262 < ← mul [id BNY] 'o2' < ├─ 5.0 [id BNZ] < └─ exp [id BNS] 't1' --- > ← mul [id BNA] 'o2' > ├─ 5.0 [id BNB] > └─ exp [id BMU] 't9' 2271,2276c2266,2271 < ← sub [id BOA] 'o0' < ├─ mul [id BOB] < │ ├─ -0.5 [id BOC] < │ └─ sqr [id BOD] < │ └─ i0 [id BOE] < └─ 0.9189385332046727 [id BOF] --- > ← sub [id BNC] 'o0' > ├─ mul [id BND] > │ ├─ -0.5 [id BNE] > │ └─ sqr [id BNF] > │ └─ i0 [id BNG] > └─ 0.9189385332046727 [id BNH] 2279,2284c2274,2279 < ← sub [id BOG] 'o0' < ├─ mul [id BOH] < │ ├─ -0.5 [id BOI] < │ └─ sqr [id BOJ] < │ └─ i0 [id BOK] < └─ 0.9189385332046727 [id BOL] --- > ← sub [id BNI] 'o0' > ├─ mul [id BNJ] > │ ├─ -0.5 [id BNK] > │ └─ sqr [id BNL] > │ └─ i0 [id BNM] > └─ 0.9189385332046727 [id BNN] 2287,2291c2282,2286 < ← exp [id BOM] 'o0' < └─ i0 [id BON] < ← Switch [id BOO] 'o1' < ├─ GE [id BOP] 't7' < │ ├─ exp [id BOM] 'o0' --- > ← exp [id BNO] 'o0' > └─ i0 [id BNP] > ← Switch [id BNQ] 'o1' > ├─ GE [id BNR] 't11' > │ ├─ exp [id BNO] 'o0' 2293,2295c2288,2290 < │ └─ 0.0 [id BOQ] < ├─ neg [id BOR] < │ └─ exp [id BOM] 'o0' --- > │ └─ 0.0 [id BNS] > ├─ neg [id BNT] > │ └─ exp [id BNO] 'o0' 2297,2300c2292,2295 < └─ 0 [id BOS] < ← add [id BOT] 'o2' < ├─ Switch [id BOU] < │ ├─ GE [id BOP] 't7' --- > └─ 0 [id BNU] > ← add [id BNV] 'o2' > ├─ Switch [id BNW] > │ ├─ GE [id BNR] 't11' 2302,2306c2297,2301 < │ ├─ sub [id BOV] < │ │ ├─ mul [id BOW] < │ │ │ ├─ -0.5 [id BOX] < │ │ │ └─ sqr [id BOY] < │ │ │ └─ exp [id BOM] 'o0' --- > │ ├─ sub [id BNX] > │ │ ├─ mul [id BNY] > │ │ │ ├─ -0.5 [id BNZ] > │ │ │ └─ sqr [id BOA] > │ │ │ └─ exp [id BNO] 'o0' 2308,2310c2303,2305 < │ │ └─ 0.22579135264472738 [id BOZ] < │ └─ -inf [id BPA] < └─ i0 [id BON] --- > │ │ └─ 0.22579135264472738 [id BOB] > │ └─ -inf [id BOC] > └─ i0 [id BNP] 2313,2319c2308,2314 < ← Switch [id BPB] 'o0' < ├─ i2 [id BPC] < ├─ add [id BPD] < │ ├─ i1 [id BPE] < │ └─ log [id BPF] < │ └─ i0 [id BPG] < └─ -inf [id BPH] --- > ← Switch [id BOD] 'o0' > ├─ i2 [id BOE] > ├─ add [id BOF] > │ ├─ i1 [id BOG] > │ └─ log [id BOH] > │ └─ i0 [id BOI] > └─ -inf [id BOJ] 2322,2329c2317,2324 < ← Switch [id BPI] 'o0' < ├─ i2 [id BPJ] < ├─ i3 [id BPK] < └─ exp [id BPL] 'o1' < └─ sub [id BPM] < ├─ i0 [id BPN] < └─ i1 [id BPO] < ← exp [id BPL] 'o1' --- > ← Switch [id BOK] 'o0' > ├─ i2 [id BOL] > ├─ i3 [id BOM] > └─ exp [id BON] 'o1' > └─ sub [id BOO] > ├─ i0 [id BOP] > └─ i1 [id BOQ] > ← exp [id BON] 'o1' 2331,2332c2326,2327 < ← neg [id BPP] 'o2' < └─ exp [id BPL] 'o1' --- > ← neg [id BOR] 'o2' > └─ exp [id BON] 'o1' 2336,2345c2331,2340 < ← Switch [id BPQ] 'o0' < ├─ i2 [id BPR] < ├─ sub [id BPS] < │ ├─ sub [id BPT] < │ │ ├─ -1.8378770664093453 [id BPU] < │ │ └─ mul [id BPV] < │ │ ├─ 0.5 [id BPW] < │ │ └─ i0 [id BPX] < │ └─ i1 [id BPY] < └─ -inf [id BPZ] --- > ← Switch [id BOS] 'o0' > ├─ i2 [id BOT] > ├─ sub [id BOU] > │ ├─ sub [id BOV] > │ │ ├─ -1.8378770664093453 [id BOW] > │ │ └─ mul [id BOX] > │ │ ├─ 0.5 [id BOY] > │ │ └─ i0 [id BOZ] > │ └─ i1 [id BPA] > └─ -inf [id BPB] 2348,2351c2343,2346 < ← add [id BQA] 'o0' < ├─ i0 [id BQB] < └─ sqr [id BQC] < └─ i1 [id BQD] --- > ← add [id BPC] 'o0' > ├─ i0 [id BPD] > └─ sqr [id BPE] > └─ i1 [id BPF] 2354,2466c2349,2461 < ← add [id BQE] 'o0' < ├─ i0 [id BQF] < └─ log [id BQG] < └─ i1 [id BQH] < < Composite{switch(i2, ((-1.8378770664093453 - (0.5 * i0)) - i1), -inf)} [id DC] < ← Switch [id BQI] 'o0' < ├─ i2 [id BQJ] < ├─ sub [id BQK] < │ ├─ sub [id BQL] < │ │ ├─ -1.8378770664093453 [id BQM] < │ │ └─ mul [id BQN] < │ │ ├─ 0.5 [id BQO] < │ │ └─ i0 [id BQP] < │ └─ i1 [id BQQ] < └─ -inf [id BQR] < < CAReduce{Composite{(i0 + sqr(i1))}, axis=1} [id DE] < ← add [id BQS] 'o0' < ├─ i0 [id BQT] < └─ sqr [id BQU] < └─ i1 [id BQV] < < CAReduce{Composite{(i0 + log(i1))}, axes=None} [id DU] < ← add [id BQW] 'o0' < ├─ i0 [id BQX] < └─ log [id BQY] < └─ i1 [id BQZ] < < Composite{switch(i2, ((-1.8378770664093453 - (0.5 * i0)) - i1), -inf)} [id DW] < ← Switch [id BRA] 'o0' < ├─ i2 [id BRB] < ├─ sub [id BRC] < │ ├─ sub [id BRD] < │ │ ├─ -1.8378770664093453 [id BRE] < │ │ └─ mul [id BRF] < │ │ ├─ 0.5 [id BRG] < │ │ └─ i0 [id BRH] < │ └─ i1 [id BRI] < └─ -inf [id BRJ] < < CAReduce{Composite{(i0 + sqr(i1))}, axis=1} [id DY] < ← add [id BRK] 'o0' < ├─ i0 [id BRL] < └─ sqr [id BRM] < └─ i1 [id BRN] < < CAReduce{Composite{(i0 + log(i1))}, axes=None} [id EO] < ← add [id BRO] 'o0' < ├─ i0 [id BRP] < └─ log [id BRQ] < └─ i1 [id BRR] < < Composite{switch(i2, ((-1.8378770664093453 - (0.5 * i0)) - i1), -inf)} [id EQ] < ← Switch [id BRS] 'o0' < ├─ i2 [id BRT] < ├─ sub [id BRU] < │ ├─ sub [id BRV] < │ │ ├─ -1.8378770664093453 [id BRW] < │ │ └─ mul [id BRX] < │ │ ├─ 0.5 [id BRY] < │ │ └─ i0 [id BRZ] < │ └─ i1 [id BSA] < └─ -inf [id BSB] < < CAReduce{Composite{(i0 + sqr(i1))}, axis=1} [id ES] < ← add [id BSC] 'o0' < ├─ i0 [id BSD] < └─ sqr [id BSE] < └─ i1 [id BSF] < < CAReduce{Composite{(i0 + log(i1))}, axes=None} [id FI] < ← add [id BSG] 'o0' < ├─ i0 [id BSH] < └─ log [id BSI] < └─ i1 [id BSJ] < < Composite{switch(i2, ((-1.8378770664093453 - (0.5 * i0)) - i1), -inf)} [id FK] < ← Switch [id BSK] 'o0' < ├─ i2 [id BSL] < ├─ sub [id BSM] < │ ├─ sub [id BSN] < │ │ ├─ -1.8378770664093453 [id BSO] < │ │ └─ mul [id BSP] < │ │ ├─ 0.5 [id BSQ] < │ │ └─ i0 [id BSR] < │ └─ i1 [id BSS] < └─ -inf [id BST] < < CAReduce{Composite{(i0 + sqr(i1))}, axis=1} [id FM] < ← add [id BSU] 'o0' < ├─ i0 [id BSV] < └─ sqr [id BSW] < └─ i1 [id BSX] < < CAReduce{Composite{(i0 + log(i1))}, axes=None} [id GC] < ← add [id BSY] 'o0' < ├─ i0 [id BSZ] < └─ log [id BTA] < └─ i1 [id BTB] < < Composite{...} [id GV] < ← sub [id BTC] 'o0' < ├─ Switch [id BTD] < │ ├─ EQ [id BTE] 'o1' < │ │ ├─ i1 [id BTF] < │ │ └─ 0 [id BTG] < │ ├─ -inf [id BTH] < │ └─ mul [id BTI] < │ ├─ 9.0 [id BTJ] < │ └─ i0 [id BTK] < └─ 12.80182748008147 [id BTL] < ← EQ [id BTE] 'o1' --- > ← add [id BPG] 'o0' > ├─ i0 [id BPH] > └─ log [id BPI] > └─ i1 [id BPJ] > > Composite{switch(i2, ((-1.8378770664093453 - (0.5 * i0)) - i1), -inf)} [id DB] > ← Switch [id BPK] 'o0' > ├─ i2 [id BPL] > ├─ sub [id BPM] > │ ├─ sub [id BPN] > │ │ ├─ -1.8378770664093453 [id BPO] > │ │ └─ mul [id BPP] > │ │ ├─ 0.5 [id BPQ] > │ │ └─ i0 [id BPR] > │ └─ i1 [id BPS] > └─ -inf [id BPT] > > CAReduce{Composite{(i0 + sqr(i1))}, axis=1} [id DD] > ← add [id BPU] 'o0' > ├─ i0 [id BPV] > └─ sqr [id BPW] > └─ i1 [id BPX] > > CAReduce{Composite{(i0 + log(i1))}, axes=None} [id DT] > ← add [id BPY] 'o0' > ├─ i0 [id BPZ] > └─ log [id BQA] > └─ i1 [id BQB] > > Composite{switch(i2, ((-1.8378770664093453 - (0.5 * i0)) - i1), -inf)} [id DU] > ← Switch [id BQC] 'o0' > ├─ i2 [id BQD] > ├─ sub [id BQE] > │ ├─ sub [id BQF] > │ │ ├─ -1.8378770664093453 [id BQG] > │ │ └─ mul [id BQH] > │ │ ├─ 0.5 [id BQI] > │ │ └─ i0 [id BQJ] > │ └─ i1 [id BQK] > └─ -inf [id BQL] > > CAReduce{Composite{(i0 + sqr(i1))}, axis=1} [id DW] > ← add [id BQM] 'o0' > ├─ i0 [id BQN] > └─ sqr [id BQO] > └─ i1 [id BQP] > > CAReduce{Composite{(i0 + log(i1))}, axes=None} [id EM] > ← add [id BQQ] 'o0' > ├─ i0 [id BQR] > └─ log [id BQS] > └─ i1 [id BQT] > > Composite{switch(i2, ((-1.8378770664093453 - (0.5 * i0)) - i1), -inf)} [id EN] > ← Switch [id BQU] 'o0' > ├─ i2 [id BQV] > ├─ sub [id BQW] > │ ├─ sub [id BQX] > │ │ ├─ -1.8378770664093453 [id BQY] > │ │ └─ mul [id BQZ] > │ │ ├─ 0.5 [id BRA] > │ │ └─ i0 [id BRB] > │ └─ i1 [id BRC] > └─ -inf [id BRD] > > CAReduce{Composite{(i0 + sqr(i1))}, axis=1} [id EP] > ← add [id BRE] 'o0' > ├─ i0 [id BRF] > └─ sqr [id BRG] > └─ i1 [id BRH] > > CAReduce{Composite{(i0 + log(i1))}, axes=None} [id FF] > ← add [id BRI] 'o0' > ├─ i0 [id BRJ] > └─ log [id BRK] > └─ i1 [id BRL] > > Composite{switch(i2, ((-1.8378770664093453 - (0.5 * i0)) - i1), -inf)} [id FG] > ← Switch [id BRM] 'o0' > ├─ i2 [id BRN] > ├─ sub [id BRO] > │ ├─ sub [id BRP] > │ │ ├─ -1.8378770664093453 [id BRQ] > │ │ └─ mul [id BRR] > │ │ ├─ 0.5 [id BRS] > │ │ └─ i0 [id BRT] > │ └─ i1 [id BRU] > └─ -inf [id BRV] > > CAReduce{Composite{(i0 + sqr(i1))}, axis=1} [id FI] > ← add [id BRW] 'o0' > ├─ i0 [id BRX] > └─ sqr [id BRY] > └─ i1 [id BRZ] > > CAReduce{Composite{(i0 + log(i1))}, axes=None} [id FY] > ← add [id BSA] 'o0' > ├─ i0 [id BSB] > └─ log [id BSC] > └─ i1 [id BSD] > > Composite{...} [id GQ] > ← sub [id BSE] 'o0' > ├─ Switch [id BSF] > │ ├─ EQ [id BSG] 'o1' > │ │ ├─ i1 [id BSH] > │ │ └─ 0 [id BSI] > │ ├─ -inf [id BSJ] > │ └─ mul [id BSK] > │ ├─ 9.0 [id BSL] > │ └─ i0 [id BSM] > └─ 12.80182748008147 [id BSN] > ← EQ [id BSG] 'o1' 2469,2523c2464,2548 < Composite{(i3 + (-1.0 * eq(i0, i1) * i2))} [id HE] < ← add [id BTM] 'o0' < ├─ i3 [id BTN] < └─ mul [id BTO] < ├─ -1.0 [id BTP] < ├─ EQ [id BTQ] < │ ├─ i0 [id BTR] < │ └─ i1 [id BTS] < └─ i2 [id BTT] < < Composite{(i3 + (i2 - (i0 * i1)))} [id HH] < ← add [id BTU] 'o0' < ├─ i3 [id BTV] < └─ sub [id BTW] < ├─ i2 [id BTX] < └─ mul [id BTY] < ├─ i0 [id BTZ] < └─ i1 [id BUA] < < Composite{(switch(i0, 0, i1) + i2)} [id HK] < ← add [id BUB] 'o0' < ├─ Switch [id BUC] < │ ├─ i0 [id BUD] < │ ├─ 0 [id BUE] < │ └─ i1 [id BUF] < └─ i2 [id BUG] < < Composite{((eq(i4, i5) * i3) + switch(i2, 0, (i0 / i1)))} [id HP] < ← add [id BUH] 'o0' < ├─ mul [id BUI] < │ ├─ EQ [id BUJ] < │ │ ├─ i4 [id BUK] < │ │ └─ i5 [id BUL] < │ └─ i3 [id BUM] < └─ Switch [id BUN] < ├─ i2 [id BUO] < ├─ 0 [id BUP] < └─ true_div [id BUQ] < ├─ i0 [id BUR] < └─ i1 [id BUS] < < Composite{(1.0 + (i0 * i1) + i2)} [id HR] < ← add [id BUT] 'o0' < ├─ 1.0 [id BUU] < ├─ mul [id BUV] < │ ├─ i0 [id BUW] < │ └─ i1 [id BUX] < └─ i2 [id BUY] < < Composite{switch(i1, reciprocal(i0), 0.0)} [id HV] < ← Switch [id BUZ] 'o0' < ├─ i1 [id BVA] < ├─ reciprocal [id BVB] < │ └─ i0 [id BVC] < └─ 0.0 [id BVD] --- > Composite{(i3 + (-1.0 * eq(i0, i1) * i2))} [id GZ] > ← add [id BSO] 'o0' > ├─ i3 [id BSP] > └─ mul [id BSQ] > ├─ -1.0 [id BSR] > ├─ EQ [id BSS] > │ ├─ i0 [id BST] > │ └─ i1 [id BSU] > └─ i2 [id BSV] > > Composite{(i3 + (i2 - (i0 * i1)))} [id HC] > ← add [id BSW] 'o0' > ├─ i3 [id BSX] > └─ sub [id BSY] > ├─ i2 [id BSZ] > └─ mul [id BTA] > ├─ i0 [id BTB] > └─ i1 [id BTC] > > Composite{(switch(i0, 0, i1) + i2)} [id HF] > ← add [id BTD] 'o0' > ├─ Switch [id BTE] > │ ├─ i0 [id BTF] > │ ├─ 0 [id BTG] > │ └─ i1 [id BTH] > └─ i2 [id BTI] > > Composite{((eq(i4, i5) * i3) + switch(i2, 0, (i0 / i1)))} [id HK] > ← add [id BTJ] 'o0' > ├─ mul [id BTK] > │ ├─ EQ [id BTL] > │ │ ├─ i4 [id BTM] > │ │ └─ i5 [id BTN] > │ └─ i3 [id BTO] > └─ Switch [id BTP] > ├─ i2 [id BTQ] > ├─ 0 [id BTR] > └─ true_div [id BTS] > ├─ i0 [id BTT] > └─ i1 [id BTU] > > Composite{(1.0 + (i0 * i1) + i2)} [id HM] > ← add [id BTV] 'o0' > ├─ 1.0 [id BTW] > ├─ mul [id BTX] > │ ├─ i0 [id BTY] > │ └─ i1 [id BTZ] > └─ i2 [id BUA] > > Composite{switch(i1, reciprocal(i0), 0.0)} [id HQ] > ← Switch [id BUB] 'o0' > ├─ i1 [id BUC] > ├─ reciprocal [id BUD] > │ └─ i0 [id BUE] > └─ 0.0 [id BUF] > > Composite{switch(i2, 0, (i0 / i1))} [id HV] > ← Switch [id BUG] 'o0' > ├─ i2 [id BUH] > ├─ 0 [id BUI] > └─ true_div [id BUJ] > ├─ i0 [id BUK] > └─ i1 [id BUL] > > Composite{((eq(i4, i5) * i3) + switch(i2, 0, (i0 / i1)))} [id IF] > ← add [id BUM] 'o0' > ├─ mul [id BUN] > │ ├─ EQ [id BUO] > │ │ ├─ i4 [id BUP] > │ │ └─ i5 [id BUQ] > │ └─ i3 [id BUR] > └─ Switch [id BUS] > ├─ i2 [id BUT] > ├─ 0 [id BUU] > └─ true_div [id BUV] > ├─ i0 [id BUW] > └─ i1 [id BUX] > > Composite{(-5.0 + (i0 * i1) + i2)} [id IH] > ← add [id BUY] 'o0' > ├─ -5.0 [id BUZ] > ├─ mul [id BVA] > │ ├─ i0 [id BVB] > │ └─ i1 [id BVC] > └─ i2 [id BVD] 2525c2550 < Composite{switch(i2, 0, (i0 / i1))} [id IA] --- > Composite{switch(i1, (-5.0 / i0), 0.0)} [id IL] 2527,2569c2552,2564 < ├─ i2 [id BVF] < ├─ 0 [id BVG] < └─ true_div [id BVH] < ├─ i0 [id BVI] < └─ i1 [id BVJ] < < Composite{((eq(i4, i5) * i3) + switch(i2, 0, (i0 / i1)))} [id IK] < ← add [id BVK] 'o0' < ├─ mul [id BVL] < │ ├─ EQ [id BVM] < │ │ ├─ i4 [id BVN] < │ │ └─ i5 [id BVO] < │ └─ i3 [id BVP] < └─ Switch [id BVQ] < ├─ i2 [id BVR] < ├─ 0 [id BVS] < └─ true_div [id BVT] < ├─ i0 [id BVU] < └─ i1 [id BVV] < < Composite{(-5.0 + (i0 * i1) + i2)} [id IM] < ← add [id BVW] 'o0' < ├─ -5.0 [id BVX] < ├─ mul [id BVY] < │ ├─ i0 [id BVZ] < │ └─ i1 [id BWA] < └─ i2 [id BWB] < < Composite{switch(i1, (-5.0 / i0), 0.0)} [id IQ] < ← Switch [id BWC] 'o0' < ├─ i1 [id BWD] < ├─ true_div [id BWE] < │ ├─ -5.0 [id BWF] < │ └─ i0 [id BWG] < └─ 0.0 [id BWH] < < Composite{switch(i2, 0, (i0 / i1))} [id IT] < ← Switch [id BWI] 'o0' < ├─ i2 [id BWJ] < ├─ 0 [id BWK] < └─ true_div [id BWL] < ├─ i0 [id BWM] < └─ i1 [id BWN] --- > ├─ i1 [id BVF] > ├─ true_div [id BVG] > │ ├─ -5.0 [id BVH] > │ └─ i0 [id BVI] > └─ 0.0 [id BVJ] > > Composite{switch(i2, 0, (i0 / i1))} [id IO] > ← Switch [id BVK] 'o0' > ├─ i2 [id BVL] > ├─ 0 [id BVM] > └─ true_div [id BVN] > ├─ i0 [id BVO] > └─ i1 [id BVP] 2572,2576c2567,2571 < ← add [id BWO] 'o0' < ├─ 1.0 [id BWP] < └─ mul [id BWQ] < ├─ i0 [id BWR] < └─ i1 [id BWS] --- > ← add [id BVQ] 'o0' > ├─ 1.0 [id BVR] > └─ mul [id BVS] > ├─ i0 [id BVT] > └─ i1 [id BVU] 2579,2583c2574,2578 < ← add [id BWT] 'o0' < ├─ 1.0 [id BWU] < └─ mul [id BWV] < ├─ i0 [id BWW] < └─ i1 [id BWX] --- > ← add [id BVV] 'o0' > ├─ 1.0 [id BVW] > └─ mul [id BVX] > ├─ i0 [id BVY] > └─ i1 [id BVZ] 2586,2624c2581,2649 < ← Switch [id BWY] 'o0' < ├─ i3 [id BWZ] < ├─ sub [id BXA] < │ ├─ mul [id BXB] < │ │ ├─ i0 [id BXC] < │ │ └─ i1 [id BXD] < │ └─ i2 [id BXE] < └─ i4 [id BXF] < < Composite{((i0 * i1) - i2)} [id MB] < ← sub [id BXG] 'o0' < ├─ mul [id BXH] < │ ├─ i0 [id BXI] < │ └─ i1 [id BXJ] < └─ i2 [id BXK] < < Composite{switch(i3, (i2 + switch(i1, (-i0), 0.0)), 1)} [id ME] < ← Switch [id BXL] 'o0' < ├─ i3 [id BXM] < ├─ add [id BXN] < │ ├─ i2 [id BXO] < │ └─ Switch [id BXP] < │ ├─ i1 [id BXQ] < │ ├─ neg [id BXR] < │ │ └─ i0 [id BXS] < │ └─ 0.0 [id BXT] < └─ 1 [id BXU] < < Composite{switch(i4, (((i0 + i1) * i2) - i3), i5)} [id OO] < ← Switch [id BXV] 'o0' < ├─ i4 [id BXW] < ├─ sub [id BXX] < │ ├─ mul [id BXY] < │ │ ├─ add [id BXZ] < │ │ │ ├─ i0 [id BYA] < │ │ │ └─ i1 [id BYB] < │ │ └─ i2 [id BYC] < │ └─ i3 [id BYD] < └─ i5 [id BYE] --- > ← Switch [id BWA] 'o0' > ├─ i3 [id BWB] > ├─ sub [id BWC] > │ ├─ mul [id BWD] > │ │ ├─ i0 [id BWE] > │ │ └─ i1 [id BWF] > │ └─ i2 [id BWG] > └─ i4 [id BWH] > > Composite{((i0 * i1) - i2)} [id MA] > ← sub [id BWI] 'o0' > ├─ mul [id BWJ] > │ ├─ i0 [id BWK] > │ └─ i1 [id BWL] > └─ i2 [id BWM] > > Composite{switch(i4, (i3 + switch(i2, (-1.0 * i0 * i1), 0.0)), 1)} [id MD] > ← Switch [id BWN] 'o0' > ├─ i4 [id BWO] > ├─ add [id BWP] > │ ├─ i3 [id BWQ] > │ └─ Switch [id BWR] > │ ├─ i2 [id BWS] > │ ├─ mul [id BWT] > │ │ ├─ -1.0 [id BWU] > │ │ ├─ i0 [id BWV] > │ │ └─ i1 [id BWW] > │ └─ 0.0 [id BWX] > └─ 1 [id BWY] > > Composite{switch(i4, (((i0 + i1) * i2) - i3), i5)} [id OG] > ← Switch [id BWZ] 'o0' > ├─ i4 [id BXA] > ├─ sub [id BXB] > │ ├─ mul [id BXC] > │ │ ├─ add [id BXD] > │ │ │ ├─ i0 [id BXE] > │ │ │ └─ i1 [id BXF] > │ │ └─ i2 [id BXG] > │ └─ i3 [id BXH] > └─ i5 [id BXI] > > Composite{((i0 * i1) - i2)} [id OQ] > ← sub [id BXJ] 'o0' > ├─ mul [id BXK] > │ ├─ i0 [id BXL] > │ └─ i1 [id BXM] > └─ i2 [id BXN] > > Composite{switch(i4, (i3 + switch(i2, (-1.0 * i0 * i1), 0.0)), 1)} [id OT] > ← Switch [id BXO] 'o0' > ├─ i4 [id BXP] > ├─ add [id BXQ] > │ ├─ i3 [id BXR] > │ └─ Switch [id BXS] > │ ├─ i2 [id BXT] > │ ├─ mul [id BXU] > │ │ ├─ -1.0 [id BXV] > │ │ ├─ i0 [id BXW] > │ │ └─ i1 [id BXX] > │ └─ 0.0 [id BXY] > └─ 1 [id BXZ] > > Composite{((i0 * i1) - i2)} [id QA] > ← sub [id BYA] 'o0' > ├─ mul [id BYB] > │ ├─ i0 [id BYC] > │ └─ i1 [id BYD] > └─ i2 [id BYE] 2626c2651 < Composite{((i0 * i1) - i2)} [id OY] --- > Composite{((i0 * i1) - i2)} [id QO] 2633,2643c2658,2670 < Composite{switch(i3, (i2 + switch(i1, (-i0), 0.0)), 1)} [id PB] < ← Switch [id BYK] 'o0' < ├─ i3 [id BYL] < ├─ add [id BYM] < │ ├─ i2 [id BYN] < │ └─ Switch [id BYO] < │ ├─ i1 [id BYP] < │ ├─ neg [id BYQ] < │ │ └─ i0 [id BYR] < │ └─ 0.0 [id BYS] < └─ 1 [id BYT] --- > Composite{((i0 * i1) - i2)} [id QY] > ← sub [id BYK] 'o0' > ├─ mul [id BYL] > │ ├─ i0 [id BYM] > │ └─ i1 [id BYN] > └─ i2 [id BYO] > > Composite{((i0 * i1) - i2)} [id RL] > ← sub [id BYP] 'o0' > ├─ mul [id BYQ] > │ ├─ i0 [id BYR] > │ └─ i1 [id BYS] > └─ i2 [id BYT] 2645c2672 < Composite{((i0 * i1) - i2)} [id QM] --- > Composite{((i0 * i1) - i2)} [id RV] 2652c2679 < Composite{((i0 * i1) - i2)} [id RA] --- > Composite{((i0 * i1) - i2)} [id SJ] 2659c2686 < Composite{((i0 * i1) - i2)} [id RK] --- > Composite{((i0 * i1) - i2)} [id SV] 2666c2693 < Composite{((i0 * i1) - i2)} [id RX] --- > Composite{((i0 * i1) - i2)} [id TF] 2673c2700 < Composite{((i0 * i1) - i2)} [id SH] --- > Composite{((i0 * i1) - i2)} [id TS] 2680,2692c2707,2717 < Composite{((i0 * i1) - i2)} [id SV] < ← sub [id BZT] 'o0' < ├─ mul [id BZU] < │ ├─ i0 [id BZV] < │ └─ i1 [id BZW] < └─ i2 [id BZX] < < Composite{((i0 * i1) - i2)} [id TH] < ← sub [id BZY] 'o0' < ├─ mul [id BZZ] < │ ├─ i0 [id CAA] < │ └─ i1 [id CAB] < └─ i2 [id CAC] --- > Composite{switch(i4, (((i0 + i1) * i2) - i3), i5)} [id UD] > ← Switch [id BZT] 'o0' > ├─ i4 [id BZU] > ├─ sub [id BZV] > │ ├─ mul [id BZW] > │ │ ├─ add [id BZX] > │ │ │ ├─ i0 [id BZY] > │ │ │ └─ i1 [id BZZ] > │ │ └─ i2 [id CAA] > │ └─ i3 [id CAB] > └─ i5 [id CAC] 2694c2719 < Composite{((i0 * i1) - i2)} [id TR] --- > Composite{((i0 * i1) - i2)} [id UN] 2701,2925c2726,2944 < Composite{((i0 * i1) - i2)} [id UE] < ← sub [id CAI] 'o0' < ├─ mul [id CAJ] < │ ├─ i0 [id CAK] < │ └─ i1 [id CAL] < └─ i2 [id CAM] < < Composite{switch(i4, (((i0 + i1) * i2) - i3), i5)} [id UP] < ← Switch [id CAN] 'o0' < ├─ i4 [id CAO] < ├─ sub [id CAP] < │ ├─ mul [id CAQ] < │ │ ├─ add [id CAR] < │ │ │ ├─ i0 [id CAS] < │ │ │ └─ i1 [id CAT] < │ │ └─ i2 [id CAU] < │ └─ i3 [id CAV] < └─ i5 [id CAW] < < Composite{((i0 * i1) - i2)} [id UZ] < ← sub [id CAX] 'o0' < ├─ mul [id CAY] < │ ├─ i0 [id CAZ] < │ └─ i1 [id CBA] < └─ i2 [id CBB] < < Composite{switch(i3, (i2 + switch(i1, (-i0), 0.0)), 1)} [id VC] < ← Switch [id CBC] 'o0' < ├─ i3 [id CBD] < ├─ add [id CBE] < │ ├─ i2 [id CBF] < │ └─ Switch [id CBG] < │ ├─ i1 [id CBH] < │ ├─ neg [id CBI] < │ │ └─ i0 [id CBJ] < │ └─ 0.0 [id CBK] < └─ 1 [id CBL] < < Composite{((i0 * i1) - i2)} [id WN] < ← sub [id CBM] 'o0' < ├─ mul [id CBN] < │ ├─ i0 [id CBO] < │ └─ i1 [id CBP] < └─ i2 [id CBQ] < < Composite{((i0 * i1) - i2)} [id XB] < ← sub [id CBR] 'o0' < ├─ mul [id CBS] < │ ├─ i0 [id CBT] < │ └─ i1 [id CBU] < └─ i2 [id CBV] < < Composite{((i0 * i1) - i2)} [id XL] < ← sub [id CBW] 'o0' < ├─ mul [id CBX] < │ ├─ i0 [id CBY] < │ └─ i1 [id CBZ] < └─ i2 [id CCA] < < Composite{((i0 * i1) - i2)} [id XY] < ← sub [id CCB] 'o0' < ├─ mul [id CCC] < │ ├─ i0 [id CCD] < │ └─ i1 [id CCE] < └─ i2 [id CCF] < < Composite{((i0 * i1) - i2)} [id YI] < ← sub [id CCG] 'o0' < ├─ mul [id CCH] < │ ├─ i0 [id CCI] < │ └─ i1 [id CCJ] < └─ i2 [id CCK] < < Composite{((i0 * i1) - i2)} [id YW] < ← sub [id CCL] 'o0' < ├─ mul [id CCM] < │ ├─ i0 [id CCN] < │ └─ i1 [id CCO] < └─ i2 [id CCP] < < Composite{((i0 * i1) - i2)} [id ZI] < ← sub [id CCQ] 'o0' < ├─ mul [id CCR] < │ ├─ i0 [id CCS] < │ └─ i1 [id CCT] < └─ i2 [id CCU] < < Composite{((i0 * i1) - i2)} [id ZS] < ← sub [id CCV] 'o0' < ├─ mul [id CCW] < │ ├─ i0 [id CCX] < │ └─ i1 [id CCY] < └─ i2 [id CCZ] < < Composite{((i0 * i1) - i2)} [id BAF] < ← sub [id CDA] 'o0' < ├─ mul [id CDB] < │ ├─ i0 [id CDC] < │ └─ i1 [id CDD] < └─ i2 [id CDE] < < Composite{switch(i4, (((i0 + i1) * i2) - i3), i5)} [id BAQ] < ← Switch [id CDF] 'o0' < ├─ i4 [id CDG] < ├─ sub [id CDH] < │ ├─ mul [id CDI] < │ │ ├─ add [id CDJ] < │ │ │ ├─ i0 [id CDK] < │ │ │ └─ i1 [id CDL] < │ │ └─ i2 [id CDM] < │ └─ i3 [id CDN] < └─ i5 [id CDO] < < Composite{((i0 * i1) - i2)} [id BBA] < ← sub [id CDP] 'o0' < ├─ mul [id CDQ] < │ ├─ i0 [id CDR] < │ └─ i1 [id CDS] < └─ i2 [id CDT] < < Composite{switch(i3, (i2 + switch(i1, (-i0), 0.0)), 1)} [id BBD] < ← Switch [id CDU] 'o0' < ├─ i3 [id CDV] < ├─ add [id CDW] < │ ├─ i2 [id CDX] < │ └─ Switch [id CDY] < │ ├─ i1 [id CDZ] < │ ├─ neg [id CEA] < │ │ └─ i0 [id CEB] < │ └─ 0.0 [id CEC] < └─ 1 [id CED] < < Composite{((i0 * i1) - i2)} [id BCO] < ← sub [id CEE] 'o0' < ├─ mul [id CEF] < │ ├─ i0 [id CEG] < │ └─ i1 [id CEH] < └─ i2 [id CEI] < < Composite{((i0 * i1) - i2)} [id BDC] < ← sub [id CEJ] 'o0' < ├─ mul [id CEK] < │ ├─ i0 [id CEL] < │ └─ i1 [id CEM] < └─ i2 [id CEN] < < Composite{((i0 * i1) - i2)} [id BDM] < ← sub [id CEO] 'o0' < ├─ mul [id CEP] < │ ├─ i0 [id CEQ] < │ └─ i1 [id CER] < └─ i2 [id CES] < < Composite{((i0 * i1) - i2)} [id BDZ] < ← sub [id CET] 'o0' < ├─ mul [id CEU] < │ ├─ i0 [id CEV] < │ └─ i1 [id CEW] < └─ i2 [id CEX] < < Composite{((i0 * i1) - i2)} [id BEJ] < ← sub [id CEY] 'o0' < ├─ mul [id CEZ] < │ ├─ i0 [id CFA] < │ └─ i1 [id CFB] < └─ i2 [id CFC] < < Composite{((i0 * i1) - i2)} [id BEX] < ← sub [id CFD] 'o0' < ├─ mul [id CFE] < │ ├─ i0 [id CFF] < │ └─ i1 [id CFG] < └─ i2 [id CFH] < < Composite{((i0 * i1) - i2)} [id BFJ] < ← sub [id CFI] 'o0' < ├─ mul [id CFJ] < │ ├─ i0 [id CFK] < │ └─ i1 [id CFL] < └─ i2 [id CFM] < < Composite{((i0 * i1) - i2)} [id BFT] < ← sub [id CFN] 'o0' < ├─ mul [id CFO] < │ ├─ i0 [id CFP] < │ └─ i1 [id CFQ] < └─ i2 [id CFR] < < Composite{((i0 * i1) - i2)} [id BGG] < ← sub [id CFS] 'o0' < ├─ mul [id CFT] < │ ├─ i0 [id CFU] < │ └─ i1 [id CFV] < └─ i2 [id CFW] < < Composite{switch(i4, (((i0 + i1) * i2) - i3), i5)} [id BGR] < ← Switch [id CFX] 'o0' < ├─ i4 [id CFY] < ├─ sub [id CFZ] < │ ├─ mul [id CGA] < │ │ ├─ add [id CGB] < │ │ │ ├─ i0 [id CGC] < │ │ │ └─ i1 [id CGD] < │ │ └─ i2 [id CGE] < │ └─ i3 [id CGF] < └─ i5 [id CGG] < < Composite{((i0 * i1) - i2)} [id BHB] < ← sub [id CGH] 'o0' < ├─ mul [id CGI] < │ ├─ i0 [id CGJ] < │ └─ i1 [id CGK] < └─ i2 [id CGL] < < Composite{switch(i3, (i2 + switch(i1, (-i0), 0.0)), 1)} [id BHE] < ← Switch [id CGM] 'o0' < ├─ i3 [id CGN] < ├─ add [id CGO] < │ ├─ i2 [id CGP] < │ └─ Switch [id CGQ] < │ ├─ i1 [id CGR] < │ ├─ neg [id CGS] < │ │ └─ i0 [id CGT] < │ └─ 0.0 [id CGU] < └─ 1 [id CGV] --- > Composite{switch(i4, (i3 + switch(i2, (-1.0 * i0 * i1), 0.0)), 1)} [id UQ] > ← Switch [id CAI] 'o0' > ├─ i4 [id CAJ] > ├─ add [id CAK] > │ ├─ i3 [id CAL] > │ └─ Switch [id CAM] > │ ├─ i2 [id CAN] > │ ├─ mul [id CAO] > │ │ ├─ -1.0 [id CAP] > │ │ ├─ i0 [id CAQ] > │ │ └─ i1 [id CAR] > │ └─ 0.0 [id CAS] > └─ 1 [id CAT] > > Composite{((i0 * i1) - i2)} [id VX] > ← sub [id CAU] 'o0' > ├─ mul [id CAV] > │ ├─ i0 [id CAW] > │ └─ i1 [id CAX] > └─ i2 [id CAY] > > Composite{((i0 * i1) - i2)} [id WL] > ← sub [id CAZ] 'o0' > ├─ mul [id CBA] > │ ├─ i0 [id CBB] > │ └─ i1 [id CBC] > └─ i2 [id CBD] > > Composite{((i0 * i1) - i2)} [id WV] > ← sub [id CBE] 'o0' > ├─ mul [id CBF] > │ ├─ i0 [id CBG] > │ └─ i1 [id CBH] > └─ i2 [id CBI] > > Composite{((i0 * i1) - i2)} [id XI] > ← sub [id CBJ] 'o0' > ├─ mul [id CBK] > │ ├─ i0 [id CBL] > │ └─ i1 [id CBM] > └─ i2 [id CBN] > > Composite{((i0 * i1) - i2)} [id XS] > ← sub [id CBO] 'o0' > ├─ mul [id CBP] > │ ├─ i0 [id CBQ] > │ └─ i1 [id CBR] > └─ i2 [id CBS] > > Composite{((i0 * i1) - i2)} [id YG] > ← sub [id CBT] 'o0' > ├─ mul [id CBU] > │ ├─ i0 [id CBV] > │ └─ i1 [id CBW] > └─ i2 [id CBX] > > Composite{((i0 * i1) - i2)} [id YS] > ← sub [id CBY] 'o0' > ├─ mul [id CBZ] > │ ├─ i0 [id CCA] > │ └─ i1 [id CCB] > └─ i2 [id CCC] > > Composite{((i0 * i1) - i2)} [id ZC] > ← sub [id CCD] 'o0' > ├─ mul [id CCE] > │ ├─ i0 [id CCF] > │ └─ i1 [id CCG] > └─ i2 [id CCH] > > Composite{((i0 * i1) - i2)} [id ZP] > ← sub [id CCI] 'o0' > ├─ mul [id CCJ] > │ ├─ i0 [id CCK] > │ └─ i1 [id CCL] > └─ i2 [id CCM] > > Composite{switch(i4, (((i0 + i1) * i2) - i3), i5)} [id BAA] > ← Switch [id CCN] 'o0' > ├─ i4 [id CCO] > ├─ sub [id CCP] > │ ├─ mul [id CCQ] > │ │ ├─ add [id CCR] > │ │ │ ├─ i0 [id CCS] > │ │ │ └─ i1 [id CCT] > │ │ └─ i2 [id CCU] > │ └─ i3 [id CCV] > └─ i5 [id CCW] > > Composite{((i0 * i1) - i2)} [id BAK] > ← sub [id CCX] 'o0' > ├─ mul [id CCY] > │ ├─ i0 [id CCZ] > │ └─ i1 [id CDA] > └─ i2 [id CDB] > > Composite{switch(i4, (i3 + switch(i2, (-1.0 * i0 * i1), 0.0)), 1)} [id BAN] > ← Switch [id CDC] 'o0' > ├─ i4 [id CDD] > ├─ add [id CDE] > │ ├─ i3 [id CDF] > │ └─ Switch [id CDG] > │ ├─ i2 [id CDH] > │ ├─ mul [id CDI] > │ │ ├─ -1.0 [id CDJ] > │ │ ├─ i0 [id CDK] > │ │ └─ i1 [id CDL] > │ └─ 0.0 [id CDM] > └─ 1 [id CDN] > > Composite{((i0 * i1) - i2)} [id BBU] > ← sub [id CDO] 'o0' > ├─ mul [id CDP] > │ ├─ i0 [id CDQ] > │ └─ i1 [id CDR] > └─ i2 [id CDS] > > Composite{((i0 * i1) - i2)} [id BCI] > ← sub [id CDT] 'o0' > ├─ mul [id CDU] > │ ├─ i0 [id CDV] > │ └─ i1 [id CDW] > └─ i2 [id CDX] > > Composite{((i0 * i1) - i2)} [id BCS] > ← sub [id CDY] 'o0' > ├─ mul [id CDZ] > │ ├─ i0 [id CEA] > │ └─ i1 [id CEB] > └─ i2 [id CEC] > > Composite{((i0 * i1) - i2)} [id BDF] > ← sub [id CED] 'o0' > ├─ mul [id CEE] > │ ├─ i0 [id CEF] > │ └─ i1 [id CEG] > └─ i2 [id CEH] > > Composite{((i0 * i1) - i2)} [id BDP] > ← sub [id CEI] 'o0' > ├─ mul [id CEJ] > │ ├─ i0 [id CEK] > │ └─ i1 [id CEL] > └─ i2 [id CEM] > > Composite{((i0 * i1) - i2)} [id BED] > ← sub [id CEN] 'o0' > ├─ mul [id CEO] > │ ├─ i0 [id CEP] > │ └─ i1 [id CEQ] > └─ i2 [id CER] > > Composite{((i0 * i1) - i2)} [id BEP] > ← sub [id CES] 'o0' > ├─ mul [id CET] > │ ├─ i0 [id CEU] > │ └─ i1 [id CEV] > └─ i2 [id CEW] > > Composite{((i0 * i1) - i2)} [id BEZ] > ← sub [id CEX] 'o0' > ├─ mul [id CEY] > │ ├─ i0 [id CEZ] > │ └─ i1 [id CFA] > └─ i2 [id CFB] > > Composite{((i0 * i1) - i2)} [id BFM] > ← sub [id CFC] 'o0' > ├─ mul [id CFD] > │ ├─ i0 [id CFE] > │ └─ i1 [id CFF] > └─ i2 [id CFG] > > Composite{switch(i4, (((i0 + i1) * i2) - i3), i5)} [id BFX] > ← Switch [id CFH] 'o0' > ├─ i4 [id CFI] > ├─ sub [id CFJ] > │ ├─ mul [id CFK] > │ │ ├─ add [id CFL] > │ │ │ ├─ i0 [id CFM] > │ │ │ └─ i1 [id CFN] > │ │ └─ i2 [id CFO] > │ └─ i3 [id CFP] > └─ i5 [id CFQ] > > Composite{((i0 * i1) - i2)} [id BGH] > ← sub [id CFR] 'o0' > ├─ mul [id CFS] > │ ├─ i0 [id CFT] > │ └─ i1 [id CFU] > └─ i2 [id CFV] > > Composite{switch(i4, (i3 + switch(i2, (-1.0 * i0 * i1), 0.0)), 1)} [id BGK] > ← Switch [id CFW] 'o0' > ├─ i4 [id CFX] > ├─ add [id CFY] > │ ├─ i3 [id CFZ] > │ └─ Switch [id CGA] > │ ├─ i2 [id CGB] > │ ├─ mul [id CGC] > │ │ ├─ -1.0 [id CGD] > │ │ ├─ i0 [id CGE] > │ │ └─ i1 [id CGF] > │ └─ 0.0 [id CGG] > └─ 1 [id CGH] > > Composite{((i0 * i1) - i2)} [id BHR] > ← sub [id CGI] 'o0' > ├─ mul [id CGJ] > │ ├─ i0 [id CGK] > │ └─ i1 [id CGL] > └─ i2 [id CGM] > > Composite{((i0 * i1) - i2)} [id BIF] > ← sub [id CGN] 'o0' > ├─ mul [id CGO] > │ ├─ i0 [id CGP] > │ └─ i1 [id CGQ] > └─ i2 [id CGR] 2928,2946c2947,2965 < ← sub [id CGW] 'o0' < ├─ mul [id CGX] < │ ├─ i0 [id CGY] < │ └─ i1 [id CGZ] < └─ i2 [id CHA] < < Composite{((i0 * i1) - i2)} [id BJD] < ← sub [id CHB] 'o0' < ├─ mul [id CHC] < │ ├─ i0 [id CHD] < │ └─ i1 [id CHE] < └─ i2 [id CHF] < < Composite{((i0 * i1) - i2)} [id BJN] < ← sub [id CHG] 'o0' < ├─ mul [id CHH] < │ ├─ i0 [id CHI] < │ └─ i1 [id CHJ] < └─ i2 [id CHK] --- > ← sub [id CGS] 'o0' > ├─ mul [id CGT] > │ ├─ i0 [id CGU] > │ └─ i1 [id CGV] > └─ i2 [id CGW] > > Composite{((i0 * i1) - i2)} [id BJC] > ← sub [id CGX] 'o0' > ├─ mul [id CGY] > │ ├─ i0 [id CGZ] > │ └─ i1 [id CHA] > └─ i2 [id CHB] > > Composite{((i0 * i1) - i2)} [id BJM] > ← sub [id CHC] 'o0' > ├─ mul [id CHD] > │ ├─ i0 [id CHE] > │ └─ i1 [id CHF] > └─ i2 [id CHG] 2949,2988c2968,2993 < ← sub [id CHL] 'o0' < ├─ mul [id CHM] < │ ├─ i0 [id CHN] < │ └─ i1 [id CHO] < └─ i2 [id CHP] < < Composite{((i0 * i1) - i2)} [id BKK] < ← sub [id CHQ] 'o0' < ├─ mul [id CHR] < │ ├─ i0 [id CHS] < │ └─ i1 [id CHT] < └─ i2 [id CHU] < < Composite{((i0 * i1) - i2)} [id BKY] < ← sub [id CHV] 'o0' < ├─ mul [id CHW] < │ ├─ i0 [id CHX] < │ └─ i1 [id CHY] < └─ i2 [id CHZ] < < Composite{((i0 * i1) - i2)} [id BLK] < ← sub [id CIA] 'o0' < ├─ mul [id CIB] < │ ├─ i0 [id CIC] < │ └─ i1 [id CID] < └─ i2 [id CIE] < < Composite{((i0 * i1) - i2)} [id BLU] < ← sub [id CIF] 'o0' < ├─ mul [id CIG] < │ ├─ i0 [id CIH] < │ └─ i1 [id CII] < └─ i2 [id CIJ] < < Composite{((i0 * i1) - i2)} [id BMH] < ← sub [id CIK] 'o0' < ├─ mul [id CIL] < │ ├─ i0 [id CIM] < │ └─ i1 [id CIN] < └─ i2 [id CIO] --- > ← sub [id CHH] 'o0' > ├─ mul [id CHI] > │ ├─ i0 [id CHJ] > │ └─ i1 [id CHK] > └─ i2 [id CHL] > > Composite{((i0 * i1) - i2)} [id BKM] > ← sub [id CHM] 'o0' > ├─ mul [id CHN] > │ ├─ i0 [id CHO] > │ └─ i1 [id CHP] > └─ i2 [id CHQ] > > Composite{((i0 * i1) - i2)} [id BKW] > ← sub [id CHR] 'o0' > ├─ mul [id CHS] > │ ├─ i0 [id CHT] > │ └─ i1 [id CHU] > └─ i2 [id CHV] > > Composite{((i0 * i1) - i2)} [id BLJ] > ← sub [id CHW] 'o0' > ├─ mul [id CHX] > │ ├─ i0 [id CHY] > │ └─ i1 [id CHZ] > └─ i2 [id CIA]