| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | Bucketization.cpp | 117 void dispatch(Tensor& result, const Tensor& input, const Tensor& boundaries, bool out_int32, bool r… in dispatch()
147 bool out_int32, in searchsorted_out_cpu()
194 bool out_int32, in searchsorted_out_cpu()
206 bool out_int32, in searchsorted_cpu()
220 bool out_int32, in searchsorted_cpu()
228 Tensor& bucketize_out_cpu(const Tensor& self, const Tensor& boundaries, bool out_int32, bool right,… in bucketize_out_cpu()
234 Tensor bucketize_cpu(const Tensor& self, const Tensor& boundaries, bool out_int32, bool right) { in bucketize_cpu()
242 Tensor bucketize_cpu(const Scalar& self, const Tensor& boundaries, bool out_int32, bool right) { in bucketize_cpu()
|
| H A D | BucketizationUtils.h | 108 const bool out_int32, in searchsorted_pre_check()
|
| H A D | TensorConversions.cpp | 142 bool out_int32) { in compressed_to_batched_compressed_indices()
1469 bool out_int32 = (row_indices.scalar_type() == at::kInt); in coo_to_sparse_csr() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | Bucketization.cu | 115 bool out_int32, in dispatch()
135 bool out_int32, in searchsorted_out_cuda()
181 bool out_int32, in searchsorted_out_cuda()
193 bool out_int32, in searchsorted_cuda()
207 bool out_int32, in searchsorted_cuda()
215 Tensor& bucketize_out_cuda(const Tensor& self, const Tensor& boundaries, bool out_int32, bool right… in bucketize_out_cuda()
221 Tensor bucketize_cuda(const Tensor& self, const Tensor& boundaries, bool out_int32, bool right) { in bucketize_cuda()
229 Tensor bucketize_cuda(const Scalar& self, const Tensor& boundaries, bool out_int32, bool right) { in bucketize_cuda()
|
| /aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/arm_gemm/ |
| H A D | interleave_indirect_impl.hpp | 57 int32_t *out_int32 = reinterpret_cast<int32_t *>(out); in interleave_block() local
97 int32_t *out_int32 = reinterpret_cast<int32_t *>(out); in interleave_block() local
115 int32_t *out_int32 = reinterpret_cast<int32_t *>(out); in FixupRowSums() local
124 int32_t *out_int32 = reinterpret_cast<int32_t *>(out); in FixupRowSums() local
|
| H A D | interleave_indirect.cpp | 76 int32_t *out_int32 = reinterpret_cast<int32_t *>(out); in interleave_block() local
116 int32_t *out_int32 = reinterpret_cast<int32_t *>(out); in interleave_block() local
133 int32_t *out_int32 = reinterpret_cast<int32_t *>(out); in FixupRowSums() local
142 int32_t *out_int32 = reinterpret_cast<int32_t *>(out); in FixupRowSums() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/functorch/ |
| H A D | BatchRulesReduceOps.cpp | 322 bool out_int32, in searchsorted_batch_rule()
409 bool out_int32, in bucketize_decomp_Tensor()
419 bool out_int32, in bucketize_decomp_Scalar()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/sparse/ |
| H A D | SparseCsrTensorMath.cpp | 1439 bool out_int32 = crow.scalar_type() == ScalarType::Int; in _sparse_mm_reduce_impl_backward_sparse_csr_cpu() local
|
| /aosp_15_r20/external/pytorch/torch/ |
| H A D | _meta_registrations.py | 6194 def meta_bucketize(self, boundaries, *, out_int32=False, right=False): argument
6329 out_int32=False, argument
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/static/ |
| H A D | generated_ops.cpp | 3361 const auto out_int32 = p_node->Input(2).toBool(); in __anon5d9c3eb914202() local
3386 const auto out_int32 = p_node->Input(2).toBool(); in __anon5d9c3eb914402() local
|
| /aosp_15_r20/external/pytorch/test/inductor/ |
| H A D | test_torchinductor.py | 9685 def fn(input, boundaries, out_int32, right): argument
9709 def fn(input, offsets, out_int32, right): argument
|
| /aosp_15_r20/external/pytorch/torch/onnx/ |
| H A D | symbolic_opset9.py | 2981 g: jit_utils.GraphContext, self, boundaries, out_int32=False, right=False argument
|
| /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
|