Searched refs:CLFuseBatchNormalizationKernel (Results 1 – 6 of 6) sorted by relevance
35 class CLFuseBatchNormalizationKernel : public ICLKernel39 CLFuseBatchNormalizationKernel();41 CLFuseBatchNormalizationKernel(const CLFuseBatchNormalizationKernel &) = delete;43 CLFuseBatchNormalizationKernel &operator=(const CLFuseBatchNormalizationKernel &) = delete;45 CLFuseBatchNormalizationKernel(CLFuseBatchNormalizationKernel &&) = default;47 CLFuseBatchNormalizationKernel &operator=(CLFuseBatchNormalizationKernel &&) = default;49 ~CLFuseBatchNormalizationKernel() = default;
102 CLFuseBatchNormalizationKernel::CLFuseBatchNormalizationKernel() in CLFuseBatchNormalizationKernel() function in arm_compute::CLFuseBatchNormalizationKernel109 void CLFuseBatchNormalizationKernel::configure(const ICLTensor *input_weights, const ICLTensor *bn_… in configure()117 void CLFuseBatchNormalizationKernel::configure(const CLCompileContext &compile_context, const ICLTe… in configure()182 Status CLFuseBatchNormalizationKernel::validate(const ITensorInfo *input_weights, const ITensorInfo… in validate()191 void CLFuseBatchNormalizationKernel::run(const arm_compute::Window &window, cl::CommandQueue &queue) in run()
36 class CLFuseBatchNormalizationKernel; variable129 std::unique_ptr<CLFuseBatchNormalizationKernel> _fuse_bn_kernel;
38 : _fuse_bn_kernel(std::make_unique<CLFuseBatchNormalizationKernel>()) in CLFuseBatchNormalization()66 …return CLFuseBatchNormalizationKernel::validate(input_weights, bn_mean, bn_var, fused_weights, fus… in validate()
460 - @ref CLFuseBatchNormalizationKernel
135 "src/core/CL/kernels/CLFuseBatchNormalizationKernel.cpp",