Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/_dynamo/variables/
H A Dtorch_function.py271 self.torch_function_fn = kwargs.pop("torch_function_fn")
275 def from_tensor_var(cls, tx, tensor_var, class_type, torch_function_fn): argument
282 var = cls(torch_function_fn=torch_function_fn, class_type=class_type, **kwargs)
350 self.torch_function_fn,