Searched refs:boxes_data (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | non_max_suppression_op.cc | 140 typename TTypes<T, 2>::ConstTensor boxes_data = boxes.tensor<T, 2>(); in CreateIOUSuppressCheckFn() local
141 return std::bind(&IOUGreaterThanThreshold<T>, boxes_data, in CreateIOUSuppressCheckFn()
266 float* boxes_data = per_batch_boxes.unaligned_flat<float>().data(); in BatchedNonMaxSuppressionOp() local
283 boxes_data[(box * q + class_idx) * 4 + cid]); in BatchedNonMaxSuppressionOp()
285 class_boxes_data.push_back(boxes_data[box * 4 + cid]); in BatchedNonMaxSuppressionOp()
319 typename TTypes<float, 2>::ConstTensor boxes_data = in BatchedNonMaxSuppressionOp() local
331 if (IOUGreaterThanThreshold(boxes_data, next_candidate.box_index, in BatchedNonMaxSuppressionOp()
345 {boxes_data(id, 0), boxes_data(id, 1), in BatchedNonMaxSuppressionOp()
346 boxes_data(id, 2), boxes_data(id, 3)}}; in BatchedNonMaxSuppressionOp()
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | image_ops_test.py | 771 boxes_data = [[0, 0, 1, 1], [0, 0.1, 1, 1.1], [0, -0.1, 1, 0.9],
773 boxes_np = np.array(boxes_data, dtype=np.float32)
814 boxes_data = [[0, 0, 1, 1], [0, 0.1, 1, 1.1], [0, -0.1, 1, 0.9],
816 boxes_np = np.array(boxes_data, dtype=np.float32)
857 boxes_data = [[0, 0, 1, 1], [0, 0.1, 1, 1.1], [0, -0.1, 1, 0.9],
859 boxes_np = np.array(boxes_data, dtype=np.float32)
898 boxes_data = [[0, 0, 1, 1], [0, 0.2, 1, 1.2], [0, 0.4, 1, 1.4],
900 boxes_np = np.array(boxes_data, dtype=np.float32)
|