diff options
Diffstat (limited to 'tensorflow/core/kernels/attention_ops.cc')
| -rw-r--r-- | tensorflow/core/kernels/attention_ops.cc | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/tensorflow/core/kernels/attention_ops.cc b/tensorflow/core/kernels/attention_ops.cc index ebc39a1fde..768df239d6 100644 --- a/tensorflow/core/kernels/attention_ops.cc +++ b/tensorflow/core/kernels/attention_ops.cc @@ -21,11 +21,11 @@ limitations under the License. #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/platform/logging.h" -#include "tensorflow/core/platform/port.h" -#include "tensorflow/core/public/tensor.h" -#include "tensorflow/core/public/tensor_shape.h" +#include "tensorflow/core/platform/types.h" namespace tensorflow { |