Home
last modified time | relevance | path

Searched refs:AccT (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dbias_op_gpu.cu.cc128 typedef typename AccumulatorType<T>::type AccT; in BiasGradNHWC_SharedAtomics() typedef
129 AccT* s_data = reinterpret_cast<AccT*>(s_buf); in BiasGradNHWC_SharedAtomics()
131 s_data[index] = AccT(0); in BiasGradNHWC_SharedAtomics()
138 CudaAtomicAdd(s_data + bias_offset, AccT(ldg(output_backprop + index))); in BiasGradNHWC_SharedAtomics()
153 typedef typename AccumulatorType<T>::type AccT; in BiasGradNCHW_SharedAtomics() typedef
155 __shared__ AccT s_data[kSDataSize]; in BiasGradNCHW_SharedAtomics()
157 s_data[index] = AccT(0); in BiasGradNCHW_SharedAtomics()
166 AccT sum(0); in BiasGradNCHW_SharedAtomics()
173 sum += AccT(val); in BiasGradNCHW_SharedAtomics()
186 AccT data = s_data[thread_index]; in BiasGradNCHW_SharedAtomics()