fn4

fn4.py

def inner_fn(index):
    i0, i1 = index
    tmp0 = ops.index_expr(i1, dtype=torch.int64)
    tmp1 = ops.to_dtype(tmp0, torch.float32, src_dtype=torch.int64)
    tmp2 = ops.constant(225.0, torch.float32)
    tmp3 = tmp1 < tmp2
    tmp4 = ops.index_expr(i1, dtype=torch.int64)
    tmp5 = ops.to_dtype(tmp4, torch.float32, src_dtype=torch.int64)
    tmp6 = ops.constant(0.0066815144766146995, torch.float32)
    tmp7 = tmp5 * tmp6
    tmp8 = ops.constant(-2.25, torch.float32)
    tmp9 = tmp7 + tmp8
    tmp10 = ops.constant(0.75, torch.float32)
    tmp11 = ops.constant(449, torch.int64)
    tmp12 = ops.index_expr(i1, dtype=torch.int64)
    tmp13 = tmp11 - tmp12
    tmp14 = ops.to_dtype(tmp13, torch.float32, src_dtype=torch.int64)
    tmp15 = ops.constant(0.0066815144766146995, torch.float32)
    tmp16 = tmp14 * tmp15
    tmp17 = tmp10 - tmp16
    tmp18 = ops.where(tmp3, tmp9, tmp17)
    tmp19 = tmp18 * tmp18
    tmp20 = ops.index_expr(i0, dtype=torch.int64)
    tmp21 = ops.to_dtype(tmp20, torch.float32, src_dtype=torch.int64)
    tmp22 = ops.constant(187.5, torch.float32)
    tmp23 = tmp21 < tmp22
    tmp24 = ops.index_expr(i0, dtype=torch.int64)
    tmp25 = ops.to_dtype(tmp24, torch.float32, src_dtype=torch.int64)
    tmp26 = ops.constant(0.0066844919786096255, torch.float32)
    tmp27 = tmp25 * tmp26
    tmp28 = ops.constant(-1.25, torch.float32)
    tmp29 = tmp27 + tmp28
    tmp30 = ops.constant(1.25, torch.float32)
    tmp31 = ops.constant(374, torch.int64)
    tmp32 = ops.index_expr(i0, dtype=torch.int64)
    tmp33 = tmp31 - tmp32
    tmp34 = ops.to_dtype(tmp33, torch.float32, src_dtype=torch.int64)
    tmp35 = ops.constant(0.0066844919786096255, torch.float32)
    tmp36 = tmp34 * tmp35
    tmp37 = tmp30 - tmp36
    tmp38 = ops.where(tmp23, tmp29, tmp37)
    tmp39 = tmp38 * tmp38
    tmp40 = tmp19 - tmp39
    tmp41 = ops.index_expr(i1, dtype=torch.int64)
    tmp42 = ops.to_dtype(tmp41, torch.float32, src_dtype=torch.int64)
    tmp43 = ops.constant(225.0, torch.float32)
    tmp44 = tmp42 < tmp43
    tmp45 = ops.index_expr(i1, dtype=torch.int64)
    tmp46 = ops.to_dtype(tmp45, torch.float32, src_dtype=torch.int64)
    tmp47 = ops.constant(0.0066815144766146995, torch.float32)
    tmp48 = tmp46 * tmp47
    tmp49 = ops.constant(-2.25, torch.float32)
    tmp50 = tmp48 + tmp49
    tmp51 = ops.constant(0.75, torch.float32)
    tmp52 = ops.constant(449, torch.int64)
    tmp53 = ops.index_expr(i1, dtype=torch.int64)
    tmp54 = tmp52 - tmp53
    tmp55 = ops.to_dtype(tmp54, torch.float32, src_dtype=torch.int64)
    tmp56 = ops.constant(0.0066815144766146995, torch.float32)
    tmp57 = tmp55 * tmp56
    tmp58 = tmp51 - tmp57
    tmp59 = ops.where(tmp44, tmp50, tmp58)
    tmp60 = tmp40 + tmp59
    tmp61 = tmp60 * tmp60
    tmp62 = ops.index_expr(i1, dtype=torch.int64)
    tmp63 = ops.to_dtype(tmp62, torch.float32, src_dtype=torch.int64)
    tmp64 = ops.constant(225.0, torch.float32)
    tmp65 = tmp63 < tmp64
    tmp66 = ops.index_expr(i1, dtype=torch.int64)
    tmp67 = ops.to_dtype(tmp66, torch.float32, src_dtype=torch.int64)
    tmp68 = ops.constant(0.0066815144766146995, torch.float32)
    tmp69 = tmp67 * tmp68
    tmp70 = ops.constant(-2.25, torch.float32)
    tmp71 = tmp69 + tmp70
    tmp72 = ops.constant(0.75, torch.float32)
    tmp73 = ops.constant(449, torch.int64)
    tmp74 = ops.index_expr(i1, dtype=torch.int64)
    tmp75 = tmp73 - tmp74
    tmp76 = ops.to_dtype(tmp75, torch.float32, src_dtype=torch.int64)
    tmp77 = ops.constant(0.0066815144766146995, torch.float32)
    tmp78 = tmp76 * tmp77
    tmp79 = tmp72 - tmp78
    tmp80 = ops.where(tmp65, tmp71, tmp79)
    tmp81 = ops.constant(2, torch.float32)
    tmp82 = tmp80 * tmp81
    tmp83 = ops.index_expr(i0, dtype=torch.int64)
    tmp84 = ops.to_dtype(tmp83, torch.float32, src_dtype=torch.int64)
    tmp85 = ops.constant(187.5, torch.float32)
    tmp86 = tmp84 < tmp85
    tmp87 = ops.index_expr(i0, dtype=torch.int64)
    tmp88 = ops.to_dtype(tmp87, torch.float32, src_dtype=torch.int64)
    tmp89 = ops.constant(0.0066844919786096255, torch.float32)
    tmp90 = tmp88 * tmp89
    tmp91 = ops.constant(-1.25, torch.float32)
    tmp92 = tmp90 + tmp91
    tmp93 = ops.constant(1.25, torch.float32)
    tmp94 = ops.constant(374, torch.int64)
    tmp95 = ops.index_expr(i0, dtype=torch.int64)
    tmp96 = tmp94 - tmp95
    tmp97 = ops.to_dtype(tmp96, torch.float32, src_dtype=torch.int64)
    tmp98 = ops.constant(0.0066844919786096255, torch.float32)
    tmp99 = tmp97 * tmp98
    tmp100 = tmp93 - tmp99
    tmp101 = ops.where(tmp86, tmp92, tmp100)
    tmp102 = tmp82 * tmp101
    tmp103 = ops.index_expr(i0, dtype=torch.int64)
    tmp104 = ops.to_dtype(tmp103, torch.float32, src_dtype=torch.int64)
    tmp105 = ops.constant(187.5, torch.float32)
    tmp106 = tmp104 < tmp105
    tmp107 = ops.index_expr(i0, dtype=torch.int64)
    tmp108 = ops.to_dtype(tmp107, torch.float32, src_dtype=torch.int64)
    tmp109 = ops.constant(0.0066844919786096255, torch.float32)
    tmp110 = tmp108 * tmp109
    tmp111 = ops.constant(-1.25, torch.float32)
    tmp112 = tmp110 + tmp111
    tmp113 = ops.constant(1.25, torch.float32)
    tmp114 = ops.constant(374, torch.int64)
    tmp115 = ops.index_expr(i0, dtype=torch.int64)
    tmp116 = tmp114 - tmp115
    tmp117 = ops.to_dtype(tmp116, torch.float32, src_dtype=torch.int64)
    tmp118 = ops.constant(0.0066844919786096255, torch.float32)
    tmp119 = tmp117 * tmp118
    tmp120 = tmp113 - tmp119
    tmp121 = ops.where(tmp106, tmp112, tmp120)
    tmp122 = tmp102 + tmp121
    tmp123 = tmp122 * tmp122
    tmp124 = tmp61 + tmp123
    tmp125 = ops.sqrt(tmp124)
    tmp126 = ops.constant(2, torch.float32)
    tmp127 = tmp125 <= tmp126
    tmp128 = ops.to_dtype(tmp127, torch.float32, src_dtype=torch.bool)
    tmp129 = ops.index_expr(i1, dtype=torch.int64)
    tmp130 = ops.to_dtype(tmp129, torch.float32, src_dtype=torch.int64)
    tmp131 = ops.constant(225.0, torch.float32)
    tmp132 = tmp130 < tmp131
    tmp133 = ops.index_expr(i1, dtype=torch.int64)
    tmp134 = ops.to_dtype(tmp133, torch.float32, src_dtype=torch.int64)
    tmp135 = ops.constant(0.0066815144766146995, torch.float32)
    tmp136 = tmp134 * tmp135
    tmp137 = ops.constant(-2.25, torch.float32)
    tmp138 = tmp136 + tmp137
    tmp139 = ops.constant(0.75, torch.float32)
    tmp140 = ops.constant(449, torch.int64)
    tmp141 = ops.index_expr(i1, dtype=torch.int64)
    tmp142 = tmp140 - tmp141
    tmp143 = ops.to_dtype(tmp142, torch.float32, src_dtype=torch.int64)
    tmp144 = ops.constant(0.0066815144766146995, torch.float32)
    tmp145 = tmp143 * tmp144
    tmp146 = tmp139 - tmp145
    tmp147 = ops.where(tmp132, tmp138, tmp146)
    tmp148 = tmp147 * tmp147
    tmp149 = ops.index_expr(i0, dtype=torch.int64)
    tmp150 = ops.to_dtype(tmp149, torch.float32, src_dtype=torch.int64)
    tmp151 = ops.constant(187.5, torch.float32)
    tmp152 = tmp150 < tmp151
    tmp153 = ops.index_expr(i0, dtype=torch.int64)
    tmp154 = ops.to_dtype(tmp153, torch.float32, src_dtype=torch.int64)
    tmp155 = ops.constant(0.0066844919786096255, torch.float32)
    tmp156 = tmp154 * tmp155
    tmp157 = ops.constant(-1.25, torch.float32)
    tmp158 = tmp156 + tmp157
    tmp159 = ops.constant(1.25, torch.float32)
    tmp160 = ops.constant(374, torch.int64)
    tmp161 = ops.index_expr(i0, dtype=torch.int64)
    tmp162 = tmp160 - tmp161
    tmp163 = ops.to_dtype(tmp162, torch.float32, src_dtype=torch.int64)
    tmp164 = ops.constant(0.0066844919786096255, torch.float32)
    tmp165 = tmp163 * tmp164
    tmp166 = tmp159 - tmp165
    tmp167 = ops.where(tmp152, tmp158, tmp166)
    tmp168 = tmp167 * tmp167
    tmp169 = tmp148 - tmp168
    tmp170 = ops.index_expr(i1, dtype=torch.int64)
    tmp171 = ops.to_dtype(tmp170, torch.float32, src_dtype=torch.int64)
    tmp172 = ops.constant(225.0, torch.float32)
    tmp173 = tmp171 < tmp172
    tmp174 = ops.index_expr(i1, dtype=torch.int64)
    tmp175 = ops.to_dtype(tmp174, torch.float32, src_dtype=torch.int64)
    tmp176 = ops.constant(0.0066815144766146995, torch.float32)
    tmp177 = tmp175 * tmp176
    tmp178 = ops.constant(-2.25, torch.float32)
    tmp179 = tmp177 + tmp178
    tmp180 = ops.constant(0.75, torch.float32)
    tmp181 = ops.constant(449, torch.int64)
    tmp182 = ops.index_expr(i1, dtype=torch.int64)
    tmp183 = tmp181 - tmp182
    tmp184 = ops.to_dtype(tmp183, torch.float32, src_dtype=torch.int64)
    tmp185 = ops.constant(0.0066815144766146995, torch.float32)
    tmp186 = tmp184 * tmp185
    tmp187 = tmp180 - tmp186
    tmp188 = ops.where(tmp173, tmp179, tmp187)
    tmp189 = tmp169 + tmp188
    tmp190 = tmp189 * tmp189
    tmp191 = ops.index_expr(i1, dtype=torch.int64)
    tmp192 = ops.to_dtype(tmp191, torch.float32, src_dtype=torch.int64)
    tmp193 = ops.constant(225.0, torch.float32)
    tmp194 = tmp192 < tmp193
    tmp195 = ops.index_expr(i1, dtype=torch.int64)
    tmp196 = ops.to_dtype(tmp195, torch.float32, src_dtype=torch.int64)
    tmp197 = ops.constant(0.0066815144766146995, torch.float32)
    tmp198 = tmp196 * tmp197
    tmp199 = ops.constant(-2.25, torch.float32)
    tmp200 = tmp198 + tmp199
    tmp201 = ops.constant(0.75, torch.float32)
    tmp202 = ops.constant(449, torch.int64)
    tmp203 = ops.index_expr(i1, dtype=torch.int64)
    tmp204 = tmp202 - tmp203
    tmp205 = ops.to_dtype(tmp204, torch.float32, src_dtype=torch.int64)
    tmp206 = ops.constant(0.0066815144766146995, torch.float32)
    tmp207 = tmp205 * tmp206
    tmp208 = tmp201 - tmp207
    tmp209 = ops.where(tmp194, tmp200, tmp208)
    tmp210 = ops.constant(2, torch.float32)
    tmp211 = tmp209 * tmp210
    tmp212 = ops.index_expr(i0, dtype=torch.int64)
    tmp213 = ops.to_dtype(tmp212, torch.float32, src_dtype=torch.int64)
    tmp214 = ops.constant(187.5, torch.float32)
    tmp215 = tmp213 < tmp214
    tmp216 = ops.index_expr(i0, dtype=torch.int64)
    tmp217 = ops.to_dtype(tmp216, torch.float32, src_dtype=torch.int64)
    tmp218 = ops.constant(0.0066844919786096255, torch.float32)
    tmp219 = tmp217 * tmp218
    tmp220 = ops.constant(-1.25, torch.float32)
    tmp221 = tmp219 + tmp220
    tmp222 = ops.constant(1.25, torch.float32)
    tmp223 = ops.constant(374, torch.int64)
    tmp224 = ops.index_expr(i0, dtype=torch.int64)
    tmp225 = tmp223 - tmp224
    tmp226 = ops.to_dtype(tmp225, torch.float32, src_dtype=torch.int64)
    tmp227 = ops.constant(0.0066844919786096255, torch.float32)
    tmp228 = tmp226 * tmp227
    tmp229 = tmp222 - tmp228
    tmp230 = ops.where(tmp215, tmp221, tmp229)
    tmp231 = tmp211 * tmp230
    tmp232 = ops.index_expr(i0, dtype=torch.int64)
    tmp233 = ops.to_dtype(tmp232, torch.float32, src_dtype=torch.int64)
    tmp234 = ops.constant(187.5, torch.float32)
    tmp235 = tmp233 < tmp234
    tmp236 = ops.index_expr(i0, dtype=torch.int64)
    tmp237 = ops.to_dtype(tmp236, torch.float32, src_dtype=torch.int64)
    tmp238 = ops.constant(0.0066844919786096255, torch.float32)
    tmp239 = tmp237 * tmp238
    tmp240 = ops.constant(-1.25, torch.float32)
    tmp241 = tmp239 + tmp240
    tmp242 = ops.constant(1.25, torch.float32)
    tmp243 = ops.constant(374, torch.int64)
    tmp244 = ops.index_expr(i0, dtype=torch.int64)
    tmp245 = tmp243 - tmp244
    tmp246 = ops.to_dtype(tmp245, torch.float32, src_dtype=torch.int64)
    tmp247 = ops.constant(0.0066844919786096255, torch.float32)
    tmp248 = tmp246 * tmp247
    tmp249 = tmp242 - tmp248
    tmp250 = ops.where(tmp235, tmp241, tmp249)
    tmp251 = tmp231 + tmp250
    tmp252 = tmp251 * tmp251
    tmp253 = tmp190 - tmp252
    tmp254 = ops.index_expr(i1, dtype=torch.int64)
    tmp255 = ops.to_dtype(tmp254, torch.float32, src_dtype=torch.int64)
    tmp256 = ops.constant(225.0, torch.float32)
    tmp257 = tmp255 < tmp256
    tmp258 = ops.index_expr(i1, dtype=torch.int64)
    tmp259 = ops.to_dtype(tmp258, torch.float32, src_dtype=torch.int64)
    tmp260 = ops.constant(0.0066815144766146995, torch.float32)
    tmp261 = tmp259 * tmp260
    tmp262 = ops.constant(-2.25, torch.float32)
    tmp263 = tmp261 + tmp262
    tmp264 = ops.constant(0.75, torch.float32)
    tmp265 = ops.constant(449, torch.int64)
    tmp266 = ops.index_expr(i1, dtype=torch.int64)
    tmp267 = tmp265 - tmp266
    tmp268 = ops.to_dtype(tmp267, torch.float32, src_dtype=torch.int64)
    tmp269 = ops.constant(0.0066815144766146995, torch.float32)
    tmp270 = tmp268 * tmp269
    tmp271 = tmp264 - tmp270
    tmp272 = ops.where(tmp257, tmp263, tmp271)
    tmp273 = tmp253 + tmp272
    tmp274 = tmp273 * tmp273
    tmp275 = ops.index_expr(i1, dtype=torch.int64)
    tmp276 = ops.to_dtype(tmp275, torch.float32, src_dtype=torch.int64)
    tmp277 = ops.constant(225.0, torch.float32)
    tmp278 = tmp276 < tmp277
    tmp279 = ops.index_expr(i1, dtype=torch.int64)
    tmp280 = ops.to_dtype(tmp279, torch.float32, src_dtype=torch.int64)
    tmp281 = ops.constant(0.0066815144766146995, torch.float32)
    tmp282 = tmp280 * tmp281
    tmp283 = ops.constant(-2.25, torch.float32)
    tmp284 = tmp282 + tmp283
    tmp285 = ops.constant(0.75, torch.float32)
    tmp286 = ops.constant(449, torch.int64)
    tmp287 = ops.index_expr(i1, dtype=torch.int64)
    tmp288 = tmp286 - tmp287
    tmp289 = ops.to_dtype(tmp288, torch.float32, src_dtype=torch.int64)
    tmp290 = ops.constant(0.0066815144766146995, torch.float32)
    tmp291 = tmp289 * tmp290
    tmp292 = tmp285 - tmp291
    tmp293 = ops.where(tmp278, tmp284, tmp292)
    tmp294 = tmp293 * tmp293
    tmp295 = ops.index_expr(i0, dtype=torch.int64)
    tmp296 = ops.to_dtype(tmp295, torch.float32, src_dtype=torch.int64)
    tmp297 = ops.constant(187.5, torch.float32)
    tmp298 = tmp296 < tmp297
    tmp299 = ops.index_expr(i0, dtype=torch.int64)
    tmp300 = ops.to_dtype(tmp299, torch.float32, src_dtype=torch.int64)
    tmp301 = ops.constant(0.0066844919786096255, torch.float32)
    tmp302 = tmp300 * tmp301
    tmp303 = ops.constant(-1.25, torch.float32)
    tmp304 = tmp302 + tmp303
    tmp305 = ops.constant(1.25, torch.float32)
    tmp306 = ops.constant(374, torch.int64)
    tmp307 = ops.index_expr(i0, dtype=torch.int64)
    tmp308 = tmp306 - tmp307
    tmp309 = ops.to_dtype(tmp308, torch.float32, src_dtype=torch.int64)
    tmp310 = ops.constant(0.0066844919786096255, torch.float32)
    tmp311 = tmp309 * tmp310
    tmp312 = tmp305 - tmp311
    tmp313 = ops.where(tmp298, tmp304, tmp312)
    tmp314 = tmp313 * tmp313
    tmp315 = tmp294 - tmp314
    tmp316 = ops.index_expr(i1, dtype=torch.int64)
    tmp317 = ops.to_dtype(tmp316, torch.float32, src_dtype=torch.int64)
    tmp318 = ops.constant(225.0, torch.float32)
    tmp319 = tmp317 < tmp318
    tmp320 = ops.index_expr(i1, dtype=torch.int64)
    tmp321 = ops.to_dtype(tmp320, torch.float32, src_dtype=torch.int64)
    tmp322 = ops.constant(0.0066815144766146995, torch.float32)
    tmp323 = tmp321 * tmp322
    tmp324 = ops.constant(-2.25, torch.float32)
    tmp325 = tmp323 + tmp324
    tmp326 = ops.constant(0.75, torch.float32)
    tmp327 = ops.constant(449, torch.int64)
    tmp328 = ops.index_expr(i1, dtype=torch.int64)
    tmp329 = tmp327 - tmp328
    tmp330 = ops.to_dtype(tmp329, torch.float32, src_dtype=torch.int64)
    tmp331 = ops.constant(0.0066815144766146995, torch.float32)
    tmp332 = tmp330 * tmp331
    tmp333 = tmp326 - tmp332
    tmp334 = ops.where(tmp319, tmp325, tmp333)
    tmp335 = tmp315 + tmp334
    tmp336 = ops.constant(2, torch.float32)
    tmp337 = tmp335 * tmp336
    tmp338 = ops.index_expr(i1, dtype=torch.int64)
    tmp339 = ops.to_dtype(tmp338, torch.float32, src_dtype=torch.int64)
    tmp340 = ops.constant(225.0, torch.float32)
    tmp341 = tmp339 < tmp340
    tmp342 = ops.index_expr(i1, dtype=torch.int64)
    tmp343 = ops.to_dtype(tmp342, torch.float32, src_dtype=torch.int64)
    tmp344 = ops.constant(0.0066815144766146995, torch.float32)
    tmp345 = tmp343 * tmp344
    tmp346 = ops.constant(-2.25, torch.float32)
    tmp347 = tmp345 + tmp346
    tmp348 = ops.constant(0.75, torch.float32)
    tmp349 = ops.constant(449, torch.int64)
    tmp350 = ops.index_expr(i1, dtype=torch.int64)
    tmp351 = tmp349 - tmp350
    tmp352 = ops.to_dtype(tmp351, torch.float32, src_dtype=torch.int64)
    tmp353 = ops.constant(0.0066815144766146995, torch.float32)
    tmp354 = tmp352 * tmp353
    tmp355 = tmp348 - tmp354
    tmp356 = ops.where(tmp341, tmp347, tmp355)
    tmp357 = ops.constant(2, torch.float32)
    tmp358 = tmp356 * tmp357
    tmp359 = ops.index_expr(i0, dtype=torch.int64)
    tmp360 = ops.to_dtype(tmp359, torch.float32, src_dtype=torch.int64)
    tmp361 = ops.constant(187.5, torch.float32)
    tmp362 = tmp360 < tmp361
    tmp363 = ops.index_expr(i0, dtype=torch.int64)
    tmp364 = ops.to_dtype(tmp363, torch.float32, src_dtype=torch.int64)
    tmp365 = ops.constant(0.0066844919786096255, torch.float32)
    tmp366 = tmp364 * tmp365
    tmp367 = ops.constant(-1.25, torch.float32)
    tmp368 = tmp366 + tmp367
    tmp369 = ops.constant(1.25, torch.float32)
    tmp370 = ops.constant(374, torch.int64)
    tmp371 = ops.index_expr(i0, dtype=torch.int64)
    tmp372 = tmp370 - tmp371
    tmp373 = ops.to_dtype(tmp372, torch.float32, src_dtype=torch.int64)
    tmp374 = ops.constant(0.0066844919786096255, torch.float32)
    tmp375 = tmp373 * tmp374
    tmp376 = tmp369 - tmp375
    tmp377 = ops.where(tmp362, tmp368, tmp376)
    tmp378 = tmp358 * tmp377
    tmp379 = ops.index_expr(i0, dtype=torch.int64)
    tmp380 = ops.to_dtype(tmp379, torch.float32, src_dtype=torch.int64)
    tmp381 = ops.constant(187.5, torch.float32)
    tmp382 = tmp380 < tmp381
    tmp383 = ops.index_expr(i0, dtype=torch.int64)
    tmp384 = ops.to_dtype(tmp383, torch.float32, src_dtype=torch.int64)
    tmp385 = ops.constant(0.0066844919786096255, torch.float32)
    tmp386 = tmp384 * tmp385
    tmp387 = ops.constant(-1.25, torch.float32)
    tmp388 = tmp386 + tmp387
    tmp389 = ops.constant(1.25, torch.float32)
    tmp390 = ops.constant(374, torch.int64)
    tmp391 = ops.index_expr(i0, dtype=torch.int64)
    tmp392 = tmp390 - tmp391
    tmp393 = ops.to_dtype(tmp392, torch.float32, src_dtype=torch.int64)
    tmp394 = ops.constant(0.0066844919786096255, torch.float32)
    tmp395 = tmp393 * tmp394
    tmp396 = tmp389 - tmp395
    tmp397 = ops.where(tmp382, tmp388, tmp396)
    tmp398 = tmp378 + tmp397
    tmp399 = tmp337 * tmp398
    tmp400 = ops.index_expr(i0, dtype=torch.int64)
    tmp401 = ops.to_dtype(tmp400, torch.float32, src_dtype=torch.int64)
    tmp402 = ops.constant(187.5, torch.float32)
    tmp403 = tmp401 < tmp402
    tmp404 = ops.index_expr(i0, dtype=torch.int64)
    tmp405 = ops.to_dtype(tmp404, torch.float32, src_dtype=torch.int64)
    tmp406 = ops.constant(0.0066844919786096255, torch.float32)
    tmp407 = tmp405 * tmp406
    tmp408 = ops.constant(-1.25, torch.float32)
    tmp409 = tmp407 + tmp408
    tmp410 = ops.constant(1.25, torch.float32)
    tmp411 = ops.constant(374, torch.int64)
    tmp412 = ops.index_expr(i0, dtype=torch.int64)
    tmp413 = tmp411 - tmp412
    tmp414 = ops.to_dtype(tmp413, torch.float32, src_dtype=torch.int64)
    tmp415 = ops.constant(0.0066844919786096255, torch.float32)
    tmp416 = tmp414 * tmp415
    tmp417 = tmp410 - tmp416
    tmp418 = ops.where(tmp403, tmp409, tmp417)
    tmp419 = tmp399 + tmp418
    tmp420 = tmp419 * tmp419
    tmp421 = tmp274 + tmp420
    tmp422 = ops.sqrt(tmp421)
    tmp423 = ops.constant(2, torch.float32)
    tmp424 = tmp422 <= tmp423
    tmp425 = ops.to_dtype(tmp424, torch.float32, src_dtype=torch.bool)
    tmp426 = tmp128 + tmp425
    tmp427 = ops.index_expr(i1, dtype=torch.int64)
    tmp428 = ops.to_dtype(tmp427, torch.float32, src_dtype=torch.int64)
    tmp429 = ops.constant(225.0, torch.float32)
    tmp430 = tmp428 < tmp429
    tmp431 = ops.index_expr(i1, dtype=torch.int64)
    tmp432 = ops.to_dtype(tmp431, torch.float32, src_dtype=torch.int64)
    tmp433 = ops.constant(0.0066815144766146995, torch.float32)
    tmp434 = tmp432 * tmp433
    tmp435 = ops.constant(-2.25, torch.float32)
    tmp436 = tmp434 + tmp435
    tmp437 = ops.constant(0.75, torch.float32)
    tmp438 = ops.constant(449, torch.int64)
    tmp439 = ops.index_expr(i1, dtype=torch.int64)
    tmp440 = tmp438 - tmp439
    tmp441 = ops.to_dtype(tmp440, torch.float32, src_dtype=torch.int64)
    tmp442 = ops.constant(0.0066815144766146995, torch.float32)
    tmp443 = tmp441 * tmp442
    tmp444 = tmp437 - tmp443
    tmp445 = ops.where(tmp430, tmp436, tmp444)
    tmp446 = tmp445 * tmp445
    tmp447 = ops.index_expr(i0, dtype=torch.int64)
    tmp448 = ops.to_dtype(tmp447, torch.float32, src_dtype=torch.int64)
    tmp449 = ops.constant(187.5, torch.float32)
    tmp450 = tmp448 < tmp449
    tmp451 = ops.index_expr(i0, dtype=torch.int64)
    tmp452 = ops.to_dtype(tmp451, torch.float32, src_dtype=torch.int64)
    tmp453 = ops.constant(0.0066844919786096255, torch.float32)
    tmp454 = tmp452 * tmp453
    tmp455 = ops.constant(-1.25, torch.float32)
    tmp456 = tmp454 + tmp455
    tmp457 = ops.constant(1.25, torch.float32)
    tmp458 = ops.constant(374, torch.int64)
    tmp459 = ops.index_expr(i0, dtype=torch.int64)
    tmp460 = tmp458 - tmp459
    tmp461 = ops.to_dtype(tmp460, torch.float32, src_dtype=torch.int64)
    tmp462 = ops.constant(0.0066844919786096255, torch.float32)
    tmp463 = tmp461 * tmp462
    tmp464 = tmp457 - tmp463
    tmp465 = ops.where(tmp450, tmp456, tmp464)
    tmp466 = tmp465 * tmp465
    tmp467 = tmp446 - tmp466
    tmp468 = ops.index_expr(i1, dtype=torch.int64)
    tmp469 = ops.to_dtype(tmp468, torch.float32, src_dtype=torch.int64)
    tmp470 = ops.constant(225.0, torch.float32)
    tmp471 = tmp469 < tmp470
    tmp472 = ops.index_expr(i1, dtype=torch.int64)
    tmp473 = ops.to_dtype(tmp472, torch.float32, src_dtype=torch.int64)
    tmp474 = ops.constant(0.0066815144766146995, torch.float32)
    tmp475 = tmp473 * tmp474
    tmp476 = ops.constant(-2.25, torch.float32)
    tmp477 = tmp475 + tmp476
    tmp478 = ops.constant(0.75, torch.float32)
    tmp479 = ops.constant(449, torch.int64)
    tmp480 = ops.index_expr(i1, dtype=torch.int64)
    tmp481 = tmp479 - tmp480
    tmp482 = ops.to_dtype(tmp481, torch.float32, src_dtype=torch.int64)
    tmp483 = ops.constant(0.0066815144766146995, torch.float32)
    tmp484 = tmp482 * tmp483
    tmp485 = tmp478 - tmp484
    tmp486 = ops.where(tmp471, tmp477, tmp485)
    tmp487 = tmp467 + tmp486
    tmp488 = tmp487 * tmp487
    tmp489 = ops.index_expr(i1, dtype=torch.int64)
    tmp490 = ops.to_dtype(tmp489, torch.float32, src_dtype=torch.int64)
    tmp491 = ops.constant(225.0, torch.float32)
    tmp492 = tmp490 < tmp491
    tmp493 = ops.index_expr(i1, dtype=torch.int64)
    tmp494 = ops.to_dtype(tmp493, torch.float32, src_dtype=torch.int64)
    tmp495 = ops.constant(0.0066815144766146995, torch.float32)
    tmp496 = tmp494 * tmp495
    tmp497 = ops.constant(-2.25, torch.float32)
    tmp498 = tmp496 + tmp497
    tmp499 = ops.constant(0.75, torch.float32)
    tmp500 = ops.constant(449, torch.int64)
    tmp501 = ops.index_expr(i1, dtype=torch.int64)
    tmp502 = tmp500 - tmp501
    tmp503 = ops.to_dtype(tmp502, torch.float32, src_dtype=torch.int64)
    tmp504 = ops.constant(0.0066815144766146995, torch.float32)
    tmp505 = tmp503 * tmp504
    tmp506 = tmp499 - tmp505
    tmp507 = ops.where(tmp492, tmp498, tmp506)
    tmp508 = ops.constant(2, torch.float32)
    tmp509 = tmp507 * tmp508
    tmp510 = ops.index_expr(i0, dtype=torch.int64)
    tmp511 = ops.to_dtype(tmp510, torch.float32, src_dtype=torch.int64)
    tmp512 = ops.constant(187.5, torch.float32)
    tmp513 = tmp511 < tmp512
    tmp514 = ops.index_expr(i0, dtype=torch.int64)
    tmp515 = ops.to_dtype(tmp514, torch.float32, src_dtype=torch.int64)
    tmp516 = ops.constant(0.0066844919786096255, torch.float32)
    tmp517 = tmp515 * tmp516
    tmp518 = ops.constant(-1.25, torch.float32)
    tmp519 = tmp517 + tmp518
    tmp520 = ops.constant(1.25, torch.float32)
    tmp521 = ops.constant(374, torch.int64)
    tmp522 = ops.index_expr(i0, dtype=torch.int64)
    tmp523 = tmp521 - tmp522
    tmp524 = ops.to_dtype(tmp523, torch.float32, src_dtype=torch.int64)
    tmp525 = ops.constant(0.0066844919786096255, torch.float32)
    tmp526 = tmp524 * tmp525
    tmp527 = tmp520 - tmp526
    tmp528 = ops.where(tmp513, tmp519, tmp527)
    tmp529 = tmp509 * tmp528
    tmp530 = ops.index_expr(i0, dtype=torch.int64)
    tmp531 = ops.to_dtype(tmp530, torch.float32, src_dtype=torch.int64)
    tmp532 = ops.constant(187.5, torch.float32)
    tmp533 = tmp531 < tmp532
    tmp534 = ops.index_expr(i0, dtype=torch.int64)
    tmp535 = ops.to_dtype(tmp534, torch.float32, src_dtype=torch.int64)
    tmp536 = ops.constant(0.0066844919786096255, torch.float32)
    tmp537 = tmp535 * tmp536
    tmp538 = ops.constant(-1.25, torch.float32)
    tmp539 = tmp537 + tmp538
    tmp540 = ops.constant(1.25, torch.float32)
    tmp541 = ops.constant(374, torch.int64)
    tmp542 = ops.index_expr(i0, dtype=torch.int64)
    tmp543 = tmp541 - tmp542
    tmp544 = ops.to_dtype(tmp543, torch.float32, src_dtype=torch.int64)
    tmp545 = ops.constant(0.0066844919786096255, torch.float32)
    tmp546 = tmp544 * tmp545
    tmp547 = tmp540 - tmp546
    tmp548 = ops.where(tmp533, tmp539, tmp547)
    tmp549 = tmp529 + tmp548
    tmp550 = tmp549 * tmp549
    tmp551 = tmp488 - tmp550
    tmp552 = ops.index_expr(i1, dtype=torch.int64)
    tmp553 = ops.to_dtype(tmp552, torch.float32, src_dtype=torch.int64)
    tmp554 = ops.constant(225.0, torch.float32)
    tmp555 = tmp553 < tmp554
    tmp556 = ops.index_expr(i1, dtype=torch.int64)
    tmp557 = ops.to_dtype(tmp556, torch.float32, src_dtype=torch.int64)
    tmp558 = ops.constant(0.0066815144766146995, torch.float32)
    tmp559 = tmp557 * tmp558
    tmp560 = ops.constant(-2.25, torch.float32)
    tmp561 = tmp559 + tmp560
    tmp562 = ops.constant(0.75, torch.float32)
    tmp563 = ops.constant(449, torch.int64)
    tmp564 = ops.index_expr(i1, dtype=torch.int64)
    tmp565 = tmp563 - tmp564
    tmp566 = ops.to_dtype(tmp565, torch.float32, src_dtype=torch.int64)
    tmp567 = ops.constant(0.0066815144766146995, torch.float32)
    tmp568 = tmp566 * tmp567
    tmp569 = tmp562 - tmp568
    tmp570 = ops.where(tmp555, tmp561, tmp569)
    tmp571 = tmp551 + tmp570
    tmp572 = tmp571 * tmp571
    tmp573 = ops.index_expr(i1, dtype=torch.int64)
    tmp574 = ops.to_dtype(tmp573, torch.float32, src_dtype=torch.int64)
    tmp575 = ops.constant(225.0, torch.float32)
    tmp576 = tmp574 < tmp575
    tmp577 = ops.index_expr(i1, dtype=torch.int64)
    tmp578 = ops.to_dtype(tmp577, torch.float32, src_dtype=torch.int64)
    tmp579 = ops.constant(0.0066815144766146995, torch.float32)
    tmp580 = tmp578 * tmp579
    tmp581 = ops.constant(-2.25, torch.float32)
    tmp582 = tmp580 + tmp581
    tmp583 = ops.constant(0.75, torch.float32)
    tmp584 = ops.constant(449, torch.int64)
    tmp585 = ops.index_expr(i1, dtype=torch.int64)
    tmp586 = tmp584 - tmp585
    tmp587 = ops.to_dtype(tmp586, torch.float32, src_dtype=torch.int64)
    tmp588 = ops.constant(0.0066815144766146995, torch.float32)
    tmp589 = tmp587 * tmp588
    tmp590 = tmp583 - tmp589
    tmp591 = ops.where(tmp576, tmp582, tmp590)
    tmp592 = tmp591 * tmp591
    tmp593 = ops.index_expr(i0, dtype=torch.int64)
    tmp594 = ops.to_dtype(tmp593, torch.float32, src_dtype=torch.int64)
    tmp595 = ops.constant(187.5, torch.float32)
    tmp596 = tmp594 < tmp595
    tmp597 = ops.index_expr(i0, dtype=torch.int64)
    tmp598 = ops.to_dtype(tmp597, torch.float32, src_dtype=torch.int64)
    tmp599 = ops.constant(0.0066844919786096255, torch.float32)
    tmp600 = tmp598 * tmp599
    tmp601 = ops.constant(-1.25, torch.float32)
    tmp602 = tmp600 + tmp601
    tmp603 = ops.constant(1.25, torch.float32)
    tmp604 = ops.constant(374, torch.int64)
    tmp605 = ops.index_expr(i0, dtype=torch.int64)
    tmp606 = tmp604 - tmp605
    tmp607 = ops.to_dtype(tmp606, torch.float32, src_dtype=torch.int64)
    tmp608 = ops.constant(0.0066844919786096255, torch.float32)
    tmp609 = tmp607 * tmp608
    tmp610 = tmp603 - tmp609
    tmp611 = ops.where(tmp596, tmp602, tmp610)
    tmp612 = tmp611 * tmp611
    tmp613 = tmp592 - tmp612
    tmp614 = ops.index_expr(i1, dtype=torch.int64)
    tmp615 = ops.to_dtype(tmp614, torch.float32, src_dtype=torch.int64)
    tmp616 = ops.constant(225.0, torch.float32)
    tmp617 = tmp615 < tmp616
    tmp618 = ops.index_expr(i1, dtype=torch.int64)
    tmp619 = ops.to_dtype(tmp618, torch.float32, src_dtype=torch.int64)
    tmp620 = ops.constant(0.0066815144766146995, torch.float32)
    tmp621 = tmp619 * tmp620
    tmp622 = ops.constant(-2.25, torch.float32)
    tmp623 = tmp621 + tmp622
    tmp624 = ops.constant(0.75, torch.float32)
    tmp625 = ops.constant(449, torch.int64)
    tmp626 = ops.index_expr(i1, dtype=torch.int64)
    tmp627 = tmp625 - tmp626
    tmp628 = ops.to_dtype(tmp627, torch.float32, src_dtype=torch.int64)
    tmp629 = ops.constant(0.0066815144766146995, torch.float32)
    tmp630 = tmp628 * tmp629
    tmp631 = tmp624 - tmp630
    tmp632 = ops.where(tmp617, tmp623, tmp631)
    tmp633 = tmp613 + tmp632
    tmp634 = ops.constant(2, torch.float32)
    tmp635 = tmp633 * tmp634
    tmp636 = ops.index_expr(i1, dtype=torch.int64)
    tmp637 = ops.to_dtype(tmp636, torch.float32, src_dtype=torch.int64)
    tmp638 = ops.constant(225.0, torch.float32)
    tmp639 = tmp637 < tmp638
    tmp640 = ops.index_expr(i1, dtype=torch.int64)
    tmp641 = ops.to_dtype(tmp640, torch.float32, src_dtype=torch.int64)
    tmp642 = ops.constant(0.0066815144766146995, torch.float32)
    tmp643 = tmp641 * tmp642
    tmp644 = ops.constant(-2.25, torch.float32)
    tmp645 = tmp643 + tmp644
    tmp646 = ops.constant(0.75, torch.float32)
    tmp647 = ops.constant(449, torch.int64)
    tmp648 = ops.index_expr(i1, dtype=torch.int64)
    tmp649 = tmp647 - tmp648
    tmp650 = ops.to_dtype(tmp649, torch.float32, src_dtype=torch.int64)
    tmp651 = ops.constant(0.0066815144766146995, torch.float32)
    tmp652 = tmp650 * tmp651
    tmp653 = tmp646 - tmp652
    tmp654 = ops.where(tmp639, tmp645, tmp653)
    tmp655 = ops.constant(2, torch.float32)
    tmp656 = tmp654 * tmp655
    tmp657 = ops.index_expr(i0, dtype=torch.int64)
    tmp658 = ops.to_dtype(tmp657, torch.float32, src_dtype=torch.int64)
    tmp659 = ops.constant(187.5, torch.float32)
    tmp660 = tmp658 < tmp659
    tmp661 = ops.index_expr(i0, dtype=torch.int64)
    tmp662 = ops.to_dtype(tmp661, torch.float32, src_dtype=torch.int64)
    tmp663 = ops.constant(0.0066844919786096255, torch.float32)
    tmp664 = tmp662 * tmp663
    tmp665 = ops.constant(-1.25, torch.float32)
    tmp666 = tmp664 + tmp665
    tmp667 = ops.constant(1.25, torch.float32)
    tmp668 = ops.constant(374, torch.int64)
    tmp669 = ops.index_expr(i0, dtype=torch.int64)
    tmp670 = tmp668 - tmp669
    tmp671 = ops.to_dtype(tmp670, torch.float32, src_dtype=torch.int64)
    tmp672 = ops.constant(0.0066844919786096255, torch.float32)
    tmp673 = tmp671 * tmp672
    tmp674 = tmp667 - tmp673
    tmp675 = ops.where(tmp660, tmp666, tmp674)
    tmp676 = tmp656 * tmp675
    tmp677 = ops.index_expr(i0, dtype=torch.int64)
    tmp678 = ops.to_dtype(tmp677, torch.float32, src_dtype=torch.int64)
    tmp679 = ops.constant(187.5, torch.float32)
    tmp680 = tmp678 < tmp679
    tmp681 = ops.index_expr(i0, dtype=torch.int64)
    tmp682 = ops.to_dtype(tmp681, torch.float32, src_dtype=torch.int64)
    tmp683 = ops.constant(0.0066844919786096255, torch.float32)
    tmp684 = tmp682 * tmp683
    tmp685 = ops.constant(-1.25, torch.float32)
    tmp686 = tmp684 + tmp685
    tmp687 = ops.constant(1.25, torch.float32)
    tmp688 = ops.constant(374, torch.int64)
    tmp689 = ops.index_expr(i0, dtype=torch.int64)
    tmp690 = tmp688 - tmp689
    tmp691 = ops.to_dtype(tmp690, torch.float32, src_dtype=torch.int64)
    tmp692 = ops.constant(0.0066844919786096255, torch.float32)
    tmp693 = tmp691 * tmp692
    tmp694 = tmp687 - tmp693
    tmp695 = ops.where(tmp680, tmp686, tmp694)
    tmp696 = tmp676 + tmp695
    tmp697 = tmp635 * tmp696
    tmp698 = ops.index_expr(i0, dtype=torch.int64)
    tmp699 = ops.to_dtype(tmp698, torch.float32, src_dtype=torch.int64)
    tmp700 = ops.constant(187.5, torch.float32)
    tmp701 = tmp699 < tmp700
    tmp702 = ops.index_expr(i0, dtype=torch.int64)
    tmp703 = ops.to_dtype(tmp702, torch.float32, src_dtype=torch.int64)
    tmp704 = ops.constant(0.0066844919786096255, torch.float32)
    tmp705 = tmp703 * tmp704
    tmp706 = ops.constant(-1.25, torch.float32)
    tmp707 = tmp705 + tmp706
    tmp708 = ops.constant(1.25, torch.float32)
    tmp709 = ops.constant(374, torch.int64)
    tmp710 = ops.index_expr(i0, dtype=torch.int64)
    tmp711 = tmp709 - tmp710
    tmp712 = ops.to_dtype(tmp711, torch.float32, src_dtype=torch.int64)
    tmp713 = ops.constant(0.0066844919786096255, torch.float32)
    tmp714 = tmp712 * tmp713
    tmp715 = tmp708 - tmp714
    tmp716 = ops.where(tmp701, tmp707, tmp715)
    tmp717 = tmp697 + tmp716
    tmp718 = tmp717 * tmp717
    tmp719 = tmp572 - tmp718
    tmp720 = ops.index_expr(i1, dtype=torch.int64)
    tmp721 = ops.to_dtype(tmp720, torch.float32, src_dtype=torch.int64)
    tmp722 = ops.constant(225.0, torch.float32)
    tmp723 = tmp721 < tmp722
    tmp724 = ops.index_expr(i1, dtype=torch.int64)
    tmp725 = ops.to_dtype(tmp724, torch.float32, src_dtype=torch.int64)
    tmp726 = ops.constant(0.0066815144766146995, torch.float32)
    tmp727 = tmp725 * tmp726
    tmp728 = ops.constant(-2.25, torch.float32)
    tmp729 = tmp727 + tmp728
    tmp730 = ops.constant(0.75, torch.float32)
    tmp731 = ops.constant(449, torch.int64)
    tmp732 = ops.index_expr(i1, dtype=torch.int64)
    tmp733 = tmp731 - tmp732
    tmp734 = ops.to_dtype(tmp733, torch.float32, src_dtype=torch.int64)
    tmp735 = ops.constant(0.0066815144766146995, torch.float32)
    tmp736 = tmp734 * tmp735
    tmp737 = tmp730 - tmp736
    tmp738 = ops.where(tmp723, tmp729, tmp737)
    tmp739 = tmp719 + tmp738
    tmp740 = tmp739 * tmp739
    tmp741 = ops.index_expr(i1, dtype=torch.int64)
    tmp742 = ops.to_dtype(tmp741, torch.float32, src_dtype=torch.int64)
    tmp743 = ops.constant(225.0, torch.float32)
    tmp744 = tmp742 < tmp743
    tmp745 = ops.index_expr(i1, dtype=torch.int64)
    tmp746 = ops.to_dtype(tmp745, torch.float32, src_dtype=torch.int64)
    tmp747 = ops.constant(0.0066815144766146995, torch.float32)
    tmp748 = tmp746 * tmp747
    tmp749 = ops.constant(-2.25, torch.float32)
    tmp750 = tmp748 + tmp749
    tmp751 = ops.constant(0.75, torch.float32)
    tmp752 = ops.constant(449, torch.int64)
    tmp753 = ops.index_expr(i1, dtype=torch.int64)
    tmp754 = tmp752 - tmp753
    tmp755 = ops.to_dtype(tmp754, torch.float32, src_dtype=torch.int64)
    tmp756 = ops.constant(0.0066815144766146995, torch.float32)
    tmp757 = tmp755 * tmp756
    tmp758 = tmp751 - tmp757
    tmp759 = ops.where(tmp744, tmp750, tmp758)
    tmp760 = tmp759 * tmp759
    tmp761 = ops.index_expr(i0, dtype=torch.int64)
    tmp762 = ops.to_dtype(tmp761, torch.float32, src_dtype=torch.int64)
    tmp763 = ops.constant(187.5, torch.float32)
    tmp764 = tmp762 < tmp763
    tmp765 = ops.index_expr(i0, dtype=torch.int64)
    tmp766 = ops.to_dtype(tmp765, torch.float32, src_dtype=torch.int64)
    tmp767 = ops.constant(0.0066844919786096255, torch.float32)
    tmp768 = tmp766 * tmp767
    tmp769 = ops.constant(-1.25, torch.float32)
    tmp770 = tmp768 + tmp769
    tmp771 = ops.constant(1.25, torch.float32)
    tmp772 = ops.constant(374, torch.int64)
    tmp773 = ops.index_expr(i0, dtype=torch.int64)
    tmp774 = tmp772 - tmp773
    tmp775 = ops.to_dtype(tmp774, torch.float32, src_dtype=torch.int64)
    tmp776 = ops.constant(0.0066844919786096255, torch.float32)
    tmp777 = tmp775 * tmp776
    tmp778 = tmp771 - tmp777
    tmp779 = ops.where(tmp764, tmp770, tmp778)
    tmp780 = tmp779 * tmp779
    tmp781 = tmp760 - tmp780
    tmp782 = ops.index_expr(i1, dtype=torch.int64)
    tmp783 = ops.to_dtype(tmp782, torch.float32, src_dtype=torch.int64)
    tmp784 = ops.constant(225.0, torch.float32)
    tmp785 = tmp783 < tmp784
    tmp786 = ops.index_expr(i1, dtype=torch.int64)
    tmp787 = ops.to_dtype(tmp786, torch.float32, src_dtype=torch.int64)
    tmp788 = ops.constant(0.0066815144766146995, torch.float32)
    tmp789 = tmp787 * tmp788
    tmp790 = ops.constant(-2.25, torch.float32)
    tmp791 = tmp789 + tmp790
    tmp792 = ops.constant(0.75, torch.float32)
    tmp793 = ops.constant(449, torch.int64)
    tmp794 = ops.index_expr(i1, dtype=torch.int64)
    tmp795 = tmp793 - tmp794
    tmp796 = ops.to_dtype(tmp795, torch.float32, src_dtype=torch.int64)
    tmp797 = ops.constant(0.0066815144766146995, torch.float32)
    tmp798 = tmp796 * tmp797
    tmp799 = tmp792 - tmp798
    tmp800 = ops.where(tmp785, tmp791, tmp799)
    tmp801 = tmp781 + tmp800
    tmp802 = tmp801 * tmp801
    tmp803 = ops.index_expr(i1, dtype=torch.int64)
    tmp804 = ops.to_dtype(tmp803, torch.float32, src_dtype=torch.int64)
    tmp805 = ops.constant(225.0, torch.float32)
    tmp806 = tmp804 < tmp805
    tmp807 = ops.index_expr(i1, dtype=torch.int64)
    tmp808 = ops.to_dtype(tmp807, torch.float32, src_dtype=torch.int64)
    tmp809 = ops.constant(0.0066815144766146995, torch.float32)
    tmp810 = tmp808 * tmp809
    tmp811 = ops.constant(-2.25, torch.float32)
    tmp812 = tmp810 + tmp811
    tmp813 = ops.constant(0.75, torch.float32)
    tmp814 = ops.constant(449, torch.int64)
    tmp815 = ops.index_expr(i1, dtype=torch.int64)
    tmp816 = tmp814 - tmp815
    tmp817 = ops.to_dtype(tmp816, torch.float32, src_dtype=torch.int64)
    tmp818 = ops.constant(0.0066815144766146995, torch.float32)
    tmp819 = tmp817 * tmp818
    tmp820 = tmp813 - tmp819
    tmp821 = ops.where(tmp806, tmp812, tmp820)
    tmp822 = ops.constant(2, torch.float32)
    tmp823 = tmp821 * tmp822
    tmp824 = ops.index_expr(i0, dtype=torch.int64)
    tmp825 = ops.to_dtype(tmp824, torch.float32, src_dtype=torch.int64)
    tmp826 = ops.constant(187.5, torch.float32)
    tmp827 = tmp825 < tmp826
    tmp828 = ops.index_expr(i0, dtype=torch.int64)
    tmp829 = ops.to_dtype(tmp828, torch.float32, src_dtype=torch.int64)
    tmp830 = ops.constant(0.0066844919786096255, torch.float32)
    tmp831 = tmp829 * tmp830
    tmp832 = ops.constant(-1.25, torch.float32)
    tmp833 = tmp831 + tmp832
    tmp834 = ops.constant(1.25, torch.float32)
    tmp835 = ops.constant(374, torch.int64)
    tmp836 = ops.index_expr(i0, dtype=torch.int64)
    tmp837 = tmp835 - tmp836
    tmp838 = ops.to_dtype(tmp837, torch.float32, src_dtype=torch.int64)
    tmp839 = ops.constant(0.0066844919786096255, torch.float32)
    tmp840 = tmp838 * tmp839
    tmp841 = tmp834 - tmp840
    tmp842 = ops.where(tmp827, tmp833, tmp841)
    tmp843 = tmp823 * tmp842
    tmp844 = ops.index_expr(i0, dtype=torch.int64)
    tmp845 = ops.to_dtype(tmp844, torch.float32, src_dtype=torch.int64)
    tmp846 = ops.constant(187.5, torch.float32)
    tmp847 = tmp845 < tmp846
    tmp848 = ops.index_expr(i0, dtype=torch.int64)
    tmp849 = ops.to_dtype(tmp848, torch.float32, src_dtype=torch.int64)
    tmp850 = ops.constant(0.0066844919786096255, torch.float32)
    tmp851 = tmp849 * tmp850
    tmp852 = ops.constant(-1.25, torch.float32)
    tmp853 = tmp851 + tmp852
    tmp854 = ops.constant(1.25, torch.float32)
    tmp855 = ops.constant(374, torch.int64)
    tmp856 = ops.index_expr(i0, dtype=torch.int64)
    tmp857 = tmp855 - tmp856
    tmp858 = ops.to_dtype(tmp857, torch.float32, src_dtype=torch.int64)
    tmp859 = ops.constant(0.0066844919786096255, torch.float32)
    tmp860 = tmp858 * tmp859
    tmp861 = tmp854 - tmp860
    tmp862 = ops.where(tmp847, tmp853, tmp861)
    tmp863 = tmp843 + tmp862
    tmp864 = tmp863 * tmp863
    tmp865 = tmp802 - tmp864
    tmp866 = ops.index_expr(i1, dtype=torch.int64)
    tmp867 = ops.to_dtype(tmp866, torch.float32, src_dtype=torch.int64)
    tmp868 = ops.constant(225.0, torch.float32)
    tmp869 = tmp867 < tmp868
    tmp870 = ops.index_expr(i1, dtype=torch.int64)
    tmp871 = ops.to_dtype(tmp870, torch.float32, src_dtype=torch.int64)
    tmp872 = ops.constant(0.0066815144766146995, torch.float32)
    tmp873 = tmp871 * tmp872
    tmp874 = ops.constant(-2.25, torch.float32)
    tmp875 = tmp873 + tmp874
    tmp876 = ops.constant(0.75, torch.float32)
    tmp877 = ops.constant(449, torch.int64)
    tmp878 = ops.index_expr(i1, dtype=torch.int64)
    tmp879 = tmp877 - tmp878
    tmp880 = ops.to_dtype(tmp879, torch.float32, src_dtype=torch.int64)
    tmp881 = ops.constant(0.0066815144766146995, torch.float32)
    tmp882 = tmp880 * tmp881
    tmp883 = tmp876 - tmp882
    tmp884 = ops.where(tmp869, tmp875, tmp883)
    tmp885 = tmp865 + tmp884
    tmp886 = ops.constant(2, torch.float32)
    tmp887 = tmp885 * tmp886
    tmp888 = ops.index_expr(i1, dtype=torch.int64)
    tmp889 = ops.to_dtype(tmp888, torch.float32, src_dtype=torch.int64)
    tmp890 = ops.constant(225.0, torch.float32)
    tmp891 = tmp889 < tmp890
    tmp892 = ops.index_expr(i1, dtype=torch.int64)
    tmp893 = ops.to_dtype(tmp892, torch.float32, src_dtype=torch.int64)
    tmp894 = ops.constant(0.0066815144766146995, torch.float32)
    tmp895 = tmp893 * tmp894
    tmp896 = ops.constant(-2.25, torch.float32)
    tmp897 = tmp895 + tmp896
    tmp898 = ops.constant(0.75, torch.float32)
    tmp899 = ops.constant(449, torch.int64)
    tmp900 = ops.index_expr(i1, dtype=torch.int64)
    tmp901 = tmp899 - tmp900
    tmp902 = ops.to_dtype(tmp901, torch.float32, src_dtype=torch.int64)
    tmp903 = ops.constant(0.0066815144766146995, torch.float32)
    tmp904 = tmp902 * tmp903
    tmp905 = tmp898 - tmp904
    tmp906 = ops.where(tmp891, tmp897, tmp905)
    tmp907 = tmp906 * tmp906
    tmp908 = ops.index_expr(i0, dtype=torch.int64)
    tmp909 = ops.to_dtype(tmp908, torch.float32, src_dtype=torch.int64)
    tmp910 = ops.constant(187.5, torch.float32)
    tmp911 = tmp909 < tmp910
    tmp912 = ops.index_expr(i0, dtype=torch.int64)
    tmp913 = ops.to_dtype(tmp912, torch.float32, src_dtype=torch.int64)
    tmp914 = ops.constant(0.0066844919786096255, torch.float32)
    tmp915 = tmp913 * tmp914
    tmp916 = ops.constant(-1.25, torch.float32)
    tmp917 = tmp915 + tmp916
    tmp918 = ops.constant(1.25, torch.float32)
    tmp919 = ops.constant(374, torch.int64)
    tmp920 = ops.index_expr(i0, dtype=torch.int64)
    tmp921 = tmp919 - tmp920
    tmp922 = ops.to_dtype(tmp921, torch.float32, src_dtype=torch.int64)
    tmp923 = ops.constant(0.0066844919786096255, torch.float32)
    tmp924 = tmp922 * tmp923
    tmp925 = tmp918 - tmp924
    tmp926 = ops.where(tmp911, tmp917, tmp925)
    tmp927 = tmp926 * tmp926
    tmp928 = tmp907 - tmp927
    tmp929 = ops.index_expr(i1, dtype=torch.int64)
    tmp930 = ops.to_dtype(tmp929, torch.float32, src_dtype=torch.int64)
    tmp931 = ops.constant(225.0, torch.float32)
    tmp932 = tmp930 < tmp931
    tmp933 = ops.index_expr(i1, dtype=torch.int64)
    tmp934 = ops.to_dtype(tmp933, torch.float32, src_dtype=torch.int64)
    tmp935 = ops.constant(0.0066815144766146995, torch.float32)
    tmp936 = tmp934 * tmp935
    tmp937 = ops.constant(-2.25, torch.float32)
    tmp938 = tmp936 + tmp937
    tmp939 = ops.constant(0.75, torch.float32)
    tmp940 = ops.constant(449, torch.int64)
    tmp941 = ops.index_expr(i1, dtype=torch.int64)
    tmp942 = tmp940 - tmp941
    tmp943 = ops.to_dtype(tmp942, torch.float32, src_dtype=torch.int64)
    tmp944 = ops.constant(0.0066815144766146995, torch.float32)
    tmp945 = tmp943 * tmp944
    tmp946 = tmp939 - tmp945
    tmp947 = ops.where(tmp932, tmp938, tmp946)
    tmp948 = tmp928 + tmp947
    tmp949 = ops.constant(2, torch.float32)
    tmp950 = tmp948 * tmp949
    tmp951 = ops.index_expr(i1, dtype=torch.int64)
    tmp952 = ops.to_dtype(tmp951, torch.float32, src_dtype=torch.int64)
    tmp953 = ops.constant(225.0, torch.float32)
    tmp954 = tmp952 < tmp953
    tmp955 = ops.index_expr(i1, dtype=torch.int64)
    tmp956 = ops.to_dtype(tmp955, torch.float32, src_dtype=torch.int64)
    tmp957 = ops.constant(0.0066815144766146995, torch.float32)
    tmp958 = tmp956 * tmp957
    tmp959 = ops.constant(-2.25, torch.float32)
    tmp960 = tmp958 + tmp959
    tmp961 = ops.constant(0.75, torch.float32)
    tmp962 = ops.constant(449, torch.int64)
    tmp963 = ops.index_expr(i1, dtype=torch.int64)
    tmp964 = tmp962 - tmp963
    tmp965 = ops.to_dtype(tmp964, torch.float32, src_dtype=torch.int64)
    tmp966 = ops.constant(0.0066815144766146995, torch.float32)
    tmp967 = tmp965 * tmp966
    tmp968 = tmp961 - tmp967
    tmp969 = ops.where(tmp954, tmp960, tmp968)
    tmp970 = ops.constant(2, torch.float32)
    tmp971 = tmp969 * tmp970
    tmp972 = ops.index_expr(i0, dtype=torch.int64)
    tmp973 = ops.to_dtype(tmp972, torch.float32, src_dtype=torch.int64)
    tmp974 = ops.constant(187.5, torch.float32)
    tmp975 = tmp973 < tmp974
    tmp976 = ops.index_expr(i0, dtype=torch.int64)
    tmp977 = ops.to_dtype(tmp976, torch.float32, src_dtype=torch.int64)
    tmp978 = ops.constant(0.0066844919786096255, torch.float32)
    tmp979 = tmp977 * tmp978
    tmp980 = ops.constant(-1.25, torch.float32)
    tmp981 = tmp979 + tmp980
    tmp982 = ops.constant(1.25, torch.float32)
    tmp983 = ops.constant(374, torch.int64)
    tmp984 = ops.index_expr(i0, dtype=torch.int64)
    tmp985 = tmp983 - tmp984
    tmp986 = ops.to_dtype(tmp985, torch.float32, src_dtype=torch.int64)
    tmp987 = ops.constant(0.0066844919786096255, torch.float32)
    tmp988 = tmp986 * tmp987
    tmp989 = tmp982 - tmp988
    tmp990 = ops.where(tmp975, tmp981, tmp989)
    tmp991 = tmp971 * tmp990
    tmp992 = ops.index_expr(i0, dtype=torch.int64)
    tmp993 = ops.to_dtype(tmp992, torch.float32, src_dtype=torch.int64)
    tmp994 = ops.constant(187.5, torch.float32)
    tmp995 = tmp993 < tmp994
    tmp996 = ops.index_expr(i0, dtype=torch.int64)
    tmp997 = ops.to_dtype(tmp996, torch.float32, src_dtype=torch.int64)
    tmp998 = ops.constant(0.0066844919786096255, torch.float32)
    tmp999 = tmp997 * tmp998
    tmp1000 = ops.constant(-1.25, torch.float32)
    tmp1001 = tmp999 + tmp1000
    tmp1002 = ops.constant(1.25, torch.float32)
    tmp1003 = ops.constant(374, torch.int64)
    tmp1004 = ops.index_expr(i0, dtype=torch.int64)
    tmp1005 = tmp1003 - tmp1004
    tmp1006 = ops.to_dtype(tmp1005, torch.float32, src_dtype=torch.int64)
    tmp1007 = ops.constant(0.0066844919786096255, torch.float32)
    tmp1008 = tmp1006 * tmp1007
    tmp1009 = tmp1002 - tmp1008
    tmp1010 = ops.where(tmp995, tmp1001, tmp1009)
    tmp1011 = tmp991 + tmp1010
    tmp1012 = tmp950 * tmp1011
    tmp1013 = ops.index_expr(i0, dtype=torch.int64)
    tmp1014 = ops.to_dtype(tmp1013, torch.float32, src_dtype=torch.int64)
    tmp1015 = ops.constant(187.5, torch.float32)
    tmp1016 = tmp1014 < tmp1015
    tmp1017 = ops.index_expr(i0, dtype=torch.int64)
    tmp1018 = ops.to_dtype(tmp1017, torch.float32, src_dtype=torch.int64)
    tmp1019 = ops.constant(0.0066844919786096255, torch.float32)
    tmp1020 = tmp1018 * tmp1019
    tmp1021 = ops.constant(-1.25, torch.float32)
    tmp1022 = tmp1020 + tmp1021
    tmp1023 = ops.constant(1.25, torch.float32)
    tmp1024 = ops.constant(374, torch.int64)
    tmp1025 = ops.index_expr(i0, dtype=torch.int64)
    tmp1026 = tmp1024 - tmp1025
    tmp1027 = ops.to_dtype(tmp1026, torch.float32, src_dtype=torch.int64)
    tmp1028 = ops.constant(0.0066844919786096255, torch.float32)
    tmp1029 = tmp1027 * tmp1028
    tmp1030 = tmp1023 - tmp1029
    tmp1031 = ops.where(tmp1016, tmp1022, tmp1030)
    tmp1032 = tmp1012 + tmp1031
    tmp1033 = tmp887 * tmp1032
    tmp1034 = ops.index_expr(i0, dtype=torch.int64)
    tmp1035 = ops.to_dtype(tmp1034, torch.float32, src_dtype=torch.int64)
    tmp1036 = ops.constant(187.5, torch.float32)
    tmp1037 = tmp1035 < tmp1036
    tmp1038 = ops.index_expr(i0, dtype=torch.int64)
    tmp1039 = ops.to_dtype(tmp1038, torch.float32, src_dtype=torch.int64)
    tmp1040 = ops.constant(0.0066844919786096255, torch.float32)
    tmp1041 = tmp1039 * tmp1040
    tmp1042 = ops.constant(-1.25, torch.float32)
    tmp1043 = tmp1041 + tmp1042
    tmp1044 = ops.constant(1.25, torch.float32)
    tmp1045 = ops.constant(374, torch.int64)
    tmp1046 = ops.index_expr(i0, dtype=torch.int64)
    tmp1047 = tmp1045 - tmp1046
    tmp1048 = ops.to_dtype(tmp1047, torch.float32, src_dtype=torch.int64)
    tmp1049 = ops.constant(0.0066844919786096255, torch.float32)
    tmp1050 = tmp1048 * tmp1049
    tmp1051 = tmp1044 - tmp1050
    tmp1052 = ops.where(tmp1037, tmp1043, tmp1051)
    tmp1053 = tmp1033 + tmp1052
    tmp1054 = tmp1053 * tmp1053
    tmp1055 = tmp740 + tmp1054
    tmp1056 = ops.sqrt(tmp1055)
    tmp1057 = ops.constant(2, torch.float32)
    tmp1058 = tmp1056 <= tmp1057
    tmp1059 = ops.to_dtype(tmp1058, torch.float32, src_dtype=torch.bool)
    tmp1060 = tmp426 + tmp1059
    tmp1061 = ops.index_expr(i1, dtype=torch.int64)
    tmp1062 = ops.to_dtype(tmp1061, torch.float32, src_dtype=torch.int64)
    tmp1063 = ops.constant(225.0, torch.float32)
    tmp1064 = tmp1062 < tmp1063
    tmp1065 = ops.index_expr(i1, dtype=torch.int64)
    tmp1066 = ops.to_dtype(tmp1065, torch.float32, src_dtype=torch.int64)
    tmp1067 = ops.constant(0.0066815144766146995, torch.float32)
    tmp1068 = tmp1066 * tmp1067
    tmp1069 = ops.constant(-2.25, torch.float32)
    tmp1070 = tmp1068 + tmp1069
    tmp1071 = ops.constant(0.75, torch.float32)
    tmp1072 = ops.constant(449, torch.int64)
    tmp1073 = ops.index_expr(i1, dtype=torch.int64)
    tmp1074 = tmp1072 - tmp1073
    tmp1075 = ops.to_dtype(tmp1074, torch.float32, src_dtype=torch.int64)
    tmp1076 = ops.constant(0.0066815144766146995, torch.float32)
    tmp1077 = tmp1075 * tmp1076
    tmp1078 = tmp1071 - tmp1077
    tmp1079 = ops.where(tmp1064, tmp1070, tmp1078)
    tmp1080 = tmp1079 * tmp1079
    tmp1081 = ops.index_expr(i0, dtype=torch.int64)
    tmp1082 = ops.to_dtype(tmp1081, torch.float32, src_dtype=torch.int64)
    tmp1083 = ops.constant(187.5, torch.float32)
    tmp1084 = tmp1082 < tmp1083
    tmp1085 = ops.index_expr(i0, dtype=torch.int64)
    tmp1086 = ops.to_dtype(tmp1085, torch.float32, src_dtype=torch.int64)
    tmp1087 = ops.constant(0.0066844919786096255, torch.float32)
    tmp1088 = tmp1086 * tmp1087
    tmp1089 = ops.constant(-1.25, torch.float32)
    tmp1090 = tmp1088 + tmp1089
    tmp1091 = ops.constant(1.25, torch.float32)
    tmp1092 = ops.constant(374, torch.int64)
    tmp1093 = ops.index_expr(i0, dtype=torch.int64)
    tmp1094 = tmp1092 - tmp1093
    tmp1095 = ops.to_dtype(tmp1094, torch.float32, src_dtype=torch.int64)
    tmp1096 = ops.constant(0.0066844919786096255, torch.float32)
    tmp1097 = tmp1095 * tmp1096
    tmp1098 = tmp1091 - tmp1097
    tmp1099 = ops.where(tmp1084, tmp1090, tmp1098)
    tmp1100 = tmp1099 * tmp1099
    tmp1101 = tmp1080 - tmp1100
    tmp1102 = ops.index_expr(i1, dtype=torch.int64)
    tmp1103 = ops.to_dtype(tmp1102, torch.float32, src_dtype=torch.int64)
    tmp1104 = ops.constant(225.0, torch.float32)
    tmp1105 = tmp1103 < tmp1104
    tmp1106 = ops.index_expr(i1, dtype=torch.int64)
    tmp1107 = ops.to_dtype(tmp1106, torch.float32, src_dtype=torch.int64)
    tmp1108 = ops.constant(0.0066815144766146995, torch.float32)
    tmp1109 = tmp1107 * tmp1108
    tmp1110 = ops.constant(-2.25, torch.float32)
    tmp1111 = tmp1109 + tmp1110
    tmp1112 = ops.constant(0.75, torch.float32)
    tmp1113 = ops.constant(449, torch.int64)
    tmp1114 = ops.index_expr(i1, dtype=torch.int64)
    tmp1115 = tmp1113 - tmp1114
    tmp1116 = ops.to_dtype(tmp1115, torch.float32, src_dtype=torch.int64)
    tmp1117 = ops.constant(0.0066815144766146995, torch.float32)
    tmp1118 = tmp1116 * tmp1117
    tmp1119 = tmp1112 - tmp1118
    tmp1120 = ops.where(tmp1105, tmp1111, tmp1119)
    tmp1121 = tmp1101 + tmp1120
    tmp1122 = tmp1121 * tmp1121
    tmp1123 = ops.index_expr(i1, dtype=torch.int64)
    tmp1124 = ops.to_dtype(tmp1123, torch.float32, src_dtype=torch.int64)
    tmp1125 = ops.constant(225.0, torch.float32)
    tmp1126 = tmp1124 < tmp1125
    tmp1127 = ops.index_expr(i1, dtype=torch.int64)
    tmp1128 = ops.to_dtype(tmp1127, torch.float32, src_dtype=torch.int64)
    tmp1129 = ops.constant(0.0066815144766146995, torch.float32)
    tmp1130 = tmp1128 * tmp1129
    tmp1131 = ops.constant(-2.25, torch.float32)
    tmp1132 = tmp1130 + tmp1131
    tmp1133 = ops.constant(0.75, torch.float32)
    tmp1134 = ops.constant(449, torch.int64)
    tmp1135 = ops.index_expr(i1, dtype=torch.int64)
    tmp1136 = tmp1134 - tmp1135
    tmp1137 = ops.to_dtype(tmp1136, torch.float32, src_dtype=torch.int64)
    tmp1138 = ops.constant(0.0066815144766146995, torch.float32)
    tmp1139 = tmp1137 * tmp1138
    tmp1140 = tmp1133 - tmp1139
    tmp1141 = ops.where(tmp1126, tmp1132, tmp1140)
    tmp1142 = ops.constant(2, torch.float32)
    tmp1143 = tmp1141 * tmp1142
    tmp1144 = ops.index_expr(i0, dtype=torch.int64)
    tmp1145 = ops.to_dtype(tmp1144, torch.float32, src_dtype=torch.int64)
    tmp1146 = ops.constant(187.5, torch.float32)
    tmp1147 = tmp1145 < tmp1146
    tmp1148 = ops.index_expr(i0, dtype=torch.int64)
    tmp1149 = ops.to_dtype(tmp1148, torch.float32, src_dtype=torch.int64)
    tmp1150 = ops.constant(0.0066844919786096255, torch.float32)
    tmp1151 = tmp1149 * tmp1150
    tmp1152 = ops.constant(-1.25, torch.float32)
    tmp1153 = tmp1151 + tmp1152
    tmp1154 = ops.constant(1.25, torch.float32)
    tmp1155 = ops.constant(374, torch.int64)
    tmp1156 = ops.index_expr(i0, dtype=torch.int64)
    tmp1157 = tmp1155 - tmp1156
    tmp1158 = ops.to_dtype(tmp1157, torch.float32, src_dtype=torch.int64)
    tmp1159 = ops.constant(0.0066844919786096255, torch.float32)
    tmp1160 = tmp1158 * tmp1159
    tmp1161 = tmp1154 - tmp1160
    tmp1162 = ops.where(tmp1147, tmp1153, tmp1161)
    tmp1163 = tmp1143 * tmp1162
    tmp1164 = ops.index_expr(i0, dtype=torch.int64)
    tmp1165 = ops.to_dtype(tmp1164, torch.float32, src_dtype=torch.int64)
    tmp1166 = ops.constant(187.5, torch.float32)
    tmp1167 = tmp1165 < tmp1166
    tmp1168 = ops.index_expr(i0, dtype=torch.int64)
    tmp1169 = ops.to_dtype(tmp1168, torch.float32, src_dtype=torch.int64)
    tmp1170 = ops.constant(0.0066844919786096255, torch.float32)
    tmp1171 = tmp1169 * tmp1170
    tmp1172 = ops.constant(-1.25, torch.float32)
    tmp1173 = tmp1171 + tmp1172
    tmp1174 = ops.constant(1.25, torch.float32)
    tmp1175 = ops.constant(374, torch.int64)
    tmp1176 = ops.index_expr(i0, dtype=torch.int64)
    tmp1177 = tmp1175 - tmp1176
    tmp1178 = ops.to_dtype(tmp1177, torch.float32, src_dtype=torch.int64)
    tmp1179 = ops.constant(0.0066844919786096255, torch.float32)
    tmp1180 = tmp1178 * tmp1179
    tmp1181 = tmp1174 - tmp1180
    tmp1182 = ops.where(tmp1167, tmp1173, tmp1181)
    tmp1183 = tmp1163 + tmp1182
    tmp1184 = tmp1183 * tmp1183
    tmp1185 = tmp1122 - tmp1184
    tmp1186 = ops.index_expr(i1, dtype=torch.int64)
    tmp1187 = ops.to_dtype(tmp1186, torch.float32, src_dtype=torch.int64)
    tmp1188 = ops.constant(225.0, torch.float32)
    tmp1189 = tmp1187 < tmp1188
    tmp1190 = ops.index_expr(i1, dtype=torch.int64)
    tmp1191 = ops.to_dtype(tmp1190, torch.float32, src_dtype=torch.int64)
    tmp1192 = ops.constant(0.0066815144766146995, torch.float32)
    tmp1193 = tmp1191 * tmp1192
    tmp1194 = ops.constant(-2.25, torch.float32)
    tmp1195 = tmp1193 + tmp1194
    tmp1196 = ops.constant(0.75, torch.float32)
    tmp1197 = ops.constant(449, torch.int64)
    tmp1198 = ops.index_expr(i1, dtype=torch.int64)
    tmp1199 = tmp1197 - tmp1198
    tmp1200 = ops.to_dtype(tmp1199, torch.float32, src_dtype=torch.int64)
    tmp1201 = ops.constant(0.0066815144766146995, torch.float32)
    tmp1202 = tmp1200 * tmp1201
    tmp1203 = tmp1196 - tmp1202
    tmp1204 = ops.where(tmp1189, tmp1195, tmp1203)
    tmp1205 = tmp1185 + tmp1204
    tmp1206 = tmp1205 * tmp1205
    tmp1207 = ops.index_expr(i1, dtype=torch.int64)
    tmp1208 = ops.to_dtype(tmp1207, torch.float32, src_dtype=torch.int64)
    tmp1209 = ops.constant(225.0, torch.float32)
    tmp1210 = tmp1208 < tmp1209
    tmp1211 = ops.index_expr(i1, dtype=torch.int64)
    tmp1212 = ops.to_dtype(tmp1211, torch.float32, src_dtype=torch.int64)
    tmp1213 = ops.constant(0.0066815144766146995, torch.float32)
    tmp1214 = tmp1212 * tmp1213
    tmp1215 = ops.constant(-2.25, torch.float32)
    tmp1216 = tmp1214 + tmp1215
    tmp1217 = ops.constant(0.75, torch.float32)
    tmp1218 = ops.constant(449, torch.int64)
    tmp1219 = ops.index_expr(i1, dtype=torch.int64)
    tmp1220 = tmp1218 - tmp1219
    tmp1221 = ops.to_dtype(tmp1220, torch.float32, src_dtype=torch.int64)
    tmp1222 = ops.constant(0.0066815144766146995, torch.float32)
    tmp1223 = tmp1221 * tmp1222
    tmp1224 = tmp1217 - tmp1223
    tmp1225 = ops.where(tmp1210, tmp1216, tmp1224)
    tmp1226 = tmp1225 * tmp1225
    tmp1227 = ops.index_expr(i0, dtype=torch.int64)
    tmp1228 = ops.to_dtype(tmp1227, torch.float32, src_dtype=torch.int64)
    tmp1229 = ops.constant(187.5, torch.float32)
    tmp1230 = tmp1228 < tmp1229
    tmp1231 = ops.index_expr(i0, dtype=torch.int64)
    tmp1232 = ops.to_dtype(tmp1231, torch.float32, src_dtype=torch.int64)
    tmp1233 = ops.constant(0.0066844919786096255, torch.float32)
    tmp1234 = tmp1232 * tmp1233
    tmp1235 = ops.constant(-1.25, torch.float32)
    tmp1236 = tmp1234 + tmp1235
    tmp1237 = ops.constant(1.25, torch.float32)
    tmp1238 = ops.constant(374, torch.int64)
    tmp1239 = ops.index_expr(i0, dtype=torch.int64)
    tmp1240 = tmp1238 - tmp1239
    tmp1241 = ops.to_dtype(tmp1240, torch.float32, src_dtype=torch.int64)
    tmp1242 = ops.constant(0.0066844919786096255, torch.float32)
    tmp1243 = tmp1241 * tmp1242
    tmp1244 = tmp1237 - tmp1243
    tmp1245 = ops.where(tmp1230, tmp1236, tmp1244)
    tmp1246 = tmp1245 * tmp1245
    tmp1247 = tmp1226 - tmp1246
    tmp1248 = ops.index_expr(i1, dtype=torch.int64)
    tmp1249 = ops.to_dtype(tmp1248, torch.float32, src_dtype=torch.int64)
    tmp1250 = ops.constant(225.0, torch.float32)
    tmp1251 = tmp1249 < tmp1250
    tmp1252 = ops.index_expr(i1, dtype=torch.int64)
    tmp1253 = ops.to_dtype(tmp1252, torch.float32, src_dtype=torch.int64)
    tmp1254 = ops.constant(0.0066815144766146995, torch.float32)
    tmp1255 = tmp1253 * tmp1254
    tmp1256 = ops.constant(-2.25, torch.float32)
    tmp1257 = tmp1255 + tmp1256
    tmp1258 = ops.constant(0.75, torch.float32)
    tmp1259 = ops.constant(449, torch.int64)
    tmp1260 = ops.index_expr(i1, dtype=torch.int64)
    tmp1261 = tmp1259 - tmp1260
    tmp1262 = ops.to_dtype(tmp1261, torch.float32, src_dtype=torch.int64)
    tmp1263 = ops.constant(0.0066815144766146995, torch.float32)
    tmp1264 = tmp1262 * tmp1263
    tmp1265 = tmp1258 - tmp1264
    tmp1266 = ops.where(tmp1251, tmp1257, tmp1265)
    tmp1267 = tmp1247 + tmp1266
    tmp1268 = ops.constant(2, torch.float32)
    tmp1269 = tmp1267 * tmp1268
    tmp1270 = ops.index_expr(i1, dtype=torch.int64)
    tmp1271 = ops.to_dtype(tmp1270, torch.float32, src_dtype=torch.int64)
    tmp1272 = ops.constant(225.0, torch.float32)
    tmp1273 = tmp1271 < tmp1272
    tmp1274 = ops.index_expr(i1, dtype=torch.int64)
    tmp1275 = ops.to_dtype(tmp1274, torch.float32, src_dtype=torch.int64)
    tmp1276 = ops.constant(0.0066815144766146995, torch.float32)
    tmp1277 = tmp1275 * tmp1276
    tmp1278 = ops.constant(-2.25, torch.float32)
    tmp1279 = tmp1277 + tmp1278
    tmp1280 = ops.constant(0.75, torch.float32)
    tmp1281 = ops.constant(449, torch.int64)
    tmp1282 = ops.index_expr(i1, dtype=torch.int64)
    tmp1283 = tmp1281 - tmp1282
    tmp1284 = ops.to_dtype(tmp1283, torch.float32, src_dtype=torch.int64)
    tmp1285 = ops.constant(0.0066815144766146995, torch.float32)
    tmp1286 = tmp1284 * tmp1285
    tmp1287 = tmp1280 - tmp1286
    tmp1288 = ops.where(tmp1273, tmp1279, tmp1287)
    tmp1289 = ops.constant(2, torch.float32)
    tmp1290 = tmp1288 * tmp1289
    tmp1291 = ops.index_expr(i0, dtype=torch.int64)
    tmp1292 = ops.to_dtype(tmp1291, torch.float32, src_dtype=torch.int64)
    tmp1293 = ops.constant(187.5, torch.float32)
    tmp1294 = tmp1292 < tmp1293
    tmp1295 = ops.index_expr(i0, dtype=torch.int64)
    tmp1296 = ops.to_dtype(tmp1295, torch.float32, src_dtype=torch.int64)
    tmp1297 = ops.constant(0.0066844919786096255, torch.float32)
    tmp1298 = tmp1296 * tmp1297
    tmp1299 = ops.constant(-1.25, torch.float32)
    tmp1300 = tmp1298 + tmp1299
    tmp1301 = ops.constant(1.25, torch.float32)
    tmp1302 = ops.constant(374, torch.int64)
    tmp1303 = ops.index_expr(i0, dtype=torch.int64)
    tmp1304 = tmp1302 - tmp1303
    tmp1305 = ops.to_dtype(tmp1304, torch.float32, src_dtype=torch.int64)
    tmp1306 = ops.constant(0.0066844919786096255, torch.float32)
    tmp1307 = tmp1305 * tmp1306
    tmp1308 = tmp1301 - tmp1307
    tmp1309 = ops.where(tmp1294, tmp1300, tmp1308)
    tmp1310 = tmp1290 * tmp1309
    tmp1311 = ops.index_expr(i0, dtype=torch.int64)
    tmp1312 = ops.to_dtype(tmp1311, torch.float32, src_dtype=torch.int64)
    tmp1313 = ops.constant(187.5, torch.float32)
    tmp1314 = tmp1312 < tmp1313
    tmp1315 = ops.index_expr(i0, dtype=torch.int64)
    tmp1316 = ops.to_dtype(tmp1315, torch.float32, src_dtype=torch.int64)
    tmp1317 = ops.constant(0.0066844919786096255, torch.float32)
    tmp1318 = tmp1316 * tmp1317
    tmp1319 = ops.constant(-1.25, torch.float32)
    tmp1320 = tmp1318 + tmp1319
    tmp1321 = ops.constant(1.25, torch.float32)
    tmp1322 = ops.constant(374, torch.int64)
    tmp1323 = ops.index_expr(i0, dtype=torch.int64)
    tmp1324 = tmp1322 - tmp1323
    tmp1325 = ops.to_dtype(tmp1324, torch.float32, src_dtype=torch.int64)
    tmp1326 = ops.constant(0.0066844919786096255, torch.float32)
    tmp1327 = tmp1325 * tmp1326
    tmp1328 = tmp1321 - tmp1327
    tmp1329 = ops.where(tmp1314, tmp1320, tmp1328)
    tmp1330 = tmp1310 + tmp1329
    tmp1331 = tmp1269 * tmp1330
    tmp1332 = ops.index_expr(i0, dtype=torch.int64)
    tmp1333 = ops.to_dtype(tmp1332, torch.float32, src_dtype=torch.int64)
    tmp1334 = ops.constant(187.5, torch.float32)
    tmp1335 = tmp1333 < tmp1334
    tmp1336 = ops.index_expr(i0, dtype=torch.int64)
    tmp1337 = ops.to_dtype(tmp1336, torch.float32, src_dtype=torch.int64)
    tmp1338 = ops.constant(0.0066844919786096255, torch.float32)
    tmp1339 = tmp1337 * tmp1338
    tmp1340 = ops.constant(-1.25, torch.float32)
    tmp1341 = tmp1339 + tmp1340
    tmp1342 = ops.constant(1.25, torch.float32)
    tmp1343 = ops.constant(374, torch.int64)
    tmp1344 = ops.index_expr(i0, dtype=torch.int64)
    tmp1345 = tmp1343 - tmp1344
    tmp1346 = ops.to_dtype(tmp1345, torch.float32, src_dtype=torch.int64)
    tmp1347 = ops.constant(0.0066844919786096255, torch.float32)
    tmp1348 = tmp1346 * tmp1347
    tmp1349 = tmp1342 - tmp1348
    tmp1350 = ops.where(tmp1335, tmp1341, tmp1349)
    tmp1351 = tmp1331 + tmp1350
    tmp1352 = tmp1351 * tmp1351
    tmp1353 = tmp1206 - tmp1352
    tmp1354 = ops.index_expr(i1, dtype=torch.int64)
    tmp1355 = ops.to_dtype(tmp1354, torch.float32, src_dtype=torch.int64)
    tmp1356 = ops.constant(225.0, torch.float32)
    tmp1357 = tmp1355 < tmp1356
    tmp1358 = ops.index_expr(i1, dtype=torch.int64)
    tmp1359 = ops.to_dtype(tmp1358, torch.float32, src_dtype=torch.int64)
    tmp1360 = ops.constant(0.0066815144766146995, torch.float32)
    tmp1361 = tmp1359 * tmp1360
    tmp1362 = ops.constant(-2.25, torch.float32)
    tmp1363 = tmp1361 + tmp1362
    tmp1364 = ops.constant(0.75, torch.float32)
    tmp1365 = ops.constant(449, torch.int64)
    tmp1366 = ops.index_expr(i1, dtype=torch.int64)
    tmp1367 = tmp1365 - tmp1366
    tmp1368 = ops.to_dtype(tmp1367, torch.float32, src_dtype=torch.int64)
    tmp1369 = ops.constant(0.0066815144766146995, torch.float32)
    tmp1370 = tmp1368 * tmp1369
    tmp1371 = tmp1364 - tmp1370
    tmp1372 = ops.where(tmp1357, tmp1363, tmp1371)
    tmp1373 = tmp1353 + tmp1372
    tmp1374 = tmp1373 * tmp1373
    tmp1375 = ops.index_expr(i1, dtype=torch.int64)
    tmp1376 = ops.to_dtype(tmp1375, torch.float32, src_dtype=torch.int64)
    tmp1377 = ops.constant(225.0, torch.float32)
    tmp1378 = tmp1376 < tmp1377
    tmp1379 = ops.index_expr(i1, dtype=torch.int64)
    tmp1380 = ops.to_dtype(tmp1379, torch.float32, src_dtype=torch.int64)
    tmp1381 = ops.constant(0.0066815144766146995, torch.float32)
    tmp1382 = tmp1380 * tmp1381
    tmp1383 = ops.constant(-2.25, torch.float32)
    tmp1384 = tmp1382 + tmp1383
    tmp1385 = ops.constant(0.75, torch.float32)
    tmp1386 = ops.constant(449, torch.int64)
    tmp1387 = ops.index_expr(i1, dtype=torch.int64)
    tmp1388 = tmp1386 - tmp1387
    tmp1389 = ops.to_dtype(tmp1388, torch.float32, src_dtype=torch.int64)
    tmp1390 = ops.constant(0.0066815144766146995, torch.float32)
    tmp1391 = tmp1389 * tmp1390
    tmp1392 = tmp1385 - tmp1391
    tmp1393 = ops.where(tmp1378, tmp1384, tmp1392)
    tmp1394 = tmp1393 * tmp1393
    tmp1395 = ops.index_expr(i0, dtype=torch.int64)
    tmp1396 = ops.to_dtype(tmp1395, torch.float32, src_dtype=torch.int64)
    tmp1397 = ops.constant(187.5, torch.float32)
    tmp1398 = tmp1396 < tmp1397
    tmp1399 = ops.index_expr(i0, dtype=torch.int64)
    tmp1400 = ops.to_dtype(tmp1399, torch.float32, src_dtype=torch.int64)
    tmp1401 = ops.constant(0.0066844919786096255, torch.float32)
    tmp1402 = tmp1400 * tmp1401
    tmp1403 = ops.constant(-1.25, torch.float32)
    tmp1404 = tmp1402 + tmp1403
    tmp1405 = ops.constant(1.25, torch.float32)
    tmp1406 = ops.constant(374, torch.int64)
    tmp1407 = ops.index_expr(i0, dtype=torch.int64)
    tmp1408 = tmp1406 - tmp1407
    tmp1409 = ops.to_dtype(tmp1408, torch.float32, src_dtype=torch.int64)
    tmp1410 = ops.constant(0.0066844919786096255, torch.float32)
    tmp1411 = tmp1409 * tmp1410
    tmp1412 = tmp1405 - tmp1411
    tmp1413 = ops.where(tmp1398, tmp1404, tmp1412)
    tmp1414 = tmp1413 * tmp1413
    tmp1415 = tmp1394 - tmp1414
    tmp1416 = ops.index_expr(i1, dtype=torch.int64)
    tmp1417 = ops.to_dtype(tmp1416, torch.float32, src_dtype=torch.int64)
    tmp1418 = ops.constant(225.0, torch.float32)
    tmp1419 = tmp1417 < tmp1418
    tmp1420 = ops.index_expr(i1, dtype=torch.int64)
    tmp1421 = ops.to_dtype(tmp1420, torch.float32, src_dtype=torch.int64)
    tmp1422 = ops.constant(0.0066815144766146995, torch.float32)
    tmp1423 = tmp1421 * tmp1422
    tmp1424 = ops.constant(-2.25, torch.float32)
    tmp1425 = tmp1423 + tmp1424
    tmp1426 = ops.constant(0.75, torch.float32)
    tmp1427 = ops.constant(449, torch.int64)
    tmp1428 = ops.index_expr(i1, dtype=torch.int64)
    tmp1429 = tmp1427 - tmp1428
    tmp1430 = ops.to_dtype(tmp1429, torch.float32, src_dtype=torch.int64)
    tmp1431 = ops.constant(0.0066815144766146995, torch.float32)
    tmp1432 = tmp1430 * tmp1431
    tmp1433 = tmp1426 - tmp1432
    tmp1434 = ops.where(tmp1419, tmp1425, tmp1433)
    tmp1435 = tmp1415 + tmp1434
    tmp1436 = tmp1435 * tmp1435
    tmp1437 = ops.index_expr(i1, dtype=torch.int64)
    tmp1438 = ops.to_dtype(tmp1437, torch.float32, src_dtype=torch.int64)
    tmp1439 = ops.constant(225.0, torch.float32)
    tmp1440 = tmp1438 < tmp1439
    tmp1441 = ops.index_expr(i1, dtype=torch.int64)
    tmp1442 = ops.to_dtype(tmp1441, torch.float32, src_dtype=torch.int64)
    tmp1443 = ops.constant(0.0066815144766146995, torch.float32)
    tmp1444 = tmp1442 * tmp1443
    tmp1445 = ops.constant(-2.25, torch.float32)
    tmp1446 = tmp1444 + tmp1445
    tmp1447 = ops.constant(0.75, torch.float32)
    tmp1448 = ops.constant(449, torch.int64)
    tmp1449 = ops.index_expr(i1, dtype=torch.int64)
    tmp1450 = tmp1448 - tmp1449
    tmp1451 = ops.to_dtype(tmp1450, torch.float32, src_dtype=torch.int64)
    tmp1452 = ops.constant(0.0066815144766146995, torch.float32)
    tmp1453 = tmp1451 * tmp1452
    tmp1454 = tmp1447 - tmp1453
    tmp1455 = ops.where(tmp1440, tmp1446, tmp1454)
    tmp1456 = ops.constant(2, torch.float32)
    tmp1457 = tmp1455 * tmp1456
    tmp1458 = ops.index_expr(i0, dtype=torch.int64)
    tmp1459 = ops.to_dtype(tmp1458, torch.float32, src_dtype=torch.int64)
    tmp1460 = ops.constant(187.5, torch.float32)
    tmp1461 = tmp1459 < tmp1460
    tmp1462 = ops.index_expr(i0, dtype=torch.int64)
    tmp1463 = ops.to_dtype(tmp1462, torch.float32, src_dtype=torch.int64)
    tmp1464 = ops.constant(0.0066844919786096255, torch.float32)
    tmp1465 = tmp1463 * tmp1464
    tmp1466 = ops.constant(-1.25, torch.float32)
    tmp1467 = tmp1465 + tmp1466
    tmp1468 = ops.constant(1.25, torch.float32)
    tmp1469 = ops.constant(374, torch.int64)
    tmp1470 = ops.index_expr(i0, dtype=torch.int64)
    tmp1471 = tmp1469 - tmp1470
    tmp1472 = ops.to_dtype(tmp1471, torch.float32, src_dtype=torch.int64)
    tmp1473 = ops.constant(0.0066844919786096255, torch.float32)
    tmp1474 = tmp1472 * tmp1473
    tmp1475 = tmp1468 - tmp1474
    tmp1476 = ops.where(tmp1461, tmp1467, tmp1475)
    tmp1477 = tmp1457 * tmp1476
    tmp1478 = ops.index_expr(i0, dtype=torch.int64)
    tmp1479 = ops.to_dtype(tmp1478, torch.float32, src_dtype=torch.int64)
    tmp1480 = ops.constant(187.5, torch.float32)
    tmp1481 = tmp1479 < tmp1480
    tmp1482 = ops.index_expr(i0, dtype=torch.int64)
    tmp1483 = ops.to_dtype(tmp1482, torch.float32, src_dtype=torch.int64)
    tmp1484 = ops.constant(0.0066844919786096255, torch.float32)
    tmp1485 = tmp1483 * tmp1484
    tmp1486 = ops.constant(-1.25, torch.float32)
    tmp1487 = tmp1485 + tmp1486
    tmp1488 = ops.constant(1.25, torch.float32)
    tmp1489 = ops.constant(374, torch.int64)
    tmp1490 = ops.index_expr(i0, dtype=torch.int64)
    tmp1491 = tmp1489 - tmp1490
    tmp1492 = ops.to_dtype(tmp1491, torch.float32, src_dtype=torch.int64)
    tmp1493 = ops.constant(0.0066844919786096255, torch.float32)
    tmp1494 = tmp1492 * tmp1493
    tmp1495 = tmp1488 - tmp1494
    tmp1496 = ops.where(tmp1481, tmp1487, tmp1495)
    tmp1497 = tmp1477 + tmp1496
    tmp1498 = tmp1497 * tmp1497
    tmp1499 = tmp1436 - tmp1498
    tmp1500 = ops.index_expr(i1, dtype=torch.int64)
    tmp1501 = ops.to_dtype(tmp1500, torch.float32, src_dtype=torch.int64)
    tmp1502 = ops.constant(225.0, torch.float32)
    tmp1503 = tmp1501 < tmp1502
    tmp1504 = ops.index_expr(i1, dtype=torch.int64)
    tmp1505 = ops.to_dtype(tmp1504, torch.float32, src_dtype=torch.int64)
    tmp1506 = ops.constant(0.0066815144766146995, torch.float32)
    tmp1507 = tmp1505 * tmp1506
    tmp1508 = ops.constant(-2.25, torch.float32)
    tmp1509 = tmp1507 + tmp1508
    tmp1510 = ops.constant(0.75, torch.float32)
    tmp1511 = ops.constant(449, torch.int64)
    tmp1512 = ops.index_expr(i1, dtype=torch.int64)
    tmp1513 = tmp1511 - tmp1512
    tmp1514 = ops.to_dtype(tmp1513, torch.float32, src_dtype=torch.int64)
    tmp1515 = ops.constant(0.0066815144766146995, torch.float32)
    tmp1516 = tmp1514 * tmp1515
    tmp1517 = tmp1510 - tmp1516
    tmp1518 = ops.where(tmp1503, tmp1509, tmp1517)
    tmp1519 = tmp1499 + tmp1518
    tmp1520 = ops.constant(2, torch.float32)
    tmp1521 = tmp1519 * tmp1520
    tmp1522 = ops.index_expr(i1, dtype=torch.int64)
    tmp1523 = ops.to_dtype(tmp1522, torch.float32, src_dtype=torch.int64)
    tmp1524 = ops.constant(225.0, torch.float32)
    tmp1525 = tmp1523 < tmp1524
    tmp1526 = ops.index_expr(i1, dtype=torch.int64)
    tmp1527 = ops.to_dtype(tmp1526, torch.float32, src_dtype=torch.int64)
    tmp1528 = ops.constant(0.0066815144766146995, torch.float32)
    tmp1529 = tmp1527 * tmp1528
    tmp1530 = ops.constant(-2.25, torch.float32)
    tmp1531 = tmp1529 + tmp1530
    tmp1532 = ops.constant(0.75, torch.float32)
    tmp1533 = ops.constant(449, torch.int64)
    tmp1534 = ops.index_expr(i1, dtype=torch.int64)
    tmp1535 = tmp1533 - tmp1534
    tmp1536 = ops.to_dtype(tmp1535, torch.float32, src_dtype=torch.int64)
    tmp1537 = ops.constant(0.0066815144766146995, torch.float32)
    tmp1538 = tmp1536 * tmp1537
    tmp1539 = tmp1532 - tmp1538
    tmp1540 = ops.where(tmp1525, tmp1531, tmp1539)
    tmp1541 = tmp1540 * tmp1540
    tmp1542 = ops.index_expr(i0, dtype=torch.int64)
    tmp1543 = ops.to_dtype(tmp1542, torch.float32, src_dtype=torch.int64)
    tmp1544 = ops.constant(187.5, torch.float32)
    tmp1545 = tmp1543 < tmp1544
    tmp1546 = ops.index_expr(i0, dtype=torch.int64)
    tmp1547 = ops.to_dtype(tmp1546, torch.float32, src_dtype=torch.int64)
    tmp1548 = ops.constant(0.0066844919786096255, torch.float32)
    tmp1549 = tmp1547 * tmp1548
    tmp1550 = ops.constant(-1.25, torch.float32)
    tmp1551 = tmp1549 + tmp1550
    tmp1552 = ops.constant(1.25, torch.float32)
    tmp1553 = ops.constant(374, torch.int64)
    tmp1554 = ops.index_expr(i0, dtype=torch.int64)
    tmp1555 = tmp1553 - tmp1554
    tmp1556 = ops.to_dtype(tmp1555, torch.float32, src_dtype=torch.int64)
    tmp1557 = ops.constant(0.0066844919786096255, torch.float32)
    tmp1558 = tmp1556 * tmp1557
    tmp1559 = tmp1552 - tmp1558
    tmp1560 = ops.where(tmp1545, tmp1551, tmp1559)
    tmp1561 = tmp1560 * tmp1560
    tmp1562 = tmp1541 - tmp1561
    tmp1563 = ops.index_expr(i1, dtype=torch.int64)
    tmp1564 = ops.to_dtype(tmp1563, torch.float32, src_dtype=torch.int64)
    tmp1565 = ops.constant(225.0, torch.float32)
    tmp1566 = tmp1564 < tmp1565
    tmp1567 = ops.index_expr(i1, dtype=torch.int64)
    tmp1568 = ops.to_dtype(tmp1567, torch.float32, src_dtype=torch.int64)
    tmp1569 = ops.constant(0.0066815144766146995, torch.float32)
    tmp1570 = tmp1568 * tmp1569
    tmp1571 = ops.constant(-2.25, torch.float32)
    tmp1572 = tmp1570 + tmp1571
    tmp1573 = ops.constant(0.75, torch.float32)
    tmp1574 = ops.constant(449, torch.int64)
    tmp1575 = ops.index_expr(i1, dtype=torch.int64)
    tmp1576 = tmp1574 - tmp1575
    tmp1577 = ops.to_dtype(tmp1576, torch.float32, src_dtype=torch.int64)
    tmp1578 = ops.constant(0.0066815144766146995, torch.float32)
    tmp1579 = tmp1577 * tmp1578
    tmp1580 = tmp1573 - tmp1579
    tmp1581 = ops.where(tmp1566, tmp1572, tmp1580)
    tmp1582 = tmp1562 + tmp1581
    tmp1583 = ops.constant(2, torch.float32)
    tmp1584 = tmp1582 * tmp1583
    tmp1585 = ops.index_expr(i1, dtype=torch.int64)
    tmp1586 = ops.to_dtype(tmp1585, torch.float32, src_dtype=torch.int64)
    tmp1587 = ops.constant(225.0, torch.float32)
    tmp1588 = tmp1586 < tmp1587
    tmp1589 = ops.index_expr(i1, dtype=torch.int64)
    tmp1590 = ops.to_dtype(tmp1589, torch.float32, src_dtype=torch.int64)
    tmp1591 = ops.constant(0.0066815144766146995, torch.float32)
    tmp1592 = tmp1590 * tmp1591
    tmp1593 = ops.constant(-2.25, torch.float32)
    tmp1594 = tmp1592 + tmp1593
    tmp1595 = ops.constant(0.75, torch.float32)
    tmp1596 = ops.constant(449, torch.int64)
    tmp1597 = ops.index_expr(i1, dtype=torch.int64)
    tmp1598 = tmp1596 - tmp1597
    tmp1599 = ops.to_dtype(tmp1598, torch.float32, src_dtype=torch.int64)
    tmp1600 = ops.constant(0.0066815144766146995, torch.float32)
    tmp1601 = tmp1599 * tmp1600
    tmp1602 = tmp1595 - tmp1601
    tmp1603 = ops.where(tmp1588, tmp1594, tmp1602)
    tmp1604 = ops.constant(2, torch.float32)
    tmp1605 = tmp1603 * tmp1604
    tmp1606 = ops.index_expr(i0, dtype=torch.int64)
    tmp1607 = ops.to_dtype(tmp1606, torch.float32, src_dtype=torch.int64)
    tmp1608 = ops.constant(187.5, torch.float32)
    tmp1609 = tmp1607 < tmp1608
    tmp1610 = ops.index_expr(i0, dtype=torch.int64)
    tmp1611 = ops.to_dtype(tmp1610, torch.float32, src_dtype=torch.int64)
    tmp1612 = ops.constant(0.0066844919786096255, torch.float32)
    tmp1613 = tmp1611 * tmp1612
    tmp1614 = ops.constant(-1.25, torch.float32)
    tmp1615 = tmp1613 + tmp1614
    tmp1616 = ops.constant(1.25, torch.float32)
    tmp1617 = ops.constant(374, torch.int64)
    tmp1618 = ops.index_expr(i0, dtype=torch.int64)
    tmp1619 = tmp1617 - tmp1618
    tmp1620 = ops.to_dtype(tmp1619, torch.float32, src_dtype=torch.int64)
    tmp1621 = ops.constant(0.0066844919786096255, torch.float32)
    tmp1622 = tmp1620 * tmp1621
    tmp1623 = tmp1616 - tmp1622
    tmp1624 = ops.where(tmp1609, tmp1615, tmp1623)
    tmp1625 = tmp1605 * tmp1624
    tmp1626 = ops.index_expr(i0, dtype=torch.int64)
    tmp1627 = ops.to_dtype(tmp1626, torch.float32, src_dtype=torch.int64)
    tmp1628 = ops.constant(187.5, torch.float32)
    tmp1629 = tmp1627 < tmp1628
    tmp1630 = ops.index_expr(i0, dtype=torch.int64)
    tmp1631 = ops.to_dtype(tmp1630, torch.float32, src_dtype=torch.int64)
    tmp1632 = ops.constant(0.0066844919786096255, torch.float32)
    tmp1633 = tmp1631 * tmp1632
    tmp1634 = ops.constant(-1.25, torch.float32)
    tmp1635 = tmp1633 + tmp1634
    tmp1636 = ops.constant(1.25, torch.float32)
    tmp1637 = ops.constant(374, torch.int64)
    tmp1638 = ops.index_expr(i0, dtype=torch.int64)
    tmp1639 = tmp1637 - tmp1638
    tmp1640 = ops.to_dtype(tmp1639, torch.float32, src_dtype=torch.int64)
    tmp1641 = ops.constant(0.0066844919786096255, torch.float32)
    tmp1642 = tmp1640 * tmp1641
    tmp1643 = tmp1636 - tmp1642
    tmp1644 = ops.where(tmp1629, tmp1635, tmp1643)
    tmp1645 = tmp1625 + tmp1644
    tmp1646 = tmp1584 * tmp1645
    tmp1647 = ops.index_expr(i0, dtype=torch.int64)
    tmp1648 = ops.to_dtype(tmp1647, torch.float32, src_dtype=torch.int64)
    tmp1649 = ops.constant(187.5, torch.float32)
    tmp1650 = tmp1648 < tmp1649
    tmp1651 = ops.index_expr(i0, dtype=torch.int64)
    tmp1652 = ops.to_dtype(tmp1651, torch.float32, src_dtype=torch.int64)
    tmp1653 = ops.constant(0.0066844919786096255, torch.float32)
    tmp1654 = tmp1652 * tmp1653
    tmp1655 = ops.constant(-1.25, torch.float32)
    tmp1656 = tmp1654 + tmp1655
    tmp1657 = ops.constant(1.25, torch.float32)
    tmp1658 = ops.constant(374, torch.int64)
    tmp1659 = ops.index_expr(i0, dtype=torch.int64)
    tmp1660 = tmp1658 - tmp1659
    tmp1661 = ops.to_dtype(tmp1660, torch.float32, src_dtype=torch.int64)
    tmp1662 = ops.constant(0.0066844919786096255, torch.float32)
    tmp1663 = tmp1661 * tmp1662
    tmp1664 = tmp1657 - tmp1663
    tmp1665 = ops.where(tmp1650, tmp1656, tmp1664)
    tmp1666 = tmp1646 + tmp1665
    tmp1667 = tmp1521 * tmp1666
    tmp1668 = ops.index_expr(i0, dtype=torch.int64)
    tmp1669 = ops.to_dtype(tmp1668, torch.float32, src_dtype=torch.int64)
    tmp1670 = ops.constant(187.5, torch.float32)
    tmp1671 = tmp1669 < tmp1670
    tmp1672 = ops.index_expr(i0, dtype=torch.int64)
    tmp1673 = ops.to_dtype(tmp1672, torch.float32, src_dtype=torch.int64)
    tmp1674 = ops.constant(0.0066844919786096255, torch.float32)
    tmp1675 = tmp1673 * tmp1674
    tmp1676 = ops.constant(-1.25, torch.float32)
    tmp1677 = tmp1675 + tmp1676
    tmp1678 = ops.constant(1.25, torch.float32)
    tmp1679 = ops.constant(374, torch.int64)
    tmp1680 = ops.index_expr(i0, dtype=torch.int64)
    tmp1681 = tmp1679 - tmp1680
    tmp1682 = ops.to_dtype(tmp1681, torch.float32, src_dtype=torch.int64)
    tmp1683 = ops.constant(0.0066844919786096255, torch.float32)
    tmp1684 = tmp1682 * tmp1683
    tmp1685 = tmp1678 - tmp1684
    tmp1686 = ops.where(tmp1671, tmp1677, tmp1685)
    tmp1687 = tmp1667 + tmp1686
    tmp1688 = tmp1687 * tmp1687
    tmp1689 = tmp1374 - tmp1688
    tmp1690 = ops.index_expr(i1, dtype=torch.int64)
    tmp1691 = ops.to_dtype(tmp1690, torch.float32, src_dtype=torch.int64)
    tmp1692 = ops.constant(225.0, torch.float32)
    tmp1693 = tmp1691 < tmp1692
    tmp1694 = ops.index_expr(i1, dtype=torch.int64)
    tmp1695 = ops.to_dtype(tmp1694, torch.float32, src_dtype=torch.int64)
    tmp1696 = ops.constant(0.0066815144766146995, torch.float32)
    tmp1697 = tmp1695 * tmp1696
    tmp1698 = ops.constant(-2.25, torch.float32)
    tmp1699 = tmp1697 + tmp1698
    tmp1700 = ops.constant(0.75, torch.float32)
    tmp1701 = ops.constant(449, torch.int64)
    tmp1702 = ops.index_expr(i1, dtype=torch.int64)
    tmp1703 = tmp1701 - tmp1702
    tmp1704 = ops.to_dtype(tmp1703, torch.float32, src_dtype=torch.int64)
    tmp1705 = ops.constant(0.0066815144766146995, torch.float32)
    tmp1706 = tmp1704 * tmp1705
    tmp1707 = tmp1700 - tmp1706
    tmp1708 = ops.where(tmp1693, tmp1699, tmp1707)
    tmp1709 = tmp1689 + tmp1708
    tmp1710 = tmp1709 * tmp1709
    tmp1711 = ops.index_expr(i1, dtype=torch.int64)
    tmp1712 = ops.to_dtype(tmp1711, torch.float32, src_dtype=torch.int64)
    tmp1713 = ops.constant(225.0, torch.float32)
    tmp1714 = tmp1712 < tmp1713
    tmp1715 = ops.index_expr(i1, dtype=torch.int64)
    tmp1716 = ops.to_dtype(tmp1715, torch.float32, src_dtype=torch.int64)
    tmp1717 = ops.constant(0.0066815144766146995, torch.float32)
    tmp1718 = tmp1716 * tmp1717
    tmp1719 = ops.constant(-2.25, torch.float32)
    tmp1720 = tmp1718 + tmp1719
    tmp1721 = ops.constant(0.75, torch.float32)
    tmp1722 = ops.constant(449, torch.int64)
    tmp1723 = ops.index_expr(i1, dtype=torch.int64)
    tmp1724 = tmp1722 - tmp1723
    tmp1725 = ops.to_dtype(tmp1724, torch.float32, src_dtype=torch.int64)
    tmp1726 = ops.constant(0.0066815144766146995, torch.float32)
    tmp1727 = tmp1725 * tmp1726
    tmp1728 = tmp1721 - tmp1727
    tmp1729 = ops.where(tmp1714, tmp1720, tmp1728)
    tmp1730 = tmp1729 * tmp1729
    tmp1731 = ops.index_expr(i0, dtype=torch.int64)
    tmp1732 = ops.to_dtype(tmp1731, torch.float32, src_dtype=torch.int64)
    tmp1733 = ops.constant(187.5, torch.float32)
    tmp1734 = tmp1732 < tmp1733
    tmp1735 = ops.index_expr(i0, dtype=torch.int64)
    tmp1736 = ops.to_dtype(tmp1735, torch.float32, src_dtype=torch.int64)
    tmp1737 = ops.constant(0.0066844919786096255, torch.float32)
    tmp1738 = tmp1736 * tmp1737
    tmp1739 = ops.constant(-1.25, torch.float32)
    tmp1740 = tmp1738 + tmp1739
    tmp1741 = ops.constant(1.25, torch.float32)
    tmp1742 = ops.constant(374, torch.int64)
    tmp1743 = ops.index_expr(i0, dtype=torch.int64)
    tmp1744 = tmp1742 - tmp1743
    tmp1745 = ops.to_dtype(tmp1744, torch.float32, src_dtype=torch.int64)
    tmp1746 = ops.constant(0.0066844919786096255, torch.float32)
    tmp1747 = tmp1745 * tmp1746
    tmp1748 = tmp1741 - tmp1747
    tmp1749 = ops.where(tmp1734, tmp1740, tmp1748)
    tmp1750 = tmp1749 * tmp1749
    tmp1751 = tmp1730 - tmp1750
    tmp1752 = ops.index_expr(i1, dtype=torch.int64)
    tmp1753 = ops.to_dtype(tmp1752, torch.float32, src_dtype=torch.int64)
    tmp1754 = ops.constant(225.0, torch.float32)
    tmp1755 = tmp1753 < tmp1754
    tmp1756 = ops.index_expr(i1, dtype=torch.int64)
    tmp1757 = ops.to_dtype(tmp1756, torch.float32, src_dtype=torch.int64)
    tmp1758 = ops.constant(0.0066815144766146995, torch.float32)
    tmp1759 = tmp1757 * tmp1758
    tmp1760 = ops.constant(-2.25, torch.float32)
    tmp1761 = tmp1759 + tmp1760
    tmp1762 = ops.constant(0.75, torch.float32)
    tmp1763 = ops.constant(449, torch.int64)
    tmp1764 = ops.index_expr(i1, dtype=torch.int64)
    tmp1765 = tmp1763 - tmp1764
    tmp1766 = ops.to_dtype(tmp1765, torch.float32, src_dtype=torch.int64)
    tmp1767 = ops.constant(0.0066815144766146995, torch.float32)
    tmp1768 = tmp1766 * tmp1767
    tmp1769 = tmp1762 - tmp1768
    tmp1770 = ops.where(tmp1755, tmp1761, tmp1769)
    tmp1771 = tmp1751 + tmp1770
    tmp1772 = tmp1771 * tmp1771
    tmp1773 = ops.index_expr(i1, dtype=torch.int64)
    tmp1774 = ops.to_dtype(tmp1773, torch.float32, src_dtype=torch.int64)
    tmp1775 = ops.constant(225.0, torch.float32)
    tmp1776 = tmp1774 < tmp1775
    tmp1777 = ops.index_expr(i1, dtype=torch.int64)
    tmp1778 = ops.to_dtype(tmp1777, torch.float32, src_dtype=torch.int64)
    tmp1779 = ops.constant(0.0066815144766146995, torch.float32)
    tmp1780 = tmp1778 * tmp1779
    tmp1781 = ops.constant(-2.25, torch.float32)
    tmp1782 = tmp1780 + tmp1781
    tmp1783 = ops.constant(0.75, torch.float32)
    tmp1784 = ops.constant(449, torch.int64)
    tmp1785 = ops.index_expr(i1, dtype=torch.int64)
    tmp1786 = tmp1784 - tmp1785
    tmp1787 = ops.to_dtype(tmp1786, torch.float32, src_dtype=torch.int64)
    tmp1788 = ops.constant(0.0066815144766146995, torch.float32)
    tmp1789 = tmp1787 * tmp1788
    tmp1790 = tmp1783 - tmp1789
    tmp1791 = ops.where(tmp1776, tmp1782, tmp1790)
    tmp1792 = ops.constant(2, torch.float32)
    tmp1793 = tmp1791 * tmp1792
    tmp1794 = ops.index_expr(i0, dtype=torch.int64)
    tmp1795 = ops.to_dtype(tmp1794, torch.float32, src_dtype=torch.int64)
    tmp1796 = ops.constant(187.5, torch.float32)
    tmp1797 = tmp1795 < tmp1796
    tmp1798 = ops.index_expr(i0, dtype=torch.int64)
    tmp1799 = ops.to_dtype(tmp1798, torch.float32, src_dtype=torch.int64)
    tmp1800 = ops.constant(0.0066844919786096255, torch.float32)
    tmp1801 = tmp1799 * tmp1800
    tmp1802 = ops.constant(-1.25, torch.float32)
    tmp1803 = tmp1801 + tmp1802
    tmp1804 = ops.constant(1.25, torch.float32)
    tmp1805 = ops.constant(374, torch.int64)
    tmp1806 = ops.index_expr(i0, dtype=torch.int64)
    tmp1807 = tmp1805 - tmp1806
    tmp1808 = ops.to_dtype(tmp1807, torch.float32, src_dtype=torch.int64)
    tmp1809 = ops.constant(0.0066844919786096255, torch.float32)
    tmp1810 = tmp1808 * tmp1809
    tmp1811 = tmp1804 - tmp1810
    tmp1812 = ops.where(tmp1797, tmp1803, tmp1811)
    tmp1813 = tmp1793 * tmp1812
    tmp1814 = ops.index_expr(i0, dtype=torch.int64)
    tmp1815 = ops.to_dtype(tmp1814, torch.float32, src_dtype=torch.int64)
    tmp1816 = ops.constant(187.5, torch.float32)
    tmp1817 = tmp1815 < tmp1816
    tmp1818 = ops.index_expr(i0, dtype=torch.int64)
    tmp1819 = ops.to_dtype(tmp1818, torch.float32, src_dtype=torch.int64)
    tmp1820 = ops.constant(0.0066844919786096255, torch.float32)
    tmp1821 = tmp1819 * tmp1820
    tmp1822 = ops.constant(-1.25, torch.float32)
    tmp1823 = tmp1821 + tmp1822
    tmp1824 = ops.constant(1.25, torch.float32)
    tmp1825 = ops.constant(374, torch.int64)
    tmp1826 = ops.index_expr(i0, dtype=torch.int64)
    tmp1827 = tmp1825 - tmp1826
    tmp1828 = ops.to_dtype(tmp1827, torch.float32, src_dtype=torch.int64)
    tmp1829 = ops.constant(0.0066844919786096255, torch.float32)
    tmp1830 = tmp1828 * tmp1829
    tmp1831 = tmp1824 - tmp1830
    tmp1832 = ops.where(tmp1817, tmp1823, tmp1831)
    tmp1833 = tmp1813 + tmp1832
    tmp1834 = tmp1833 * tmp1833
    tmp1835 = tmp1772 - tmp1834
    tmp1836 = ops.index_expr(i1, dtype=torch.int64)
    tmp1837 = ops.to_dtype(tmp1836, torch.float32, src_dtype=torch.int64)
    tmp1838 = ops.constant(225.0, torch.float32)
    tmp1839 = tmp1837 < tmp1838
    tmp1840 = ops.index_expr(i1, dtype=torch.int64)
    tmp1841 = ops.to_dtype(tmp1840, torch.float32, src_dtype=torch.int64)
    tmp1842 = ops.constant(0.0066815144766146995, torch.float32)
    tmp1843 = tmp1841 * tmp1842
    tmp1844 = ops.constant(-2.25, torch.float32)
    tmp1845 = tmp1843 + tmp1844
    tmp1846 = ops.constant(0.75, torch.float32)
    tmp1847 = ops.constant(449, torch.int64)
    tmp1848 = ops.index_expr(i1, dtype=torch.int64)
    tmp1849 = tmp1847 - tmp1848
    tmp1850 = ops.to_dtype(tmp1849, torch.float32, src_dtype=torch.int64)
    tmp1851 = ops.constant(0.0066815144766146995, torch.float32)
    tmp1852 = tmp1850 * tmp1851
    tmp1853 = tmp1846 - tmp1852
    tmp1854 = ops.where(tmp1839, tmp1845, tmp1853)
    tmp1855 = tmp1835 + tmp1854
    tmp1856 = tmp1855 * tmp1855
    tmp1857 = ops.index_expr(i1, dtype=torch.int64)
    tmp1858 = ops.to_dtype(tmp1857, torch.float32, src_dtype=torch.int64)
    tmp1859 = ops.constant(225.0, torch.float32)
    tmp1860 = tmp1858 < tmp1859
    tmp1861 = ops.index_expr(i1, dtype=torch.int64)
    tmp1862 = ops.to_dtype(tmp1861, torch.float32, src_dtype=torch.int64)
    tmp1863 = ops.constant(0.0066815144766146995, torch.float32)
    tmp1864 = tmp1862 * tmp1863
    tmp1865 = ops.constant(-2.25, torch.float32)
    tmp1866 = tmp1864 + tmp1865
    tmp1867 = ops.constant(0.75, torch.float32)
    tmp1868 = ops.constant(449, torch.int64)
    tmp1869 = ops.index_expr(i1, dtype=torch.int64)
    tmp1870 = tmp1868 - tmp1869
    tmp1871 = ops.to_dtype(tmp1870, torch.float32, src_dtype=torch.int64)
    tmp1872 = ops.constant(0.0066815144766146995, torch.float32)
    tmp1873 = tmp1871 * tmp1872
    tmp1874 = tmp1867 - tmp1873
    tmp1875 = ops.where(tmp1860, tmp1866, tmp1874)
    tmp1876 = tmp1875 * tmp1875
    tmp1877 = ops.index_expr(i0, dtype=torch.int64)
    tmp1878 = ops.to_dtype(tmp1877, torch.float32, src_dtype=torch.int64)
    tmp1879 = ops.constant(187.5, torch.float32)
    tmp1880 = tmp1878 < tmp1879
    tmp1881 = ops.index_expr(i0, dtype=torch.int64)
    tmp1882 = ops.to_dtype(tmp1881, torch.float32, src_dtype=torch.int64)
    tmp1883 = ops.constant(0.0066844919786096255, torch.float32)
    tmp1884 = tmp1882 * tmp1883
    tmp1885 = ops.constant(-1.25, torch.float32)
    tmp1886 = tmp1884 + tmp1885
    tmp1887 = ops.constant(1.25, torch.float32)
    tmp1888 = ops.constant(374, torch.int64)
    tmp1889 = ops.index_expr(i0, dtype=torch.int64)
    tmp1890 = tmp1888 - tmp1889
    tmp1891 = ops.to_dtype(tmp1890, torch.float32, src_dtype=torch.int64)
    tmp1892 = ops.constant(0.0066844919786096255, torch.float32)
    tmp1893 = tmp1891 * tmp1892
    tmp1894 = tmp1887 - tmp1893
    tmp1895 = ops.where(tmp1880, tmp1886, tmp1894)
    tmp1896 = tmp1895 * tmp1895
    tmp1897 = tmp1876 - tmp1896
    tmp1898 = ops.index_expr(i1, dtype=torch.int64)
    tmp1899 = ops.to_dtype(tmp1898, torch.float32, src_dtype=torch.int64)
    tmp1900 = ops.constant(225.0, torch.float32)
    tmp1901 = tmp1899 < tmp1900
    tmp1902 = ops.index_expr(i1, dtype=torch.int64)
    tmp1903 = ops.to_dtype(tmp1902, torch.float32, src_dtype=torch.int64)
    tmp1904 = ops.constant(0.0066815144766146995, torch.float32)
    tmp1905 = tmp1903 * tmp1904
    tmp1906 = ops.constant(-2.25, torch.float32)
    tmp1907 = tmp1905 + tmp1906
    tmp1908 = ops.constant(0.75, torch.float32)
    tmp1909 = ops.constant(449, torch.int64)
    tmp1910 = ops.index_expr(i1, dtype=torch.int64)
    tmp1911 = tmp1909 - tmp1910
    tmp1912 = ops.to_dtype(tmp1911, torch.float32, src_dtype=torch.int64)
    tmp1913 = ops.constant(0.0066815144766146995, torch.float32)
    tmp1914 = tmp1912 * tmp1913
    tmp1915 = tmp1908 - tmp1914
    tmp1916 = ops.where(tmp1901, tmp1907, tmp1915)
    tmp1917 = tmp1897 + tmp1916
    tmp1918 = ops.constant(2, torch.float32)
    tmp1919 = tmp1917 * tmp1918
    tmp1920 = ops.index_expr(i1, dtype=torch.int64)
    tmp1921 = ops.to_dtype(tmp1920, torch.float32, src_dtype=torch.int64)
    tmp1922 = ops.constant(225.0, torch.float32)
    tmp1923 = tmp1921 < tmp1922
    tmp1924 = ops.index_expr(i1, dtype=torch.int64)
    tmp1925 = ops.to_dtype(tmp1924, torch.float32, src_dtype=torch.int64)
    tmp1926 = ops.constant(0.0066815144766146995, torch.float32)
    tmp1927 = tmp1925 * tmp1926
    tmp1928 = ops.constant(-2.25, torch.float32)
    tmp1929 = tmp1927 + tmp1928
    tmp1930 = ops.constant(0.75, torch.float32)
    tmp1931 = ops.constant(449, torch.int64)
    tmp1932 = ops.index_expr(i1, dtype=torch.int64)
    tmp1933 = tmp1931 - tmp1932
    tmp1934 = ops.to_dtype(tmp1933, torch.float32, src_dtype=torch.int64)
    tmp1935 = ops.constant(0.0066815144766146995, torch.float32)
    tmp1936 = tmp1934 * tmp1935
    tmp1937 = tmp1930 - tmp1936
    tmp1938 = ops.where(tmp1923, tmp1929, tmp1937)
    tmp1939 = ops.constant(2, torch.float32)
    tmp1940 = tmp1938 * tmp1939
    tmp1941 = ops.index_expr(i0, dtype=torch.int64)
    tmp1942 = ops.to_dtype(tmp1941, torch.float32, src_dtype=torch.int64)
    tmp1943 = ops.constant(187.5, torch.float32)
    tmp1944 = tmp1942 < tmp1943
    tmp1945 = ops.index_expr(i0, dtype=torch.int64)
    tmp1946 = ops.to_dtype(tmp1945, torch.float32, src_dtype=torch.int64)
    tmp1947 = ops.constant(0.0066844919786096255, torch.float32)
    tmp1948 = tmp1946 * tmp1947
    tmp1949 = ops.constant(-1.25, torch.float32)
    tmp1950 = tmp1948 + tmp1949
    tmp1951 = ops.constant(1.25, torch.float32)
    tmp1952 = ops.constant(374, torch.int64)
    tmp1953 = ops.index_expr(i0, dtype=torch.int64)
    tmp1954 = tmp1952 - tmp1953
    tmp1955 = ops.to_dtype(tmp1954, torch.float32, src_dtype=torch.int64)
    tmp1956 = ops.constant(0.0066844919786096255, torch.float32)
    tmp1957 = tmp1955 * tmp1956
    tmp1958 = tmp1951 - tmp1957
    tmp1959 = ops.where(tmp1944, tmp1950, tmp1958)
    tmp1960 = tmp1940 * tmp1959
    tmp1961 = ops.index_expr(i0, dtype=torch.int64)
    tmp1962 = ops.to_dtype(tmp1961, torch.float32, src_dtype=torch.int64)
    tmp1963 = ops.constant(187.5, torch.float32)
    tmp1964 = tmp1962 < tmp1963
    tmp1965 = ops.index_expr(i0, dtype=torch.int64)
    tmp1966 = ops.to_dtype(tmp1965, torch.float32, src_dtype=torch.int64)
    tmp1967 = ops.constant(0.0066844919786096255, torch.float32)
    tmp1968 = tmp1966 * tmp1967
    tmp1969 = ops.constant(-1.25, torch.float32)
    tmp1970 = tmp1968 + tmp1969
    tmp1971 = ops.constant(1.25, torch.float32)
    tmp1972 = ops.constant(374, torch.int64)
    tmp1973 = ops.index_expr(i0, dtype=torch.int64)
    tmp1974 = tmp1972 - tmp1973
    tmp1975 = ops.to_dtype(tmp1974, torch.float32, src_dtype=torch.int64)
    tmp1976 = ops.constant(0.0066844919786096255, torch.float32)
    tmp1977 = tmp1975 * tmp1976
    tmp1978 = tmp1971 - tmp1977
    tmp1979 = ops.where(tmp1964, tmp1970, tmp1978)
    tmp1980 = tmp1960 + tmp1979
    tmp1981 = tmp1919 * tmp1980
    tmp1982 = ops.index_expr(i0, dtype=torch.int64)
    tmp1983 = ops.to_dtype(tmp1982, torch.float32, src_dtype=torch.int64)
    tmp1984 = ops.constant(187.5, torch.float32)
    tmp1985 = tmp1983 < tmp1984
    tmp1986 = ops.index_expr(i0, dtype=torch.int64)
    tmp1987 = ops.to_dtype(tmp1986, torch.float32, src_dtype=torch.int64)
    tmp1988 = ops.constant(0.0066844919786096255, torch.float32)
    tmp1989 = tmp1987 * tmp1988
    tmp1990 = ops.constant(-1.25, torch.float32)
    tmp1991 = tmp1989 + tmp1990
    tmp1992 = ops.constant(1.25, torch.float32)
    tmp1993 = ops.constant(374, torch.int64)
    tmp1994 = ops.index_expr(i0, dtype=torch.int64)
    tmp1995 = tmp1993 - tmp1994
    tmp1996 = ops.to_dtype(tmp1995, torch.float32, src_dtype=torch.int64)
    tmp1997 = ops.constant(0.0066844919786096255, torch.float32)
    tmp1998 = tmp1996 * tmp1997
    tmp1999 = tmp1992 - tmp1998
    tmp2000 = ops.where(tmp1985, tmp1991, tmp1999)
    tmp2001 = tmp1981 + tmp2000
    tmp2002 = tmp2001 * tmp2001
    tmp2003 = tmp1856 - tmp2002
    tmp2004 = ops.index_expr(i1, dtype=torch.int64)
    tmp2005 = ops.to_dtype(tmp2004, torch.float32, src_dtype=torch.int64)
    tmp2006 = ops.constant(225.0, torch.float32)
    tmp2007 = tmp2005 < tmp2006
    tmp2008 = ops.index_expr(i1, dtype=torch.int64)
    tmp2009 = ops.to_dtype(tmp2008, torch.float32, src_dtype=torch.int64)
    tmp2010 = ops.constant(0.0066815144766146995, torch.float32)
    tmp2011 = tmp2009 * tmp2010
    tmp2012 = ops.constant(-2.25, torch.float32)
    tmp2013 = tmp2011 + tmp2012
    tmp2014 = ops.constant(0.75, torch.float32)
    tmp2015 = ops.constant(449, torch.int64)
    tmp2016 = ops.index_expr(i1, dtype=torch.int64)
    tmp2017 = tmp2015 - tmp2016
    tmp2018 = ops.to_dtype(tmp2017, torch.float32, src_dtype=torch.int64)
    tmp2019 = ops.constant(0.0066815144766146995, torch.float32)
    tmp2020 = tmp2018 * tmp2019
    tmp2021 = tmp2014 - tmp2020
    tmp2022 = ops.where(tmp2007, tmp2013, tmp2021)
    tmp2023 = tmp2003 + tmp2022
    tmp2024 = ops.constant(2, torch.float32)
    tmp2025 = tmp2023 * tmp2024
    tmp2026 = ops.index_expr(i1, dtype=torch.int64)
    tmp2027 = ops.to_dtype(tmp2026, torch.float32, src_dtype=torch.int64)
    tmp2028 = ops.constant(225.0, torch.float32)
    tmp2029 = tmp2027 < tmp2028
    tmp2030 = ops.index_expr(i1, dtype=torch.int64)
    tmp2031 = ops.to_dtype(tmp2030, torch.float32, src_dtype=torch.int64)
    tmp2032 = ops.constant(0.0066815144766146995, torch.float32)
    tmp2033 = tmp2031 * tmp2032
    tmp2034 = ops.constant(-2.25, torch.float32)
    tmp2035 = tmp2033 + tmp2034
    tmp2036 = ops.constant(0.75, torch.float32)
    tmp2037 = ops.constant(449, torch.int64)
    tmp2038 = ops.index_expr(i1, dtype=torch.int64)
    tmp2039 = tmp2037 - tmp2038
    tmp2040 = ops.to_dtype(tmp2039, torch.float32, src_dtype=torch.int64)
    tmp2041 = ops.constant(0.0066815144766146995, torch.float32)
    tmp2042 = tmp2040 * tmp2041
    tmp2043 = tmp2036 - tmp2042
    tmp2044 = ops.where(tmp2029, tmp2035, tmp2043)
    tmp2045 = tmp2044 * tmp2044
    tmp2046 = ops.index_expr(i0, dtype=torch.int64)
    tmp2047 = ops.to_dtype(tmp2046, torch.float32, src_dtype=torch.int64)
    tmp2048 = ops.constant(187.5, torch.float32)
    tmp2049 = tmp2047 < tmp2048
    tmp2050 = ops.index_expr(i0, dtype=torch.int64)
    tmp2051 = ops.to_dtype(tmp2050, torch.float32, src_dtype=torch.int64)
    tmp2052 = ops.constant(0.0066844919786096255, torch.float32)
    tmp2053 = tmp2051 * tmp2052
    tmp2054 = ops.constant(-1.25, torch.float32)
    tmp2055 = tmp2053 + tmp2054
    tmp2056 = ops.constant(1.25, torch.float32)
    tmp2057 = ops.constant(374, torch.int64)
    tmp2058 = ops.index_expr(i0, dtype=torch.int64)
    tmp2059 = tmp2057 - tmp2058
    tmp2060 = ops.to_dtype(tmp2059, torch.float32, src_dtype=torch.int64)
    tmp2061 = ops.constant(0.0066844919786096255, torch.float32)
    tmp2062 = tmp2060 * tmp2061
    tmp2063 = tmp2056 - tmp2062
    tmp2064 = ops.where(tmp2049, tmp2055, tmp2063)
    tmp2065 = tmp2064 * tmp2064
    tmp2066 = tmp2045 - tmp2065
    tmp2067 = ops.index_expr(i1, dtype=torch.int64)
    tmp2068 = ops.to_dtype(tmp2067, torch.float32, src_dtype=torch.int64)
    tmp2069 = ops.constant(225.0, torch.float32)
    tmp2070 = tmp2068 < tmp2069
    tmp2071 = ops.index_expr(i1, dtype=torch.int64)
    tmp2072 = ops.to_dtype(tmp2071, torch.float32, src_dtype=torch.int64)
    tmp2073 = ops.constant(0.0066815144766146995, torch.float32)
    tmp2074 = tmp2072 * tmp2073
    tmp2075 = ops.constant(-2.25, torch.float32)
    tmp2076 = tmp2074 + tmp2075
    tmp2077 = ops.constant(0.75, torch.float32)
    tmp2078 = ops.constant(449, torch.int64)
    tmp2079 = ops.index_expr(i1, dtype=torch.int64)
    tmp2080 = tmp2078 - tmp2079
    tmp2081 = ops.to_dtype(tmp2080, torch.float32, src_dtype=torch.int64)
    tmp2082 = ops.constant(0.0066815144766146995, torch.float32)
    tmp2083 = tmp2081 * tmp2082
    tmp2084 = tmp2077 - tmp2083
    tmp2085 = ops.where(tmp2070, tmp2076, tmp2084)
    tmp2086 = tmp2066 + tmp2085
    tmp2087 = tmp2086 * tmp2086
    tmp2088 = ops.index_expr(i1, dtype=torch.int64)
    tmp2089 = ops.to_dtype(tmp2088, torch.float32, src_dtype=torch.int64)
    tmp2090 = ops.constant(225.0, torch.float32)
    tmp2091 = tmp2089 < tmp2090
    tmp2092 = ops.index_expr(i1, dtype=torch.int64)
    tmp2093 = ops.to_dtype(tmp2092, torch.float32, src_dtype=torch.int64)
    tmp2094 = ops.constant(0.0066815144766146995, torch.float32)
    tmp2095 = tmp2093 * tmp2094
    tmp2096 = ops.constant(-2.25, torch.float32)
    tmp2097 = tmp2095 + tmp2096
    tmp2098 = ops.constant(0.75, torch.float32)
    tmp2099 = ops.constant(449, torch.int64)
    tmp2100 = ops.index_expr(i1, dtype=torch.int64)
    tmp2101 = tmp2099 - tmp2100
    tmp2102 = ops.to_dtype(tmp2101, torch.float32, src_dtype=torch.int64)
    tmp2103 = ops.constant(0.0066815144766146995, torch.float32)
    tmp2104 = tmp2102 * tmp2103
    tmp2105 = tmp2098 - tmp2104
    tmp2106 = ops.where(tmp2091, tmp2097, tmp2105)
    tmp2107 = ops.constant(2, torch.float32)
    tmp2108 = tmp2106 * tmp2107
    tmp2109 = ops.index_expr(i0, dtype=torch.int64)
    tmp2110 = ops.to_dtype(tmp2109, torch.float32, src_dtype=torch.int64)
    tmp2111 = ops.constant(187.5, torch.float32)
    tmp2112 = tmp2110 < tmp2111
    tmp2113 = ops.index_expr(i0, dtype=torch.int64)
    tmp2114 = ops.to_dtype(tmp2113, torch.float32, src_dtype=torch.int64)
    tmp2115 = ops.constant(0.0066844919786096255, torch.float32)
    tmp2116 = tmp2114 * tmp2115
    tmp2117 = ops.constant(-1.25, torch.float32)
    tmp2118 = tmp2116 + tmp2117
    tmp2119 = ops.constant(1.25, torch.float32)
    tmp2120 = ops.constant(374, torch.int64)
    tmp2121 = ops.index_expr(i0, dtype=torch.int64)
    tmp2122 = tmp2120 - tmp2121
    tmp2123 = ops.to_dtype(tmp2122, torch.float32, src_dtype=torch.int64)
    tmp2124 = ops.constant(0.0066844919786096255, torch.float32)
    tmp2125 = tmp2123 * tmp2124
    tmp2126 = tmp2119 - tmp2125
    tmp2127 = ops.where(tmp2112, tmp2118, tmp2126)
    tmp2128 = tmp2108 * tmp2127
    tmp2129 = ops.index_expr(i0, dtype=torch.int64)
    tmp2130 = ops.to_dtype(tmp2129, torch.float32, src_dtype=torch.int64)
    tmp2131 = ops.constant(187.5, torch.float32)
    tmp2132 = tmp2130 < tmp2131
    tmp2133 = ops.index_expr(i0, dtype=torch.int64)
    tmp2134 = ops.to_dtype(tmp2133, torch.float32, src_dtype=torch.int64)
    tmp2135 = ops.constant(0.0066844919786096255, torch.float32)
    tmp2136 = tmp2134 * tmp2135
    tmp2137 = ops.constant(-1.25, torch.float32)
    tmp2138 = tmp2136 + tmp2137
    tmp2139 = ops.constant(1.25, torch.float32)
    tmp2140 = ops.constant(374, torch.int64)
    tmp2141 = ops.index_expr(i0, dtype=torch.int64)
    tmp2142 = tmp2140 - tmp2141
    tmp2143 = ops.to_dtype(tmp2142, torch.float32, src_dtype=torch.int64)
    tmp2144 = ops.constant(0.0066844919786096255, torch.float32)
    tmp2145 = tmp2143 * tmp2144
    tmp2146 = tmp2139 - tmp2145
    tmp2147 = ops.where(tmp2132, tmp2138, tmp2146)
    tmp2148 = tmp2128 + tmp2147
    tmp2149 = tmp2148 * tmp2148
    tmp2150 = tmp2087 - tmp2149
    tmp2151 = ops.index_expr(i1, dtype=torch.int64)
    tmp2152 = ops.to_dtype(tmp2151, torch.float32, src_dtype=torch.int64)
    tmp2153 = ops.constant(225.0, torch.float32)
    tmp2154 = tmp2152 < tmp2153
    tmp2155 = ops.index_expr(i1, dtype=torch.int64)
    tmp2156 = ops.to_dtype(tmp2155, torch.float32, src_dtype=torch.int64)
    tmp2157 = ops.constant(0.0066815144766146995, torch.float32)
    tmp2158 = tmp2156 * tmp2157
    tmp2159 = ops.constant(-2.25, torch.float32)
    tmp2160 = tmp2158 + tmp2159
    tmp2161 = ops.constant(0.75, torch.float32)
    tmp2162 = ops.constant(449, torch.int64)
    tmp2163 = ops.index_expr(i1, dtype=torch.int64)
    tmp2164 = tmp2162 - tmp2163
    tmp2165 = ops.to_dtype(tmp2164, torch.float32, src_dtype=torch.int64)
    tmp2166 = ops.constant(0.0066815144766146995, torch.float32)
    tmp2167 = tmp2165 * tmp2166
    tmp2168 = tmp2161 - tmp2167
    tmp2169 = ops.where(tmp2154, tmp2160, tmp2168)
    tmp2170 = tmp2150 + tmp2169
    tmp2171 = ops.constant(2, torch.float32)
    tmp2172 = tmp2170 * tmp2171
    tmp2173 = ops.index_expr(i1, dtype=torch.int64)
    tmp2174 = ops.to_dtype(tmp2173, torch.float32, src_dtype=torch.int64)
    tmp2175 = ops.constant(225.0, torch.float32)
    tmp2176 = tmp2174 < tmp2175
    tmp2177 = ops.index_expr(i1, dtype=torch.int64)
    tmp2178 = ops.to_dtype(tmp2177, torch.float32, src_dtype=torch.int64)
    tmp2179 = ops.constant(0.0066815144766146995, torch.float32)
    tmp2180 = tmp2178 * tmp2179
    tmp2181 = ops.constant(-2.25, torch.float32)
    tmp2182 = tmp2180 + tmp2181
    tmp2183 = ops.constant(0.75, torch.float32)
    tmp2184 = ops.constant(449, torch.int64)
    tmp2185 = ops.index_expr(i1, dtype=torch.int64)
    tmp2186 = tmp2184 - tmp2185
    tmp2187 = ops.to_dtype(tmp2186, torch.float32, src_dtype=torch.int64)
    tmp2188 = ops.constant(0.0066815144766146995, torch.float32)
    tmp2189 = tmp2187 * tmp2188
    tmp2190 = tmp2183 - tmp2189
    tmp2191 = ops.where(tmp2176, tmp2182, tmp2190)
    tmp2192 = tmp2191 * tmp2191
    tmp2193 = ops.index_expr(i0, dtype=torch.int64)
    tmp2194 = ops.to_dtype(tmp2193, torch.float32, src_dtype=torch.int64)
    tmp2195 = ops.constant(187.5, torch.float32)
    tmp2196 = tmp2194 < tmp2195
    tmp2197 = ops.index_expr(i0, dtype=torch.int64)
    tmp2198 = ops.to_dtype(tmp2197, torch.float32, src_dtype=torch.int64)
    tmp2199 = ops.constant(0.0066844919786096255, torch.float32)
    tmp2200 = tmp2198 * tmp2199
    tmp2201 = ops.constant(-1.25, torch.float32)
    tmp2202 = tmp2200 + tmp2201
    tmp2203 = ops.constant(1.25, torch.float32)
    tmp2204 = ops.constant(374, torch.int64)
    tmp2205 = ops.index_expr(i0, dtype=torch.int64)
    tmp2206 = tmp2204 - tmp2205
    tmp2207 = ops.to_dtype(tmp2206, torch.float32, src_dtype=torch.int64)
    tmp2208 = ops.constant(0.0066844919786096255, torch.float32)
    tmp2209 = tmp2207 * tmp2208
    tmp2210 = tmp2203 - tmp2209
    tmp2211 = ops.where(tmp2196, tmp2202, tmp2210)
    tmp2212 = tmp2211 * tmp2211
    tmp2213 = tmp2192 - tmp2212
    tmp2214 = ops.index_expr(i1, dtype=torch.int64)
    tmp2215 = ops.to_dtype(tmp2214, torch.float32, src_dtype=torch.int64)
    tmp2216 = ops.constant(225.0, torch.float32)
    tmp2217 = tmp2215 < tmp2216
    tmp2218 = ops.index_expr(i1, dtype=torch.int64)
    tmp2219 = ops.to_dtype(tmp2218, torch.float32, src_dtype=torch.int64)
    tmp2220 = ops.constant(0.0066815144766146995, torch.float32)
    tmp2221 = tmp2219 * tmp2220
    tmp2222 = ops.constant(-2.25, torch.float32)
    tmp2223 = tmp2221 + tmp2222
    tmp2224 = ops.constant(0.75, torch.float32)
    tmp2225 = ops.constant(449, torch.int64)
    tmp2226 = ops.index_expr(i1, dtype=torch.int64)
    tmp2227 = tmp2225 - tmp2226
    tmp2228 = ops.to_dtype(tmp2227, torch.float32, src_dtype=torch.int64)
    tmp2229 = ops.constant(0.0066815144766146995, torch.float32)
    tmp2230 = tmp2228 * tmp2229
    tmp2231 = tmp2224 - tmp2230
    tmp2232 = ops.where(tmp2217, tmp2223, tmp2231)
    tmp2233 = tmp2213 + tmp2232
    tmp2234 = ops.constant(2, torch.float32)
    tmp2235 = tmp2233 * tmp2234
    tmp2236 = ops.index_expr(i1, dtype=torch.int64)
    tmp2237 = ops.to_dtype(tmp2236, torch.float32, src_dtype=torch.int64)
    tmp2238 = ops.constant(225.0, torch.float32)
    tmp2239 = tmp2237 < tmp2238
    tmp2240 = ops.index_expr(i1, dtype=torch.int64)
    tmp2241 = ops.to_dtype(tmp2240, torch.float32, src_dtype=torch.int64)
    tmp2242 = ops.constant(0.0066815144766146995, torch.float32)
    tmp2243 = tmp2241 * tmp2242
    tmp2244 = ops.constant(-2.25, torch.float32)
    tmp2245 = tmp2243 + tmp2244
    tmp2246 = ops.constant(0.75, torch.float32)
    tmp2247 = ops.constant(449, torch.int64)
    tmp2248 = ops.index_expr(i1, dtype=torch.int64)
    tmp2249 = tmp2247 - tmp2248
    tmp2250 = ops.to_dtype(tmp2249, torch.float32, src_dtype=torch.int64)
    tmp2251 = ops.constant(0.0066815144766146995, torch.float32)
    tmp2252 = tmp2250 * tmp2251
    tmp2253 = tmp2246 - tmp2252
    tmp2254 = ops.where(tmp2239, tmp2245, tmp2253)
    tmp2255 = ops.constant(2, torch.float32)
    tmp2256 = tmp2254 * tmp2255
    tmp2257 = ops.index_expr(i0, dtype=torch.int64)
    tmp2258 = ops.to_dtype(tmp2257, torch.float32, src_dtype=torch.int64)
    tmp2259 = ops.constant(187.5, torch.float32)
    tmp2260 = tmp2258 < tmp2259
    tmp2261 = ops.index_expr(i0, dtype=torch.int64)
    tmp2262 = ops.to_dtype(tmp2261, torch.float32, src_dtype=torch.int64)
    tmp2263 = ops.constant(0.0066844919786096255, torch.float32)
    tmp2264 = tmp2262 * tmp2263
    tmp2265 = ops.constant(-1.25, torch.float32)
    tmp2266 = tmp2264 + tmp2265
    tmp2267 = ops.constant(1.25, torch.float32)
    tmp2268 = ops.constant(374, torch.int64)
    tmp2269 = ops.index_expr(i0, dtype=torch.int64)
    tmp2270 = tmp2268 - tmp2269
    tmp2271 = ops.to_dtype(tmp2270, torch.float32, src_dtype=torch.int64)
    tmp2272 = ops.constant(0.0066844919786096255, torch.float32)
    tmp2273 = tmp2271 * tmp2272
    tmp2274 = tmp2267 - tmp2273
    tmp2275 = ops.where(tmp2260, tmp2266, tmp2274)
    tmp2276 = tmp2256 * tmp2275
    tmp2277 = ops.index_expr(i0, dtype=torch.int64)
    tmp2278 = ops.to_dtype(tmp2277, torch.float32, src_dtype=torch.int64)
    tmp2279 = ops.constant(187.5, torch.float32)
    tmp2280 = tmp2278 < tmp2279
    tmp2281 = ops.index_expr(i0, dtype=torch.int64)
    tmp2282 = ops.to_dtype(tmp2281, torch.float32, src_dtype=torch.int64)
    tmp2283 = ops.constant(0.0066844919786096255, torch.float32)
    tmp2284 = tmp2282 * tmp2283
    tmp2285 = ops.constant(-1.25, torch.float32)
    tmp2286 = tmp2284 + tmp2285
    tmp2287 = ops.constant(1.25, torch.float32)
    tmp2288 = ops.constant(374, torch.int64)
    tmp2289 = ops.index_expr(i0, dtype=torch.int64)
    tmp2290 = tmp2288 - tmp2289
    tmp2291 = ops.to_dtype(tmp2290, torch.float32, src_dtype=torch.int64)
    tmp2292 = ops.constant(0.0066844919786096255, torch.float32)
    tmp2293 = tmp2291 * tmp2292
    tmp2294 = tmp2287 - tmp2293
    tmp2295 = ops.where(tmp2280, tmp2286, tmp2294)
    tmp2296 = tmp2276 + tmp2295
    tmp2297 = tmp2235 * tmp2296
    tmp2298 = ops.index_expr(i0, dtype=torch.int64)
    tmp2299 = ops.to_dtype(tmp2298, torch.float32, src_dtype=torch.int64)
    tmp2300 = ops.constant(187.5, torch.float32)
    tmp2301 = tmp2299 < tmp2300
    tmp2302 = ops.index_expr(i0, dtype=torch.int64)
    tmp2303 = ops.to_dtype(tmp2302, torch.float32, src_dtype=torch.int64)
    tmp2304 = ops.constant(0.0066844919786096255, torch.float32)
    tmp2305 = tmp2303 * tmp2304
    tmp2306 = ops.constant(-1.25, torch.float32)
    tmp2307 = tmp2305 + tmp2306
    tmp2308 = ops.constant(1.25, torch.float32)
    tmp2309 = ops.constant(374, torch.int64)
    tmp2310 = ops.index_expr(i0, dtype=torch.int64)
    tmp2311 = tmp2309 - tmp2310
    tmp2312 = ops.to_dtype(tmp2311, torch.float32, src_dtype=torch.int64)
    tmp2313 = ops.constant(0.0066844919786096255, torch.float32)
    tmp2314 = tmp2312 * tmp2313
    tmp2315 = tmp2308 - tmp2314
    tmp2316 = ops.where(tmp2301, tmp2307, tmp2315)
    tmp2317 = tmp2297 + tmp2316
    tmp2318 = tmp2172 * tmp2317
    tmp2319 = ops.index_expr(i0, dtype=torch.int64)
    tmp2320 = ops.to_dtype(tmp2319, torch.float32, src_dtype=torch.int64)
    tmp2321 = ops.constant(187.5, torch.float32)
    tmp2322 = tmp2320 < tmp2321
    tmp2323 = ops.index_expr(i0, dtype=torch.int64)
    tmp2324 = ops.to_dtype(tmp2323, torch.float32, src_dtype=torch.int64)
    tmp2325 = ops.constant(0.0066844919786096255, torch.float32)
    tmp2326 = tmp2324 * tmp2325
    tmp2327 = ops.constant(-1.25, torch.float32)
    tmp2328 = tmp2326 + tmp2327
    tmp2329 = ops.constant(1.25, torch.float32)
    tmp2330 = ops.constant(374, torch.int64)
    tmp2331 = ops.index_expr(i0, dtype=torch.int64)
    tmp2332 = tmp2330 - tmp2331
    tmp2333 = ops.to_dtype(tmp2332, torch.float32, src_dtype=torch.int64)
    tmp2334 = ops.constant(0.0066844919786096255, torch.float32)
    tmp2335 = tmp2333 * tmp2334
    tmp2336 = tmp2329 - tmp2335
    tmp2337 = ops.where(tmp2322, tmp2328, tmp2336)
    tmp2338 = tmp2318 + tmp2337
    tmp2339 = tmp2025 * tmp2338
    tmp2340 = ops.index_expr(i0, dtype=torch.int64)
    tmp2341 = ops.to_dtype(tmp2340, torch.float32, src_dtype=torch.int64)
    tmp2342 = ops.constant(187.5, torch.float32)
    tmp2343 = tmp2341 < tmp2342
    tmp2344 = ops.index_expr(i0, dtype=torch.int64)
    tmp2345 = ops.to_dtype(tmp2344, torch.float32, src_dtype=torch.int64)
    tmp2346 = ops.constant(0.0066844919786096255, torch.float32)
    tmp2347 = tmp2345 * tmp2346
    tmp2348 = ops.constant(-1.25, torch.float32)
    tmp2349 = tmp2347 + tmp2348
    tmp2350 = ops.constant(1.25, torch.float32)
    tmp2351 = ops.constant(374, torch.int64)
    tmp2352 = ops.index_expr(i0, dtype=torch.int64)
    tmp2353 = tmp2351 - tmp2352
    tmp2354 = ops.to_dtype(tmp2353, torch.float32, src_dtype=torch.int64)
    tmp2355 = ops.constant(0.0066844919786096255, torch.float32)
    tmp2356 = tmp2354 * tmp2355
    tmp2357 = tmp2350 - tmp2356
    tmp2358 = ops.where(tmp2343, tmp2349, tmp2357)
    tmp2359 = tmp2339 + tmp2358
    tmp2360 = tmp2359 * tmp2359
    tmp2361 = tmp1710 + tmp2360
    tmp2362 = ops.sqrt(tmp2361)
    tmp2363 = ops.constant(2, torch.float32)
    tmp2364 = tmp2362 <= tmp2363
    tmp2365 = ops.to_dtype(tmp2364, torch.float32, src_dtype=torch.bool)
    tmp2366 = tmp1060 + tmp2365
    return tmp2366