Home
last modified time | relevance | path

Searched defs:temp_storage_bytes (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dgpu_prim_helpers.h109 size_t temp_storage_bytes = 0; variable
184 size_t temp_storage_bytes; in GpuInclusivePrefixSum() local
220 size_t temp_storage_bytes; in GpuSegmentedReduce() local
261 size_t temp_storage_bytes; variable
Dhistogram_op_gpu.cu.cc63 size_t temp_storage_bytes = 0; in Compute() local
Ddynamic_partition_op_gpu.cu.cc339 size_t temp_storage_bytes = 0; in RadixSort() local
415 size_t temp_storage_bytes = 0; in CountAndSortParts() local
Dbincount_op_gpu.cu.cc62 size_t temp_storage_bytes = 0; in Compute() local
Dtopk_op_gpu.h449 size_t temp_storage_bytes = -1; in LaunchSortKernel() local
Dsparse_fill_empty_rows_op_gpu.cu.cc546 std::size_t temp_storage_bytes = 0; in operator ()() local
/external/pytorch/aten/src/ATen/native/cuda/
DNonzero.cu66 size_t temp_storage_bytes=0; in nonzero_cuda_out_impl() local
/external/tensorflow/tensorflow/core/kernels/sparse/
Dkernels_gpu.cu.cc71 std::size_t temp_storage_bytes = 0; in operator ()() local