(batch_size, num_threads, shape)
| 37 | |
| 38 | |
| 39 | def check_deserialization_with_params(batch_size, num_threads, shape): |
| 40 | init_pipe = TestPipeline(batch_size=batch_size, num_threads=num_threads, shape=shape) |
| 41 | serialized = init_pipe.serialize() |
| 42 | ref_pipe = TestPipeline(batch_size=batch_size**2, num_threads=num_threads + 1, shape=shape) |
| 43 | test_pipe = Pipeline.deserialize( |
| 44 | serialized, batch_size=batch_size**2, num_threads=num_threads + 1 |
| 45 | ) |
| 46 | test_utils.compare_pipelines(ref_pipe, test_pipe, batch_size=batch_size**2, N_iterations=3) |
| 47 | |
| 48 | |
| 49 | def check_deserialization_from_file(batch_size, num_threads, shape): |
nothing calls this directly
no test coverage detected