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

Add support for grouped convolutions to the HloEvaluator. 

Add a missing check to InferConvolveShape(), the output feature dimension needs to be divisible by feature_group_count.

Also fix some tests which took a const reference to the return value of
a function which doesn't return a reference.

PiperOrigin-RevId: 211592011
parent 568b7637
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