/aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
H A D | Bucketization.cpp | 145 const Tensor& sorted_sequence, in searchsorted_out_cpu() argument 155 searchsorted_pre_check(sorted_sequence, self, result, out_int32, right, side_opt, sorter); in searchsorted_out_cpu() 170 …if (sorted_sequence.is_contiguous() && self.is_contiguous() && sorted_sequence.dtype() == self.dty… in searchsorted_out_cpu() 171 dispatch(out, self, sorted_sequence, out_int32, is_right, sorter); in searchsorted_out_cpu() 177 …im_input_tensors(trimmed_input, trimmed_boundaries, trimmed_sorter, self, sorted_sequence, sorter); in searchsorted_out_cpu() 179 …nst Tensor& final_boundaries = trimmed_boundaries.defined() ? trimmed_boundaries : sorted_sequence; in searchsorted_out_cpu() 192 const Tensor& sorted_sequence, in searchsorted_out_cpu() argument 199 const Tensor& scalar_tensor = searchsorted_scalar_tensor(self, sorted_sequence.device()); in searchsorted_out_cpu() 200 …return searchsorted_out_cpu(sorted_sequence, scalar_tensor, out_int32, right, side_opt, sorter_opt… in searchsorted_out_cpu() 204 const Tensor& sorted_sequence, in searchsorted_cpu() argument [all …]
|
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
H A D | Bucketization.cu | 133 const Tensor& sorted_sequence, in searchsorted_out_cuda() argument 143 searchsorted_pre_check(sorted_sequence, self, result, out_int32, right, side_opt, sorter); in searchsorted_out_cuda() 157 …if (sorted_sequence.is_contiguous() && self.is_contiguous() && sorted_sequence.dtype() == self.dty… in searchsorted_out_cuda() 158 dispatch(out, self, sorted_sequence, out_int32, is_right, sorter); in searchsorted_out_cuda() 164 …im_input_tensors(trimmed_input, trimmed_boundaries, trimmed_sorter, self, sorted_sequence, sorter); in searchsorted_out_cuda() 166 …nst Tensor& final_boundaries = trimmed_boundaries.defined() ? trimmed_boundaries : sorted_sequence; in searchsorted_out_cuda() 179 const Tensor& sorted_sequence, in searchsorted_out_cuda() argument 186 const Tensor& scalar_tensor = searchsorted_scalar_tensor(self, sorted_sequence.device()); in searchsorted_out_cuda() 187 …return searchsorted_out_cuda(sorted_sequence, scalar_tensor, out_int32, right, side_opt, sorter_op… in searchsorted_out_cuda() 191 const Tensor& sorted_sequence, in searchsorted_cuda() argument [all …]
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tests/ |
H A D | searchsorted_op_test.py | 36 sorted_sequence = np.sort(unsorted) 37 np_ans = np.searchsorted(sorted_sequence, values, side=side) 41 tf_ans = array_ops.searchsorted(sorted_sequence, values, side=side) 45 def _test2DExample(self, dtype, side, sorted_sequence, values, correct_ans): argument 49 tf_ans = array_ops.searchsorted(sorted_sequence, values, side=side) 56 sorted_sequence = np.array([[0, 3, 9, 9, 10], [1, 2, 3, 4, 5]], dtype) 59 self._test2DExample(dtype, 'left', sorted_sequence, values, correct_ans) 64 sorted_sequence = np.array([[0, 3, 9, 9, 10], [1, 2, 3, 4, 5]], dtype) 67 self._test2DExample(dtype, 'right', sorted_sequence, values, correct_ans)
|
/aosp_15_r20/external/pytorch/aten/src/ATen/native/mps/operations/ |
H A D | Bucketization.mm | 254 Tensor& searchsorted_out_mps(const Tensor& sorted_sequence, 264 searchsorted_pre_check(sorted_sequence, self, result, out_int32, right, side_opt, sorter); 277 …if (sorted_sequence.is_contiguous() && self.is_contiguous() && sorted_sequence.dtype() == self.dty… 279 mps::searchsorted_mps_contiguous(out, self, sorted_sequence, right, sorter); 285 trimmed_input, trimmed_boundaries, trimmed_sorter, self, sorted_sequence, sorter); 287 …nst Tensor& final_boundaries = trimmed_boundaries.defined() ? trimmed_boundaries : sorted_sequence; 299 Tensor& searchsorted_out_mps(const Tensor& sorted_sequence, 306 const Tensor& scalar_tensor = mps::wrapped_scalar_tensor_mps(self, sorted_sequence.device()); 307 …return searchsorted_out_mps(sorted_sequence, scalar_tensor, out_int32, right, side_opt, sorter_opt… 310 Tensor searchsorted_mps(const Tensor& sorted_sequence, [all …]
|
/aosp_15_r20/external/tensorflow/tensorflow/core/api_def/base_api/ |
H A D | api_def_UpperBound.pbtxt | 35 sorted_sequence = [[0, 3, 9, 9, 10], 40 result = UpperBound(sorted_sequence, values)
|
H A D | api_def_LowerBound.pbtxt | 35 sorted_sequence = [[0, 3, 9, 9, 10], 40 result = LowerBound(sorted_sequence, values)
|
/aosp_15_r20/external/pytorch/aten/src/ATen/functorch/ |
H A D | BatchRulesReduceOps.cpp | 318 const Tensor& sorted_sequence, in searchsorted_batch_rule() argument 327 auto buckets_logical_rank = rankWithoutBatchDim(sorted_sequence, sorted_sequence_bdim); in searchsorted_batch_rule() 333 auto buckets = moveBatchDimToFront(sorted_sequence, sorted_sequence_bdim); in searchsorted_batch_rule() 344 sorted_sequence, sorted_sequence_bdim, in searchsorted_batch_rule()
|
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/ |
H A D | array_ops.py | 6393 def searchsorted(sorted_sequence, argument 6458 sequence_size = shape_internal(sorted_sequence)[-1] 6460 sorted_sequence_2d = reshape(sorted_sequence, [-1, sequence_size])
|
/aosp_15_r20/external/tensorflow/tensorflow/tools/api/golden/v2/ |
H A D | tensorflow.pbtxt | 993 …argspec: "args=[\'sorted_sequence\', \'values\', \'side\', \'out_type\', \'name\'], varargs=None, …
|
/aosp_15_r20/external/tensorflow/tensorflow/python/feature_column/ |
H A D | feature_column_v2_test.py | 95 sorted_sequence = ['0', a, b, c, 'd'] 96 reversed_sequence = sorted_sequence[::-1] 97 self.assertAllEqual(sorted(reversed_sequence), sorted_sequence)
|
/aosp_15_r20/external/pytorch/torch/testing/_internal/ |
H A D | common_methods_invocations.py | 11429 def reference_searchsorted(sorted_sequence, boundary, out_int32=False, right=False, side='left', so… argument 11431 if len(sorted_sequence.shape) == 1 : 11432 ret = np.searchsorted(sorted_sequence, boundary, side=side, sorter=sorter) 11434 elif sorted_sequence.shape[0] == 0: 11437 … ret = np.searchsorted(sorted_sequence.flatten(), boundary.flatten(), side=side, sorter=sorter) 11443 num_splits = np.prod(sorted_sequence.shape[:-1]) 11445 …sorted_sequence, boundary = sorted_sequence.reshape(num_splits, -1), boundary.reshape(num_splits, … 11449 split_sequence = [sorted_sequence[i] for i in splits]
|
/aosp_15_r20/external/pytorch/torch/ |
H A D | _meta_registrations.py | 6326 sorted_sequence, argument 6338 return torch.empty((), dtype=dtype, device=sorted_sequence.device)
|
H A D | overrides.py | 1099 … torch.searchsorted: lambda sorted_sequence, input, out_int32=False, right=False, out=None: -1,
|
/aosp_15_r20/external/tensorflow/tensorflow/tools/api/golden/v1/ |
H A D | tensorflow.pbtxt | 2053 …argspec: "args=[\'sorted_sequence\', \'values\', \'side\', \'out_type\', \'name\'], varargs=None, …
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
H A D | tf_generated_ops.td | 7905 sorted_sequence = [[0, 3, 9, 9, 10], 7910 result = LowerBound(sorted_sequence, values) 19534 sorted_sequence = [[0, 3, 9, 9, 10], 19539 result = UpperBound(sorted_sequence, values)
|