diff options
Diffstat (limited to 'src/gpu/cl/kernels/ClWeightsReshapeKernel.h')
| -rw-r--r-- | src/gpu/cl/kernels/ClWeightsReshapeKernel.h | 11 |
1 files changed, 8 insertions, 3 deletions
diff --git a/src/gpu/cl/kernels/ClWeightsReshapeKernel.h b/src/gpu/cl/kernels/ClWeightsReshapeKernel.h index 7364eb97ae..5e05f8d006 100644 --- a/src/gpu/cl/kernels/ClWeightsReshapeKernel.h +++ b/src/gpu/cl/kernels/ClWeightsReshapeKernel.h @@ -75,14 +75,19 @@ public: * @param[in] num_groups (Optional) Number of groups when performing a grouped convolution. num_groups != 1 is only supported for NCHW data layout * Number of groups greater than one are only supported for NCHW data layout, and the number of weights must be a multiple of it. */ - void configure(const ClCompileContext &compile_context, const ITensorInfo *src, const ITensorInfo *biases, ITensorInfo *dst, unsigned int num_groups = 1); + void configure(const ClCompileContext &compile_context, + const ITensorInfo *src, + const ITensorInfo *biases, + ITensorInfo *dst, + unsigned int num_groups = 1); /** Static function to check if given info will lead to a valid configuration * * Similar to ClWeightsReshapeKernel::configure() * * @return a status */ - static Status validate(const ITensorInfo *src, const ITensorInfo *biases, const ITensorInfo *dst, unsigned int num_groups = 1); + static Status + validate(const ITensorInfo *src, const ITensorInfo *biases, const ITensorInfo *dst, unsigned int num_groups = 1); // Inherited methods overridden: void run_op(ITensorPack &tensors, const Window &window, cl::CommandQueue &queue) override; @@ -90,4 +95,4 @@ public: } // namespace kernels } // namespace opencl } // namespace arm_compute -#endif /*ARM_COMPUTE_CL_WEIGHTSRESHAPE_KERNEL_H */
\ No newline at end of file +#endif /*ARM_COMPUTE_CL_WEIGHTSRESHAPE_KERNEL_H */ |