()
| 94 | |
| 95 | @pipeline_def(enable_conditionals=True, num_threads=4, batch_size=8, device_id=0) |
| 96 | def pipeline_op(): |
| 97 | pred = fn.external_source(source=lambda x: np.array(x.idx_in_batch % 2), batch=False) |
| 98 | data = types.Constant(np.array(42), device="cpu") |
| 99 | if pred: |
| 100 | out = {"out": data - 1} |
| 101 | else: |
| 102 | out = {"out": data + 1} |
| 103 | return out["out"] |
| 104 | |
| 105 | pipe_op = pipeline_op() |
| 106 | (out,) = pipe_op.run() |