diff options
author | A. Unique TensorFlower <nobody@tensorflow.org> | 2016-03-17 08:30:51 -0800 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2016-03-18 08:44:53 -0700 |
commit | 6ff94239ef72159d50edcba664c4081bb70b4fde (patch) | |
tree | 67f108efe8cbabb8fd27939dcffb94109ada296b /tensorflow/core/kernels/relu_op_gpu.cu.cc | |
parent | 307f58ccbb567ba030de66850ecadb1dd2d4868d (diff) |
Move Relu and SoftMax Op declarations into header files so they can be registered by experimental devices.
Right now tensorflow/core/kernels explicitly depends on all Eigen devices that might want to implement any of the templated Eigen Ops. This is because the template classes that need to be specialized are defined in .cc files, so the specializations themselves have to appear there too. Moving the classes to .h files allows us to use arbitrary Eigen devices defined outside of tensorflow/core, which fits better with the intent behind core/kernels.
Over time more kernels may need to be refactored this way for the same reason.
Change: 117452814
Diffstat (limited to 'tensorflow/core/kernels/relu_op_gpu.cu.cc')
-rw-r--r-- | tensorflow/core/kernels/relu_op_gpu.cu.cc | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/core/kernels/relu_op_gpu.cu.cc b/tensorflow/core/kernels/relu_op_gpu.cu.cc index 6451619768..0a12c854b8 100644 --- a/tensorflow/core/kernels/relu_op_gpu.cu.cc +++ b/tensorflow/core/kernels/relu_op_gpu.cu.cc @@ -19,7 +19,7 @@ limitations under the License. #include <stdio.h> -#include "tensorflow/core/kernels/relu_op.h" +#include "tensorflow/core/kernels/relu_op_functor.h" #include "tensorflow/core/framework/register_types.h" #include "tensorflow/core/framework/tensor_types.h" |