Home
last modified time | relevance | path

Searched refs:output_lambda (Results 1 – 1 of 1) sorted by relevance

/aosp_15_r20/external/pytorch/test/
H A Dtest_namedtensor.py1234 def test_out_variant(op, output_lambda, device): argument
1236 if output_lambda:
1237 out = output_lambda(t)
1290 test_out_variant(op, testcase.output_lambda, device)