Searched refs:InstanceNormalizationLayerKernelInfo (Results 1 – 7 of 7) sorted by relevance
| /aosp_15_r20/external/ComputeLibrary/arm_compute/core/ |
| H A D | KernelDescriptors.h | 142 struct InstanceNormalizationLayerKernelInfo struct
145 InstanceNormalizationLayerKernelInfo() in InstanceNormalizationLayerKernelInfo() function
146 : InstanceNormalizationLayerKernelInfo(1.f, 0.f, 1e-12, true) in InstanceNormalizationLayerKernelInfo()
156 …InstanceNormalizationLayerKernelInfo(float gamma, float beta, float epsilon, bool use_mixed_precis… in InstanceNormalizationLayerKernelInfo() argument
|
| /aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/ |
| H A D | NEInstanceNormalizationLayerKernel.h | 32 struct InstanceNormalizationLayerKernelInfo;
61 … void configure(ITensor *input, ITensor *output, const InstanceNormalizationLayerKernelInfo &info);
71 …onst ITensorInfo *input, const ITensorInfo *output, const InstanceNormalizationLayerKernelInfo &in…
|
| H A D | NEInstanceNormalizationLayerKernel.cpp | 135 …rKernel::configure(ITensor *input, ITensor *output, const InstanceNormalizationLayerKernelInfo &in… in configure()
155 …onst ITensorInfo *input, const ITensorInfo *output, const InstanceNormalizationLayerKernelInfo &in… in validate()
|
| /aosp_15_r20/external/ComputeLibrary/src/core/CL/kernels/ |
| H A D | CLInstanceNormalizationLayerKernel.h | 62 …sor *input, ICLTensor *mean_var, ICLTensor *output, const InstanceNormalizationLayerKernelInfo &in…
72 …onst ITensorInfo *input, const ITensorInfo *output, const InstanceNormalizationLayerKernelInfo &in…
|
| H A D | CLInstanceNormalizationLayerKernel.cpp | 41 …onst ITensorInfo *input, const ITensorInfo *output, const InstanceNormalizationLayerKernelInfo &in… in validate_arguments()
159 …sor *input, ICLTensor *mean_var, ICLTensor *output, const InstanceNormalizationLayerKernelInfo &in… in configure()
199 …onst ITensorInfo *input, const ITensorInfo *output, const InstanceNormalizationLayerKernelInfo &in… in validate()
|
| /aosp_15_r20/external/ComputeLibrary/src/runtime/CL/functions/ |
| H A D | CLInstanceNormalizationLayer.cpp | 61 …figure(compile_context, input, &_mean_var_tensor, output, InstanceNormalizationLayerKernelInfo(gam… in configure()
68 …InstanceNormalizationLayerKernel::validate(input, output, InstanceNormalizationLayerKernelInfo(gam… in validate()
|
| /aosp_15_r20/external/ComputeLibrary/src/runtime/NEON/functions/ |
| H A D | NEInstanceNormalizationLayer.cpp | 46 …const auto kernel_descriptor = InstanceNormalizationLayerKernelInfo{ gamma, beta, epsilon, t… in configure()
79 … InstanceNormalizationLayerKernelInfo{ gamma, beta, epsilon, true }); in validate()
|