Searched defs:use_mixed_precision (Results 1 – 5 of 5) sorted by relevance
| /aosp_15_r20/external/ComputeLibrary/src/runtime/CL/functions/ |
| H A D | CLInstanceNormalizationLayer.cpp | 49 …Tensor *input, ICLTensor *output, float gamma, float beta, float epsilon, bool use_mixed_precision) in configure()
54 …Tensor *input, ICLTensor *output, float gamma, float beta, float epsilon, bool use_mixed_precision) in configure()
66 …input, const ITensorInfo *output, float gamma, float beta, float epsilon, bool use_mixed_precision) in validate()
|
| /aosp_15_r20/external/ComputeLibrary/src/cpu/kernels/instancenorm/generic/neon/ |
| H A D | fp16.cpp | 30 …input, ITensor *output, float gamma, float beta, float epsilon, bool use_mixed_precision, const Wi… in neon_fp16_instancenorm()
|
| H A D | fp32.cpp | 29 …input, ITensor *output, float gamma, float beta, float epsilon, bool use_mixed_precision, const Wi… in neon_fp32_instancenorm()
|
| /aosp_15_r20/external/ComputeLibrary/arm_compute/core/ |
| H A D | KernelDescriptors.h | 164 … bool use_mixed_precision; /**< Use mixed precision in case of FP16 execution. Defaults to true */ member
|
| /aosp_15_r20/external/ComputeLibrary/src/core/CL/kernels/ |
| H A D | CLInstanceNormalizationLayerKernel.cpp | 78 …t CLCompileContext &compile_context, ICLTensor *input, ICLTensor *output, bool use_mixed_precision) in configure()
|