| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cudnn/ |
| H A D | ConvShared.cpp | 112 bool allow_tf32, in setConvolutionParams()
216 bool allow_tf32) { in cudnn_convolution_forward_out()
249 bool allow_tf32) { in cudnn_convolution()
285 bool allow_tf32, in cudnn_convolution_out()
318 bool allow_tf32) { in cudnn_convolution_transpose_backward_input()
376 bool allow_tf32) { in cudnn_convolution_backward_input()
418 bool allow_tf32) { in cudnn_convolution_transpose_forward()
451 bool allow_tf32) { in cudnn_convolution_backward_input()
478 bool allow_tf32) { in cudnn_convolution_transpose()
513 bool allow_tf32) { in cudnn_convolution_backward_weight()
[all …]
|
| H A D | ConvPlaceholders.cpp | 37 bool allow_tf32) { in cudnn_convolution()
50 bool allow_tf32, in cudnn_convolution_out()
65 bool allow_tf32) { in cudnn_convolution_backward_input()
80 bool allow_tf32) { in cudnn_convolution_backward_weight()
95 bool allow_tf32, in cudnn_convolution_backward()
110 bool allow_tf32) { in cudnn_convolution_transpose()
123 bool allow_tf32) { in cudnn_convolution_transpose_backward_input()
138 bool allow_tf32) { in cudnn_convolution_transpose_backward_weight()
154 bool allow_tf32, in cudnn_convolution_transpose_backward()
170 bool allow_tf32) { in raw_cudnn_convolution_forward_out()
[all …]
|
| H A D | Conv_v8.cpp | 142 bool allow_tf32, in filterEngineConfigs()
566 const bool allow_tf32, in get_generator_sources()
707 const bool allow_tf32) { in get_plans_from_find()
756 const bool allow_tf32) { in get_plans_from_find_fused()
812 const bool allow_tf32, in get_configs_from_heuristics()
843 const bool allow_tf32, in get_configs_from_heuristics_fused()
985 const bool allow_tf32) { in run_single_conv()
1083 const bool allow_tf32) { in run_fused_conv()
1182 const bool allow_tf32) { in raw_cudnn_convolution_forward_out()
1224 const bool allow_tf32) { in raw_cudnn_convolution_backward_input_out()
[all …]
|
| H A D | Conv_v7.cpp | 616 bool allow_tf32, in split_batch_dim_to_32bit_out()
707 bool allow_tf32) { in raw_cudnn_convolution_forward_out_32bit()
790 bool allow_tf32) { in raw_cudnn_convolution_forward_out_v7()
822 bool allow_tf32) { in raw_cudnn_convolution_backward_input_out_32bit()
908 bool allow_tf32) { in raw_cudnn_convolution_backward_input_out_v7()
940 bool allow_tf32) { in raw_cudnn_convolution_backward_weight_out_32bit()
1027 bool allow_tf32) { in raw_cudnn_convolution_backward_weight_out_v7()
1117 bool allow_tf32) { in raw_cudnn_convolution_add_relu_out_v7()
1215 bool allow_tf32) { in raw_cudnn_convolution_add_relu_fallback_out()
|
| H A D | ConvShared.h | 32 bool allow_tf32; member
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cudnn/ |
| H A D | BinaryOps.cpp | 39 bool allow_tf32; member
50 bool deterministic, bool allow_tf32) { in setAddParams()
130 bool allow_tf32{false}; in add() local
|
| H A D | Linear.cpp | 48 bool allow_tf32; member
61 bool deterministic, bool allow_tf32) { in setLinearParams()
124 bool allow_tf32{false}; in apply_impl_helper() local
|
| H A D | utils.h | 300 bool deterministic, bool allow_tf32, c10::ScalarType scalar_type) in filterEngineConfigs()
|
| H A D | Conv.cpp | 98 bool allow_tf32{false}; in apply_impl_helper() local
|
| /aosp_15_r20/external/pytorch/torch/backends/cudnn/ |
| H A D | __init__.py | 158 allow_tf32=True, argument
194 allow_tf32 = ContextProp( variable in CudnnModule
|
| /aosp_15_r20/external/pytorch/torch/_inductor/kernel/ |
| H A D | conv.py | 664 allow_tf32, argument
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | Convolution.cpp | 295 bool allow_tf32{}; member
1472 bool benchmark, bool deterministic, bool cudnn_enabled, bool allow_tf32) { in _convolution()
|
| /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/onnx/ |
| H A D | symbolic_opset9.py | 2270 allow_tf32=None, argument
|
| /aosp_15_r20/external/pytorch/torch/csrc/autograd/ |
| H A D | FunctionsManual.cpp | 1343 bool allow_tf32) { in _convolution_jvp()
|
| /aosp_15_r20/external/pytorch/test/onnx/ |
| H A D | test_pytorch_onnx_onnxruntime.py | 13390 def __init__(self, allow_tf32): argument
|