Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/docs/source/notes/
H A Dextending.rst150 class LinearFunction(Function):
196 linear = LinearFunction.apply
200 return LinearFunction.apply(input, weight, bias)
334 class LinearFunction(Function):
455 return LinearFunction.apply(input, self.weight, self.bias)
/aosp_15_r20/external/pytorch/test/dynamo/
H A Dtest_autograd_function.py88 class LinearFunction(torch.autograd.Function): class
121 return LinearFunction.apply(input, weight, bias)