(model, input_data)
| 225 | |
| 226 | |
| 227 | def tune_input_shape(model, input_data): |
| 228 | param_shapes = model.get_parameter_shapes() |
| 229 | input_shapes = {} |
| 230 | for name, s in param_shapes.items(): |
| 231 | assert name in input_data |
| 232 | data_shape = list(input_data[name].shape) |
| 233 | if not np.array_equal(data_shape, s.lens()): |
| 234 | input_shapes[name] = data_shape |
| 235 | |
| 236 | return input_shapes |
| 237 | |
| 238 | |
| 239 | def main(): |
no test coverage detected