Home
last modified time | relevance | path

Searched defs:values_ptr (Results 1 – 25 of 34) sorted by relevance

12

/external/pytorch/aten/src/ATen/native/cuda/
DSortStable.cu109 scalar_t* values_ptr, in segmented_sort_large_segments()
138 scalar_t* const values_ptr, in segmented_sort_pairs_by_full_sort()
189 scalar_t* values_ptr, in segmented_sort_pairs()
239 scalar_t* values_ptr = values.mutable_data_ptr<scalar_t>(); in launch_stable_sort_kernel() local
/external/pytorch/aten/src/ATen/mkl/
DSparseDescriptors.h95 auto values_ptr = values_->data_ptr<scalar_t>(); in MklSparseCsrDescriptor() local
/external/openscreen/third_party/abseil/src/absl/container/internal/
Dinlined_vector.h81 ValueAdapter* values_ptr, SizeType construct_size) { in ConstructElements()
94 void AssignElements(Pointer assign_first, ValueAdapter* values_ptr, in AssignElements()
236 void Construct(Pointer data, ValueAdapter* values_ptr, SizeType size) { in Construct()
/external/pytorch/aten/src/ATen/cuda/
DCUDASparseDescriptors.cpp87 void* values_ptr = static_cast<char*>(data_ptr) + in createRawDnMatDescriptor() local
/external/pytorch/torch/sparse/
D_triton_ops.py1263 values_ptr, argument
1383 values_ptr, argument
1714 values_ptr, argument
2249 values_ptr, argument
/external/tensorflow/tensorflow/core/kernels/
Dsparse_to_dense_op_gpu.cu.cc150 const T* values_ptr = values.flat<T>().data(); in operator ()() local
Dserialize_sparse_op.cc306 const T* values_ptr = values.data(); in operator ()() local
/external/pytorch/aten/src/ATen/native/sparse/
DSparseCsrTensorMath.cpp1084 scalar_t* values_ptr = values.data_ptr<scalar_t>(); in reduce_sparse_csr_dim0_cpu_template() local
1189 scalar_t* values_ptr = values.data_ptr<scalar_t>(); in reduce_sparse_csr_dim1_cpu_template() local
1244 scalar_t* values_ptr = values.data_ptr<scalar_t>(); in reduce_sparse_csr_dim01_cpu_template() local
DSparseTensor.cpp671 scalar_t* values_ptr = values.data_ptr<scalar_t>(); in _coalesce_sparse_cpu() local
/external/pytorch/aten/src/ATen/native/sparse/cuda/
DSparseCsrTensorMath.cu495 scalar_t* values_ptr = values.data_ptr<scalar_t>(); in reduce_sparse_csr_dim0_cuda_template() local
607 scalar_t* values_ptr = values.data_ptr<scalar_t>(); in reduce_sparse_csr_dim1_cuda_template() local
/external/libjpeg-turbo/simd/arm/
Djcphuff-neon.c47 UJCOEF *values_ptr = values; in jsimd_encode_mcu_AC_first_prepare_neon() local
/external/igt-gpu-tools/include/drm-uapi/
Dvc4_drm.h435 __u64 values_ptr; member
Ddrm_mode.h416 __u64 values_ptr; /* values and blob lengths */ member
/external/kernel-headers/original/uapi/drm/
Dvc4_drm.h435 __u64 values_ptr; member
Dv3d_drm.h731 __u64 values_ptr; member
Ddrm_mode.h592 __u64 values_ptr; member
/external/libdrm/include/drm/
Dvc4_drm.h435 __u64 values_ptr; member
Ddrm_mode.h592 __u64 values_ptr; member
Di915_drm.h1315 __u64 values_ptr; member
/external/mesa3d/include/drm-uapi/
Dvc4_drm.h435 __u64 values_ptr; member
Dv3d_drm.h735 __u64 values_ptr; member
Ddrm_mode.h592 __u64 values_ptr; member
/external/pytorch/aten/src/ATen/native/nested/cuda/
DNestedTensorTransformerFunctions.cu1086 __half* values_ptr = reinterpret_cast<__half*>(&values[real_row][0]); in jagged_dense_dense_elementwise_jagged_output_opt_gather_kernel_() local
/external/intel-media-driver/media_softlet/linux/common/os/i915/include/uapi/
Ddrm_mode.h589 __u64 values_ptr; member
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.cc2406 llvm::Value* values_ptr = in HandleTopK() local

12