Closed wwwind closed 2 years ago
When the data_format is 'channels_first', clustering per channel is not working correctly due to mismatch in arguments. This CR fixed this case + added test that we set the argument correctly.
The reported bug is https://github.com/tensorflow/model-optimization/issues/939
When the data_format is 'channels_first', clustering per channel is not working correctly due to mismatch in arguments. This CR fixed this case + added test that we set the argument correctly.
The reported bug is https://github.com/tensorflow/model-optimization/issues/939