Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Drelu_op_gpu.cu.cc178 constexpr int32 kThreadInBlock = 512; in operator ()() local
228 constexpr int32 kThreadInBlock = 512; in operator ()() local
H A Dfused_batch_norm_op.cu.cc333 constexpr int32 kThreadInBlock = 1024; in operator ()() local
335 constexpr int32 kThreadInBlock = 512; in operator ()() local