Commit 3a3b7530 authored by Yuxin Wu's avatar Yuxin Wu Committed by drpngx
Browse files

Support empty input tensor for some ops (fix #14657) (#15264) 

* Support empty input tensor for FusedBatchNorm,FusedBatchNormGrad,Conv2DBackpropFilter (fix #14657)

* Also fix pooling ops

* Add some comments in ops

* Add tests for conv/pooling/bn.

* Return NaN mean/variance when input is empty

* update comments

* fix typo

* Move fill_functor implementations to :fill_functor
parent 1c5f27a7
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