Searched defs:FunctionalTensorMode (Results 1 – 1 of 1) sorted by relevance
314 class FunctionalTensorMode(TorchDispatchMode): class605 def dispatch_functionalize(func, mode: FunctionalTensorMode = FunctionalTensorMode()):674 self, mode: Optional[FunctionalTensorMode] = None, pre_dispatch: bool = False