Commit f16b50e3 authored by ImSheridan's avatar ImSheridan Committed by Sourabh Bajaj
Browse files

Support a single Tensor in StagingArea.put() (#17863) 

* Support a single Tensor in a StagingArea.put()

* Fix the input values parameter could also be dictionary type
parent 10878ede
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