Searched refs:DoRowReduction (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | softmax_op_gpu.cu.cc | 116 void DoRowReduction(OpKernelContext* context, T* output, InputIter input, in DoRowReduction() function
160 DoRowReduction<T, cub::Max, const T*>( in Compute()
179 DoRowReduction<acc_type, cub::Sum, InputIterType>( in Compute()
|
| D | bias_op_gpu.h | 44 static void DoRowReduction(OpKernelContext* context, T* output,
|
| D | bias_op_gpu.cu.cc | 252 void BiasGradGPU<T>::DoRowReduction(OpKernelContext* context, T* output, in DoRowReduction() function in tensorflow::BiasGradGPU
|
| D | bias_op.cc | 513 BiasGradGPU<T>::DoRowReduction( in ComputeWithReduceSum()
|