normalization.py
5.58 KB
-
Fix the device inference in eager execution · 8dbb73a7
Summary: This commit computes the correct device for an existing output tensor.
Ting PAN committed
Summary: This commit computes the correct device for an existing output tensor.