| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | SortStable.cu | 110 int64_t* indices_ptr) { in segmented_sort_large_segments()
139 int64_t* const indices_ptr) { in segmented_sort_pairs_by_full_sort()
190 int64_t* indices_ptr) { in segmented_sort_pairs()
234 int64_t* indices_ptr = indices.mutable_data_ptr<int64_t>(); in launch_stable_sort_kernel() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/ |
| H A D | AdaptiveMaxPoolKernel.cpp | 44 int64_t* indices_ptr = indices_data + c * output_height * output_width; in cpu_adaptive_max_pool2d() local
367 const int64_t* indices_ptr = indices_data + c * output_height * output_width; in cpu_adaptive_max_pool2d_backward() local
415 const int64_t* indices_ptr = indices_data + n * output_height * output_width * channels; in cpu_adaptive_max_pool2d_backward_channels_last() local
512 int64_t* indices_ptr = indices_data + c * output_depth * output_height * output_width; in cpu_adaptive_max_pool3d() local
859 int64_t* indices_ptr = indices_data + c * output_depth * output_height * output_width; in cpu_adaptive_max_pool3d_backward() local
911 … int64_t* indices_ptr = indices_data + n * output_depth * output_height * output_width * channels; in cpu_adaptive_max_pool3d_backward_channels_last() local
|
| H A D | MaxPoolKernel.cpp | 298 int64_t* indices_ptr = indices_data + c * output_depth * output_height * output_width; in cpu_max_pool() local
514 const int64_t * indices_ptr = indices_data + c * output_depth * output_height * output_width; in cpu_max_pool_backward() local
574 …const int64_t* indices_ptr = indices_data + n * output_depth * output_height * output_width * chan… in cpu_max_pool_backward_channels_last() local
|
| H A D | MaxUnpoolKernel.cpp | 131 const int64_t* indices_ptr = indices_data + i * channels; in cpu_max_unpool_channels_last() local
|
| /aosp_15_r20/external/executorch/kernels/portable/cpu/util/ |
| H A D | kernel_ops_util.h | 154 int64_t* const indices_ptr, in kernel_reduction_then_map_2d()
347 int64_t* indices_ptr = nullptr; variable
|
| /aosp_15_r20/external/executorch/backends/cadence/reference/operators/ |
| H A D | op_embedding.cpp | 29 const int64_t* indices_ptr = indices.const_data_ptr<int64_t>(); in embedding_out() local
|
| /aosp_15_r20/external/ComputeLibrary/tests/validation/reference/ |
| H A D | Gather.cpp | 42 const auto *indices_ptr = static_cast<const uint32_t *>(indices.data()); in gather() local
|
| /aosp_15_r20/external/executorch/kernels/portable/cpu/ |
| H A D | op_embedding.cpp | 38 const CTYPE* indices_ptr = indices.const_data_ptr<CTYPE>(); in embedding_kernel() local
|
| /aosp_15_r20/external/ComputeLibrary/tests/validation/fixtures/ |
| H A D | GatherFixture.h | 68 uint32_t *indices_ptr = static_cast<uint32_t *>(indices.data()); in generate_indices() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ops/ |
| H A D | data_format_vec_permute.cc | 152 int32* indices_ptr = indices_weights->GetPointer<int32>(); in Convert() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | sparse_to_dense_op_gpu.cu.cc | 149 const Index* indices_ptr = indices.flat<Index>().data(); in operator ()() local
|
| H A D | serialize_sparse_op.cc | 303 const int64_t* indices_ptr = indices.data(); in operator ()() local
|
| H A D | scatter_nd_op.cc | 880 se::DeviceMemoryBase indices_ptr( in DoScatterNdOnCpu() local
|
| /aosp_15_r20/external/executorch/kernels/quantized/cpu/ |
| H A D | op_embedding.cpp | 146 const int64_t* indices_ptr = indices.const_data_ptr<int64_t>(); in embedding_byte_per_channel() local
|
| H A D | embeddingxb.cpp | 186 const int64_t* indices_ptr = indices.const_data_ptr<int64_t>(); in embedding_xbit_per_channel() local
|