(axes)
| 427 | @tc.returns(core.Axes) |
| 428 | @tc.accepts(core.Axes) |
| 429 | def output_axes(axes): |
| 430 | if enqueue_many: |
| 431 | if 'batch' not in axes or list(axes.keys()).index('batch') != 0: |
| 432 | raise ValueError( |
| 433 | 'When enqueue_many is True, input tensors must have an axis ' |
| 434 | 'called "batch" as their first dimension, ' |
| 435 | 'but axes were %s' % axes) |
| 436 | culled_axes = axes.remove('batch') |
| 437 | return core.Axes([('batch', batch_size)] + list(culled_axes.values())) |
| 438 | else: |
| 439 | return core.Axes([('batch', batch_size)] + list(axes.values())) |
| 440 | |
| 441 | output_labeled_tensors = [] |
| 442 | for i, tensor in enumerate(batch_ops): |
no test coverage detected