Searched refs:bias_index (Results 1 – 2 of 2) sorted by relevance
163 int32 bias_index = blockIdx.x % bias_size; in BiasGradNCHW_SharedAtomics() local172 (batch * bias_size + bias_index) * image_size + image_offset); in BiasGradNCHW_SharedAtomics()191 CudaAtomicAdd(bias_backprop + bias_index, T(data)); in BiasGradNCHW_SharedAtomics()
292 virtual inline void GetBiasSizeInMklOrder(size_t bias_index, in GetBiasSizeInMklOrder() argument294 const Tensor& bias = MklGetInput(context_, bias_index); in GetBiasSizeInMklOrder()