HomeSort by relevance Sort by last modified time
    Searched full:cfg1d (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/core/util/
cuda_kernel_helper_test.cu.cc 201 CudaLaunchConfig cfg1d; local
205 cfg1d = GetCudaLaunchConfig(bufsize, d); \
206 SetOutbufZero<<<cfg1d.block_count, cfg1d.thread_per_block, 0, d.stream()>>>( \
207 cfg1d, outbuf); \
215 cfg1d = GetCudaLaunchConfig(bufsize, d, SetOutbufZero, 0, 0); \
216 SetOutbufZero<<<cfg1d.block_count, cfg1d.thread_per_block, 0, d.stream()>>>( \
217 cfg1d, outbuf); \
241 CudaLaunchConfig cfg1d; local
    [all...]
  /external/tensorflow/tensorflow/core/kernels/
svd_op_gpu.cu.cc 165 CudaLaunchConfig cfg1D = GetCudaLaunchConfig(batch_size, d);
166 ExtractSignOfVKernel<<<cfg1D.block_count, cfg1D.thread_per_block, 0,
167 d.stream()>>>(cfg1D, outputVT_ptr);

Completed in 350 milliseconds