blob: f7a5be5c2bd46bcd1224f7f0444e12f75de4763a (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
|
#ifndef TENSORFLOW_KERNELS_TRANSPOSE_OP_H_
#define TENSORFLOW_KERNELS_TRANSPOSE_OP_H_
#include "third_party/eigen3/unsupported/Eigen/CXX11/Tensor"
#include "tensorflow/core/framework/op_kernel.h"
#include "tensorflow/core/framework/tensor_types.h"
namespace tensorflow {
template <typename Device, typename T>
class TransposeOp : public OpKernel {
public:
explicit TransposeOp(OpKernelConstruction* context);
void Compute(OpKernelContext* context) override;
};
} // namespace tensorflow
#endif // TENSORFLOW_KERNELS_TRANSPOSE_OP_H_
|
