OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
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