Skip to content

Commit

Permalink
Replace eigen3 NeuralNetworks include with eigen_attention.h in atten…
Browse files Browse the repository at this point in the history 
…tion_ops.cc to fix build error

Change: 118964563
  • Loading branch information
@andrewharp @tensorflower-gardener
andrewharp authored and tensorflower-gardener committed Apr 4, 2016
1 parent 81b25ec commit 2814fd9
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion tensorflow/core/kernels/attention_ops.cc
View file
Original file line number Diff line number Diff line change
Expand Up @@ -18,12 +18,12 @@ limitations under the License.
#define EIGEN_USE_THREADS

#include <vector>
#include "third_party/eigen3/unsupported/Eigen/CXX11/NeuralNetworks"
#include "tensorflow/core/framework/op.h"
#include "tensorflow/core/framework/op_kernel.h"
#include "tensorflow/core/framework/tensor.h"
#include "tensorflow/core/framework/tensor_shape.h"
#include "tensorflow/core/framework/types.h"
#include "tensorflow/core/kernels/eigen_attention.h"
#include "tensorflow/core/platform/logging.h"
#include "tensorflow/core/platform/types.h"

Expand Down

0 comments on commit 2814fd9

Please sign in to comment.