Searched refs:BiasNHWCKernel (Results 1 – 1 of 1) sorted by relevance
56 __global__ void BiasNHWCKernel(int32 nthreads, const T* input, const T* bias, in BiasNHWCKernel() function88 TF_CHECK_OK(CudaLaunchKernel(BiasNHWCKernel<T>, config.block_count, in compute()