Searched refs:sorted_data (Results 1 – 6 of 6) sorted by relevance
| /external/pytorch/aten/src/ATen/native/cuda/ |
| D | LegacyThrustHelpers.cu | 40 auto sorted_data = device_ptr(sorted_indices.mutable_data_ptr<int64_t>()); in index_put_with_sort_kernel_thrust_helper() local
41 thrust::sort_by_key(policy, sorted_data, sorted_data + num_indices, orig_data, LTOp<int64_t>()); in index_put_with_sort_kernel_thrust_helper()
57 auto sorted_data = thrust::device_ptr<const index_t>(sorted_indices.const_data_ptr<index_t>()); in embedding_dense_backward_cuda_scan() local
61 sorted_data, in embedding_dense_backward_cuda_scan()
62 sorted_data + num_indices, in embedding_dense_backward_cuda_scan()
72 thrust::make_reverse_iterator(sorted_data + num_indices), in embedding_dense_backward_cuda_scan()
73 thrust::make_reverse_iterator(sorted_data), in embedding_dense_backward_cuda_scan()
|
| D | Embedding.cu | 316 auto sorted_data = sorted_indices.const_data_ptr<index_t>(); in embedding_dense_backward_cuda() local
319 sorted_data, in embedding_dense_backward_cuda()
329 thrust::make_reverse_iterator(sorted_data + num_indices), in embedding_dense_backward_cuda()
|
| D | EmbeddingBag.cu | 208 auto sorted_data = sorted_indices.const_data_ptr<index_t>(); in embedding_bag_backward_cuda_sum_avg() local
211 sorted_data, in embedding_bag_backward_cuda_sum_avg()
221 thrust::make_reverse_iterator(sorted_data + num_indices), in embedding_bag_backward_cuda_sum_avg()
|
| /external/rust/android-crates-io/crates/rayon-core/src/join/ |
| D | test.rs | 43 let mut sorted_data = data.clone(); in sort() localVariable
44 sorted_data.sort(); in sort()
46 assert_eq!(data, sorted_data); in sort()
56 let mut sorted_data = data.clone(); in sort_in_pool() localVariable
57 sorted_data.sort(); in sort_in_pool()
59 assert_eq!(data, sorted_data); in sort_in_pool()
|
| /external/perfetto/tools/ |
| D | diff_test_trace_processor.py | 106 sorted_data = sorted(
108 for perf_args in sorted_data:
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | dynamic_partition_op.cc | 74 xla::XlaOp sorted_data = xla::GetTupleElement(sorted, 1); in DynamicPartition1D() local
96 xla::Pad(sorted_data, xla::Zero(ctx->builder(), ctx->input_xla_type(0)), in DynamicPartition1D()
|