OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
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