Commit 3f08f82a authored by Adrian Kuegel's avatar Adrian Kuegel Committed by TensorFlower Gardener
Browse files

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
parent b7f74b43
Show whitespace changes
Inline Side-by-side
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment