Searched refs:input_boxes (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | non_max_suppression.cc | 82 const TfLiteTensor* input_boxes; in Prepare() local
84 context, GetInputSafe(context, node, kInputTensorBoxes, &input_boxes)); in Prepare()
85 TF_LITE_ENSURE_EQ(context, input_boxes->type, kTfLiteFloat32); in Prepare()
86 TF_LITE_ENSURE_EQ(context, NumDimensions(input_boxes), 2); in Prepare()
87 TF_LITE_ENSURE_EQ(context, SizeOfDimension(input_boxes, 1), 4); in Prepare()
88 const int num_boxes = SizeOfDimension(input_boxes, 0); in Prepare()
204 const TfLiteTensor* input_boxes; in Eval() local
206 context, GetInputSafe(context, node, kInputTensorBoxes, &input_boxes)); in Eval()
207 const int num_boxes = SizeOfDimension(input_boxes, 0); in Eval()
260 input_boxes->data.f, num_boxes, input_scores->data.f, in Eval()
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/image/ |
| D | sample_distorted_bounding_box_op.cc | 281 const Tensor& input_boxes = context->input(1); in DoCompute() local
282 OP_REQUIRES(context, input_boxes.dims() == 3, in DoCompute()
285 input_boxes.shape().DebugString())); in DoCompute()
286 OP_REQUIRES(context, input_boxes.dim_size(input_boxes.dims() - 1) == 4, in DoCompute()
289 input_boxes.shape().DebugString())); in DoCompute()
315 if (input_boxes.NumElements() > 0) { in DoCompute()
316 TTypes<float>::ConstMatrix boxes = input_boxes.flat_inner_dims<float>(); in DoCompute()
|