Searched defs:cudnn_enable (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/pytorch/test/onnx/ |
| H A D | test_onnxscript_no_runtime.py | 72 g, input, normalized_shape, weight, bias, eps, cudnn_enable argument
|
| H A D | test_onnxscript_runtime.py | 111 g, input, normalized_shape, weight, bias, eps, cudnn_enable argument
|