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

  /external/tensorflow/tensorflow/core/kernels/
tile_functor_gpu.cu.cc 52 const int64 out_nelem = out->NumElements(); local
53 CHECK_LT(out_nelem, kint32max) << "Tensor too large to transpose on GPU";
74 CudaLaunchConfig cfg = GetCudaLaunchConfig(out_nelem, d);

Completed in 63 milliseconds