Searched defs:bias_backprop (Results 1 – 1 of 1) sorted by relevance
108 T* __restrict__ bias_backprop, in BiasGradNHWC_Naive()120 T* __restrict__ bias_backprop, in BiasGradNCHW_Naive()132 T* __restrict__ bias_backprop, int32 bias_size) { in BiasGradNHWC_SharedAtomics()155 const T* __restrict__ output_backprop, T* __restrict__ bias_backprop, in BiasGradNCHW_SharedAtomics()216 T* bias_backprop, int32 batch, int32 height, in compute()