Searched defs:bias_backprop (Results 1 – 2 of 2) sorted by relevance
103 T* bias_backprop, int32 bias_size) { in BiasGradNHWC_Naive()113 T* bias_backprop, int32 bias_size, in BiasGradNCHW_Naive()127 T* bias_backprop, int32 bias_size) { in BiasGradNHWC_SharedAtomics()149 T* bias_backprop, int32 batch, in BiasGradNCHW_SharedAtomics()198 T* bias_backprop, int32 batch, int32 height, in compute()
91 Tensor* bias_backprop = nullptr; in Compute() local