| /external/ComputeLibrary/src/core/CL/kernels/ |
| D | CLNormalizationLayerKernel.cpp | 44 …te_arguments(const ITensorInfo *input, const ITensorInfo *output, NormalizationLayerInfo norm_info) in validate_arguments() argument
51 … ARM_COMPUTE_RETURN_ERROR_ON_MSG(!(norm_info.norm_size() % 2), "Normalization size should be odd"); in validate_arguments()
64 …ate_and_configure_window(ITensorInfo *input, ITensorInfo *output, NormalizationLayerInfo norm_info) in validate_and_configure_window() argument
75 …gned int norm_idx = get_normalization_dimension_index(input->data_layout(), norm_info); in validate_and_configure_window()
78 const unsigned int norm_radius = norm_info.norm_size() / 2; in validate_and_configure_window()
113 if(norm_info.is_cross_map()) in validate_and_configure_window()
135 …LayerKernel::configure(const ICLTensor *input, ICLTensor *output, NormalizationLayerInfo norm_info) in configure() argument
137 configure(CLKernelLibrary::get().get_compile_context(), input, output, norm_info); in configure()
140 …text &compile_context, const ICLTensor *input, ICLTensor *output, NormalizationLayerInfo norm_info) in configure() argument
146 ARM_COMPUTE_ERROR_THROW_ON(validate_arguments(input->info(), output->info(), norm_info)); in configure()
[all …]
|
| D | CLNormalizationLayerKernel.h | 56 void configure(const ICLTensor *input, ICLTensor *output, NormalizationLayerInfo norm_info);
66 …ext &compile_context, const ICLTensor *input, ICLTensor *output, NormalizationLayerInfo norm_info);
77 …us validate(const ITensorInfo *input, const ITensorInfo *output, NormalizationLayerInfo norm_info);
|
| /external/ComputeLibrary/src/runtime/CL/functions/ |
| D | CLNormalizationLayer.cpp | 48 …ationLayer::configure(ICLTensor *input, ICLTensor *output, const NormalizationLayerInfo &norm_info) in configure() argument
50 configure(CLKernelLibrary::get().get_compile_context(), input, output, norm_info); in configure()
53 …ext &compile_context, ICLTensor *input, ICLTensor *output, const NormalizationLayerInfo &norm_info) in configure() argument
56 ARM_COMPUTE_LOG_PARAMS(input, output, norm_info); in configure()
59 _norm_kernel->configure(compile_context, input, output, norm_info); in configure()
68 …idate(const ITensorInfo *input, const ITensorInfo *output, const NormalizationLayerInfo &norm_info) in validate() argument
70 return CLNormalizationLayerKernel::validate(input, output, norm_info); in validate()
|
| /external/ComputeLibrary/src/core/NEON/kernels/ |
| D | NENormalizationLayerKernel.cpp | 44 …nst ITensorInfo *input_squared, const ITensorInfo *output, const NormalizationLayerInfo &norm_info) in validate_arguments() argument
52 … ARM_COMPUTE_RETURN_ERROR_ON_MSG(!(norm_info.norm_size() % 2), "Normalization size should be odd"); in validate_arguments()
72 …st ITensor *input, const ITensor *input_squared, ITensor *output, NormalizationLayerInfo norm_info) in configure() argument
79 …RROR_THROW_ON(validate_arguments(input->info(), input_squared->info(), output->info(), norm_info)); in configure()
81 …unsigned int norm_idx = get_normalization_dimension_index(input->info()->data_layout(), norm_info); in configure()
86 _norm_info = norm_info; in configure()
96 if(norm_info.type() == NormType::IN_MAP_2D) in configure()
107 if(norm_info.type() == NormType::IN_MAP_2D) in configure()
131 if(norm_info.type() == NormType::IN_MAP_2D) in configure()
142 if(norm_info.type() == NormType::IN_MAP_2D) in configure()
[all …]
|
| D | NENormalizationLayerKernel.h | 63 …t ITensor *input, const ITensor *input_squared, ITensor *output, NormalizationLayerInfo norm_info);
75 …ut, const ITensorInfo *input_squared, const ITensorInfo *output, NormalizationLayerInfo norm_info);
|
| /external/ComputeLibrary/src/runtime/NEON/functions/ |
| D | NENormalizationLayer.cpp | 44 …ionLayer::configure(const ITensor *input, ITensor *output, const NormalizationLayerInfo &norm_info) in configure() argument
47 ARM_COMPUTE_LOG_PARAMS(input, output, norm_info); in configure()
57 _norm_kernel->configure(input, &_input_squared, output, norm_info); in configure()
64 …idate(const ITensorInfo *input, const ITensorInfo *output, const NormalizationLayerInfo &norm_info) in validate() argument
69 …ARM_COMPUTE_RETURN_ON_ERROR(NENormalizationLayerKernel::validate(input, input, output, norm_info)); in validate()
|
| /external/ComputeLibrary/arm_compute/runtime/CL/functions/ |
| D | CLNormalizationLayer.h | 81 void configure(ICLTensor *input, ICLTensor *output, const NormalizationLayerInfo &norm_info);
92 …xt &compile_context, ICLTensor *input, ICLTensor *output, const NormalizationLayerInfo &norm_info);
103 …date(const ITensorInfo *input, const ITensorInfo *output, const NormalizationLayerInfo &norm_info);
|
| /external/ComputeLibrary/arm_compute/runtime/NEON/functions/ |
| D | NENormalizationLayer.h | 81 void configure(const ITensor *input, ITensor *output, const NormalizationLayerInfo &norm_info);
91 …date(const ITensorInfo *input, const ITensorInfo *output, const NormalizationLayerInfo &norm_info);
|
| /external/ComputeLibrary/src/graph/nodes/ |
| D | NormalizationLayerNode.cpp | 34 NormalizationLayerNode::NormalizationLayerNode(NormalizationLayerInfo norm_info) in NormalizationLayerNode() argument
35 : _info(norm_info) in NormalizationLayerNode()
|
| /external/ComputeLibrary/src/graph/backends/NEON/ |
| D | NEFunctionFactory.cpp | 100 const NormalizationLayerInfo norm_info = node.normalization_info(); in create_normalization_layer() local
106 func->configure(input, output, norm_info); in create_normalization_layer()
116 << " Normalization info: " << norm_info.type() in create_normalization_layer()
|
| /external/ComputeLibrary/tests/validation/NEON/ |
| D | NormalizationLayer.cpp | 86 input_info, output_info, norm_info, expected)
88 …_info.clone()->set_is_resizable(false), &output_info.clone()->set_is_resizable(false), norm_info));
|
| /external/ComputeLibrary/tests/validation/CL/ |
| D | NormalizationLayer.cpp | 87 input_info, output_info, norm_info, expected)
89 …set_is_resizable(false), &output_info.clone()->set_is_resizable(false), norm_info)) == expected, f…
|
| /external/ComputeLibrary/arm_compute/graph/nodes/ |
| D | NormalizationLayerNode.h | 41 NormalizationLayerNode(NormalizationLayerInfo norm_info);
|
| /external/ComputeLibrary/arm_compute/graph/ |
| D | GraphBuilder.h | 361 …rmalization_node(Graph &g, NodeParams params, NodeIdxPair input, NormalizationLayerInfo norm_info);
|
| /external/ComputeLibrary/src/graph/ |
| D | GraphBuilder.cpp | 560 …ormalization_node(Graph &g, NodeParams params, NodeIdxPair input, NormalizationLayerInfo norm_info) in add_normalization_node() argument
562 …return create_simple_single_input_output_node<NormalizationLayerNode>(g, params, input, norm_info); in add_normalization_node()
|
| /external/ComputeLibrary/arm_compute/graph/frontend/ |
| D | Layers.h | 875 NormalizationLayer(NormalizationLayerInfo norm_info) in NormalizationLayer() argument
876 : _norm_info(norm_info) in NormalizationLayer()
|
| /external/ComputeLibrary/arm_compute/graph/backends/ |
| D | FunctionHelpers.h | 1336 const NormalizationLayerInfo norm_info = node.normalization_info(); in create_normalization_layer() local
1342 func->configure(input, output, norm_info); in create_normalization_layer()
1352 << " Normalization info: " << norm_info.type() in create_normalization_layer()
|