Home
last modified time | relevance | path

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

12

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/image/
H A 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()
196 int num_boxes, const Tensor& max_output_size, in DoNonMaxSuppressionOp()
340 const float* scores_data, int num_boxes, int q, in DoNMSPerClass()
466 int num_boxes, const int max_size_per_class, const int total_size_per_batch, in BatchedNonMaxSuppressionOp()
642 int num_boxes = 0; in Compute() local
689 int num_boxes = 0; in Compute() local
[all …]
H A Dcrop_and_resize_op.cc61 int* num_boxes) { in ParseAndCheckBoxSizes()
98 const int num_boxes = box_index.dimension(0); in RunIfBoxIndexIsValid() local
151 int num_boxes = 0; in ComputeAsync() local
222 const int num_boxes = crops.dimension(0); in operator ()() local
384 int num_boxes = 0; in ComputeAsync() local
468 const int num_boxes = grads.dimension(0); in operator ()() local
622 int num_boxes = 0; in ComputeAsync() local
682 const int num_boxes = grads.dimension(0); in operator ()() local
828 const int num_boxes = box_index.dimension(0); in RunIfBoxIndexIsValid() local
H A Dnon_max_suppression_op.cu.cc120 const int num_boxes, const int max_boxes, in NMSReduce()
166 void NMSKernel(const Box* d_desc_sorted_boxes, const int num_boxes, in NMSKernel()
319 int num_boxes = boxes.dim_size(0); in DoNMS() local
503 int num_boxes = boxes.dim_size(0); in CheckValidInputs() local
544 int num_boxes = boxes.dim_size(0); in Compute() local
590 int num_boxes = boxes.dim_size(0); in Compute() local
641 int num_boxes = boxes.dim_size(0); in Compute() local
667 Status NmsGpu(const float* d_sorted_boxes_float_ptr, const int num_boxes, in NmsGpu()
H A Dcrop_and_resize_op_gpu.cu.cc43 int num_boxes, int batch, int image_height, int image_width, in CropAndResizeKernel()
147 int num_boxes, int batch, int image_height, int image_width, in CropAndResizeBackpropImageKernel()
243 const int32* __restrict__ box_ind_ptr, int num_boxes, int batch, in CropAndResizeBackpropBoxesKernel()
372 const int num_boxes = crops.dimension(0); in operator ()() local
410 const int num_boxes = grads.dimension(0); in operator ()() local
461 const int num_boxes = grads.dimension(0); in operator ()() local
H A Dnon_max_suppression_op_test.cc149 int num_boxes = 10; in TEST_F() local
307 int num_boxes = 10; in TEST_F() local
550 int num_boxes = 10; in TYPED_TEST() local
792 size_t num_boxes = boxes.size() / 4; in AddIoUInput() local
923 int num_boxes = 10; in TEST_F() local
H A Dnon_max_suppression_op_gpu_test.cc165 int num_boxes = 10; in TEST_F() local
396 int num_boxes = 10; in TEST_F() local
H A Ddraw_bounding_box_op.cc121 const int64_t num_boxes = boxes.dim_size(1); in Compute() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A Ddetection_postprocess.cc279 const int num_boxes = input_box_encodings->dims->data[1]; in DecodeCenterSizeBoxes() local
388 bool ValidateBoxes(const TfLiteTensor* decoded_boxes, const int num_boxes) { in ValidateBoxes()
434 const int num_boxes = input_box_encodings->dims->data[1]; in NonMaxSuppressionSingleClassHelper() local
508 int num_boxes; member
636 const int num_boxes = input_box_encodings->dims->data[1]; in NonMaxSuppressionMultiClassRegularHelper() local
775 const int num_boxes = input_box_encodings->dims->data[1]; in NonMaxSuppressionMultiClassFastHelper() local
831 const int num_boxes, in DequantizeClassPredictions()
857 const int num_boxes = input_box_encodings->dims->data[1]; in NonMaxSuppressionMultiClass() local
H A Dnon_max_suppression_test.cc73 const int num_boxes = 6; in NonMaxSuppressionV4OpModel() local
164 const int num_boxes = 6; in NonMaxSuppressionV5OpModel() local
H A 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
/aosp_15_r20/external/pytorch/benchmarks/functional_autograd_benchmark/
H A Dtorchvision_models.py687 def loss_labels(self, outputs, targets, indices, num_boxes, log=True): argument
717 def loss_cardinality(self, outputs, targets, indices, num_boxes): argument
732 def loss_boxes(self, outputs, targets, indices, num_boxes): argument
757 def loss_masks(self, outputs, targets, indices, num_boxes): argument
810 def get_loss(self, loss, outputs, targets, indices, num_boxes, **kwargs): argument
/aosp_15_r20/external/ComputeLibrary/tests/validation/fixtures/
H A DBoundingBoxTransformFixture.h45 …e_deltas(std::vector<float> &boxes, const TensorShape &image_shape, size_t num_boxes, size_t num_c… in generate_deltas()
79 std::vector<float> generate_boxes(const TensorShape &image_shape, size_t num_boxes, std::mt19937 &g… in generate_boxes()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
H A Dimage_ops.cc341 const int64_t num_boxes; member
364 const int64_t num_boxes; member
431 const int64_t num_boxes = boxes_shape.dim_size(0); in ComputeResult() local
/aosp_15_r20/external/tensorflow/tensorflow/examples/multibox_detector/
H A Dmain.cc267 const int num_boxes, in PrintTopDetections()
349 int32_t num_boxes = 784; in main() local
/aosp_15_r20/external/armnn/tests/TfLiteYoloV3Big-Armnn/
H A DNMS.hpp15 unsigned int num_boxes{0}; /**< Number of detected boxes */ member
/aosp_15_r20/external/mesa3d/src/gallium/frontends/xa/
H A Dxa_yuv.c118 unsigned int num_boxes, in xa_yuv_planar_blit()
H A Dxa_context.c102 int to_surface, struct xa_box *boxes, unsigned int num_boxes) in xa_surface_dma()
/aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/
H A DNEBoundingBoxTransformKernel.cpp159 const unsigned int num_boxes = boxes->info()->dimension(1); in configure() local
/aosp_15_r20/external/ComputeLibrary/tests/validation/reference/
H A DBoundingBoxTransform.cpp46 const size_t num_boxes = deltas.shape()[1]; in bounding_box_transform() local
H A DNonMaxSuppression.cpp142 const size_t num_boxes = bboxes.shape().y(); in non_max_suppression() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/reference/
H A Dnon_max_suppression.h87 inline void NonMaxSuppression(const float* boxes, const int num_boxes, in NonMaxSuppression()
/aosp_15_r20/external/mesa3d/src/gallium/drivers/zink/
H A Dzink_resource.c2754 unsigned num_boxes = util_dynarray_num_elements(&res->obj->copies[level], struct pipe_box); in zink_resource_copy_box_intersects() local
2792 unsigned num_boxes = util_dynarray_num_elements(&res->obj->copies[level], struct pipe_box); in zink_resource_copy_box_add() local
2964 unsigned num_boxes = util_dynarray_num_elements(&res->obj->copies[0], struct pipe_box); in zink_resource_copies_reset() local
/aosp_15_r20/external/ComputeLibrary/tests/
H A DAssetsLibrary.h542 const uint32_t num_boxes = tensor.num_elements() / 4; in fill_boxes() local
/aosp_15_r20/external/mesa3d/src/intel/compiler/elk/
H A Delk_fs_combine_constants.cpp897 unsigned num_boxes; member
/aosp_15_r20/external/mesa3d/src/intel/compiler/
H A Dbrw_fs_combine_constants.cpp851 unsigned num_boxes; member

12