Home
last modified time | relevance | path

Searched defs:check_gradients (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/torch/testing/_internal/optests/
H A Daot_autograd.py39 check_gradients=True, argument
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/
H A Dnn_test.py1128 check_gradients=False): argument
/aosp_15_r20/external/pytorch/test/
H A Dtest_custom_ops.py91 self, device, dynamic, check_gradients argument