Home
last modified time | relevance | path

Searched defs:num_boxes (Results 1 – 18 of 18) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/image/
Dnon_max_suppression_op.cc42 static inline void CheckScoreSizes(OpKernelContext* context, int num_boxes, in CheckScoreSizes()
58 int* num_boxes) { in ParseAndCheckOverlapSizes()
71 const Tensor& boxes, int* num_boxes) { in ParseAndCheckBoxSizes()
85 int num_boxes, in CheckCombinedNMSScoreSizes()
97 int* num_boxes, in ParseAndCheckCombinedNMSBoxSizes()
163 int num_boxes, const Tensor& max_output_size, in DoNonMaxSuppressionOp()
305 const float* scores_data, int num_boxes, int q, in DoNMSPerClass()
443 int num_boxes, const int max_size_per_class, const int total_size_per_batch, in BatchedNonMaxSuppressionOp()
598 int num_boxes = 0; in Compute() local
645 int num_boxes = 0; in Compute() local
[all …]
Dcrop_and_resize_op.cc60 int* num_boxes) { in ParseAndCheckBoxSizes()
97 const int num_boxes = box_index.dimension(0); in RunIfBoxIndexIsValid() local
150 int num_boxes = 0; in ComputeAsync() local
220 const int num_boxes = crops.dimension(0); in operator ()() local
382 int num_boxes = 0; in ComputeAsync() local
456 const int num_boxes = grads.dimension(0); in operator ()() local
604 int num_boxes = 0; in ComputeAsync() local
655 const int num_boxes = grads.dimension(0); in operator ()() local
801 const int num_boxes = box_index.dimension(0); in RunIfBoxIndexIsValid() local
Dnon_max_suppression_op.cu.cc118 const int num_boxes, const int max_boxes, in NMSReduce()
160 void NMSKernel(const Box* d_desc_sorted_boxes, const int num_boxes, in NMSKernel()
252 Status NmsGpu(const float* d_sorted_boxes_float_ptr, const int num_boxes, in NmsGpu()
412 int num_boxes = boxes.dim_size(0); in DoNMS() local
574 int num_boxes = boxes.dim_size(0); in CheckValidInputs() local
615 int num_boxes = boxes.dim_size(0); in Compute() local
661 int num_boxes = boxes.dim_size(0); in Compute() local
712 int num_boxes = boxes.dim_size(0); in Compute() local
Dcrop_and_resize_op_gpu.cu.cc43 int num_boxes, int batch, int image_height, int image_width, in CropAndResizeKernel()
136 int num_boxes, int batch, int image_height, int image_width, in CropAndResizeBackpropImageKernel()
232 const int32* __restrict__ box_ind_ptr, int num_boxes, int batch, in CropAndResizeBackpropBoxesKernel()
361 const int num_boxes = crops.dimension(0); in operator ()() local
399 const int num_boxes = grads.dimension(0); in operator ()() local
450 const int num_boxes = grads.dimension(0); in operator ()() local
Dnon_max_suppression_op_gpu_test.cc147 int num_boxes = 10; in TEST_F() local
361 int num_boxes = 10; in TEST_F() local
Dnon_max_suppression_op_test.cc149 int num_boxes = 10; in TEST_F() local
307 int num_boxes = 10; in TEST_F() local
514 int num_boxes = 10; in TEST_F() local
733 size_t num_boxes = boxes.size() / 4; in AddIoUInput() local
864 int num_boxes = 10; in TEST_F() local
Ddraw_bounding_box_op.cc115 const int64 num_boxes = boxes.dim_size(1); in Compute() local
/external/tensorflow/tensorflow/lite/micro/kernels/
Ddetection_postprocess.cc170 const int num_boxes = input_box_encodings->dims->data[1]; in Prepare() local
275 const int num_boxes = input_box_encodings->dims->data[1]; in DecodeCenterSizeBoxes() local
369 bool ValidateBoxes(const float* decoded_boxes, const int num_boxes) { in ValidateBoxes()
409 const int num_boxes = input_box_encodings->dims->data[1]; in NonMaxSuppressionSingleClassHelper() local
501 const int num_boxes = input_box_encodings->dims->data[1]; in NonMaxSuppressionMultiClassRegularHelper() local
635 const int num_boxes = input_box_encodings->dims->data[1]; in NonMaxSuppressionMultiClassFastHelper() local
705 const int num_boxes, in DequantizeClassPredictions()
727 const int num_boxes = input_box_encodings->dims->data[1]; in NonMaxSuppressionMultiClass() local
/external/tensorflow/tensorflow/lite/kernels/
Dnon_max_suppression_test.cc73 const int num_boxes = 6; in NonMaxSuppressionV4OpModel() local
164 const int num_boxes = 6; in NonMaxSuppressionV5OpModel() local
Ddetection_postprocess.cc287 const int num_boxes = input_box_encodings->dims->data[1]; in DecodeCenterSizeBoxes() local
382 bool ValidateBoxes(const TfLiteTensor* decoded_boxes, const int num_boxes) { in ValidateBoxes()
428 const int num_boxes = input_box_encodings->dims->data[1]; in NonMaxSuppressionSingleClassHelper() local
533 const int num_boxes = input_box_encodings->dims->data[1]; in NonMaxSuppressionMultiClassRegularHelper() local
675 const int num_boxes = input_box_encodings->dims->data[1]; in NonMaxSuppressionMultiClassFastHelper() local
730 const int num_boxes, in DequantizeClassPredictions()
754 const int num_boxes = input_box_encodings->dims->data[1]; in NonMaxSuppressionMultiClass() local
Dnon_max_suppression.cc88 const int num_boxes = SizeOfDimension(input_boxes, 0); in Prepare() local
207 const int num_boxes = SizeOfDimension(input_boxes, 0); in Eval() local
/external/mesa3d/src/gallium/frontends/xa/
Dxa_yuv.c119 unsigned int num_boxes, in xa_yuv_planar_blit()
Dxa_context.c102 int to_surface, struct xa_box *boxes, unsigned int num_boxes) in xa_surface_dma()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dimage_ops.cc334 const int64 num_boxes; member
357 const int64 num_boxes; member
413 const int64 num_boxes = boxes_shape.dim_size(0); in Compile() local
/external/tensorflow/tensorflow/examples/multibox_detector/
Dmain.cc267 const int num_boxes, in PrintTopDetections()
349 int32 num_boxes = 784; in main() local
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dnon_max_suppression.h87 inline void NonMaxSuppression(const float* boxes, const int num_boxes, in NonMaxSuppression()
/external/tensorflow/tensorflow/core/grappler/costs/
Dop_level_cost_estimator.cc2622 const int64 num_boxes = op_context.op_info.inputs(1).shape().dim(0).size(); in PredictCropAndResize() local
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes.cc5946 const int num_boxes = boxes_dims.d[0 + offset]; in ConvertCombinedNMS() local