HomeSort by relevance Sort by last modified time
    Searched refs:d_temp_storage (Results 1 - 3 of 3) sorted by null

  /external/tensorflow/tensorflow/core/kernels/
bincount_op_gpu.cu.cc 61 // with d_temp_storage = NULL passed to the call.
63 /* d_temp_storage */ NULL,
82 void* d_temp_storage = temp_storage.flat<int8>().data(); local
83 // The second HistogramEven is to actual run with d_temp_storage
86 /* d_temp_storage */ d_temp_storage,
where_op_gpu.cu.h 69 cudaError_t operator()(void* d_temp_storage, size_t& temp_storage_bytes,
76 return cub::DeviceReduce::Sum(d_temp_storage, temp_storage_bytes,
84 cudaError_t operator()(void* d_temp_storage, size_t& temp_storage_bytes,
88 return cub::DeviceReduce::Sum(d_temp_storage, temp_storage_bytes, d_in,
100 cudaError_t operator()(void* d_temp_storage, size_t& temp_storage_bytes,
110 d_temp_storage, temp_storage_bytes, select_counter /*d_in*/,
119 cudaError_t operator()(void* d_temp_storage, size_t& temp_storage_bytes,
126 d_temp_storage, temp_storage_bytes, select_counter /*d_in*/, d_flags,
histogram_op_gpu.cu.cc 72 // with d_temp_storage = NULL passed to the call.
74 /* d_temp_storage */ NULL,
93 void* d_temp_storage = temp_storage.flat<int8>().data(); local
95 // The second HistogramRange is to actual run with d_temp_storage
98 /* d_temp_storage */ d_temp_storage,

Completed in 74 milliseconds