diff options
author | Jean Flaherty <kobejean@me.com> | 2018-01-29 17:03:31 -0700 |
---|---|---|
committer | Rasmus Munk Larsen <rmlarsen@google.com> | 2018-01-29 16:03:31 -0800 |
commit | 4ab2d8531c461169cd6a33bc0fef1129b419e9df (patch) | |
tree | f32970777a805b9ce6e57e933be813dd6d5e1247 /tensorflow/cc | |
parent | 8c1f3b3fcdc5f897dbce3b7315c1b7000069f9c4 (diff) |
Tensor roll op implementation (#14953)
* Half migrated to manip
Half migrated to tf.manip
* Roll op: polymorphism & GPU attempt
* Roll op: Added support for gradients
Added compile script
* Rebase for roll op code
* Roll op: Migrated to manip namespace
* Roll op: Supports CPU thread pooling
fix namespace error
* Remove roll from user_ops
* Roll op: Optimization
* Roll op: Pylint fix
* Roll op: Updated documentation
* Roll op: Two versions for CPU thread pooling
* Roll op: Huge CPU speed up
Fixed thread pooling issue that was due to a bad cost_per_unit parameter
Also improved readability
* Roll op: Rough draft of DoRollV2
DoRollV2 copies memory in groups instead of element by element.
Not thoroughly tested yet.
Polished DoRollV2 algorithm
* Roll op: Restrict tensor size for GPU implementation
* Roll op: Fixed clang-format and missing include
* Roll op: Minor change
* Roll op GPU bug fix
Roll op GPU bug fix
GPU bug fix
Roll op GPU bug fix
Roll op GPU fix
Roll GPU test BUILD update
* Roll op: Remove GPU code
Fully remove roll op GPU code
Remove compile_cpu.sh
* Roll op: Fixes problems with array_ops_test.py and a size 1 dimension bug
* Roll op: Migrated to manip
Migrated to tf.manip
Roll op registered
Roll op uses InlinedVector
Small improvements
* Roll op: Revert array op changes
* Roll op: Api def fix
* Roll op: review changes
* Roll op: API review changes
Roll op: Docstring fix
* Roll op: Review changes round 1
* Roll op: resolve conflicts
* Roll op: Resolve conflicts
* Roll op: clang-tidy
* Roll op: Review round 2 changes
Roll op: fixed BUILD file
Roll op: api docs update
* Roll op: failure fixes 1
- updates goldens and fixes api compatibility issue
- fixes python op test issue for windows
- fixes makefile issues
* Roll op: Windows CMake failure fix
Windows CMake checks were failing because numpy was on an older version that did not support np.roll on multiple shifts that was use to check the correctness of tf.manip.roll. manip_ops_test.py now checks for numpy version 1.12.0 before testing multiple shifts, otherwise it'll just test single shift roll.
* Roll op: pylint changes
Diffstat (limited to 'tensorflow/cc')
-rw-r--r-- | tensorflow/cc/BUILD | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/tensorflow/cc/BUILD b/tensorflow/cc/BUILD index c9ade5fb83..9060c19e9d 100644 --- a/tensorflow/cc/BUILD +++ b/tensorflow/cc/BUILD @@ -433,6 +433,7 @@ tf_gen_op_wrappers_cc( "linalg_ops", "logging_ops", "lookup_ops", + "manip_ops", "math_ops", "nn_ops", "no_op", |