Searched refs:DoColReduction (Results 1 – 3 of 3) sorted by relevance
47 static void DoColReduction(OpKernelContext* context, T* output,
519 BiasGradGPU<T>::DoColReduction(context, output->flat<T>().data(), in ComputeWithReduceSum()526 BiasGradGPU<T>::DoColReduction( in ComputeWithReduceSum()
262 void BiasGradGPU<T>::DoColReduction(OpKernelContext* context, T* output, in DoColReduction() function in tensorflow::BiasGradGPU