diff options
author | Moritz Pflanzer <moritz.pflanzer@arm.com> | 2017-09-07 09:48:04 +0100 |
---|---|---|
committer | Anthony Barbier <anthony.barbier@arm.com> | 2018-11-02 16:35:24 +0000 |
commit | c186b574e52b81c75e551cee46a6c4cc7d500c90 (patch) | |
tree | 682add7d1b705eb6d0b79b1f9a7a039dbaa4287b /arm_compute/core/NEON/kernels/NESoftmaxLayerKernel.h | |
parent | 3e05e4e85912e745b8555102e1bcef13478d2ceb (diff) | |
download | ComputeLibrary-c186b574e52b81c75e551cee46a6c4cc7d500c90.tar.gz |
COMPMID-481: Add thread info parameter
Change-Id: Iebb50a88d017445b6b37a86563ebd4abd86c5cf5
Reviewed-on: http://mpd-gerrit.cambridge.arm.com/86788
Tested-by: Kaizen <jeremy.johnson+kaizengerrit@arm.com>
Reviewed-by: Anthony Barbier <anthony.barbier@arm.com>
Diffstat (limited to 'arm_compute/core/NEON/kernels/NESoftmaxLayerKernel.h')
-rw-r--r-- | arm_compute/core/NEON/kernels/NESoftmaxLayerKernel.h | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/arm_compute/core/NEON/kernels/NESoftmaxLayerKernel.h b/arm_compute/core/NEON/kernels/NESoftmaxLayerKernel.h index 2caef55498..4897a39421 100644 --- a/arm_compute/core/NEON/kernels/NESoftmaxLayerKernel.h +++ b/arm_compute/core/NEON/kernels/NESoftmaxLayerKernel.h @@ -45,7 +45,7 @@ public: void configure(const ITensor *input, ITensor *output); // Inherited methods overridden: - void run(const Window &window) override; + void run(const Window &window, const ThreadInfo &info) override; BorderSize border_size() const override; private: @@ -82,7 +82,7 @@ public: void configure(const ITensor *input, const ITensor *max, ITensor *output, ITensor *sum); // Inherited methods overridden: - void run(const Window &window) override; + void run(const Window &window, const ThreadInfo &info) override; private: using Logits1DShiftExpSumFunction = void(const ITensor *in, const ITensor *max, ITensor *out, ITensor *sum, const Window &window); @@ -120,7 +120,7 @@ public: void configure(const ITensor *input, const ITensor *sum, ITensor *output); // Inherited methods overridden: - void run(const Window &window) override; + void run(const Window &window, const ThreadInfo &info) override; private: using Logits1DNormFunction = void(const ITensor *in, const ITensor *sum, ITensor *out, const Window &window); |