Commit 03fb5daa authored by Tim Shen's avatar Tim Shen Committed by TensorFlower Gardener
Browse files

Simplify PrepareForConvolution: 

* Change the virtual functions overloaded on DeviceMemory<T> to a single function DeviceMemoryBase.
* Create a non-virtual, template wrapper that takes DeviceMemory<T>.

PiperOrigin-RevId: 228818212
parent ed7a8b4d
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