-
- Downloads
Enhance support for 'tf.IndexedSlices' in 'TensorflowVector'.
* Implement a public util to wrap tensorflow operations in order to preserve `tf.IndexedSlices` structures and run appropriate computations with them. * Deploy the former wrapper to cover all usual operations in the backend of `TensorflowVector`. * Refactor the use of the device-placement-handling wrapper in the backend of `TensorflowVector` to reduce runtime overheads and factor it with the new indexed-slices-handling wrapper.
parent
71debe68
No related branches found
No related tags found
Showing
- declearn/model/tensorflow/_vector.py 69 additions, 20 deletionsdeclearn/model/tensorflow/_vector.py
- declearn/model/tensorflow/utils/__init__.py 5 additions, 0 deletionsdeclearn/model/tensorflow/utils/__init__.py
- declearn/model/tensorflow/utils/_slices.py 137 additions, 0 deletionsdeclearn/model/tensorflow/utils/_slices.py
- declearn/optimizer/modules/_noise.py 5 additions, 1 deletiondeclearn/optimizer/modules/_noise.py
- declearn/test_utils/_vectors.py 18 additions, 1 deletiondeclearn/test_utils/_vectors.py
- test/model/test_tflow.py 2 additions, 0 deletionstest/model/test_tflow.py
- test/optimizer/test_tflow_optim.py 9 additions, 2 deletionstest/optimizer/test_tflow_optim.py
Loading
Please register or sign in to comment