| /aosp_15_r20/external/pytorch/test/inductor/ |
| H A 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 …]
|
| H A 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
|
| /aosp_15_r20/external/pytorch/test/ |
| H A 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 …]
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A 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
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/transformers/cuda/flash_attn/ |
| H A 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
|
| H A 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
|
| H A 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
|
| /aosp_15_r20/external/pytorch/torch/autograd/ |
| H A D | __init__.py | 481 def vjp(gO): argument
|
| /aosp_15_r20/external/pytorch/test/profiler/ |
| H A D | test_profiler.py | 138 def backward(ctx, gO): argument
208 def backward(ctx, gO): argument
|
| /aosp_15_r20/external/pytorch/torch/csrc/autograd/ |
| H A 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
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | Convolution.cpp | 1751 auto gO = gO_r; in _convolution_double_backward() local
|
| /aosp_15_r20/external/pytorch/test/dynamo/ |
| H A D | test_autograd_function.py | 756 def backward(ctx, gO): argument
|
| /aosp_15_r20/external/chromium-trace/catapult/common/py_vulcanize/third_party/rjsmin/bench/ |
| H A D | apiviewer.js | 175 …",b=".prototype",L="function",K="Boolean",J="Error",I="constructor",H="warn",G="hasOwnProperty",F=…
|
| /aosp_15_r20/prebuilts/cmdline-tools/tools/lib/ |
| HD | r8.jar | resources/new_api_database.ser
LICENSE
META-INF/MANIFEST.MF
META-INF/services/ ... |
| /aosp_15_r20/prebuilts/sdk/tools/windows/lib/ |
| HD | d8.jar | META-INF/
META-INF/MANIFEST.MF
META-INF/services/com. ... |
| /aosp_15_r20/prebuilts/sdk/tools/darwin/lib/ |
| HD | d8.jar | META-INF/
META-INF/MANIFEST.MF
META-INF/services/com. ... |
| /aosp_15_r20/prebuilts/sdk/tools/linux/lib/ |
| HD | d8.jar | META-INF/
META-INF/MANIFEST.MF
META-INF/services/com. ... |
| /aosp_15_r20/out/host/common/obj/JAVA_LIBRARIES/d8_intermediates/ |
| D | classes.jar | META-INF/
META-INF/MANIFEST.MF
META-INF/services/com. ... |
| D | javalib.jar | META-INF/
META-INF/MANIFEST.MF
META-INF/services/com. ... |
| /aosp_15_r20/prebuilts/r8/ |
| HD | r8.jar | resources/new_api_database.ser
LICENSE
META-INF/services/com.android.tools.r8 ... |
| /aosp_15_r20/out/soong/.intermediates/prebuilts/r8/d8/linux_glibc_common/combined/ |
| D | d8.jar | META-INF/
META-INF/MANIFEST.MF
META-INF/services/com. ... |
| /aosp_15_r20/out/soong/.intermediates/prebuilts/r8/r8lib-prebuilt/linux_glibc_common/local-combined/ |
| D | r8lib-prebuilt.jar | META-INF/services/com.android.tools.r8.internal.IV
LICENSE
com ... |
| /aosp_15_r20/out/host/linux-x86/framework/ |
| D | d8.jar | META-INF/
META-INF/MANIFEST.MF
META-INF/services/com. ... |
| /aosp_15_r20/prebuilts/bundletool/ |
| HD | bundletool-all-20230807.jar | META-INF/
META-INF/MANIFEST.MF
com/
com/android/
com/ ... |
| /aosp_15_r20/prebuilts/bazel/common/android_tools/android_tools/ |
| D | all_android_tools_deploy.jar | META-INF/
META-INF/MANIFEST.MF
build-data.properties
com/
com ... |