STT-tensorflow/tensorflow/python/framework/experimental
2020-10-15 21:55:35 +07:00
..
BUILD Implement python binding for TapeContext. 2020-09-28 15:05:07 -07:00
context_stack.py [C++ gradients] Move thread_local_stack to its own lib so that it can re-used for tape stack. 2020-09-10 12:21:11 -07:00
def_function.py
gradient_registry.py [C++ gradients] Add experimental python bindings for unified API's Tape. 2020-09-10 12:39:44 -07:00
math_ops.cc Add Sub gradients 2020-10-15 21:55:35 +07:00
math_ops.py Add Sub gradients 2020-10-15 21:55:35 +07:00
nn_ops.cc Implement python binding for TapeContext. 2020-09-28 15:05:07 -07:00
nn_ops.py Implement python binding for TapeContext. 2020-09-28 15:05:07 -07:00
tape.cc Add Sub gradients 2020-10-15 21:55:35 +07:00
tape.py Implement python binding for TapeContext. 2020-09-28 15:05:07 -07:00
thread_local_stack.py [C++ gradients] Move thread_local_stack to its own lib so that it can re-used for tape stack. 2020-09-10 12:21:11 -07:00
unified_api_test.py Add Sub gradients 2020-10-15 21:55:35 +07:00
unified_api.cc 1. Add helper for casting EagerTensor* to AbstractTensorHandle*. This allows us to directly work with EagerTensors and will also allow using tf.data with unified APIs. 2020-09-14 17:40:33 -07:00