Searched refs:sumGrad (Results 1 – 3 of 3) sorted by relevance
| /dports/misc/py-mxnet/incubator-mxnet-1.9.0/src/operator/contrib/ |
| H A D | sync_batch_norm-inl.h | 400 Tensor<xpu, 1> sumGrad = workspace[3]; in Backward() local
402 sumGrad = sumall_except_dim<1>(grad); in Backward()
409 Tensor<cpu, 1, real_t>* grad_cpu_ptr = sharedGrad->Retrieve(sumGrad.shape_, myRank); in Backward()
410 Tensor<cpu, 1, real_t>* prod_cpu_ptr = sharedProd->Retrieve(sumGrad.shape_, myRank); in Backward()
411 mshadow::Copy(*grad_cpu_ptr, sumGrad, s); in Backward()
419 mshadow::Copy(sumGrad, grad_cpu, s); in Backward()
424 gmean = sumGrad * slope; in Backward()
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/src/operator/contrib/ |
| H A D | sync_batch_norm-inl.h | 400 Tensor<xpu, 1> sumGrad = workspace[3]; in Backward() local
402 sumGrad = sumall_except_dim<1>(grad); in Backward()
409 Tensor<cpu, 1, real_t>* grad_cpu_ptr = sharedGrad->Retrieve(sumGrad.shape_, myRank); in Backward()
410 Tensor<cpu, 1, real_t>* prod_cpu_ptr = sharedProd->Retrieve(sumGrad.shape_, myRank); in Backward()
411 mshadow::Copy(*grad_cpu_ptr, sumGrad, s); in Backward()
419 mshadow::Copy(sumGrad, grad_cpu, s); in Backward()
424 gmean = sumGrad * slope; in Backward()
|
| /dports/math/openturns/openturns-1.18/lib/src/Uncertainty/Algorithm/Simulation/ |
| H A D | SimulationSensitivityAnalysis.cxx | 141 Point sumGrad(parameterDimension); in computeEventProbabilitySensitivity() local
156 sumGrad += grad / size; in computeEventProbabilitySensitivity()
159 PointWithDescription sensitivity(sumGrad); in computeEventProbabilitySensitivity()
|