Home
last modified time | relevance | path

Searched defs:total_device_threads (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dfused_batch_norm_op.cu.cc141 const int32 total_device_threads = gridDim.x * blockDim.x; in run() local
209 const int32 total_device_threads = gridDim.x * blockDim.x; in run() local
Drelu_op_gpu.cu.cc50 const int32 total_device_threads = gridDim.x * blockDim.x; in ReluGradHalfKernel() local