From: 김수진/동작제어Lab(SR)/Engineer/삼성전자 Date: Mon, 9 Jul 2018 05:23:42 +0000 (+0900) Subject: [newrt] Remove CPU prefix of ConvolutionLayer class (#1891) X-Git-Tag: 0.2~491 X-Git-Url: http://review.tizen.org/git/?a=commitdiff_plain;h=f3c4aa9133e93804742c68d811fa7c7d54006ffb;p=platform%2Fcore%2Fml%2Fnnfw.git [newrt] Remove CPU prefix of ConvolutionLayer class (#1891) This commit removes CPU prefix of ConvolutionLayer class Signed-off-by: sjsujinkim --- diff --git a/runtimes/new_runtime/src/internal/cpu/StageGenerator.cc b/runtimes/new_runtime/src/internal/cpu/StageGenerator.cc index 189a5e6..a6deae5 100644 --- a/runtimes/new_runtime/src/internal/cpu/StageGenerator.cc +++ b/runtimes/new_runtime/src/internal/cpu/StageGenerator.cc @@ -3,7 +3,7 @@ #include #include "internal/Padding.h" -#include "internal/kernels/cpufallback/CPUConvolutionLayer.h" +#include "internal/kernels/cpufallback/ConvolutionLayer.h" #include "internal/kernels/cpufallback/AvgPoolLayer.h" #include "logging.h" @@ -95,8 +95,8 @@ Stage StageGenerator::generate(const ::internal::tflite::op::Conv2D::implicit::N 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::CPUConvolutionLayer> fn{ - new ::internal::kernels::cpu::CPUConvolutionLayer}; + std::unique_ptr<::internal::kernels::cpu::ConvolutionLayer> fn{ + new ::internal::kernels::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, diff --git a/runtimes/new_runtime/src/internal/kernels/cpufallback/CPUConvolutionLayer.cc b/runtimes/new_runtime/src/internal/kernels/cpufallback/ConvolutionLayer.cc similarity index 97% rename from runtimes/new_runtime/src/internal/kernels/cpufallback/CPUConvolutionLayer.cc rename to runtimes/new_runtime/src/internal/kernels/cpufallback/ConvolutionLayer.cc index 37eb11c..b96b906 100644 --- a/runtimes/new_runtime/src/internal/kernels/cpufallback/CPUConvolutionLayer.cc +++ b/runtimes/new_runtime/src/internal/kernels/cpufallback/ConvolutionLayer.cc @@ -1,4 +1,4 @@ -#include "CPUConvolutionLayer.h" +#include "ConvolutionLayer.h" #include "tensorflow/contrib/lite/kernels/internal/optimized/optimized_ops.h" #include "internal/kernels/cpufallback/OperationUtils.h" @@ -65,7 +65,7 @@ static char static_scratch_buffer[kStaticBufferSize]; im2colGuard.reset(im2colData); \ } -bool CPUConvolutionLayer::convFloat32() +bool ConvolutionLayer::convFloat32() { ANDROID_NN_CONV_PARAMETERS(float) @@ -82,7 +82,7 @@ bool CPUConvolutionLayer::convFloat32() return true; } -bool CPUConvolutionLayer::convQuant8() +bool ConvolutionLayer::convQuant8() { /* ANDROID_NN_CONV_PARAMETERS(uint8_t) @@ -133,7 +133,7 @@ bool CPUConvolutionLayer::convQuant8() return true; } -void CPUConvolutionLayer::configure( +void ConvolutionLayer::configure( uint8_t *inputData, const internal::tflite::operand::Shape inputShape, uint8_t *kernelData, const internal::tflite::operand::Shape kernelShape, uint8_t *biasData, const internal::tflite::operand::Shape biasShape, const uint32_t paddingLeft, @@ -158,7 +158,7 @@ void CPUConvolutionLayer::configure( _outputShape = convertShape(outputShape); } -void CPUConvolutionLayer::run() +void ConvolutionLayer::run() { convFloat32(); /* diff --git a/runtimes/new_runtime/src/internal/kernels/cpufallback/CPUConvolutionLayer.h b/runtimes/new_runtime/src/internal/kernels/cpufallback/ConvolutionLayer.h similarity index 84% rename from runtimes/new_runtime/src/internal/kernels/cpufallback/CPUConvolutionLayer.h rename to runtimes/new_runtime/src/internal/kernels/cpufallback/ConvolutionLayer.h index 0c784eb..365e55d 100644 --- a/runtimes/new_runtime/src/internal/kernels/cpufallback/CPUConvolutionLayer.h +++ b/runtimes/new_runtime/src/internal/kernels/cpufallback/ConvolutionLayer.h @@ -1,5 +1,5 @@ -#ifndef __INTERNAL_KERNELS_CPU_CPUCONVOLUTIONLAYER_H__ -#define __INTERNAL_KERNELS_CPU_CPUCONVOLUTIONLAYER_H__ +#ifndef __INTERNAL_KERNELS_CPU_CONVOLUTIONLAYER_H__ +#define __INTERNAL_KERNELS_CPU_CONVOLUTIONLAYER_H__ #include @@ -17,10 +17,10 @@ namespace kernels namespace cpu { -class CPUConvolutionLayer : public ::arm_compute::IFunction +class ConvolutionLayer : public ::arm_compute::IFunction { public: - CPUConvolutionLayer() {} + ConvolutionLayer() {} public: bool convFloat32(); @@ -63,4 +63,4 @@ private: } // namespace kernels } // namespace internal -#endif // __INTERNAL_KERNELS_CPU_CPUCONVOLUTIONLAYER_H__ +#endif // __INTERNAL_KERNELS_CPU_CONVOLUTIONLAYER_H__