HomeSort by relevance Sort by last modified time
    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