diff options
Diffstat (limited to 'src/core/NEON/kernels/arm_conv/depthwise/depthwise_implementation.hpp')
-rw-r--r-- | src/core/NEON/kernels/arm_conv/depthwise/depthwise_implementation.hpp | 13 |
1 files changed, 4 insertions, 9 deletions
diff --git a/src/core/NEON/kernels/arm_conv/depthwise/depthwise_implementation.hpp b/src/core/NEON/kernels/arm_conv/depthwise/depthwise_implementation.hpp index 1ee19e5075..0f91fe363c 100644 --- a/src/core/NEON/kernels/arm_conv/depthwise/depthwise_implementation.hpp +++ b/src/core/NEON/kernels/arm_conv/depthwise/depthwise_implementation.hpp @@ -55,7 +55,9 @@ struct DepthwiseImplementation DepthwiseCommon<TInput, TWeight, TOutput> *get_instance(const DepthwiseArgs &args, const OutputStage &os) const { - return initialise(args, os); + auto impl = initialise(args, os); + impl->set_name(std::string(name)); + return impl; } }; @@ -136,14 +138,7 @@ UniqueDepthwiseCommon<TInput, TWeight, TOutput> depthwise(const DepthwiseArgs &a { const DepthwiseImplementation<TInput, TWeight, TOutput, OutputStage> *impl = nullptr; const bool success = find_implementation<TInput, TWeight, TOutput, OutputStage>(args, os, impl); - - if(success) - { - auto i = impl->get_instance(args, os); - i->set_name(impl->name); - return UniqueDepthwiseCommon<TInput, TWeight, TOutput>(i); - } - return nullptr; + return UniqueDepthwiseCommon<TInput, TWeight, TOutput>(success ? impl->get_instance(args, os) : nullptr); } } // namespace depthwise |