Searched refs:get_backing_tensor (Results 1 – 3 of 3) sorted by relevance
62 typename TargetInfo::TensorType *get_backing_tensor(arm_compute::graph::Tensor *tensor) in get_backing_tensor() function107 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 …]
135 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 …]
98 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()