Composite{...}.0 [id A] 258 ├─ Sum{axes=None} [id B] 186 │ └─ Composite{...}.0 [id C] 163 │ ├─ Join [id D] 77 │ │ ├─ 0 [id E] │ │ ├─ Add [id F] 69 │ │ │ ├─ w_simplex__ [id G] │ │ │ └─ ExpandDims{axis=0} [id H] 61 │ │ │ └─ Sum{axes=None} [id I] 8 │ │ │ └─ w_simplex__ [id G] │ │ └─ [0.] [id J] │ ├─ ExpandDims{axis=0} [id K] 121 │ │ └─ Max{axis=0} [id L] 'max' 102 │ │ └─ Join [id D] 77 │ │ └─ ··· │ ├─ Isinf [id M] 140 │ │ └─ ExpandDims{axis=0} [id K] 121 │ │ └─ ··· │ └─ Exp [id N] 139 │ └─ ExpandDims{axis=0} [id K] 121 │ └─ ··· ├─ Max{axis=0} [id L] 'max' 102 │ └─ ··· ├─ 5.0 [id O] ├─ 1.6094379124341003 [id P] ├─ Sum{axes=None} [id Q] 119 │ └─ Composite{((-0.5 * sqr(i0)) - 0.9189385332046727)} [id R] 100 │ └─ CumOp{-1, add} [id S] 75 │ └─ SetSubtensor{start:} [id T] 67 │ ├─ SetSubtensor{i} [id U] 60 │ │ ├─ Alloc [id V] 7 │ │ │ ├─ 0.0 [id W] │ │ │ └─ 5 [id X] │ │ ├─ Subtensor{i} [id Y] 6 │ │ │ ├─ x_coord_ordered__ [id Z] │ │ │ └─ 0 [id BA] │ │ └─ 0 [id BA] │ ├─ Exp [id BB] 58 │ │ └─ Subtensor{start:} [id BC] 5 │ │ ├─ x_coord_ordered__ [id Z] │ │ └─ 1 [id BD] │ └─ 1 [id BD] ├─ Sum{axes=None} [id BE] 59 │ └─ Subtensor{start:} [id BC] 5 │ └─ ··· ├─ Sum{axes=None} [id BF] 57 │ └─ Composite{((-0.5 * sqr(i0)) - 0.9189385332046727)} [id BG] 'y_coord_logprob' 4 │ └─ y_coord [id BH] ├─ Sum{axes=None} [id BI] 56 │ └─ Composite{...}.2 [id BJ] 'sigma_log___logprob' 0 │ └─ sigma_log__ [id BK] ├─ 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 │ │ │ │ ├─ Join [id BT] 76 │ │ │ │ │ ├─ 0 [id E] │ │ │ │ │ ├─ w_simplex__ [id G] │ │ │ │ │ └─ Neg [id BU] 68 │ │ │ │ │ └─ ExpandDims{axis=0} [id H] 61 │ │ │ │ │ └─ ··· │ │ │ │ └─ ExpandDims{axis=0} [id BV] 120 │ │ │ │ └─ Max{axis=0} [id BW] 'max' 101 │ │ │ │ └─ Join [id BT] 76 │ │ │ │ └─ ··· │ │ │ └─ Join [id BX] 238 │ │ │ ├─ 1 [id BY] │ │ │ ├─ 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 │ │ │ │ │ └─ ··· │ │ │ │ └─ 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 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 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 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 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 │ │ │ └─ ··· │ │ ├─ Isinf [id GG] 246 │ │ │ └─ ExpandDims{axis=1} [id GE] 244 │ │ │ └─ ··· │ │ └─ Exp [id GH] 247 │ │ └─ ExpandDims{axis=1} [id GE] 244 │ │ └─ ··· │ ├─ Max{axis=1} [id GF] 'max' 242 │ │ └─ ··· │ └─ 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 │ │ │ └─ ··· │ │ └─ [0] [id CO] │ ├─ All{axes=None} [id GO] 200 │ │ └─ Le [id GP] 184 │ │ ├─ Softmax{axis=0} [id GN] 162 │ │ │ └─ ··· │ │ └─ [1] [id GQ] │ └─ True [id GR] ├─ Sum{axes=None} [id I] 8 │ └─ ··· ├─ -inf [id GS] ├─ 144.56574394634487 [id GT] ├─ Sum{axes=None} [id GU] 199 │ └─ Composite{...}.0 [id GV] 183 │ ├─ LogSoftmax{axis=0} [id BR] 161 │ │ └─ ··· │ └─ Softmax{axis=0} [id GN] 162 │ └─ ··· ├─ Any{axis=0} [id GW] 198 │ └─ Lt [id GX] 182 │ ├─ Softmax{axis=0} [id GN] 162 │ │ └─ ··· │ └─ [0] [id CO] └─ Any{axis=0} [id GY] 197 └─ Gt [id GZ] 181 ├─ Softmax{axis=0} [id GN] 162 │ └─ ··· └─ [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 │ │ │ │ └─ ··· │ │ │ ├─ Join [id BT] 76 │ │ │ │ └─ ··· │ │ │ ├─ 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 │ │ │ │ │ │ └─ ··· │ │ │ │ │ ├─ Switch [id HL] 263 │ │ │ │ │ │ ├─ ExpandDims{axis=0} [id HM] 261 │ │ │ │ │ │ │ └─ Composite{...}.1 [id A] 258 │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ ├─ [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 │ │ │ │ │ │ └─ ··· │ │ │ │ │ ├─ ExpandDims{axis=1} [id HQ] 251 │ │ │ │ │ │ └─ Sum{axis=1} [id BN] 250 │ │ │ │ │ │ └─ ··· │ │ │ │ │ ├─ Isinf [id GG] 246 │ │ │ │ │ │ └─ ··· │ │ │ │ │ ├─ 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 │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ └─ Isinf [id GG] 246 │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ ├─ 1000 [id HW] │ │ │ │ │ │ │ └─ 5 [id HX] │ │ │ │ │ │ ├─ Exp [id GH] 247 │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ └─ 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 HQ] 251 │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ └─ Isinf [id GG] 246 │ │ │ │ │ │ └─ ··· │ │ │ │ │ ├─ ExpandDims{axis=1} [id GE] 244 │ │ │ │ │ │ └─ ··· │ │ │ │ │ └─ Add [id BP] 240 │ │ │ │ │ └─ ··· │ │ │ │ ├─ Softmax{axis=0} [id GN] 162 │ │ │ │ │ └─ ··· │ │ │ │ ├─ Composite{(switch(i0, 0, i1) + i2)} [id HK] 273 │ │ │ │ │ └─ ··· │ │ │ │ └─ SoftmaxGrad{axis=0} [id IB] 180 │ │ │ │ ├─ [0. 0. 0. 0. 0.] [id IC] │ │ │ │ └─ Softmax{axis=0} [id GN] 162 │ │ │ │ └─ ··· │ │ │ └─ Composite{(i3 + (i2 - (i0 * i1)))} [id HH] 311 │ │ │ └─ ··· │ │ ├─ 0 [id E] │ │ └─ [4 1] [id ID] │ └─ 4 [id IE] ├─ Neg [id IF] 383 │ └─ SpecifyShape [id IG] 375 │ ├─ Split{2}.1 [id HD] 362 │ │ └─ ··· │ └─ 1 [id BY] ├─ 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 IL] 202 │ │ │ └─ Sum{axes=None} [id B] 186 │ │ │ └─ ··· │ │ ├─ Isinf [id M] 140 │ │ │ └─ ··· │ │ ├─ 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 │ │ │ │ │ │ └─ ··· │ │ │ │ │ └─ Isinf [id M] 140 │ │ │ │ │ └─ ··· │ │ │ │ └─ 5 [id HX] │ │ │ ├─ Exp [id N] 139 │ │ │ │ └─ ··· │ │ │ └─ 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 IL] 202 │ │ │ │ └─ ··· │ │ │ └─ Isinf [id M] 140 │ │ │ └─ ··· │ │ ├─ ExpandDims{axis=0} [id K] 121 │ │ │ └─ ··· │ │ └─ Join [id D] 77 │ │ └─ ··· │ ├─ 0 [id E] │ └─ [4 1] [id ID] └─ Split{2}.0 [id IJ] 243 └─ ··· IncSubtensor{start:} [id IU] 'x_coord_ordered___grad' 687 ├─ IncSubtensor{i} [id IV] 645 │ ├─ Alloc [id IW] 50 │ │ ├─ [0.] [id J] │ │ └─ 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 │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ │ │ │ │ │ │ └─ 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 │ │ │ │ │ │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ │ │ │ │ │ └─ 0 [id BA] │ │ │ │ │ │ │ │ │ │ │ │ ├─ 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 │ │ │ │ │ │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ │ │ │ │ │ └─ 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 │ │ │ │ │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ │ │ │ │ └─ 1 [id DO] │ │ │ │ │ │ │ │ │ │ │ ├─ 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 │ │ │ │ │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ │ │ │ │ └─ 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 │ │ │ │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ │ │ │ └─ 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 │ │ │ │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ │ │ │ └─ 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 │ │ │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ │ │ └─ 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 │ │ │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ │ │ └─ 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 │ │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ │ └─ 4 [id FW] │ │ │ │ │ │ │ │ ├─ 0 [id E] │ │ │ │ │ │ │ │ └─ [1 1] [id KW] │ │ │ │ │ │ │ ├─ 1 [id BY] │ │ │ │ │ │ │ └─ NoneConst{None} [id JX] │ │ │ │ │ │ └─ -1 [id KX] │ │ │ │ │ └─ Subtensor{::step} [id KY] 99 │ │ │ │ │ ├─ CumOp{-1, add} [id S] 75 │ │ │ │ │ │ └─ ··· │ │ │ │ │ └─ -1 [id KX] │ │ │ │ └─ -1 [id KX] │ │ │ ├─ [0. 0. 0. 0.] [id KZ] │ │ │ └─ 1 [id BD] │ │ └─ 0 [id BA] │ └─ 0 [id BA] ├─ 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 │ │ │ │ └─ ··· │ │ │ ├─ 3 [id LD] │ │ │ ├─ -6 [id LE] │ │ │ └─ -1 [id KX] │ │ └─ 4 [id IE] │ └─ Exp [id BB] 58 │ └─ ··· └─ 1 [id BD] Sub [id LF] 'y_coord_grad' 414 ├─ DropDims{axis=0} [id LG] 405 │ └─ SpecifyShape [id LH] 398 │ ├─ Split{2}.1 [id JF] 392 │ │ └─ ··· │ ├─ 1 [id BY] │ └─ NoneConst{None} [id JX] └─ y_coord [id BH] 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 [id BJ] 0 │ │ │ │ │ │ └─ ··· │ │ │ │ │ ├─ 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 │ │ │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id LX] 85 │ │ │ │ │ │ │ │ │ │ │ └─ Cholesky{lower=True, destructive=False, on_error='nan'} [id CI] 71 │ │ │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ │ │ └─ [[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 │ │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ │ └─ Composite{((i0 * i1) - i2)} [id MB] 604 │ │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id MC] 413 │ │ │ │ │ │ │ │ │ │ └─ dot [id MD] 404 │ │ │ │ │ │ │ │ │ │ ├─ Switch [id LS] 167 │ │ │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ │ │ └─ 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 │ │ │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ │ │ ├─ 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 │ │ │ │ │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ │ │ │ │ ├─ 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 │ │ │ │ │ │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ │ │ │ │ │ └─ ExtractDiag{offset=0, axis1=0, axis2=1, view=False} [id CH] 82 │ │ │ │ │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ │ │ │ │ ├─ [0 1] [id CN] │ │ │ │ │ │ │ │ │ │ │ │ └─ [0 1] [id CN] │ │ │ │ │ │ │ │ │ │ │ ├─ 2 [id NB] │ │ │ │ │ │ │ │ │ │ │ ├─ 0 [id NC] │ │ │ │ │ │ │ │ │ │ │ └─ 2 [id NB] │ │ │ │ │ │ │ │ │ │ └─ Invert [id LT] 146 │ │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ │ ├─ 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 │ │ │ │ │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ │ │ │ │ └─ 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 NN] 11 │ │ │ │ │ │ │ │ │ │ │ └─ 1 [id NO] │ │ │ │ │ │ │ │ │ │ └─ 0 [id E] │ │ │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id NP] 598 │ │ │ │ │ │ │ │ │ └─ AdvancedSetSubtensor [id NQ] 591 │ │ │ │ │ │ │ │ │ ├─ Alloc [id NR] 540 │ │ │ │ │ │ │ │ │ │ ├─ 0.0 [id W] │ │ │ │ │ │ │ │ │ │ ├─ 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 │ │ │ │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id NX] 2 │ │ │ │ │ │ │ │ │ │ │ └─ -1 [id NY] │ │ │ │ │ │ │ │ │ │ └─ Subtensor{i} [id NS] 491 │ │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ │ ├─ Mul [id NU] 428 │ │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ │ ├─ ARange{dtype='int64'} [id NZ] 539 │ │ │ │ │ │ │ │ │ │ ├─ 0 [id E] │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id NS] 491 │ │ │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ │ │ └─ 1 [id BY] │ │ │ │ │ │ │ │ │ └─ ARange{dtype='int64'} [id NZ] 539 │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id OA] 771 │ │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id LR] 729 │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ ├─ Tri{dtype='float64'} [id OB] 894 │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id OC] 881 │ │ │ │ │ │ │ │ │ ├─ Subtensor{start:} [id OD] 858 │ │ │ │ │ │ │ │ │ │ ├─ Shape [id OE] 831 │ │ │ │ │ │ │ │ │ │ │ └─ Add [id LQ] 805 │ │ │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id NI] 12 │ │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id NK] 13 │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id OF] 880 │ │ │ │ │ │ │ │ │ ├─ Subtensor{start:} [id OD] 858 │ │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id NN] 11 │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ └─ 0 [id E] │ │ │ │ │ │ │ ├─ AdvancedSetSubtensor [id OG] 879 │ │ │ │ │ │ │ │ ├─ Alloc [id OH] 857 │ │ │ │ │ │ │ │ │ ├─ 0.0 [id W] │ │ │ │ │ │ │ │ │ ├─ 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 │ │ │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id NX] 2 │ │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ │ └─ Subtensor{i} [id OI] 830 │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ ├─ ExtractDiag{offset=0, axis1=0, axis2=1, view=False} [id OK] 770 │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ ├─ ARange{dtype='int64'} [id OL] 856 │ │ │ │ │ │ │ │ │ ├─ 0 [id E] │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id OI] 830 │ │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ │ └─ 1 [id BY] │ │ │ │ │ │ │ │ └─ ARange{dtype='int64'} [id OL] 856 │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ ├─ Invert [id LT] 146 │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ └─ [[nan]] [id OM] │ │ │ │ │ │ ├─ [0 1] [id CN] │ │ │ │ │ │ └─ [0 1] [id CN] │ │ │ │ │ └─ 0 [id BA] │ │ │ │ ├─ 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 │ │ │ │ │ │ │ │ │ └─ Cholesky{lower=True, destructive=False, on_error='nan'} [id DL] 74 │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id OV] 97 │ │ │ │ │ │ │ │ │ └─ Cholesky{lower=True, destructive=False, on_error='nan'} [id DL] 74 │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ └─ [[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 │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ └─ Composite{((i0 * i1) - i2)} [id OY] 605 │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id OZ] 418 │ │ │ │ │ │ │ │ └─ dot [id PA] 407 │ │ │ │ │ │ │ │ ├─ Switch [id OQ] 173 │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ └─ 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 │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ ├─ SetSubtensor{:stop, start:stop} [id PN] 355 │ │ │ │ │ │ │ │ │ ├─ Alloc [id MU] 9 │ │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ │ ├─ AdvancedSetSubtensor [id PO] 343 │ │ │ │ │ │ │ │ │ │ ├─ Alloc [id CK] 1 │ │ │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ │ │ ├─ 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 │ │ │ │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ │ │ │ └─ ExtractDiag{offset=0, axis1=0, axis2=1, view=False} [id DK] 95 │ │ │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ │ │ ├─ [0 1] [id CN] │ │ │ │ │ │ │ │ │ │ └─ [0 1] [id CN] │ │ │ │ │ │ │ │ │ ├─ 2 [id NB] │ │ │ │ │ │ │ │ │ ├─ 0 [id NC] │ │ │ │ │ │ │ │ │ └─ 2 [id NB] │ │ │ │ │ │ │ │ └─ Invert [id OR] 155 │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ ├─ 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 │ │ │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id NI] 12 │ │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id NK] 13 │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id PZ] 444 │ │ │ │ │ │ │ │ │ ├─ Subtensor{start:} [id PX] 432 │ │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id NN] 11 │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ └─ 0 [id E] │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id QA] 599 │ │ │ │ │ │ │ └─ AdvancedSetSubtensor [id QB] 593 │ │ │ │ │ │ │ ├─ Alloc [id QC] 544 │ │ │ │ │ │ │ │ ├─ 0.0 [id W] │ │ │ │ │ │ │ │ ├─ 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 │ │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id NX] 2 │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ └─ Subtensor{i} [id QD] 495 │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ ├─ Mul [id QF] 431 │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ ├─ ARange{dtype='int64'} [id QH] 543 │ │ │ │ │ │ │ │ ├─ 0 [id E] │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id QD] 495 │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ └─ 1 [id BY] │ │ │ │ │ │ │ └─ ARange{dtype='int64'} [id QH] 543 │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ ├─ 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 QK] 698 │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id QL] 656 │ │ │ │ │ │ │ ├─ Switch [id OQ] 173 │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ └─ Composite{((i0 * i1) - i2)} [id QM] 614 │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id OZ] 418 │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id QN] 554 │ │ │ │ │ │ │ │ └─ Tri{dtype='float64'} [id QO] 505 │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id PW] 445 │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id QP] 454 │ │ │ │ │ │ │ │ │ ├─ Subtensor{start:} [id PX] 432 │ │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id QQ] 49 │ │ │ │ │ │ │ │ │ └─ 1 [id QR] │ │ │ │ │ │ │ │ └─ 0 [id E] │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id QA] 599 │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ ├─ 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 │ │ │ │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ │ │ │ └─ 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 │ │ │ │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ │ │ │ └─ Composite{((i0 * i1) - i2)} [id RA] 613 │ │ │ │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id OZ] 418 │ │ │ │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id RB] 553 │ │ │ │ │ │ │ │ │ │ │ │ └─ Tri{dtype='float64'} [id RC] 504 │ │ │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id PW] 445 │ │ │ │ │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id RD] 453 │ │ │ │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{start:} [id PX] 432 │ │ │ │ │ │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id RE] 48 │ │ │ │ │ │ │ │ │ │ │ │ │ └─ 1 [id RF] │ │ │ │ │ │ │ │ │ │ │ │ └─ 0 [id E] │ │ │ │ │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id QA] 599 │ │ │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ │ │ └─ 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 RI] 696 │ │ │ │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id RJ] 654 │ │ │ │ │ │ │ │ │ │ ├─ Switch [id OQ] 173 │ │ │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ │ │ └─ Composite{((i0 * i1) - i2)} [id RK] 612 │ │ │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id OZ] 418 │ │ │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id RL] 552 │ │ │ │ │ │ │ │ │ │ │ └─ Tri{dtype='float64'} [id RM] 503 │ │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id PW] 445 │ │ │ │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id RN] 452 │ │ │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{start:} [id PX] 432 │ │ │ │ │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id RO] 47 │ │ │ │ │ │ │ │ │ │ │ │ └─ 1 [id RP] │ │ │ │ │ │ │ │ │ │ │ └─ 0 [id E] │ │ │ │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id QA] 599 │ │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id NI] 12 │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id NK] 13 │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ ├─ 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 │ │ │ │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ │ │ │ └─ 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 │ │ │ │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ │ │ │ └─ Composite{((i0 * i1) - i2)} [id RX] 611 │ │ │ │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id OZ] 418 │ │ │ │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id RY] 551 │ │ │ │ │ │ │ │ │ │ │ │ └─ Tri{dtype='float64'} [id RZ] 502 │ │ │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id PW] 445 │ │ │ │ │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id SA] 451 │ │ │ │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{start:} [id PX] 432 │ │ │ │ │ │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id SB] 46 │ │ │ │ │ │ │ │ │ │ │ │ │ └─ 1 [id SC] │ │ │ │ │ │ │ │ │ │ │ │ └─ 0 [id E] │ │ │ │ │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id QA] 599 │ │ │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ │ │ └─ 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 SF] 694 │ │ │ │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id SG] 652 │ │ │ │ │ │ │ │ │ │ ├─ Switch [id OQ] 173 │ │ │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ │ │ └─ Composite{((i0 * i1) - i2)} [id SH] 610 │ │ │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id OZ] 418 │ │ │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id SI] 550 │ │ │ │ │ │ │ │ │ │ │ └─ Tri{dtype='float64'} [id SJ] 501 │ │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id PW] 445 │ │ │ │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id SK] 450 │ │ │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{start:} [id PX] 432 │ │ │ │ │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id SL] 45 │ │ │ │ │ │ │ │ │ │ │ │ └─ 1 [id SM] │ │ │ │ │ │ │ │ │ │ │ └─ 0 [id E] │ │ │ │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id QA] 599 │ │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id NI] 12 │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id NN] 11 │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ └─ 0 [id E] │ │ │ │ │ │ ├─ AdvancedSetSubtensor [id SN] 882 │ │ │ │ │ │ │ ├─ Alloc [id SO] 861 │ │ │ │ │ │ │ │ ├─ 0.0 [id W] │ │ │ │ │ │ │ │ ├─ 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 │ │ │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ │ │ └─ 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 │ │ │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ │ │ └─ Composite{((i0 * i1) - i2)} [id SV] 609 │ │ │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id OZ] 418 │ │ │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id SW] 549 │ │ │ │ │ │ │ │ │ │ │ └─ Tri{dtype='float64'} [id SX] 500 │ │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id PW] 445 │ │ │ │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id SY] 449 │ │ │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{start:} [id PX] 432 │ │ │ │ │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id SZ] 44 │ │ │ │ │ │ │ │ │ │ │ │ └─ 1 [id TA] │ │ │ │ │ │ │ │ │ │ │ └─ 0 [id E] │ │ │ │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id QA] 599 │ │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id NX] 2 │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ └─ 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 │ │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ │ └─ 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 │ │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ │ └─ Composite{((i0 * i1) - i2)} [id TH] 608 │ │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id OZ] 418 │ │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id TI] 548 │ │ │ │ │ │ │ │ │ │ └─ Tri{dtype='float64'} [id TJ] 499 │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id PW] 445 │ │ │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id TK] 448 │ │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{start:} [id PX] 432 │ │ │ │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id TL] 43 │ │ │ │ │ │ │ │ │ │ │ └─ 1 [id TM] │ │ │ │ │ │ │ │ │ │ └─ 0 [id E] │ │ │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id QA] 599 │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id NX] 2 │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ ├─ 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 │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ └─ 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 │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ └─ Composite{((i0 * i1) - i2)} [id TR] 607 │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id OZ] 418 │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id TS] 547 │ │ │ │ │ │ │ │ │ └─ Tri{dtype='float64'} [id TT] 498 │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id PW] 445 │ │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id TU] 447 │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{start:} [id PX] 432 │ │ │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id TV] 42 │ │ │ │ │ │ │ │ │ │ └─ 1 [id TW] │ │ │ │ │ │ │ │ │ └─ 0 [id E] │ │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id QA] 599 │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ ├─ ARange{dtype='int64'} [id TX] 860 │ │ │ │ │ │ │ │ ├─ 0 [id E] │ │ │ │ │ │ │ │ ├─ 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 │ │ │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ │ │ └─ 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 │ │ │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ │ │ └─ Composite{((i0 * i1) - i2)} [id UE] 606 │ │ │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id OZ] 418 │ │ │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id UF] 546 │ │ │ │ │ │ │ │ │ │ │ └─ Tri{dtype='float64'} [id UG] 497 │ │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id PW] 445 │ │ │ │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id UH] 446 │ │ │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{start:} [id PX] 432 │ │ │ │ │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id UI] 41 │ │ │ │ │ │ │ │ │ │ │ │ └─ 1 [id UJ] │ │ │ │ │ │ │ │ │ │ │ └─ 0 [id E] │ │ │ │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id QA] 599 │ │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id NX] 2 │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ └─ 1 [id BY] │ │ │ │ │ │ │ └─ ARange{dtype='int64'} [id UK] 859 │ │ │ │ │ │ │ ├─ 0 [id E] │ │ │ │ │ │ │ ├─ 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 │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id NX] 2 │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ └─ 1 [id BY] │ │ │ │ │ │ ├─ Invert [id OR] 155 │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ └─ [[nan]] [id OM] │ │ │ │ │ ├─ [0 1] [id CN] │ │ │ │ │ └─ [0 1] [id CN] │ │ │ │ └─ 1 [id DO] │ │ │ ├─ 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 │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id UW] 93 │ │ │ │ │ │ │ │ └─ Cholesky{lower=True, destructive=False, on_error='nan'} [id EF] 73 │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ └─ [[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 │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ └─ Composite{((i0 * i1) - i2)} [id UZ] 615 │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id VA] 421 │ │ │ │ │ │ │ └─ dot [id VB] 408 │ │ │ │ │ │ │ ├─ Switch [id UR] 171 │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ └─ 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 │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ ├─ SetSubtensor{:stop, start:stop} [id VO] 357 │ │ │ │ │ │ │ │ ├─ Alloc [id MU] 9 │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ ├─ AdvancedSetSubtensor [id VP] 345 │ │ │ │ │ │ │ │ │ ├─ Alloc [id CK] 1 │ │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ │ ├─ 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 │ │ │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ │ │ └─ ExtractDiag{offset=0, axis1=0, axis2=1, view=False} [id EE] 90 │ │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ │ ├─ [0 1] [id CN] │ │ │ │ │ │ │ │ │ └─ [0 1] [id CN] │ │ │ │ │ │ │ │ ├─ 2 [id NB] │ │ │ │ │ │ │ │ ├─ 0 [id NC] │ │ │ │ │ │ │ │ └─ 2 [id NB] │ │ │ │ │ │ │ └─ Invert [id US] 152 │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ ├─ 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 │ │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id NI] 12 │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id NK] 13 │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ ├─ Subtensor{i} [id WA] 456 │ │ │ │ │ │ │ │ ├─ Subtensor{start:} [id VY] 434 │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id NN] 11 │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ └─ 0 [id E] │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id WB] 600 │ │ │ │ │ │ └─ AdvancedSetSubtensor [id WC] 594 │ │ │ │ │ │ ├─ Alloc [id WD] 556 │ │ │ │ │ │ │ ├─ 0.0 [id W] │ │ │ │ │ │ │ ├─ 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 │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id NX] 2 │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ └─ Subtensor{i} [id WE] 506 │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ ├─ Mul [id WG] 433 │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ ├─ ARange{dtype='int64'} [id WI] 555 │ │ │ │ │ │ │ ├─ 0 [id E] │ │ │ │ │ │ │ ├─ Subtensor{i} [id WE] 506 │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ └─ 1 [id BY] │ │ │ │ │ │ └─ ARange{dtype='int64'} [id WI] 555 │ │ │ │ │ │ └─ ··· │ │ │ │ │ ├─ 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 WL] 708 │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id WM] 666 │ │ │ │ │ │ ├─ Switch [id UR] 171 │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ └─ Composite{((i0 * i1) - i2)} [id WN] 624 │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id VA] 421 │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id WO] 566 │ │ │ │ │ │ │ └─ Tri{dtype='float64'} [id WP] 516 │ │ │ │ │ │ │ ├─ Subtensor{i} [id VX] 457 │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ ├─ Subtensor{i} [id WQ] 466 │ │ │ │ │ │ │ │ ├─ Subtensor{start:} [id VY] 434 │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id WR] 40 │ │ │ │ │ │ │ │ └─ 1 [id WS] │ │ │ │ │ │ │ └─ 0 [id E] │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id WB] 600 │ │ │ │ │ │ └─ ··· │ │ │ │ │ ├─ 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 │ │ │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ │ │ └─ 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 │ │ │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ │ │ └─ Composite{((i0 * i1) - i2)} [id XB] 623 │ │ │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id VA] 421 │ │ │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id XC] 565 │ │ │ │ │ │ │ │ │ │ │ └─ Tri{dtype='float64'} [id XD] 515 │ │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id VX] 457 │ │ │ │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id XE] 465 │ │ │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{start:} [id VY] 434 │ │ │ │ │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id XF] 39 │ │ │ │ │ │ │ │ │ │ │ │ └─ 1 [id XG] │ │ │ │ │ │ │ │ │ │ │ └─ 0 [id E] │ │ │ │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id WB] 600 │ │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ │ └─ 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 XJ] 706 │ │ │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id XK] 664 │ │ │ │ │ │ │ │ │ ├─ Switch [id UR] 171 │ │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ │ └─ Composite{((i0 * i1) - i2)} [id XL] 622 │ │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id VA] 421 │ │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id XM] 564 │ │ │ │ │ │ │ │ │ │ └─ Tri{dtype='float64'} [id XN] 514 │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id VX] 457 │ │ │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id XO] 464 │ │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{start:} [id VY] 434 │ │ │ │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id XP] 38 │ │ │ │ │ │ │ │ │ │ │ └─ 1 [id XQ] │ │ │ │ │ │ │ │ │ │ └─ 0 [id E] │ │ │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id WB] 600 │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id NI] 12 │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ └─ ScalarFromTensor [id NK] 13 │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ ├─ 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 │ │ │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ │ │ └─ 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 │ │ │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ │ │ └─ Composite{((i0 * i1) - i2)} [id XY] 621 │ │ │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id VA] 421 │ │ │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id XZ] 563 │ │ │ │ │ │ │ │ │ │ │ └─ Tri{dtype='float64'} [id YA] 513 │ │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id VX] 457 │ │ │ │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id YB] 463 │ │ │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{start:} [id VY] 434 │ │ │ │ │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id YC] 37 │ │ │ │ │ │ │ │ │ │ │ │ └─ 1 [id YD] │ │ │ │ │ │ │ │ │ │ │ └─ 0 [id E] │ │ │ │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id WB] 600 │ │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ │ └─ 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 YG] 704 │ │ │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id YH] 662 │ │ │ │ │ │ │ │ │ ├─ Switch [id UR] 171 │ │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ │ └─ Composite{((i0 * i1) - i2)} [id YI] 620 │ │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id VA] 421 │ │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id YJ] 562 │ │ │ │ │ │ │ │ │ │ └─ Tri{dtype='float64'} [id YK] 512 │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id VX] 457 │ │ │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id YL] 462 │ │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{start:} [id VY] 434 │ │ │ │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id YM] 36 │ │ │ │ │ │ │ │ │ │ │ └─ 1 [id YN] │ │ │ │ │ │ │ │ │ │ └─ 0 [id E] │ │ │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id WB] 600 │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id NI] 12 │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ └─ ScalarFromTensor [id NN] 11 │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ └─ 0 [id E] │ │ │ │ │ ├─ AdvancedSetSubtensor [id YO] 885 │ │ │ │ │ │ ├─ Alloc [id YP] 866 │ │ │ │ │ │ │ ├─ 0.0 [id W] │ │ │ │ │ │ │ ├─ 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 │ │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ │ └─ 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 │ │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ │ └─ Composite{((i0 * i1) - i2)} [id YW] 619 │ │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id VA] 421 │ │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id YX] 561 │ │ │ │ │ │ │ │ │ │ └─ Tri{dtype='float64'} [id YY] 511 │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id VX] 457 │ │ │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id YZ] 461 │ │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{start:} [id VY] 434 │ │ │ │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id ZA] 35 │ │ │ │ │ │ │ │ │ │ │ └─ 1 [id ZB] │ │ │ │ │ │ │ │ │ │ └─ 0 [id E] │ │ │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id WB] 600 │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id NX] 2 │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ └─ 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 │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ └─ 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 │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ └─ Composite{((i0 * i1) - i2)} [id ZI] 618 │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id VA] 421 │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id ZJ] 560 │ │ │ │ │ │ │ │ │ └─ Tri{dtype='float64'} [id ZK] 510 │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id VX] 457 │ │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id ZL] 460 │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{start:} [id VY] 434 │ │ │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id ZM] 34 │ │ │ │ │ │ │ │ │ │ └─ 1 [id ZN] │ │ │ │ │ │ │ │ │ └─ 0 [id E] │ │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id WB] 600 │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ └─ ScalarFromTensor [id NX] 2 │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ ├─ 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 │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ └─ 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 │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ └─ Composite{((i0 * i1) - i2)} [id ZS] 617 │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id VA] 421 │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id ZT] 559 │ │ │ │ │ │ │ │ └─ Tri{dtype='float64'} [id ZU] 509 │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id VX] 457 │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id ZV] 459 │ │ │ │ │ │ │ │ │ ├─ Subtensor{start:} [id VY] 434 │ │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id ZW] 33 │ │ │ │ │ │ │ │ │ └─ 1 [id ZX] │ │ │ │ │ │ │ │ └─ 0 [id E] │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id WB] 600 │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ ├─ ARange{dtype='int64'} [id ZY] 865 │ │ │ │ │ │ │ ├─ 0 [id E] │ │ │ │ │ │ │ ├─ 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 │ │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ │ └─ 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 │ │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ │ └─ Composite{((i0 * i1) - i2)} [id BAF] 616 │ │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id VA] 421 │ │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id BAG] 558 │ │ │ │ │ │ │ │ │ │ └─ Tri{dtype='float64'} [id BAH] 508 │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id VX] 457 │ │ │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id BAI] 458 │ │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{start:} [id VY] 434 │ │ │ │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id BAJ] 32 │ │ │ │ │ │ │ │ │ │ │ └─ 1 [id BAK] │ │ │ │ │ │ │ │ │ │ └─ 0 [id E] │ │ │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id WB] 600 │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id NX] 2 │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ └─ 1 [id BY] │ │ │ │ │ │ └─ ARange{dtype='int64'} [id BAL] 864 │ │ │ │ │ │ ├─ 0 [id E] │ │ │ │ │ │ ├─ 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 │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ └─ ScalarFromTensor [id NX] 2 │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ └─ 1 [id BY] │ │ │ │ │ ├─ Invert [id US] 152 │ │ │ │ │ │ └─ ··· │ │ │ │ │ └─ [[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 │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id BAX] 89 │ │ │ │ │ │ │ └─ Cholesky{lower=True, destructive=False, on_error='nan'} [id EZ] 72 │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ └─ [[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 │ │ │ │ │ │ └─ ··· │ │ │ │ │ └─ Composite{((i0 * i1) - i2)} [id BBA] 625 │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id BBB] 424 │ │ │ │ │ │ └─ dot [id BBC] 409 │ │ │ │ │ │ ├─ Switch [id BAS] 169 │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ └─ 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 │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ ├─ SetSubtensor{:stop, start:stop} [id BBP] 359 │ │ │ │ │ │ │ ├─ Alloc [id MU] 9 │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ ├─ AdvancedSetSubtensor [id BBQ] 347 │ │ │ │ │ │ │ │ ├─ Alloc [id CK] 1 │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ ├─ 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 │ │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ │ └─ ExtractDiag{offset=0, axis1=0, axis2=1, view=False} [id EY] 86 │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ ├─ [0 1] [id CN] │ │ │ │ │ │ │ │ └─ [0 1] [id CN] │ │ │ │ │ │ │ ├─ 2 [id NB] │ │ │ │ │ │ │ ├─ 0 [id NC] │ │ │ │ │ │ │ └─ 2 [id NB] │ │ │ │ │ │ └─ Invert [id BAT] 149 │ │ │ │ │ │ └─ ··· │ │ │ │ │ ├─ 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 │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id NI] 12 │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ └─ ScalarFromTensor [id NK] 13 │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ ├─ Subtensor{i} [id BCB] 468 │ │ │ │ │ │ │ ├─ Subtensor{start:} [id BBZ] 436 │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ └─ ScalarFromTensor [id NN] 11 │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ └─ 0 [id E] │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id BCC] 601 │ │ │ │ │ └─ AdvancedSetSubtensor [id BCD] 595 │ │ │ │ │ ├─ Alloc [id BCE] 568 │ │ │ │ │ │ ├─ 0.0 [id W] │ │ │ │ │ │ ├─ 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 │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ └─ ScalarFromTensor [id NX] 2 │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ └─ Subtensor{i} [id BCF] 517 │ │ │ │ │ │ └─ ··· │ │ │ │ │ ├─ Mul [id BCH] 435 │ │ │ │ │ │ └─ ··· │ │ │ │ │ ├─ ARange{dtype='int64'} [id BCJ] 567 │ │ │ │ │ │ ├─ 0 [id E] │ │ │ │ │ │ ├─ Subtensor{i} [id BCF] 517 │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ └─ 1 [id BY] │ │ │ │ │ └─ ARange{dtype='int64'} [id BCJ] 567 │ │ │ │ │ └─ ··· │ │ │ │ ├─ 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 BCM] 718 │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id BCN] 676 │ │ │ │ │ ├─ Switch [id BAS] 169 │ │ │ │ │ │ └─ ··· │ │ │ │ │ └─ Composite{((i0 * i1) - i2)} [id BCO] 634 │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id BBB] 424 │ │ │ │ │ │ └─ ··· │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id BCP] 578 │ │ │ │ │ │ └─ Tri{dtype='float64'} [id BCQ] 527 │ │ │ │ │ │ ├─ Subtensor{i} [id BBY] 469 │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ ├─ Subtensor{i} [id BCR] 478 │ │ │ │ │ │ │ ├─ Subtensor{start:} [id BBZ] 436 │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ └─ ScalarFromTensor [id BCS] 31 │ │ │ │ │ │ │ └─ 1 [id BCT] │ │ │ │ │ │ └─ 0 [id E] │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id BCC] 601 │ │ │ │ │ └─ ··· │ │ │ │ ├─ 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 │ │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ │ └─ 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 │ │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ │ └─ Composite{((i0 * i1) - i2)} [id BDC] 633 │ │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id BBB] 424 │ │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id BDD] 577 │ │ │ │ │ │ │ │ │ │ └─ Tri{dtype='float64'} [id BDE] 526 │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id BBY] 469 │ │ │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id BDF] 477 │ │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{start:} [id BBZ] 436 │ │ │ │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id BDG] 30 │ │ │ │ │ │ │ │ │ │ │ └─ 1 [id BDH] │ │ │ │ │ │ │ │ │ │ └─ 0 [id E] │ │ │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id BCC] 601 │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ └─ 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 BDK] 716 │ │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id BDL] 674 │ │ │ │ │ │ │ │ ├─ Switch [id BAS] 169 │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ └─ Composite{((i0 * i1) - i2)} [id BDM] 632 │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id BBB] 424 │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id BDN] 576 │ │ │ │ │ │ │ │ │ └─ Tri{dtype='float64'} [id BDO] 525 │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id BBY] 469 │ │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id BDP] 476 │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{start:} [id BBZ] 436 │ │ │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id BDQ] 29 │ │ │ │ │ │ │ │ │ │ └─ 1 [id BDR] │ │ │ │ │ │ │ │ │ └─ 0 [id E] │ │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id BCC] 601 │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ └─ ScalarFromTensor [id NI] 12 │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ └─ ScalarFromTensor [id NK] 13 │ │ │ │ │ │ └─ ··· │ │ │ │ │ ├─ 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 │ │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ │ └─ 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 │ │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ │ └─ Composite{((i0 * i1) - i2)} [id BDZ] 631 │ │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id BBB] 424 │ │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id BEA] 575 │ │ │ │ │ │ │ │ │ │ └─ Tri{dtype='float64'} [id BEB] 524 │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id BBY] 469 │ │ │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id BEC] 475 │ │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{start:} [id BBZ] 436 │ │ │ │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id BED] 28 │ │ │ │ │ │ │ │ │ │ │ └─ 1 [id BEE] │ │ │ │ │ │ │ │ │ │ └─ 0 [id E] │ │ │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id BCC] 601 │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ └─ 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 BEH] 714 │ │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id BEI] 672 │ │ │ │ │ │ │ │ ├─ Switch [id BAS] 169 │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ └─ Composite{((i0 * i1) - i2)} [id BEJ] 630 │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id BBB] 424 │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id BEK] 574 │ │ │ │ │ │ │ │ │ └─ Tri{dtype='float64'} [id BEL] 523 │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id BBY] 469 │ │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id BEM] 474 │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{start:} [id BBZ] 436 │ │ │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id BEN] 27 │ │ │ │ │ │ │ │ │ │ └─ 1 [id BEO] │ │ │ │ │ │ │ │ │ └─ 0 [id E] │ │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id BCC] 601 │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ └─ ScalarFromTensor [id NI] 12 │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ └─ ScalarFromTensor [id NN] 11 │ │ │ │ │ │ └─ ··· │ │ │ │ │ └─ 0 [id E] │ │ │ │ ├─ AdvancedSetSubtensor [id BEP] 888 │ │ │ │ │ ├─ Alloc [id BEQ] 871 │ │ │ │ │ │ ├─ 0.0 [id W] │ │ │ │ │ │ ├─ 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 │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ └─ 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 │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ └─ Composite{((i0 * i1) - i2)} [id BEX] 629 │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id BBB] 424 │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id BEY] 573 │ │ │ │ │ │ │ │ │ └─ Tri{dtype='float64'} [id BEZ] 522 │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id BBY] 469 │ │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id BFA] 473 │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{start:} [id BBZ] 436 │ │ │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id BFB] 26 │ │ │ │ │ │ │ │ │ │ └─ 1 [id BFC] │ │ │ │ │ │ │ │ │ └─ 0 [id E] │ │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id BCC] 601 │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ └─ ScalarFromTensor [id NX] 2 │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ └─ 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 │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ └─ 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 │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ └─ Composite{((i0 * i1) - i2)} [id BFJ] 628 │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id BBB] 424 │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id BFK] 572 │ │ │ │ │ │ │ │ └─ Tri{dtype='float64'} [id BFL] 521 │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id BBY] 469 │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id BFM] 472 │ │ │ │ │ │ │ │ │ ├─ Subtensor{start:} [id BBZ] 436 │ │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id BFN] 25 │ │ │ │ │ │ │ │ │ └─ 1 [id BFO] │ │ │ │ │ │ │ │ └─ 0 [id E] │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id BCC] 601 │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ └─ ScalarFromTensor [id NX] 2 │ │ │ │ │ │ └─ ··· │ │ │ │ │ ├─ 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 │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ └─ 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 │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ └─ Composite{((i0 * i1) - i2)} [id BFT] 627 │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id BBB] 424 │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id BFU] 571 │ │ │ │ │ │ │ └─ Tri{dtype='float64'} [id BFV] 520 │ │ │ │ │ │ │ ├─ Subtensor{i} [id BBY] 469 │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ ├─ Subtensor{i} [id BFW] 471 │ │ │ │ │ │ │ │ ├─ Subtensor{start:} [id BBZ] 436 │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id BFX] 24 │ │ │ │ │ │ │ │ └─ 1 [id BFY] │ │ │ │ │ │ │ └─ 0 [id E] │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id BCC] 601 │ │ │ │ │ │ └─ ··· │ │ │ │ │ ├─ ARange{dtype='int64'} [id BFZ] 870 │ │ │ │ │ │ ├─ 0 [id E] │ │ │ │ │ │ ├─ 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 │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ └─ 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 │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ └─ Composite{((i0 * i1) - i2)} [id BGG] 626 │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id BBB] 424 │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id BGH] 570 │ │ │ │ │ │ │ │ │ └─ Tri{dtype='float64'} [id BGI] 519 │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id BBY] 469 │ │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id BGJ] 470 │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{start:} [id BBZ] 436 │ │ │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id BGK] 23 │ │ │ │ │ │ │ │ │ │ └─ 1 [id BGL] │ │ │ │ │ │ │ │ │ └─ 0 [id E] │ │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id BCC] 601 │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ └─ ScalarFromTensor [id NX] 2 │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ └─ 1 [id BY] │ │ │ │ │ └─ ARange{dtype='int64'} [id BGM] 869 │ │ │ │ │ ├─ 0 [id E] │ │ │ │ │ ├─ 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 │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ └─ ScalarFromTensor [id NX] 2 │ │ │ │ │ │ └─ ··· │ │ │ │ │ └─ 1 [id BY] │ │ │ │ ├─ Invert [id BAT] 149 │ │ │ │ │ └─ ··· │ │ │ │ └─ [[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 │ │ │ │ │ │ └─ ··· │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id BGY] 81 │ │ │ │ │ │ └─ Cholesky{lower=True, destructive=False, on_error='nan'} [id FT] 70 │ │ │ │ │ │ └─ ··· │ │ │ │ │ └─ [[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 │ │ │ │ │ └─ ··· │ │ │ │ └─ Composite{((i0 * i1) - i2)} [id BHB] 635 │ │ │ │ ├─ Transpose{axes=[1, 0]} [id BHC] 427 │ │ │ │ │ └─ dot [id BHD] 410 │ │ │ │ │ ├─ Switch [id BGT] 164 │ │ │ │ │ │ └─ ··· │ │ │ │ │ └─ 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 │ │ │ │ │ │ └─ ··· │ │ │ │ │ ├─ SetSubtensor{:stop, start:stop} [id BHQ] 363 │ │ │ │ │ │ ├─ Alloc [id MU] 9 │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ ├─ AdvancedSetSubtensor [id BHR] 351 │ │ │ │ │ │ │ ├─ Alloc [id CK] 1 │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ ├─ 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 │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ └─ ExtractDiag{offset=0, axis1=0, axis2=1, view=False} [id FS] 79 │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ ├─ [0 1] [id CN] │ │ │ │ │ │ │ └─ [0 1] [id CN] │ │ │ │ │ │ ├─ 2 [id NB] │ │ │ │ │ │ ├─ 0 [id NC] │ │ │ │ │ │ └─ 2 [id NB] │ │ │ │ │ └─ Invert [id BGU] 141 │ │ │ │ │ └─ ··· │ │ │ │ ├─ 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 │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ └─ ScalarFromTensor [id NI] 12 │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ └─ ScalarFromTensor [id NK] 13 │ │ │ │ │ │ └─ ··· │ │ │ │ │ ├─ Subtensor{i} [id BIC] 480 │ │ │ │ │ │ ├─ Subtensor{start:} [id BIA] 438 │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ └─ ScalarFromTensor [id NN] 11 │ │ │ │ │ │ └─ ··· │ │ │ │ │ └─ 0 [id E] │ │ │ │ └─ Transpose{axes=[1, 0]} [id BID] 602 │ │ │ │ └─ AdvancedSetSubtensor [id BIE] 596 │ │ │ │ ├─ Alloc [id BIF] 580 │ │ │ │ │ ├─ 0.0 [id W] │ │ │ │ │ ├─ 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 │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ └─ ScalarFromTensor [id NX] 2 │ │ │ │ │ │ └─ ··· │ │ │ │ │ └─ Subtensor{i} [id BIG] 528 │ │ │ │ │ └─ ··· │ │ │ │ ├─ Mul [id BII] 437 │ │ │ │ │ └─ ··· │ │ │ │ ├─ ARange{dtype='int64'} [id BIK] 579 │ │ │ │ │ ├─ 0 [id E] │ │ │ │ │ ├─ Subtensor{i} [id BIG] 528 │ │ │ │ │ │ └─ ··· │ │ │ │ │ └─ 1 [id BY] │ │ │ │ └─ ARange{dtype='int64'} [id BIK] 579 │ │ │ │ └─ ··· │ │ │ ├─ 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 BIN] 728 │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id BIO] 686 │ │ │ │ ├─ Switch [id BGT] 164 │ │ │ │ │ └─ ··· │ │ │ │ └─ Composite{((i0 * i1) - i2)} [id BIP] 644 │ │ │ │ ├─ Transpose{axes=[1, 0]} [id BHC] 427 │ │ │ │ │ └─ ··· │ │ │ │ ├─ Transpose{axes=[1, 0]} [id BIQ] 590 │ │ │ │ │ └─ Tri{dtype='float64'} [id BIR] 538 │ │ │ │ │ ├─ Subtensor{i} [id BHZ] 481 │ │ │ │ │ │ └─ ··· │ │ │ │ │ ├─ Subtensor{i} [id BIS] 490 │ │ │ │ │ │ ├─ Subtensor{start:} [id BIA] 438 │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ └─ ScalarFromTensor [id BIT] 22 │ │ │ │ │ │ └─ 1 [id BIU] │ │ │ │ │ └─ 0 [id E] │ │ │ │ └─ Transpose{axes=[1, 0]} [id BID] 602 │ │ │ │ └─ ··· │ │ │ ├─ 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 │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ └─ 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 │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ └─ Composite{((i0 * i1) - i2)} [id BJD] 643 │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id BHC] 427 │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id BJE] 589 │ │ │ │ │ │ │ │ │ └─ Tri{dtype='float64'} [id BJF] 537 │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id BHZ] 481 │ │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id BJG] 489 │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{start:} [id BIA] 438 │ │ │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id BJH] 21 │ │ │ │ │ │ │ │ │ │ └─ 1 [id BJI] │ │ │ │ │ │ │ │ │ └─ 0 [id E] │ │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id BID] 602 │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ └─ 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 BJL] 726 │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id BJM] 684 │ │ │ │ │ │ │ ├─ Switch [id BGT] 164 │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ └─ Composite{((i0 * i1) - i2)} [id BJN] 642 │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id BHC] 427 │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id BJO] 588 │ │ │ │ │ │ │ │ └─ Tri{dtype='float64'} [id BJP] 536 │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id BHZ] 481 │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id BJQ] 488 │ │ │ │ │ │ │ │ │ ├─ Subtensor{start:} [id BIA] 438 │ │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id BJR] 20 │ │ │ │ │ │ │ │ │ └─ 1 [id BJS] │ │ │ │ │ │ │ │ └─ 0 [id E] │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id BID] 602 │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ └─ ScalarFromTensor [id NI] 12 │ │ │ │ │ │ └─ ··· │ │ │ │ │ └─ ScalarFromTensor [id NK] 13 │ │ │ │ │ └─ ··· │ │ │ │ ├─ 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 │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ └─ 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 │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ └─ Composite{((i0 * i1) - i2)} [id BKA] 641 │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id BHC] 427 │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id BKB] 587 │ │ │ │ │ │ │ │ │ └─ Tri{dtype='float64'} [id BKC] 535 │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id BHZ] 481 │ │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id BKD] 487 │ │ │ │ │ │ │ │ │ │ ├─ Subtensor{start:} [id BIA] 438 │ │ │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id BKE] 19 │ │ │ │ │ │ │ │ │ │ └─ 1 [id BKF] │ │ │ │ │ │ │ │ │ └─ 0 [id E] │ │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id BID] 602 │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ └─ 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 BKI] 724 │ │ │ │ │ │ │ └─ SolveTriangular{trans=0, unit_diagonal=False, lower=False, check_finite=True, b_ndim=2} [id BKJ] 682 │ │ │ │ │ │ │ ├─ Switch [id BGT] 164 │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ └─ Composite{((i0 * i1) - i2)} [id BKK] 640 │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id BHC] 427 │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id BKL] 586 │ │ │ │ │ │ │ │ └─ Tri{dtype='float64'} [id BKM] 534 │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id BHZ] 481 │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id BKN] 486 │ │ │ │ │ │ │ │ │ ├─ Subtensor{start:} [id BIA] 438 │ │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id BKO] 18 │ │ │ │ │ │ │ │ │ └─ 1 [id BKP] │ │ │ │ │ │ │ │ └─ 0 [id E] │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id BID] 602 │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ └─ ScalarFromTensor [id NI] 12 │ │ │ │ │ │ └─ ··· │ │ │ │ │ └─ ScalarFromTensor [id NN] 11 │ │ │ │ │ └─ ··· │ │ │ │ └─ 0 [id E] │ │ │ ├─ AdvancedSetSubtensor [id BKQ] 891 │ │ │ │ ├─ Alloc [id BKR] 876 │ │ │ │ │ ├─ 0.0 [id W] │ │ │ │ │ ├─ 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 │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ └─ 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 │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ └─ Composite{((i0 * i1) - i2)} [id BKY] 639 │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id BHC] 427 │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id BKZ] 585 │ │ │ │ │ │ │ │ └─ Tri{dtype='float64'} [id BLA] 533 │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id BHZ] 481 │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id BLB] 485 │ │ │ │ │ │ │ │ │ ├─ Subtensor{start:} [id BIA] 438 │ │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id BLC] 17 │ │ │ │ │ │ │ │ │ └─ 1 [id BLD] │ │ │ │ │ │ │ │ └─ 0 [id E] │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id BID] 602 │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ └─ ScalarFromTensor [id NX] 2 │ │ │ │ │ │ └─ ··· │ │ │ │ │ └─ 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 │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ └─ 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 │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ └─ Composite{((i0 * i1) - i2)} [id BLK] 638 │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id BHC] 427 │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id BLL] 584 │ │ │ │ │ │ │ └─ Tri{dtype='float64'} [id BLM] 532 │ │ │ │ │ │ │ ├─ Subtensor{i} [id BHZ] 481 │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ ├─ Subtensor{i} [id BLN] 484 │ │ │ │ │ │ │ │ ├─ Subtensor{start:} [id BIA] 438 │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id BLO] 16 │ │ │ │ │ │ │ │ └─ 1 [id BLP] │ │ │ │ │ │ │ └─ 0 [id E] │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id BID] 602 │ │ │ │ │ │ └─ ··· │ │ │ │ │ └─ ScalarFromTensor [id NX] 2 │ │ │ │ │ └─ ··· │ │ │ │ ├─ 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 │ │ │ │ │ │ └─ ··· │ │ │ │ │ └─ 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 │ │ │ │ │ │ └─ ··· │ │ │ │ │ └─ Composite{((i0 * i1) - i2)} [id BLU] 637 │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id BHC] 427 │ │ │ │ │ │ └─ ··· │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id BLV] 583 │ │ │ │ │ │ └─ Tri{dtype='float64'} [id BLW] 531 │ │ │ │ │ │ ├─ Subtensor{i} [id BHZ] 481 │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ ├─ Subtensor{i} [id BLX] 483 │ │ │ │ │ │ │ ├─ Subtensor{start:} [id BIA] 438 │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ └─ ScalarFromTensor [id BLY] 15 │ │ │ │ │ │ │ └─ 1 [id BLZ] │ │ │ │ │ │ └─ 0 [id E] │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id BID] 602 │ │ │ │ │ └─ ··· │ │ │ │ ├─ ARange{dtype='int64'} [id BMA] 875 │ │ │ │ │ ├─ 0 [id E] │ │ │ │ │ ├─ 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 │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ └─ 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 │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ └─ Composite{((i0 * i1) - i2)} [id BMH] 636 │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id BHC] 427 │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ ├─ Transpose{axes=[1, 0]} [id BMI] 582 │ │ │ │ │ │ │ │ └─ Tri{dtype='float64'} [id BMJ] 530 │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id BHZ] 481 │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ ├─ Subtensor{i} [id BMK] 482 │ │ │ │ │ │ │ │ │ ├─ Subtensor{start:} [id BIA] 438 │ │ │ │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ │ │ └─ ScalarFromTensor [id BML] 14 │ │ │ │ │ │ │ │ │ └─ 1 [id BMM] │ │ │ │ │ │ │ │ └─ 0 [id E] │ │ │ │ │ │ │ └─ Transpose{axes=[1, 0]} [id BID] 602 │ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ └─ ScalarFromTensor [id NX] 2 │ │ │ │ │ │ └─ ··· │ │ │ │ │ └─ 1 [id BY] │ │ │ │ └─ ARange{dtype='int64'} [id BMN] 874 │ │ │ │ ├─ 0 [id E] │ │ │ │ ├─ 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 │ │ │ │ │ │ └─ ··· │ │ │ │ │ └─ ScalarFromTensor [id NX] 2 │ │ │ │ │ └─ ··· │ │ │ │ └─ 1 [id BY] │ │ │ ├─ Invert [id BGU] 141 │ │ │ │ └─ ··· │ │ │ └─ [[nan]] [id OM] │ │ ├─ [0 1] [id CN] │ │ └─ [0 1] [id CN] │ └─ 4 [id FW] └─ Composite{...}.0 [id BJ] 0 └─ ··· Inner graphs: Composite{...} [id A] ← 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' └─ ··· Composite{...} [id C] ← Switch [id BNP] 'o0' ├─ i2 [id BNQ] ├─ i3 [id BNR] └─ exp [id BNS] 't8' └─ sub [id BNT] ├─ i0 [id BNU] └─ i1 [id BNV] ← mul [id BNW] 'o1' ├─ -5.0 [id BNX] └─ exp [id BNS] 't8' └─ ··· ← mul [id BNY] 'o2' ├─ 5.0 [id BNZ] └─ exp [id BNS] 't8' └─ ··· Composite{((-0.5 * sqr(i0)) - 0.9189385332046727)} [id R] ← sub [id BOA] 'o0' ├─ mul [id BOB] │ ├─ -0.5 [id BOC] │ └─ sqr [id BOD] │ └─ i0 [id BOE] └─ 0.9189385332046727 [id BOF] Composite{((-0.5 * sqr(i0)) - 0.9189385332046727)} [id BG] ← sub [id BOG] 'o0' ├─ mul [id BOH] │ ├─ -0.5 [id BOI] │ └─ sqr [id BOJ] │ └─ i0 [id BOK] └─ 0.9189385332046727 [id BOL] Composite{...} [id BJ] ← exp [id BOM] 'o0' └─ i0 [id BON] ← Switch [id BOO] 'o1' ├─ GE [id BOP] 't13' │ ├─ exp [id BOM] 'o0' │ │ └─ ··· │ └─ 0.0 [id BOQ] ├─ neg [id BOR] │ └─ exp [id BOM] 'o0' │ └─ ··· └─ 0 [id BOS] ← add [id BOT] 'o2' ├─ Switch [id BOU] │ ├─ GE [id BOP] 't13' │ │ └─ ··· │ ├─ sub [id BOV] │ │ ├─ mul [id BOW] │ │ │ ├─ -0.5 [id BOX] │ │ │ └─ sqr [id BOY] │ │ │ └─ exp [id BOM] 'o0' │ │ │ └─ ··· │ │ └─ 0.22579135264472738 [id BOZ] │ └─ -inf [id BPA] └─ i0 [id BON] Composite{switch(i2, (i1 + log(i0)), -inf)} [id BM] ← Switch [id BPB] 'o0' ├─ i2 [id BPC] ├─ add [id BPD] │ ├─ i1 [id BPE] │ └─ log [id BPF] │ └─ i0 [id BPG] └─ -inf [id BPH] Composite{...} [id BO] ← 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' └─ ··· ← neg [id BPP] 'o2' └─ exp [id BPL] 'o1' └─ ··· Composite{switch(i2, ((-1.8378770664093453 - (0.5 * i0)) - i1), -inf)} [id BZ] ← 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] CAReduce{Composite{(i0 + sqr(i1))}, axis=1} [id CB] ← add [id BQA] 'o0' ├─ i0 [id BQB] └─ sqr [id BQC] └─ i1 [id BQD] CAReduce{Composite{(i0 + log(i1))}, axes=None} [id DA] ← 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' └─ ··· 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{switch(i2, 0, (i0 / i1))} [id IA] ← Switch [id BVE] 'o0' ├─ 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] Composite{(1.0 + (i0 * i1))} [id LA] ← add [id BWO] 'o0' ├─ 1.0 [id BWP] └─ mul [id BWQ] ├─ i0 [id BWR] └─ i1 [id BWS] Composite{(1.0 + (i0 * i1))} [id LI] ← add [id BWT] 'o0' ├─ 1.0 [id BWU] └─ mul [id BWV] ├─ i0 [id BWW] └─ i1 [id BWX] Composite{switch(i3, ((i0 * i1) - i2), i4)} [id LP] ← 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] Composite{((i0 * i1) - i2)} [id OY] ← sub [id BYF] 'o0' ├─ mul [id BYG] │ ├─ i0 [id BYH] │ └─ i1 [id BYI] └─ i2 [id BYJ] 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 QM] ← sub [id BYU] 'o0' ├─ mul [id BYV] │ ├─ i0 [id BYW] │ └─ i1 [id BYX] └─ i2 [id BYY] Composite{((i0 * i1) - i2)} [id RA] ← sub [id BYZ] 'o0' ├─ mul [id BZA] │ ├─ i0 [id BZB] │ └─ i1 [id BZC] └─ i2 [id BZD] Composite{((i0 * i1) - i2)} [id RK] ← sub [id BZE] 'o0' ├─ mul [id BZF] │ ├─ i0 [id BZG] │ └─ i1 [id BZH] └─ i2 [id BZI] Composite{((i0 * i1) - i2)} [id RX] ← sub [id BZJ] 'o0' ├─ mul [id BZK] │ ├─ i0 [id BZL] │ └─ i1 [id BZM] └─ i2 [id BZN] Composite{((i0 * i1) - i2)} [id SH] ← sub [id BZO] 'o0' ├─ mul [id BZP] │ ├─ i0 [id BZQ] │ └─ i1 [id BZR] └─ i2 [id BZS] 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{((i0 * i1) - i2)} [id TR] ← sub [id CAD] 'o0' ├─ mul [id CAE] │ ├─ i0 [id CAF] │ └─ i1 [id CAG] └─ i2 [id CAH] 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{((i0 * i1) - i2)} [id BIP] ← 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] Composite{((i0 * i1) - i2)} [id BKA] ← 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]