| 184 | return builder.apply_const(value, dtype, device) |
| 185 | |
| 186 | def build(builder, outputs, outputs_has_grad): |
| 187 | builder = type(builder)(builder) |
| 188 | builder.outputs(outputs) |
| 189 | builder.outputs_has_grad(outputs_has_grad) |
| 190 | if jit_fusion: |
| 191 | assert gopt_level is None |
| 192 | op = lambda: builder.jit_fuse() |
| 193 | elif gopt_level is None: |
| 194 | op = lambda: builder.get() |
| 195 | else: |
| 196 | op = lambda: builder.compile(gopt_level) |
| 197 | return op |
| 198 | |
| 199 | inputs = [builder.input() for _ in range(nr_inputs)] |
| 200 | if not custom_grad: |