diff options
Diffstat (limited to 'src/gpu/cl/kernels/ClConvertFullyConnectedWeightsKernel.h')
-rw-r--r-- | src/gpu/cl/kernels/ClConvertFullyConnectedWeightsKernel.h | 11 |
1 files changed, 9 insertions, 2 deletions
diff --git a/src/gpu/cl/kernels/ClConvertFullyConnectedWeightsKernel.h b/src/gpu/cl/kernels/ClConvertFullyConnectedWeightsKernel.h index 16000e82f6..0ddb54561a 100644 --- a/src/gpu/cl/kernels/ClConvertFullyConnectedWeightsKernel.h +++ b/src/gpu/cl/kernels/ClConvertFullyConnectedWeightsKernel.h @@ -55,14 +55,21 @@ public: * @param[in] original_src_shape Shape of the original src tensor (the one entering fully connected layer). * @param[in] data_layout The data layout the weights have been trained in. */ - void configure(const CLCompileContext &compile_context, const ITensorInfo *src, ITensorInfo *dst, const TensorShape &original_src_shape, DataLayout data_layout); + void configure(const CLCompileContext &compile_context, + const ITensorInfo *src, + ITensorInfo *dst, + const TensorShape &original_src_shape, + DataLayout data_layout); /** Static function to check if given info will lead to a valid configuration * * Similar to @ref ClConvertFullyConnectedWeightsKernel::configure() * * @return a status */ - static Status validate(const ITensorInfo *src, const ITensorInfo *dst, const TensorShape &original_src_shape, DataLayout data_layout); + static Status validate(const ITensorInfo *src, + const ITensorInfo *dst, + const TensorShape &original_src_shape, + DataLayout data_layout); // Inherited methods overridden: void run_op(ITensorPack &tensors, const Window &window, ::cl::CommandQueue &queue) override; |