Searched refs:dispatch_functionalize (Results 1 – 2 of 2) sorted by relevance
12 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 …]
605 def dispatch_functionalize(func, mode: FunctionalTensorMode = FunctionalTensorMode()): function694 return dispatch_functionalize(inner_f, self.mode)