Searched refs:BiasGradGPU (Results 1 – 3 of 3) sorted by relevance
197 void BiasGradGPU<T>::compute(const GPUDevice& d, const T* output_backprop, in compute()252 void BiasGradGPU<T>::DoRowReduction(OpKernelContext* context, T* output, in DoRowReduction()262 void BiasGradGPU<T>::DoColReduction(OpKernelContext* context, T* output, in DoColReduction()273 template struct BiasGradGPU<T>;
494 BiasGradGPU<T>::compute(context->template eigen_device<Device>(), in ComputeWithCustomKernel()513 BiasGradGPU<T>::DoRowReduction( in ComputeWithReduceSum()519 BiasGradGPU<T>::DoColReduction(context, output->flat<T>().data(), in ComputeWithReduceSum()526 BiasGradGPU<T>::DoColReduction( in ComputeWithReduceSum()
39 struct BiasGradGPU { struct