Home
last modified time | relevance | path

Searched defs:gI (Results 1 – 10 of 10) sorted by relevance

/external/pytorch/aten/src/ATen/native/cuda/
DActivationGluKernel.cu78 scalar_t* gI, in glu_backward_kernel()
123 auto gI = static_cast<scalar_t*>(iter.data_ptr(0)); in launch_glu_backward_kernel() local
DSoftMax.cu958 …rd(const Tensor &grad_, const Tensor &output_, int64_t dim_, bool half_to_float, const Tensor &gI){ in host_softmax_backward()
/external/pytorch/test/inductor/
Dtest_distributed_patterns.py88 def bw_post_hook(mod, gI, gO): argument
117 def bw_post_hook(mod, gI, gO): argument
Dtest_compiled_autograd.py66 def hook3(gI, gO): argument
/external/clang/test/CodeGen/
Dxcore-stringtype.c48 void gI() {eI();}; in gI() function
/external/pytorch/aten/src/ATen/native/
DSoftMax.cpp268 const Tensor& gI, in host_softmax_backward()
DConvolution.cpp1883 Tensor gI; in _convolution_double_backward() local
/external/pytorch/test/
Dtest_autograd.py1494 def posthook(gO, gI): argument
1498 def prehook(gI): argument
1514 def posthook(gO, gI): argument
1550 def posthook(gI, gO): argument
6456 def jvp(ctx, gI, _1, _2): argument
6542 def jvp(ctx, gI): argument
8598 def jvp(ctx, gI, _): argument
8637 def jvp(ctx, gI, _): argument
8671 def jvp(ctx, gI, _): argument
/external/pytorch/torch/csrc/autograd/
DFunctionsManual.cpp2520 auto gI = (input * input - 2 * input * target + target) / in binary_cross_entropy_double_backward() local
3330 auto gI = ggI * gO * (2 * f_prime_gh + g_prime_fh + h_prime_fg); in gelu_double_backward() local
3337 auto gI = ggI * gO * dgrad_dInput; in gelu_double_backward() local
4569 auto gI = at::_fft_r2c(grad, dim, normalization, /*onesided=*/true); in fft_c2r_backward() local
4705 Tensor gI; in batchnorm_double_backward() local
4844 Tensor gI; in layer_norm_double_backward() local
/external/chromium-trace/catapult/common/py_vulcanize/third_party/rjsmin/bench/
Dapiviewer.js175 …",b=".prototype",L="function",K="Boolean",J="Error",I="constructor",H="warn",G="hasOwnProperty",F=…