| /external/tensorflow/tensorflow/core/kernels/ |
| D | non_max_suppression_op.cc | 42 static inline void CheckScoreSizes(OpKernelContext* context, int num_boxes, in CheckScoreSizes() argument
48 OP_REQUIRES(context, scores.dim_size(0) == num_boxes, in CheckScoreSizes()
54 int* num_boxes) { in ParseAndCheckOverlapSizes() argument
60 *num_boxes = overlaps.dim_size(0); in ParseAndCheckOverlapSizes()
61 OP_REQUIRES(context, overlaps.dim_size(1) == *num_boxes, in ParseAndCheckOverlapSizes()
67 const Tensor& boxes, int* num_boxes) { in ParseAndCheckBoxSizes() argument
72 *num_boxes = boxes.dim_size(0); in ParseAndCheckBoxSizes()
78 int num_boxes, in CheckCombinedNMSScoreSizes() argument
84 OP_REQUIRES(context, scores.dim_size(1) == num_boxes, in CheckCombinedNMSScoreSizes()
90 int* num_boxes, in ParseAndCheckCombinedNMSBoxSizes() argument
[all …]
|
| D | non_max_suppression_op_test.cc | 136 int num_boxes = 10; in TEST_F() local
137 std::vector<float> corners(num_boxes * 4); in TEST_F()
138 std::vector<float> scores(num_boxes); in TEST_F()
139 for (int i = 0; i < num_boxes; ++i) { in TEST_F()
146 AddInputFromArray<float>(TensorShape({num_boxes, 4}), corners); in TEST_F()
147 AddInputFromArray<float>(TensorShape({num_boxes}), scores); in TEST_F()
295 int num_boxes = 10; in TEST_F() local
296 std::vector<float> corners(num_boxes * 4); in TEST_F()
297 std::vector<float> scores(num_boxes); in TEST_F()
298 for (int i = 0; i < num_boxes; ++i) { in TEST_F()
[all …]
|
| D | crop_and_resize_op.cc | 54 int* num_boxes) { in ParseAndCheckBoxSizes() argument
56 *num_boxes = 0; in ParseAndCheckBoxSizes()
64 *num_boxes = boxes.dim_size(0); in ParseAndCheckBoxSizes()
73 if (box_index.dim_size(0) != *num_boxes) { in ParseAndCheckBoxSizes()
91 const int num_boxes = box_index.dimension(0); in RunIfBoxIndexIsValid() local
92 for (int b = 0; b < num_boxes; ++b) { in RunIfBoxIndexIsValid()
144 int num_boxes = 0; in ComputeAsync() local
146 context, ParseAndCheckBoxSizes(boxes, box_index, &num_boxes), done); in ComputeAsync()
171 0, TensorShape({num_boxes, crop_height, crop_width, depth}), in ComputeAsync()
213 const int num_boxes = crops.dimension(0); in operator ()() local
[all …]
|
| D | crop_and_resize_op_gpu.cu.cc | 43 const int32* box_ind_ptr, int num_boxes, int batch, int image_height, in CropAndResizeKernel() argument
135 const int32* box_ind_ptr, int num_boxes, int batch, int image_height, in CropAndResizeBackpropImageKernel() argument
230 const float* boxes_ptr, const int32* box_ind_ptr, int num_boxes, int batch, in CropAndResizeBackpropBoxesKernel() argument
359 const int num_boxes = crops.dimension(0); in operator ()() local
364 const int total_count = num_boxes * crop_height * crop_width * depth; in operator ()()
377 boxes.data(), box_ind.data(), num_boxes, batch, image_height, in operator ()()
397 const int num_boxes = grads.dimension(0); in operator ()() local
421 total_count = num_boxes * crop_height * crop_width * depth; in operator ()()
427 grads.data(), boxes.data(), box_ind.data(), num_boxes, batch, in operator ()()
447 const int num_boxes = grads.dimension(0); in operator ()() local
[all …]
|
| D | draw_bounding_box_op.cc | 90 const int64 num_boxes = boxes.dim_size(1); in Compute() local
92 for (int64 bb = 0; bb < num_boxes; ++bb) { in Compute()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | image_ops.cc | 332 const int64 num_boxes; member
335 explicit WhileCondFn(int64 num_boxes, int64 output_size) in WhileCondFn()
336 : num_boxes(num_boxes), output_size(output_size) {} in WhileCondFn()
342 xla::Lt(row_idx, xla::ConstantR0<int32>(cond_builder, num_boxes)); in operator ()()
355 const int64 num_boxes; member
357 explicit SuppressBodyFn(int64 num_boxes) : num_boxes(num_boxes) {} in SuppressBodyFn()
379 auto row_iou = xla::DynamicSlice(iou_mask, {row_idx, zero}, {1, num_boxes}); in operator ()()
386 row_iou = xla::Reshape(row_iou, {num_boxes}); in operator ()()
389 included_iou = xla::Select(xla::Broadcast(active_elem, {num_boxes}), in operator ()()
411 const int64 num_boxes = boxes_shape.dim_size(0); in Compile() local
[all …]
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | detection_postprocess.cc | 264 const int num_boxes = input_box_encodings->dims->data[1]; in DecodeCenterSizeBoxes() local
273 for (int idx = 0; idx < num_boxes; ++idx) { in DecodeCenterSizeBoxes()
338 bool ValidateBoxes(const TfLiteTensor* decoded_boxes, const int num_boxes) { in ValidateBoxes() argument
339 for (int i = 0; i < num_boxes; ++i) { in ValidateBoxes()
380 const int num_boxes = input_box_encodings->dims->data[1]; in NonMaxSuppressionSingleClassHelper() local
392 TF_LITE_ENSURE(context, ValidateBoxes(decoded_boxes, num_boxes)); in NonMaxSuppressionSingleClassHelper()
412 TF_LITE_ENSURE(context, (active_candidate->dims->data[0]) == num_boxes); in NonMaxSuppressionSingleClassHelper()
469 const int num_boxes = input_box_encodings->dims->data[1]; in NonMaxSuppressionMultiClassRegularHelper() local
480 std::vector<float> class_scores(num_boxes); in NonMaxSuppressionMultiClassRegularHelper()
483 num_boxes + max_detections); in NonMaxSuppressionMultiClassRegularHelper()
[all …]
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | image_grad_test.py | 336 num_boxes = 2
340 crops_shape = [num_boxes, crop_height, crop_width, depth]
352 boxes, shape=[num_boxes, 4]),
354 box_ind, shape=[num_boxes]),
414 for num_boxes in range(1, 3):
416 batch = num_boxes
419 crops_shape = [num_boxes, crop_height, crop_width, depth]
420 boxes_shape = [num_boxes, 4]
425 for _ in range(num_boxes):
439 boxes_tensor = constant_op.constant(boxes, shape=[num_boxes, 4])
[all …]
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_CropAndResizeGradBoxes.pbtxt | 6 A 4-D tensor of shape `[num_boxes, crop_height, crop_width, depth]`.
19 A 2-D tensor of shape `[num_boxes, 4]`. The `i`-th row of the tensor
34 A 1-D tensor of shape `[num_boxes]` with int32 values in `[0, batch)`.
41 A 2-D tensor of shape `[num_boxes, 4]`.
|
| D | api_def_CropAndResizeGradImage.pbtxt | 6 A 4-D tensor of shape `[num_boxes, crop_height, crop_width, depth]`.
12 A 2-D tensor of shape `[num_boxes, 4]`. The `i`-th row of the tensor
27 A 1-D tensor of shape `[num_boxes]` with int32 values in `[0, batch)`.
|
| D | api_def_CropAndResize.pbtxt | 13 A 2-D tensor of shape `[num_boxes, 4]`. The `i`-th row of the tensor
28 A 1-D tensor of shape `[num_boxes]` with int32 values in `[0, batch)`.
44 A 4-D tensor of shape `[num_boxes, crop_height, crop_width, depth]`.
73 `[num_boxes, crop_height, crop_width, depth]`. The resizing is corner aligned.
|
| D | api_def_NonMaxSuppressionWithOverlaps.pbtxt | 6 A 2-D float tensor of shape `[num_boxes, num_boxes]` representing
13 A 1-D float tensor of shape `[num_boxes]` representing a single
|
| D | api_def_NonMaxSuppression.pbtxt | 6 A 2-D float tensor of shape `[num_boxes, 4]`.
12 A 1-D float tensor of shape `[num_boxes]` representing a single
|
| D | api_def_NonMaxSuppressionV2.pbtxt | 6 A 2-D float tensor of shape `[num_boxes, 4]`.
12 A 1-D float tensor of shape `[num_boxes]` representing a single
|
| D | api_def_NonMaxSuppressionV3.pbtxt | 6 A 2-D float tensor of shape `[num_boxes, 4]`.
12 A 1-D float tensor of shape `[num_boxes]` representing a single
|
| D | api_def_NonMaxSuppressionV4.pbtxt | 6 A 2-D float tensor of shape `[num_boxes, 4]`.
12 A 1-D float tensor of shape `[num_boxes]` representing a single
|
| D | api_def_CombinedNonMaxSuppression.pbtxt | 6 A 4-D float tensor of shape `[batch_size, num_boxes, q, 4]`. If `q` is 1 then
14 A 3-D float tensor of shape `[batch_size, num_boxes, num_classes]`
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | draw_bounding_box_op_test.py | 76 for num_boxes in range(1, num_colors + 2):
79 color = color_table[(num_boxes - 1) % num_colors, 0:depth]
82 bboxes = np.vstack([bboxes for _ in range(num_boxes)])
|
| /external/mesa3d/src/gallium/state_trackers/xa/ |
| D | xa_context.h | 63 unsigned int num_boxes,
80 unsigned int num_boxes);
|
| D | xa_yuv.c | 119 unsigned int num_boxes, in xa_yuv_planar_blit() argument
143 while (num_boxes--) { in xa_yuv_planar_blit()
|
| D | xa_context.c | 101 int to_surface, struct xa_box *boxes, unsigned int num_boxes) in xa_surface_dma() argument
112 for (i = 0; i < num_boxes; ++i, ++boxes) { in xa_surface_dma()
|
| /external/tensorflow/tensorflow/examples/multibox_detector/ |
| D | main.cc | 263 const int num_boxes, in PrintTopDetections() argument
275 CHECK_EQ(label_count, num_boxes * 8); in PrintTopDetections()
345 int32 num_boxes = 784; in main() local
365 Flag("num_boxes", &num_boxes, in main()
423 Status print_status = PrintTopDetections(outputs, box_priors, num_boxes, in main()
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | image_ops_test.py | 736 num_boxes = 1024
737 boxes_np = np.random.normal(50, 10, (num_boxes, 4)).astype("f4")
738 scores_np = np.random.normal(0.5, 0.1, (num_boxes,)).astype("f4")
|