Home
last modified time | relevance | path

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

/external/pytorch/test/
Dtest_functionalization.py12 dispatch_functionalize,
2150 out_test = dispatch_functionalize(f)(x)
2156 fx_g = make_fx(dispatch_functionalize(f))(x)
2171 out_test = dispatch_functionalize(f)(x)
2185 out_test = dispatch_functionalize(f)(x)
2200 out_test = dispatch_functionalize(f)(x)
2206 fx_g = make_fx(dispatch_functionalize(f))(x)
2238 out_test = dispatch_functionalize(f)(x)
2244 fx_g = make_fx(dispatch_functionalize(f))(x)
2279 out_test = dispatch_functionalize(f)(x)
[all …]
/external/pytorch/torch/_subclasses/
Dfunctional_tensor.py605 def dispatch_functionalize(func, mode: FunctionalTensorMode = FunctionalTensorMode()): function
694 return dispatch_functionalize(inner_f, self.mode)