()
| 1243 | def impl(device, shape_a, shape_b): |
| 1244 | @pipeline_def(batch_size=1, num_threads=3, device_id=0) |
| 1245 | def pipe(): |
| 1246 | a = fn.random.uniform(range=[-1, 1], shape=shape_a) |
| 1247 | b = fn.random.uniform(range=[-1, 1], shape=shape_b) |
| 1248 | return a + b |
| 1249 | |
| 1250 | p = pipe() |
| 1251 | p.run() |