Searched defs:cudnn_enabled (Results 1 – 7 of 7) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | Normalization.cpp | 492 bool cudnn_enabled = ctx.userEnabledCuDNN(); in _select_batch_norm_backend() local
542 bool training, double momentum, double eps, bool cudnn_enabled) { in _batch_norm_impl_index()
671 bool training, double momentum, double eps, bool cudnn_enabled) { in batch_norm()
706 bool use_input_stats, double momentum, double eps, bool cudnn_enabled) { in instance_norm()
|
| H A D | Convolution.cpp | 294 bool cudnn_enabled{}; member
1472 bool benchmark, bool deterministic, bool cudnn_enabled, bool allow_tf32) { in _convolution()
1712 bool benchmark, bool deterministic, bool cudnn_enabled) in _convolution()
|
| /aosp_15_r20/external/pytorch/torch/onnx/ |
| H A D | symbolic_opset14.py | 79 cudnn_enabled, argument
|
| H A D | symbolic_opset9.py | 2269 cudnn_enabled, argument
2636 cudnn_enabled, argument
5714 g: jit_utils.GraphContext, input, num_groups, weight, bias, eps, cudnn_enabled argument
|
| /aosp_15_r20/external/pytorch/torch/_inductor/kernel/ |
| H A D | conv.py | 663 cudnn_enabled, argument
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/functorch/ |
| H A D | BatchRulesConvolution.cpp | 130 bool benchmark, bool deterministic, bool cudnn_enabled, bool allow_tf32) { in _convolution_decomp()
|
| /aosp_15_r20/external/pytorch/torch/csrc/autograd/ |
| H A D | FunctionsManual.cpp | 1342 bool cudnn_enabled, in _convolution_jvp()
|