| /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)
|