input_layer.py
2.89 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.