#include <stdint.h>
#ifdef TFLITE_KERNEL_USE_XNNPACK
#include <array>
#include <vector>
#include "xnnpack.h"
#include "tensorflow/lite/kernels/cpu_backend_context.h"
#include "tensorflow/lite/logger.h"
#include "tensorflow/lite/minimal_logging.h"
#endif
#include "tensorflow/lite/core/c/common.h"
#include "tensorflow/lite/kernels/internal/optimized/optimized_ops.h"
#include "tensorflow/lite/kernels/internal/tensor_ctypes.h"
#include "tensorflow/lite/kernels/internal/types.h"
#include "tensorflow/lite/kernels/kernel_util.h"
namespace tflite {
namespace ops {
namespace builtin {
namespace transpose {
enum KernelType { … };
struct TransposeContext { … };
TfLiteStatus ResizeOutputTensor(TfLiteContext* context,
TransposeContext* op_context) { … }
TfLiteStatus Prepare(TfLiteContext* context, TfLiteNode* node) { … }
template <KernelType kernel_type>
TfLiteStatus Eval(TfLiteContext* context, TfLiteNode* node) { … }
}
TfLiteRegistration* Register_TRANSPOSE_REF() { … }
TfLiteRegistration* Register_TRANSPOSE_GENERIC_OPTIMIZED() { … }
TfLiteRegistration* Register_TRANSPOSE() { … }
}
}
}