Searched refs:input_box_encoding (Results 1 – 7 of 7) sorted by relevance
| /external/ComputeLibrary/src/runtime/CPP/functions/ |
| D | CPPDetectionPostProcessLayer.cpp | 39 Status validate_arguments(const ITensorInfo *input_box_encoding, const ITensorInfo *input_class_sco… in validate_arguments() argument
43 ARM_COMPUTE_RETURN_ERROR_ON_NULLPTR(input_box_encoding, input_class_score, input_anchors); in validate_arguments()
44 …ARM_COMPUTE_RETURN_ERROR_ON_DATA_TYPE_CHANNEL_NOT_IN(input_box_encoding, 1, DataType::F32, DataTyp… in validate_arguments()
45 ARM_COMPUTE_RETURN_ERROR_ON_MISMATCHING_DATA_TYPES(input_box_encoding, input_anchors); in validate_arguments()
46 …ARM_COMPUTE_RETURN_ERROR_ON_MSG(input_box_encoding->num_dimensions() > 3, "The location input tens… in validate_arguments()
47 if(input_box_encoding->num_dimensions() > 2) in validate_arguments()
49 …ARM_COMPUTE_RETURN_ERROR_ON_MSG_VAR(input_box_encoding->dimension(2) != kBatchSize, "The third dim… in validate_arguments()
51 …ARM_COMPUTE_RETURN_ERROR_ON_MSG_VAR(input_box_encoding->dimension(0) != kNumCoordBox, "The first d… in validate_arguments()
60 …ARM_COMPUTE_RETURN_ERROR_ON_MSG((input_box_encoding->dimension(1) != input_class_score->dimension(… in validate_arguments()
61 … || (input_box_encoding->dimension(1) != input_anchors->dimension(1)), in validate_arguments()
[all …]
|
| /external/ComputeLibrary/src/runtime/NEON/functions/ |
| D | NEDetectionPostProcessLayer.cpp | 41 void NEDetectionPostProcessLayer::configure(const ITensor *input_box_encoding, const ITensor *input… in configure() argument
44 …ARM_COMPUTE_ERROR_ON_NULLPTR(input_box_encoding, input_scores, input_anchors, output_boxes, output… in configure()
45 …ARM_COMPUTE_ERROR_THROW_ON(NEDetectionPostProcessLayer::validate(input_box_encoding->info(), input… in configure()
51 …_run_dequantize = is_data_type_quantized(input_box_encoding->inf… in configure()
68 …_detection_post_process.configure(input_box_encoding, input_scores_to_use, input_anchors, output_b… in configure()
72 Status NEDetectionPostProcessLayer::validate(const ITensorInfo *input_box_encoding, const ITensorIn… in validate() argument
75 bool run_dequantize = is_data_type_quantized(input_box_encoding->data_type()); in validate()
81 …ARM_COMPUTE_RETURN_ON_ERROR(CPPDetectionPostProcessLayer::validate(input_box_encoding, input_score… in validate()
|
| /external/ComputeLibrary/arm_compute/runtime/NEON/functions/ |
| D | NEDetectionPostProcessLayer.h | 71 …void configure(const ITensor *input_box_encoding, const ITensor *input_score, const ITensor *input…
86 …static Status validate(const ITensorInfo *input_box_encoding, const ITensorInfo *input_class_score…
|
| /external/ComputeLibrary/arm_compute/runtime/CPP/functions/ |
| D | CPPDetectionPostProcessLayer.h | 68 …void configure(const ITensor *input_box_encoding, const ITensor *input_score, const ITensor *input…
83 …static Status validate(const ITensorInfo *input_box_encoding, const ITensorInfo *input_class_score…
|
| /external/ComputeLibrary/src/graph/ |
| D | GraphBuilder.cpp | 413 …tection_post_process_node(Graph &g, NodeParams params, NodeIdxPair input_box_encoding, NodeIdxPair… in add_detection_post_process_node() argument
416 check_nodeidx_pair(input_box_encoding, g); in add_detection_post_process_node()
420 …r input_box_encoding_tensor_desc = get_tensor_descriptor(g, g.node(input_box_encoding.node_id)->ou… in add_detection_post_process_node()
434 g.add_connection(input_box_encoding.node_id, input_box_encoding.index, detect_nid, 0); in add_detection_post_process_node()
|
| /external/ComputeLibrary/arm_compute/graph/ |
| D | GraphBuilder.h | 266 …tection_post_process_node(Graph &g, NodeParams params, NodeIdxPair input_box_encoding, NodeIdxPair…
|
| /external/ComputeLibrary/arm_compute/graph/frontend/ |
| D | Layers.h | 629 NodeIdxPair input_box_encoding = { s.tail_node(), 0 }; in create_layer() local
631 …Builder::add_detection_post_process_node(s.graph(), common_params, input_box_encoding, input_class… in create_layer()
|