Home
last modified time | relevance | path

Searched refs:sorted_sequence (Results 1 – 15 of 15) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DBucketization.cpp145 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 DBucketization.cu133 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 Dsearchsorted_op_test.py36 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 DBucketization.mm254 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 Dapi_def_UpperBound.pbtxt35 sorted_sequence = [[0, 3, 9, 9, 10],
40 result = UpperBound(sorted_sequence, values)
H A Dapi_def_LowerBound.pbtxt35 sorted_sequence = [[0, 3, 9, 9, 10],
40 result = LowerBound(sorted_sequence, values)
/aosp_15_r20/external/pytorch/aten/src/ATen/functorch/
H A DBatchRulesReduceOps.cpp318 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 Darray_ops.py6393 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 Dtensorflow.pbtxt993 …argspec: "args=[\'sorted_sequence\', \'values\', \'side\', \'out_type\', \'name\'], varargs=None, …
/aosp_15_r20/external/tensorflow/tensorflow/python/feature_column/
H A Dfeature_column_v2_test.py95 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 Dcommon_methods_invocations.py11429 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])
11445sorted_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.py6326 sorted_sequence, argument
6338 return torch.empty((), dtype=dtype, device=sorted_sequence.device)
H A Doverrides.py1099 … 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 Dtensorflow.pbtxt2053 …argspec: "args=[\'sorted_sequence\', \'values\', \'side\', \'out_type\', \'name\'], varargs=None, …
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
H A Dtf_generated_ops.td7905 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)