(shape, dtype=dtype, device=device)
| 90 | primitive_batch_norm = trace(symbolic=symbolic)(primitive_batch_norm) |
| 91 | |
| 92 | def rand_tensor(shape, dtype=dtype, device=device): |
| 93 | return megengine.tensor(np.random.random(shape), dtype=dtype, device=device) |
| 94 | |
| 95 | # skip this test because could not do several reduce sequentially with opr cache |
| 96 | return |