82 inline bool searchsorted_dims_matched_before_last_dim(const Tensor& boundaries, const Tensor& input…  in searchsorted_dims_matched_before_last_dim()  argument
83   if (boundaries.dim() != input.dim()) {  in searchsorted_dims_matched_before_last_dim()
86   const auto& dims_bd = boundaries.sizes();  in searchsorted_dims_matched_before_last_dim()
88   for (int64_t dim = 0; dim + 1 < boundaries.dim(); ++dim) {  in searchsorted_dims_matched_before_last_dim()
105     const Tensor& boundaries,  in searchsorted_pre_check()  argument
122 …TORCH_CHECK(boundaries.device() == input.device(), "torch.searchsorted(): boundaries and input val…  in searchsorted_pre_check()
123 …"should have same device type, but got boundaries tensor device type ", boundaries.device(), " and…  in searchsorted_pre_check()
127 …TORCH_CHECK(sorter.device() == boundaries.device(), "torch.searchsorted(): sorter and boundary ten…  in searchsorted_pre_check()
129       "device type ", boundaries.device());  in searchsorted_pre_check()
131 …TORCH_CHECK(sorter.sizes() == boundaries.sizes(), "torch.searchsorted(): boundary and sorter must …  in searchsorted_pre_check()
132 …   "size, but got boundary tensor ", boundaries.sizes(), "and got sorter tensor ", sorter.sizes());  in searchsorted_pre_check()
145   TORCH_CHECK(input.dim() > 0 || (input.dim() == 0 && input.numel() == 1 && boundaries.dim() == 1),  in searchsorted_pre_check()
146 …"torch.searchsorted(): input value can be a scalar only when boundaries tensor dimension is 1, but…  in searchsorted_pre_check()
147     "boundaries tensor dim(", boundaries.dim(), ") and input value's dim(", input.dim(), ") numel(",  in searchsorted_pre_check()
150 …TORCH_CHECK(boundaries.dim() != 0, "torch.searchsorted(): boundaries tensor should have positive d…  in searchsorted_pre_check()
153   TORCH_CHECK(boundaries.dim() == 1 || searchsorted_dims_matched_before_last_dim(boundaries, input),  in searchsorted_pre_check()
154 …"torch.searchsorted(): boundaries tensor should be 1 dimension or the first N-1 dimensions of boun…  in searchsorted_pre_check()
155 …"and input value tensor must match, but we got boundaries tensor ", boundaries.sizes(), " and inpu…  in searchsorted_pre_check()
167     TORCH_CHECK(boundaries.sizes().back() < INT_MAX,  in searchsorted_pre_check()
168 …"torch.searchsorted(): the size of boundaries' last dimension should be less than ", INT_MAX, ", b…  in searchsorted_pre_check()
169       boundaries.sizes().back());  in searchsorted_pre_check()