(input_shape: Sequence[int], scale_num: int)
| 79 | """ |
| 80 | |
| 81 | def scale_batch_size(input_shape: Sequence[int], scale_num: int): |
| 82 | scale_shape = [*input_shape] |
| 83 | scale_shape[0] = scale_num |
| 84 | return scale_shape |
| 85 | |
| 86 | # Use the dynamic batchsize range to generate the min, opt and max model input shape |
| 87 | if dynamic_batchsize: |
no outgoing calls
no test coverage detected
searching dependent graphs…