Home
last modified time | relevance | path

Searched refs:bn_var (Results 1 – 20 of 20) sorted by relevance

/external/ComputeLibrary/src/core/CL/kernels/
DCLFuseBatchNormalizationKernel.cpp42 …_arguments(const ITensorInfo *input_weights, const ITensorInfo *bn_mean, const ITensorInfo *bn_var, in validate_arguments() argument
48 ARM_COMPUTE_ERROR_ON_NULLPTR(input_weights, bn_mean, bn_var); in validate_arguments()
51 ARM_COMPUTE_RETURN_ERROR_ON_MISMATCHING_SHAPES(bn_mean, bn_var); in validate_arguments()
52 ARM_COMPUTE_RETURN_ERROR_ON_MISMATCHING_DATA_TYPES(input_weights, bn_mean, bn_var); in validate_arguments()
109 …ernel::configure(const ICLTensor *input_weights, const ICLTensor *bn_mean, const ICLTensor *bn_var, in configure() argument
114 …configure(CLKernelLibrary::get().get_compile_context(), input_weights, bn_mean, bn_var, fused_weig… in configure()
117 …compile_context, const ICLTensor *input_weights, const ICLTensor *bn_mean, const ICLTensor *bn_var, in configure() argument
122 ARM_COMPUTE_ERROR_ON_NULLPTR(input_weights, bn_mean, bn_var); in configure()
124 …auto padding_info = get_padding_info({ input_weights, bn_mean, bn_var, fused_weights, fused_bias, … in configure()
129 _bn_var = bn_var; in configure()
[all …]
DCLFuseBatchNormalizationKernel.h65 …nst ICLTensor *input_weights, const ICLTensor *bn_mean, const ICLTensor *bn_var, ICLTensor *fused_…
84 …nst ICLTensor *input_weights, const ICLTensor *bn_mean, const ICLTensor *bn_var, ICLTensor *fused_…
104 …s validate(const ITensorInfo *input_weights, const ITensorInfo *bn_mean, const ITensorInfo *bn_var,
/external/ComputeLibrary/src/runtime/CL/functions/
DCLFuseBatchNormalization.cpp44 …ation::configure(const ICLTensor *input_weights, const ICLTensor *bn_mean, const ICLTensor *bn_var, in configure() argument
49 …configure(CLKernelLibrary::get().get_compile_context(), input_weights, bn_mean, bn_var, fused_weig… in configure()
52 …compile_context, const ICLTensor *input_weights, const ICLTensor *bn_mean, const ICLTensor *bn_var, in configure() argument
57 …ARM_COMPUTE_LOG_PARAMS(input_weights, bn_mean, bn_var, fused_weights, fused_bias, input_bias, bn_b… in configure()
58 …_fuse_bn_kernel->configure(compile_context, input_weights, bn_mean, bn_var, fused_weights, fused_b… in configure()
61 …::validate(const ITensorInfo *input_weights, const ITensorInfo *bn_mean, const ITensorInfo *bn_var, in validate() argument
66 …return CLFuseBatchNormalizationKernel::validate(input_weights, bn_mean, bn_var, fused_weights, fus… in validate()
/external/ComputeLibrary/src/runtime/NEON/functions/
DNEFuseBatchNormalization.cpp43 …rmalization::configure(const ITensor *input_weights, const ITensor *bn_mean, const ITensor *bn_var, in configure() argument
48 ARM_COMPUTE_LOG_PARAMS(input_weights, bn_mean, bn_var, fused_weights, fused_bias, input_bias, in configure()
52 …_fuse_bn_kernel->configure(input_weights, bn_mean, bn_var, fused_weights, fused_bias, input_bias, … in configure()
55 …::validate(const ITensorInfo *input_weights, const ITensorInfo *bn_mean, const ITensorInfo *bn_var, in validate() argument
60 …return NEFuseBatchNormalizationKernel::validate(input_weights, bn_mean, bn_var, fused_weights, fus… in validate()
/external/ComputeLibrary/src/core/NEON/kernels/
DNEFuseBatchNormalizationKernel.cpp153 …_arguments(const ITensorInfo *input_weights, const ITensorInfo *bn_mean, const ITensorInfo *bn_var, in validate_arguments() argument
159 ARM_COMPUTE_RETURN_ERROR_ON_NULLPTR(input_weights, bn_mean, bn_var); in validate_arguments()
162 ARM_COMPUTE_RETURN_ERROR_ON_MISMATCHING_SHAPES(bn_mean, bn_var); in validate_arguments()
163 ARM_COMPUTE_RETURN_ERROR_ON_MISMATCHING_DATA_TYPES(input_weights, bn_mean, bn_var); in validate_arguments()
220 …ationKernel::configure(const ITensor *input_weights, const ITensor *bn_mean, const ITensor *bn_var, in configure() argument
225 ARM_COMPUTE_ERROR_ON_NULLPTR(input_weights, bn_mean, bn_var); in configure()
230 _bn_var = bn_var; in configure()
253 …ARM_COMPUTE_ERROR_THROW_ON(validate_arguments(input_weights->info(), bn_mean->info(), bn_var->info… in configure()
271 …::validate(const ITensorInfo *input_weights, const ITensorInfo *bn_mean, const ITensorInfo *bn_var, in validate() argument
276 …ARM_COMPUTE_RETURN_ON_ERROR(validate_arguments(input_weights, bn_mean, bn_var, fused_weights, fuse… in validate()
DNEFuseBatchNormalizationKernel.h69 …void configure(const ITensor *input_weights, const ITensor *bn_mean, const ITensor *bn_var, ITenso…
89 …s validate(const ITensorInfo *input_weights, const ITensorInfo *bn_mean, const ITensorInfo *bn_var,
111 …const ITensor *bn_mean, const ITensor *bn_var, const ITensor *bn_beta, const ITensor *bn_gamma, fl…
/external/ComputeLibrary/tests/validation/fixtures/
DBatchNormalizationLayerFusionFixture.h97 …TensorType bn_var = create_tensor<TensorType>(b_shape, _data_type, 1, QuantizationInfo(), _data_… in compute_target() local
110 …fuse_fn.configure(&conv_w, &bn_mean, &bn_var, &fused_w, &fused_b, conv_b_ptr, beta_ptr, gamma_ptr,… in compute_target()
117 ARM_COMPUTE_ASSERT(bn_var.info()->is_resizable()); in compute_target()
129 bn_var.allocator()->allocate(); in compute_target()
140 ARM_COMPUTE_ASSERT(!bn_var.info()->is_resizable()); in compute_target()
150 … AccessorType(bn_mean), AccessorType(bn_var), AccessorType(bn_beta), AccessorType(bn_gamma)); in compute_target()
165 SimpleTensor<T> bn_var{ b_shape, _data_type, 1 }; in compute_reference()
171 fill(src, conv_w, conv_b, bn_mean, bn_var, bn_beta, bn_gamma); in compute_reference()
175 …return reference::batch_normalization_layer(conv_res, bn_mean, bn_var, bn_beta, bn_gamma, epsilon,… in compute_reference()
/external/ComputeLibrary/src/cpu/kernels/fuse_batch_normalization/nchw/
Dall.cpp33 …const ITensor *bn_mean, const ITensor *bn_var, const ITensor *bn_beta, const ITensor *bn_gamma, fl… in fused_batch_normalization_dwc_nchw() argument
57 …const auto input_var = reinterpret_cast<const ScalarType *>(bn_var->ptr_to_element(Coordinates(0… in fused_batch_normalization_dwc_nchw()
129 …const ITensor *bn_mean, const ITensor *bn_var, const ITensor *bn_beta, const ITensor *bn_gamma, fl… in fused_batch_normalization_dwc_nchw_f32() argument
132 … bn_mean, bn_var, bn_beta, bn_gamma, epsilon, window); in fused_batch_normalization_dwc_nchw_f32()
137 …const ITensor *bn_mean, const ITensor *bn_var, const ITensor *bn_beta, const ITensor *bn_gamma, fl… in fused_batch_normalization_dwc_nchw_f16() argument
140 … bn_mean, bn_var, bn_beta, bn_gamma, epsilon, window); in fused_batch_normalization_dwc_nchw_f16()
/external/ComputeLibrary/src/cpu/kernels/fuse_batch_normalization/
Dlist.h33 …const ITensor *bn_mean, const ITensor *bn_var, const ITensor *bn_beta, const ITensor *bn_gamma, fl…
37 …const ITensor *bn_mean, const ITensor *bn_var, const ITensor *bn_beta, const ITensor *bn_gamma, fl…
41 …const ITensor *bn_mean, const ITensor *bn_var, const ITensor *bn_beta, const ITensor *bn_gamma, fl…
/external/ComputeLibrary/arm_compute/runtime/CL/functions/
DCLFuseBatchNormalization.h81 …nst ICLTensor *input_weights, const ICLTensor *bn_mean, const ICLTensor *bn_var, ICLTensor *fused_…
100 …nst ICLTensor *input_weights, const ICLTensor *bn_mean, const ICLTensor *bn_var, ICLTensor *fused_…
120 …s validate(const ITensorInfo *input_weights, const ITensorInfo *bn_mean, const ITensorInfo *bn_var,
/external/ComputeLibrary/src/cpu/kernels/fuse_batch_normalization/generic/
Dimpl.cpp32 …const ITensor *bn_mean, const ITensor *bn_var, const ITensor *bn_beta, const ITensor *bn_gamma, fl… in fused_batch_normalization_conv() argument
56 …const auto input_var = reinterpret_cast<const ScalarType *>(bn_var->ptr_to_element(Coordinates(0… in fused_batch_normalization_conv()
127 …const ITensor *bn_mean, const ITensor *bn_var, const ITensor *bn_beta, const ITensor *bn_gamma, fl…
131 …const ITensor *bn_mean, const ITensor *bn_var, const ITensor *bn_beta, const ITensor *bn_gamma, fl…
Dfp32.cpp32 …const ITensor *bn_mean, const ITensor *bn_var, const ITensor *bn_beta, const ITensor *bn_gamma, fl… in fused_batch_normalization_conv_f32() argument
35 … bn_mean, bn_var, bn_beta, bn_gamma, epsilon, window); in fused_batch_normalization_conv_f32()
Dfp16.cpp33 …const ITensor *bn_mean, const ITensor *bn_var, const ITensor *bn_beta, const ITensor *bn_gamma, fl… in fused_batch_normalization_conv_f16() argument
36 … bn_mean, bn_var, bn_beta, bn_gamma, epsilon, window); in fused_batch_normalization_conv_f16()
Dimpl.h36 …const ITensor *bn_mean, const ITensor *bn_var, const ITensor *bn_beta, const ITensor *bn_gamma, fl…
/external/ComputeLibrary/src/cpu/kernels/fuse_batch_normalization/nhwc/neon/
Dfp32.cpp33 …const ITensor *bn_mean, const ITensor *bn_var, const ITensor *bn_beta, const ITensor *bn_gamma, fl… in fused_batch_normalization_dwc_nhwc_f32() argument
36 … bn_mean, bn_var, bn_beta, bn_gamma, epsilon, window); in fused_batch_normalization_dwc_nhwc_f32()
Dfp16.cpp34 …const ITensor *bn_mean, const ITensor *bn_var, const ITensor *bn_beta, const ITensor *bn_gamma, fl… in fused_batch_normalization_dwc_nhwc_f16() argument
37 … bn_mean, bn_var, bn_beta, bn_gamma, epsilon, window); in fused_batch_normalization_dwc_nhwc_f16()
Dimpl.cpp32 …const ITensor *bn_mean, const ITensor *bn_var, const ITensor *bn_beta, const ITensor *bn_gamma, fl… in fused_batch_normalization_dwc_nhwc() argument
56 …const auto input_var = reinterpret_cast<const ScalarType *>(bn_var->ptr_to_element(Coordinates(0… in fused_batch_normalization_dwc_nhwc()
150 …const ITensor *bn_mean, const ITensor *bn_var, const ITensor *bn_beta, const ITensor *bn_gamma, fl…
154 …const ITensor *bn_mean, const ITensor *bn_var, const ITensor *bn_beta, const ITensor *bn_gamma, fl…
Dimpl.h36 …const ITensor *bn_mean, const ITensor *bn_var, const ITensor *bn_beta, const ITensor *bn_gamma, fl…
/external/ComputeLibrary/arm_compute/runtime/NEON/functions/
DNEFuseBatchNormalization.h78 …void configure(const ITensor *input_weights, const ITensor *bn_mean, const ITensor *bn_var, ITenso…
98 …s validate(const ITensorInfo *input_weights, const ITensorInfo *bn_mean, const ITensorInfo *bn_var,
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ops/
Dquantization_ops_test.cc169 auto bn_var = in __anon3b69349c0202() local
176 bn_mean, bn_var, in __anon3b69349c0202()