(self, batch_size, image_height, image_width)
| 428 | } |
| 429 | |
| 430 | def get_shape_dict(self, batch_size, image_height, image_width): |
| 431 | latent_height, latent_width = self.check_dims(batch_size, image_height, image_width) |
| 432 | return { |
| 433 | 'images': (batch_size, 3, image_height, image_width), |
| 434 | 'latent': (batch_size, 4, latent_height, latent_width) |
| 435 | } |
| 436 | |
| 437 | def get_sample_input(self, batch_size, image_height, image_width): |
| 438 | self.check_dims(batch_size, image_height, image_width) |
nothing calls this directly
no test coverage detected