| /external/pytorch/test/inductor/ |
| D | test_distributed_patterns.py | 60 def bw_pre_hook(mod, gO): argument
88 def bw_post_hook(mod, gI, gO): argument
111 def bw_pre_hook(mod, gO): argument
117 def bw_post_hook(mod, gI, gO): argument
|
| D | test_compiled_autograd.py | 66 def hook3(gI, gO): argument
648 def backward(ctx, gO): argument
698 def backward(ctx, gO): argument
928 def backward(ctx, gO): argument
973 def backward(ctx, gO): argument
995 def backward(ctx, gO): argument
1017 def backward(ctx, gO): argument
1148 def backward(ctx, gO): argument
1172 def backward(ctx, gO): argument
1183 def backward(ctx, gO): argument
[all …]
|
| /external/pytorch/aten/src/ATen/native/cuda/ |
| D | ActivationGluKernel.cu | 80 const scalar_t* gO, in glu_backward_kernel()
125 auto gO = static_cast<const scalar_t*>(iter.data_ptr(2)); in launch_glu_backward_kernel() local
|
| /external/pytorch/torch/autograd/ |
| D | __init__.py | 481 def vjp(gO): argument
|
| /external/pytorch/test/ |
| D | test_autograd.py | 614 def backward(ctx, gO): argument
693 def backward(ctx, gO): argument
712 def backward(ctx, gO): argument
737 def backward(ctx, gO, _): argument
758 def backward(ctx, gO): argument
769 def backward(ctx, gO): argument
1494 def posthook(gO, gI): argument
1514 def posthook(gO, gI): argument
1550 def posthook(gI, gO): argument
1784 def backward(ctx, gO): argument
[all …]
|
| /external/pytorch/aten/src/ATen/native/transformers/cuda/flash_attn/ |
| D | flash_fwd_kernel.h | 78 Tensor gO = local_tile(mO(_, bidh, _), Shape<Int<kBlockM>, Int<kHeadDim>>{}, in compute_attn_1rowblock() local
430 Tensor gO = local_tile(mO(_, bidh, _), Shape<Int<kBlockM>, Int<kHeadDim>>{}, in compute_attn_1rowblock() local
1242 Tensor gO = make_tensor(make_gmem_ptr(o_ptr + col), in combine_attn_seqk_parallel() local
|
| D | flash_bwd_preprocess_kernel.h | 87 Tensor gO = make_tensor(make_gmem_ptr(reinterpret_cast<Element *>(params.o_ptr) + row_offset_o), in compute_dot_do_o() local
|
| D | flash_bwd_kernel.h | 140 Tensor gO = make_tensor(make_gmem_ptr(reinterpret_cast<Element *>(params.o_ptr) + row_offset_o), in compute_dq_dk_dv_1colblock() local
|
| /external/pytorch/test/profiler/ |
| D | test_profiler.py | 138 def backward(ctx, gO): argument
208 def backward(ctx, gO): argument
|
| /external/pytorch/torch/csrc/autograd/ |
| D | FunctionsManual.cpp | 2222 auto& gO = grad_output; in glu_double_backward() local
3302 const Tensor& gO, in gelu_double_backward()
4656 const Tensor& gO, in batchnorm_double_backward()
4739 const Tensor& gamma) -> Tensor { in batchnorm_double_backward()
4811 auto gO = gO_t.reshape({M, N}); in layer_norm_double_backward() local
|
| /external/pytorch/aten/src/ATen/native/ |
| D | Convolution.cpp | 1751 auto gO = gO_r; in _convolution_double_backward() local
|
| /external/pytorch/test/dynamo/ |
| D | test_autograd_function.py | 756 def backward(ctx, gO): argument
|
| /external/chromium-trace/catapult/common/py_vulcanize/third_party/rjsmin/bench/ |
| D | apiviewer.js | 175 …",b=".prototype",L="function",K="Boolean",J="Error",I="constructor",H="warn",G="hasOwnProperty",F=…
|