OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
refs:inp_idx
(Results
1 - 2
of
2
) sorted by null
/external/tensorflow/tensorflow/core/kernels/
spacetodepth_op_gpu.cu.cc
39
CUDA_1D_KERNEL_LOOP(
inp_idx
, nthreads) {
40
//
inp_idx
= d + input_depth * (w + input_width * (h + input_height * b))
41
const int d =
inp_idx
% input_depth;
42
const int inp_idx2 =
inp_idx
/ input_depth;
57
*(output_ptr + out_idx) = ldg(input_ptr +
inp_idx
);
depthtospace_op_gpu.cu.cc
56
const int
inp_idx
=
local
58
*(output_ptr + out_idx) = ldg(input_ptr +
inp_idx
);
Completed in 99 milliseconds