Searched refs:BiasGradGPU (Results 1 – 3 of 3) sorted by relevance
212 void BiasGradGPU<T>::compute(const GPUDevice& d, const T* output_backprop, in compute()267 void BiasGradGPU<T>::DoRowReduction(OpKernelContext* context, T* output, in DoRowReduction()277 void BiasGradGPU<T>::DoColReduction(OpKernelContext* context, T* output, in DoColReduction()288 template struct BiasGradGPU<T>;
39 struct BiasGradGPU { struct
462 BiasGradGPU<T>::compute(context->template eigen_device<Device>(), in ComputeWithCustomKernel()481 BiasGradGPU<T>::DoRowReduction( in ComputeWithReduceSum()487 BiasGradGPU<T>::DoColReduction(context, output->flat<T>().data(), in ComputeWithReduceSum()494 BiasGradGPU<T>::DoColReduction( in ComputeWithReduceSum()