Home
last modified time | relevance | path

Searched refs:outgrad (Results 1 – 7 of 7) sorted by relevance

/dports/science/qmcpack/qmcpack-3.11.0/src/QMCWaveFunctions/detail/CUDA_legacy/
H A DAtomicOrbitalCuda.cu1421 outgrad[outIndex][0] = grad[i][0][0]; in evaluateHybridSplineComplexToReal_kernel()
1422 outgrad[outIndex][1] = grad[i][1][0]; in evaluateHybridSplineComplexToReal_kernel()
1423 outgrad[outIndex][2] = grad[i][2][0]; in evaluateHybridSplineComplexToReal_kernel()
1444 outgrad[outIndex][0] = grad[i][0][1]; in evaluateHybridSplineComplexToReal_kernel()
1445 outgrad[outIndex][1] = grad[i][1][1]; in evaluateHybridSplineComplexToReal_kernel()
1654 outgrad[outIndex][0] = grad[i][0][0]; in evaluateHybridPolyComplexToReal_kernel()
1655 outgrad[outIndex][1] = grad[i][1][0]; in evaluateHybridPolyComplexToReal_kernel()
1656 outgrad[outIndex][2] = grad[i][2][0]; in evaluateHybridPolyComplexToReal_kernel()
2988 outgrad[outIndex][0] = grad[i][0][0]; in evaluate3DSplineComplexToReal_kernel()
2989 outgrad[outIndex][1] = grad[i][0][1]; in evaluate3DSplineComplexToReal_kernel()
[all …]
/dports/math/py-autograd/autograd-1.3/autograd/
H A Dcore.py20 outgrad = outgrads.pop(node)
21 ingrads = node.vjp(outgrad[0])
24 return outgrad[0]
/dports/math/py-theano/Theano-1.0.5/theano/tensor/
H A Dsubtensor.py2374 outgrad, = output_gradients
2387 outgrad,
2391 gx = outgrad
2392 gy = advanced_subtensor(outgrad, *idxs)
2410 outgrad, = output_gradients
2423 outgrad,
2427 gx = outgrad
2428 gy = advanced_boolean_subtensor(outgrad, *idxs)
/dports/misc/mxnet/incubator-mxnet-1.9.0/example/extensions/lib_custom_op/
H A Drelu_lib.cu38 __global__ void relu_gpu_backward(float *ingrad, float *outgrad, float *indata, int64_t N) { in relu_gpu_backward() argument
41 ingrad[tid] = indata[tid] > 0 ? 1 * outgrad[tid] : 0; in relu_gpu_backward()
/dports/misc/py-mxnet/incubator-mxnet-1.9.0/example/extensions/lib_custom_op/
H A Drelu_lib.cu38 __global__ void relu_gpu_backward(float *ingrad, float *outgrad, float *indata, int64_t N) { in relu_gpu_backward() argument
41 ingrad[tid] = indata[tid] > 0 ? 1 * outgrad[tid] : 0; in relu_gpu_backward()
/dports/misc/py-mxnet/incubator-mxnet-1.9.0/tests/python/unittest/
H A Dtest_operator.py2886 lambda outgrad, data, outdata, axis, keepdims, keepdim_shape:
2887 outgrad.reshape(keepdim_shape),
2890 lambda outgrad, data, outdata, axis, keepdims, keepdim_shape:
2891 outgrad.reshape(keepdim_shape)/(data.size/outdata.size),
2894 lambda outgrad, data, outdata, axis, keepdims, keepdim_shape:
2898 lambda outgrad, data, outdata, axis, keepdims, keepdim_shape:
2902 lambda outgrad, data, outdata, axis, keepdims, keepdim_shape:
2909 lambda outgrad, data, outdata, axis, keepdims, keepdim_shape:
2910 outgrad.reshape(keepdim_shape) *
2914 lambda outgrad, data, outdata, axis, keepdims, keepdim_shape:
[all …]
/dports/misc/mxnet/incubator-mxnet-1.9.0/tests/python/unittest/
H A Dtest_operator.py2886 lambda outgrad, data, outdata, axis, keepdims, keepdim_shape:
2887 outgrad.reshape(keepdim_shape),
2890 lambda outgrad, data, outdata, axis, keepdims, keepdim_shape:
2891 outgrad.reshape(keepdim_shape)/(data.size/outdata.size),
2894 lambda outgrad, data, outdata, axis, keepdims, keepdim_shape:
2898 lambda outgrad, data, outdata, axis, keepdims, keepdim_shape:
2902 lambda outgrad, data, outdata, axis, keepdims, keepdim_shape:
2909 lambda outgrad, data, outdata, axis, keepdims, keepdim_shape:
2910 outgrad.reshape(keepdim_shape) *
2914 lambda outgrad, data, outdata, axis, keepdims, keepdim_shape:
[all …]