xla
ec123533 - Simplify output shape for BatchNormForward

Commit

6 years ago

Simplify output shape for BatchNormForward We know the output shape is the same as the input shape.

References

Author

asuhan

Committer

dlibenzi

Parents