diff options
Diffstat (limited to 'src/gpu/cl/operators/ClScale.h')
| -rw-r--r-- | src/gpu/cl/operators/ClScale.h | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/src/gpu/cl/operators/ClScale.h b/src/gpu/cl/operators/ClScale.h index af97cf23e7..1427bb4fdc 100644 --- a/src/gpu/cl/operators/ClScale.h +++ b/src/gpu/cl/operators/ClScale.h @@ -25,6 +25,7 @@ #define ARM_COMPUTE_CL_SCALE_H #include "arm_compute/core/KernelDescriptors.h" + #include "src/gpu/cl/ClCompileContext.h" #include "src/gpu/cl/IClOperator.h" @@ -49,7 +50,8 @@ public: * All but the lowest two dimensions must be the same size as in the input tensor, i.e. scaling is only performed within the XY-plane. * @param[in] info @ref ScaleKernelInfo descriptor to be used to configure */ - void configure(const CLCompileContext &compile_context, ITensorInfo *src, ITensorInfo *dst, const ScaleKernelInfo &info); + void + configure(const CLCompileContext &compile_context, ITensorInfo *src, ITensorInfo *dst, const ScaleKernelInfo &info); /** Static function to check if given info will lead to a valid configuration * * Similar to ClScale::configure() |