Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Dhistogram_op_gpu.cu.cc63 size_t temp_storage_bytes = 0; in Compute() local
Dbincount_op_gpu.cu.cc52 size_t temp_storage_bytes = 0; in Compute() local
Ddynamic_partition_op_gpu.cu.cc350 size_t temp_storage_bytes = 0; in RadixSort() local
426 size_t temp_storage_bytes = 0; in CountAndSortParts() local
/external/tensorflow/tensorflow/core/kernels/sparse/
Dkernels_gpu.cu.cc71 std::size_t temp_storage_bytes = 0; in operator ()() local