diff options
author | 2018-08-16 02:18:48 -0700 | |
---|---|---|
committer | 2018-08-16 02:22:18 -0700 | |
commit | 3f08f82a7c896d357262a47eda8df80b778a2672 (patch) | |
tree | b34f6644ec1be83f9b77f63d4858f5bbc3068ee0 /tensorflow/contrib/kfac/python/ops/utils.py | |
parent | b7f74b43aabd2953c06fd9aa40f8f6e2d62b7e6a (diff) |
Add a HloPass to rewrite convolutions with feature_group_count > 1.
The convolutions are rewritten to convolutions with feature_group_count = 1.
Adapt hlo_verifier and shape_inference to handle feature_group_count > 1.
Add the new pass to the backends, and write a depthwise convolution test.
PiperOrigin-RevId: 208954204
Diffstat (limited to 'tensorflow/contrib/kfac/python/ops/utils.py')
0 files changed, 0 insertions, 0 deletions