#include <stdexcept>
#include "internal/Padding.h"
-#include "internal/kernels/cpufallback/ConvolutionLayer.h"
-#include "internal/kernels/cpufallback/AvgPoolLayer.h"
-#include "internal/kernels/cpufallback/MaxPoolLayer.h"
-#include "internal/kernels/cpufallback/ConcatLayer.h"
-#include "internal/kernels/cpufallback/FullyConnectedLayer.h"
-#include "internal/kernels/cpufallback/ReshapeLayer.h"
-#include "internal/kernels/cpufallback/SoftMaxLayer.h"
+#include "internal/kernel/cpufallback/ConvolutionLayer.h"
+#include "internal/kernel/cpufallback/AvgPoolLayer.h"
+#include "internal/kernel/cpufallback/MaxPoolLayer.h"
+#include "internal/kernel/cpufallback/ConcatLayer.h"
+#include "internal/kernel/cpufallback/FullyConnectedLayer.h"
+#include "internal/kernel/cpufallback/ReshapeLayer.h"
+#include "internal/kernel/cpufallback/SoftMaxLayer.h"
#include "logging.h"
auto ker_alloc = tensors->at(::internal::tflite::operand::Index{param.ker_index});
auto bias_alloc = tensors->at(::internal::tflite::operand::Index{param.bias_index});
- std::unique_ptr<::internal::kernels::cpu::ConvolutionLayer> fn{
- new ::internal::kernels::cpu::ConvolutionLayer};
+ std::unique_ptr<::internal::kernel::cpu::ConvolutionLayer> fn{
+ new ::internal::kernel::cpu::ConvolutionLayer};
fn->configure(ifm_alloc->buffer(), param.ifm_shape, ker_alloc->buffer(), param.ker_shape,
bias_alloc->buffer(), param.bias_shape, param.padding.left, param.padding.right,
auto ofm_alloc = tensors->at(::internal::tflite::operand::Index{param.ofm_index}).get();
auto ifm_alloc = tensors->at(::internal::tflite::operand::Index{param.ifm_index}).get();
- std::unique_ptr<::internal::kernels::cpu::MaxPoolLayer> fn{
- new ::internal::kernels::cpu::MaxPoolLayer};
+ std::unique_ptr<::internal::kernel::cpu::MaxPoolLayer> fn{
+ new ::internal::kernel::cpu::MaxPoolLayer};
fn->configure(ifm_alloc->buffer(), param.ifm_shape, param.padding.left, param.padding.right,
param.padding.top, param.padding.bottom, param.stride.horizontal,
auto ofm_alloc = tensors->at(::internal::tflite::operand::Index{param.ofm_index}).get();
auto ifm_alloc = tensors->at(::internal::tflite::operand::Index{param.ifm_index}).get();
- std::unique_ptr<::internal::kernels::cpu::AvgPoolLayer> fn{
- new ::internal::kernels::cpu::AvgPoolLayer};
+ std::unique_ptr<::internal::kernel::cpu::AvgPoolLayer> fn{
+ new ::internal::kernel::cpu::AvgPoolLayer};
fn->configure(ifm_alloc->buffer(), param.ifm_shape, param.padding.left, param.padding.right,
param.padding.top, param.padding.bottom, param.stride.horizontal,
tensors->at(::internal::tflite::operand::Index{ifm_ind}).get()->buffer());
}
- std::unique_ptr<::internal::kernels::cpu::ConcatLayer> fn{
- new ::internal::kernels::cpu::ConcatLayer};
+ std::unique_ptr<::internal::kernel::cpu::ConcatLayer> fn{
+ new ::internal::kernel::cpu::ConcatLayer};
fn->configure(input_buffers, param.ifm_shapes, param.axis, output_alloc->buffer(),
param.ofm_shape);
auto weight_alloc = tensors->at(::internal::tflite::operand::Index{param.weight_index}).get();
auto bias_alloc = tensors->at(::internal::tflite::operand::Index{param.bias_index}).get();
- std::unique_ptr<::internal::kernels::cpu::FullyConnectedLayer> fn{
- new ::internal::kernels::cpu::FullyConnectedLayer};
+ std::unique_ptr<::internal::kernel::cpu::FullyConnectedLayer> fn{
+ new ::internal::kernel::cpu::FullyConnectedLayer};
fn->configure(input_alloc->buffer(), param.ifm_shape, weight_alloc->buffer(),
param.weight_shape, bias_alloc->buffer(), param.bias_shape, param.activation,
auto output_alloc = tensors->at(::internal::tflite::operand::Index{param.output_index}).get();
auto input_alloc = tensors->at(::internal::tflite::operand::Index{param.input_index}).get();
- std::unique_ptr<::internal::kernels::cpu::ReshapeLayer> fn{
- new ::internal::kernels::cpu::ReshapeLayer};
+ std::unique_ptr<::internal::kernel::cpu::ReshapeLayer> fn{
+ new ::internal::kernel::cpu::ReshapeLayer};
fn->configure(input_alloc->buffer(), param.ifm_shape, output_alloc->buffer(), param.ofm_shape);
auto output_alloc = tensors->at(::internal::tflite::operand::Index{param.output_index}).get();
auto input_alloc = tensors->at(::internal::tflite::operand::Index{param.input_index}).get();
- std::unique_ptr<::internal::kernels::cpu::SoftMaxLayer> fn{
- new ::internal::kernels::cpu::SoftMaxLayer};
+ std::unique_ptr<::internal::kernel::cpu::SoftMaxLayer> fn{
+ new ::internal::kernel::cpu::SoftMaxLayer};
fn->configure(input_alloc->buffer(), param.ifm_shape, param.scale, output_alloc->buffer(),
param.ofm_shape);