Home
last modified time | relevance | path

Searched refs:get_backing_tensor (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/ComputeLibrary/arm_compute/graph/backends/
H A DFunctionHelpers.h62 typename TargetInfo::TensorType *get_backing_tensor(arm_compute::graph::Tensor *tensor) in get_backing_tensor() function
107 typename TargetInfo::TensorType *input = get_backing_tensor<TargetInfo>(node.input(0)); in create_activation_layer()
108 typename TargetInfo::TensorType *output = get_backing_tensor<TargetInfo>(node.output(0)); in create_activation_layer()
145 typename TargetInfo::TensorType *input = get_backing_tensor<TargetInfo>(node.input(0)); in create_arg_min_max_layer()
146 typename TargetInfo::TensorType *output = get_backing_tensor<TargetInfo>(node.output(0)); in create_arg_min_max_layer()
182 typename TargetInfo::TensorType *input = get_backing_tensor<TargetInfo>(node.input(0)); in create_batch_normalization_layer()
183 typename TargetInfo::TensorType *mean = get_backing_tensor<TargetInfo>(node.input(1)); in create_batch_normalization_layer()
184 typename TargetInfo::TensorType *var = get_backing_tensor<TargetInfo>(node.input(2)); in create_batch_normalization_layer()
185 typename TargetInfo::TensorType *beta = get_backing_tensor<TargetInfo>(node.input(3)); in create_batch_normalization_layer()
186 typename TargetInfo::TensorType *gamma = get_backing_tensor<TargetInfo>(node.input(4)); in create_batch_normalization_layer()
[all …]
/aosp_15_r20/external/ComputeLibrary/src/graph/backends/CL/
H A DCLFunctionsFactory.cpp135 CLTargetInfo::TensorType *input0 = get_backing_tensor<CLTargetInfo>(node.input(0)); in create_detection_output_layer()
136 CLTargetInfo::TensorType *input1 = get_backing_tensor<CLTargetInfo>(node.input(1)); in create_detection_output_layer()
137 CLTargetInfo::TensorType *input2 = get_backing_tensor<CLTargetInfo>(node.input(2)); in create_detection_output_layer()
138 CLTargetInfo::TensorType *output = get_backing_tensor<CLTargetInfo>(node.output(0)); in create_detection_output_layer()
179 … CLTargetInfo::TensorType *input0 = get_backing_tensor<CLTargetInfo>(node.input(0)); in create_detection_post_process_layer()
180 … CLTargetInfo::TensorType *input1 = get_backing_tensor<CLTargetInfo>(node.input(1)); in create_detection_post_process_layer()
181 … CLTargetInfo::TensorType *input2 = get_backing_tensor<CLTargetInfo>(node.input(2)); in create_detection_post_process_layer()
182 …CLTargetInfo::TensorType *output0 = get_backing_tensor<CLTargetInfo>(node.output(0)); in create_detection_post_process_layer()
183 …CLTargetInfo::TensorType *output1 = get_backing_tensor<CLTargetInfo>(node.output(1)); in create_detection_post_process_layer()
184 …CLTargetInfo::TensorType *output2 = get_backing_tensor<CLTargetInfo>(node.output(2)); in create_detection_post_process_layer()
[all …]
/aosp_15_r20/external/ComputeLibrary/src/graph/backends/NEON/
H A DNEFunctionFactory.cpp98 NETargetInfo::TensorType *input = get_backing_tensor<NETargetInfo>(node.input(0)); in create_normalization_layer()
99 NETargetInfo::TensorType *output = get_backing_tensor<NETargetInfo>(node.output(0)); in create_normalization_layer()