Commit 79c68501 authored by Adrian Kuegel's avatar Adrian Kuegel Committed by TensorFlower Gardener
Browse files

Use feature_group_count parameter for depthwise convolution. 

This makes use of the newly added HLO support for grouped convolutions.

PiperOrigin-RevId: 209118249
parent 467ab9d9
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