OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
refs:col_scan
(Results
1 - 2
of
2
) sorted by null
/external/tensorflow/tensorflow/core/kernels/
concat_lib_gpu_impl.cu.cc
65
IntType*
col_scan
= GetCudaDeviceArrayOnDevice(&output_scan);
local
80
smem_col_scan[i] =
col_scan
[i];
85
col_scan
= smem_col_scan;
92
cuda_helper::upper_bound<IntType>(
col_scan
, num_inputs, gidx) - 1;
94
IntType curr_offset =
col_scan
[segment];
98
while ((curr_col_offset =
col_scan
[curr_segment + 1]) <= gidx) {
split_lib_gpu.cu.cc
115
IntType*
col_scan
= GetCudaDeviceArrayOnDevice(&output_scan);
local
130
smem_col_scan[i] =
col_scan
[i];
135
col_scan
= smem_col_scan;
142
cuda_helper::upper_bound<IntType>(
col_scan
, num_outputs, gidx) - 1;
144
IntType curr_offset =
col_scan
[segment];
148
while ((curr_col_offset =
col_scan
[curr_segment + 1]) <= gidx) {
Completed in 121 milliseconds