Lines Matching defs:box_index
54 const Tensor& box_index,
56 if (boxes.NumElements() == 0 && box_index.NumElements() == 0) {
69 // The shape of 'box_index' is [num_boxes].
70 if (box_index.dims() != 1) {
71 return errors::InvalidArgument("box_index must be 1-D",
72 box_index.shape().DebugString());
74 if (box_index.dim_size(0) != *num_boxes) {
75 return errors::InvalidArgument("box_index has incompatible shape");
80 // Conditionally calls the compute callback if all values in box_index are in
84 OpKernelContext* context, typename TTypes<int32, 1>::ConstTensor box_index,
90 OpKernelContext* context, typename TTypes<int32, 1>::ConstTensor box_index,
92 const int num_boxes = box_index.dimension(0);
95 context, FastBoundsCheck(box_index(b), batch_size),
96 errors::OutOfRange("box_index has values outside [0, batch_size)"),
128 // The shape of 'box_index' is [num_boxes].
129 const Tensor& box_index = context->input(2);
147 context, ParseAndCheckBoxSizes(boxes, box_index, &num_boxes), done);
179 const Tensor& box_index = context->input(2);
182 box_index.tensor<int32, 1>(), extrapolation_value_,
190 RunIfBoxIndexIsValid<Device>(context, box_index.tensor<int32, 1>(),
206 typename TTypes<int32, 1>::ConstTensor box_index,
226 const int32 b_in = box_index(b);
324 // The shape of 'box_index' is [num_boxes].
325 const Tensor& box_index = context->input(2);
341 context, ParseAndCheckBoxSizes(boxes, box_index, &num_boxes), done);
379 const Tensor& box_index = context->input(2);
382 boxes.tensor<float, 2>(), box_index.tensor<int32, 1>(),
390 RunIfBoxIndexIsValid<Device>(context, box_index.tensor<int32, 1>(),
403 typename TTypes<int32, 1>::ConstTensor box_index,
422 const int32 b_in = box_index(b);
493 // The shape of 'box_index' is [num_boxes].
494 const Tensor& box_index = context->input(3);
526 context, ParseAndCheckBoxSizes(boxes, box_index, &num_boxes), done);
544 const Tensor& box_index = context->input(3);
548 box_index.tensor<int32, 1>(), output->tensor<float, 2>());
555 RunIfBoxIndexIsValid<Device>(context, box_index.tensor<int32, 1>(),
569 typename TTypes<int32, 1>::ConstTensor box_index,
588 const int32 b_in = box_index(b);
709 const GPUDevice& d, typename TTypes<int32, 1>::ConstTensor box_index,
719 OpKernelContext* context, typename TTypes<int32, 1>::ConstTensor box_index,
721 const int num_boxes = box_index.dimension(0);
739 context->eigen_device<GPUDevice>(), box_index, batch_size, isvalid_dev);
780 errors::OutOfRange("box_index has values outside [0, batch_size)"),
