Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Dnon_max_suppression_op.cc42 static inline void CheckScoreSizes(OpKernelContext* context, int num_boxes, in CheckScoreSizes()
54 int* num_boxes) { in ParseAndCheckOverlapSizes()
67 const Tensor& boxes, int* num_boxes) { in ParseAndCheckBoxSizes()
78 int num_boxes, in CheckCombinedNMSScoreSizes()
90 int* num_boxes, in ParseAndCheckCombinedNMSBoxSizes()
156 OpKernelContext* context, const Tensor& scores, int num_boxes, in DoNonMaxSuppressionOp()
228 int num_boxes, const int max_size_per_class, const int total_size_per_batch, in BatchedNonMaxSuppressionOp()
460 int num_boxes = 0; in Compute() local
504 int num_boxes = 0; in Compute() local
552 int num_boxes = 0; in Compute() local
[all …]
Dcrop_and_resize_op.cc54 int* num_boxes) { in ParseAndCheckBoxSizes()
91 const int num_boxes = box_index.dimension(0); in RunIfBoxIndexIsValid() local
144 int num_boxes = 0; in ComputeAsync() local
213 const int num_boxes = crops.dimension(0); in operator ()() local
364 int num_boxes = 0; in ComputeAsync() local
438 const int num_boxes = grads.dimension(0); in operator ()() local
562 int num_boxes = 0; in ComputeAsync() local
613 const int num_boxes = grads.dimension(0); in operator ()() local
759 const int num_boxes = box_index.dimension(0); in RunIfBoxIndexIsValid() local
Dcrop_and_resize_op_gpu.cu.cc43 const int32* box_ind_ptr, int num_boxes, int batch, int image_height, in CropAndResizeKernel()
135 const int32* box_ind_ptr, int num_boxes, int batch, int image_height, in CropAndResizeBackpropImageKernel()
230 const float* boxes_ptr, const int32* box_ind_ptr, int num_boxes, int batch, in CropAndResizeBackpropBoxesKernel()
359 const int num_boxes = crops.dimension(0); in operator ()() local
397 const int num_boxes = grads.dimension(0); in operator ()() local
447 const int num_boxes = grads.dimension(0); in operator ()() local
Ddraw_bounding_box_op.cc90 const int64 num_boxes = boxes.dim_size(1); in Compute() local
Dnon_max_suppression_op_test.cc136 int num_boxes = 10; in TEST_F() local
295 int num_boxes = 10; in TEST_F() local
503 int num_boxes = 10; in TEST_F() local
647 size_t num_boxes = boxes.size() / 4; in AddIoUInput() local
778 int num_boxes = 10; in TEST_F() local
/external/tensorflow/tensorflow/lite/kernels/
Ddetection_postprocess.cc264 const int num_boxes = input_box_encodings->dims->data[1]; in DecodeCenterSizeBoxes() local
338 bool ValidateBoxes(const TfLiteTensor* decoded_boxes, const int num_boxes) { in ValidateBoxes()
380 const int num_boxes = input_box_encodings->dims->data[1]; in NonMaxSuppressionSingleClassHelper() local
469 const int num_boxes = input_box_encodings->dims->data[1]; in NonMaxSuppressionMultiClassRegularHelper() local
594 const int num_boxes = input_box_encodings->dims->data[1]; in NonMaxSuppressionMultiClassFastHelper() local
646 const int num_boxes, in DequantizeClassPredictions()
666 const int num_boxes = input_box_encodings->dims->data[1]; in NonMaxSuppressionMultiClass() local
/external/mesa3d/src/gallium/state_trackers/xa/
Dxa_yuv.c119 unsigned int num_boxes, in xa_yuv_planar_blit()
Dxa_context.c101 int to_surface, struct xa_box *boxes, unsigned int num_boxes) in xa_surface_dma()
/external/tensorflow/tensorflow/examples/multibox_detector/
Dmain.cc263 const int num_boxes, in PrintTopDetections()
345 int32 num_boxes = 784; in main() local
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dimage_ops.cc332 const int64 num_boxes; member
355 const int64 num_boxes; member
411 const int64 num_boxes = boxes_shape.dim_size(0); in Compile() local