Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/test/
H A Dtest_autocast.py195 class CustomLinear(torch.autograd.Function): class
306 class CustomLinear(torch.autograd.Function): class
/aosp_15_r20/external/pytorch/benchmarks/dynamo/
H A Ddist_util.py40 class CustomLinear(torch.nn.Module): class
/aosp_15_r20/external/pytorch/test/dynamo/
H A Dtest_modules.py2484 class CustomLinear(torch.nn.Module): class