Searched refs:kNumBoxes (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | non_max_suppression_test.cc | 29 constexpr int kNumBoxes = 6; variable
88 std::vector<float> boxes(kNumBoxes * 4); in TEST()
89 std::vector<float> scores(kNumBoxes); in TEST()
90 for (int i = 0; i < kNumBoxes; ++i) { in TEST()
99 const int max_output_size = kNumBoxes; in TEST()
107 boxes.data(), kNumBoxes, scores.data(), max_output_size, iou_threshold, in TEST()
115 boxes.data(), kNumBoxes, scores.data(), max_output_size, iou_threshold, in TEST()
137 boxes.data(), kNumBoxes, scores.data(), max_output_size, iou_threshold, in TEST()
147 boxes.data(), kNumBoxes, scores.data(), max_output_size, iou_threshold, in TEST()
157 boxes.data(), kNumBoxes, scores.data(), max_output_size, iou_threshold, in TEST()
[all …]
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | op_level_cost_estimator_test.cc | 2246 const int kNumBoxes = 10; in TEST_F() local
2248 kNumBoxes * kOutputImageDim * kOutputImageDim * kChannelSize; in TEST_F()
2254 DescribeArbitraryRankInput({kNumBoxes, 4}, DT_INT64, &op_context.op_info); in TEST_F()
2255 DescribeTensor4D(kNumBoxes, kOutputImageDim, kOutputImageDim, kChannelSize, in TEST_F()
2262 kNumBoxes * 4 * 8 / 10 + // boxes (kNumBoxes x 4) in int64. in TEST_F()
2263 kNumBoxes * kOutputImageDim * kOutputImageDim * 4); // output in float. in TEST_F()
2272 int num_ops = 28 * kNumBoxes + 4 * kNumBoxes * kOutputImageDim + in TEST_F()
2273 4 * kNumBoxes * kOutputImageDim * kOutputImageDim + in TEST_F()
2274 3 * kNumBoxes * kOutputImageDim + in TEST_F()
2275 3 * kNumBoxes * kOutputImageDim * kOutputImageDim + in TEST_F()
[all …]
|