Home
last modified time | relevance | path

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

12

/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()
164 int num_boxes, const Tensor& max_output_size, in DoNonMaxSuppressionOp()
308 const float* scores_data, int num_boxes, int q, in DoNMSPerClass()
446 int num_boxes, const int max_size_per_class, const int total_size_per_batch, in BatchedNonMaxSuppressionOp()
601 int num_boxes = 0; in Compute() local
648 int num_boxes = 0; in Compute() local
[all …]
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
221 const int num_boxes = crops.dimension(0); in operator ()() local
383 int num_boxes = 0; in ComputeAsync() local
466 const int num_boxes = grads.dimension(0); in operator ()() local
620 int num_boxes = 0; in ComputeAsync() local
680 const int num_boxes = grads.dimension(0); in operator ()() local
826 const int num_boxes = box_index.dimension(0); in RunIfBoxIndexIsValid() local
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
482 int num_boxes = boxes.dim_size(0); in CheckValidInputs() local
523 int num_boxes = boxes.dim_size(0); in Compute() local
569 int num_boxes = boxes.dim_size(0); in Compute() local
620 int num_boxes = boxes.dim_size(0); in Compute() local
646 Status NmsGpu(const float* d_sorted_boxes_float_ptr, const int num_boxes, in NmsGpu()
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_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
Dnon_max_suppression_op_gpu_test.cc147 int num_boxes = 10; in TEST_F() local
361 int num_boxes = 10; in TEST_F() local
Ddraw_bounding_box_op.cc121 const int64_t num_boxes = boxes.dim_size(1); in Compute() 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.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
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/ComputeLibrary/tests/validation/fixtures/
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()
/external/armnn/tests/TfLiteYoloV3Big-Armnn/
DNMS.hpp15 unsigned int num_boxes{0}; /**< Number of detected boxes */ member
/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/ComputeLibrary/tests/validation/reference/
DBoundingBoxTransform.cpp46 const size_t num_boxes = deltas.shape()[1]; in bounding_box_transform() local
DNonMaxSuppression.cpp142 const size_t num_boxes = bboxes.shape().y(); in non_max_suppression() local
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dimage_ops.cc334 const int64 num_boxes; member
357 const int64 num_boxes; member
413 const int64_t 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_t 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/ComputeLibrary/src/core/NEON/kernels/
DNEBoundingBoxTransformKernel.cpp102 const unsigned int num_boxes = boxes->info()->dimension(1); in configure() local
/external/ComputeLibrary/src/core/CPP/kernels/
DCPPBoxWithNonMaximaSuppressionLimitKernel.cpp220 …const int num_boxes = _batch_splits_in == nullptr ? 1 : static_cast<int>(*reinterpret_cast<T *>(_b… in run_nmslimit() local
/external/ComputeLibrary/tests/
DAssetsLibrary.h539 const uint32_t num_boxes = tensor.num_elements() / 4; in fill_boxes() local
/external/tensorflow/tensorflow/core/grappler/costs/
Dop_level_cost_estimator.cc2638 const int64_t num_boxes = op_context.op_info.inputs(1).shape().dim(0).size(); in PredictCropAndResize() local
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes.cc6407 const int num_boxes = boxes_dims.d[0 + offset]; in ConvertCombinedNMS() local
/external/libbpf/.github/actions/build-selftests/
Dvmlinux.h19958 int num_boxes; member
20318 u16 num_boxes; member

12