Searched refs:torch_function_fn (Results 1 – 1 of 1) sorted by relevance
271 self.torch_function_fn = kwargs.pop("torch_function_fn")275 def from_tensor_var(cls, tx, tensor_var, class_type, torch_function_fn): argument282 var = cls(torch_function_fn=torch_function_fn, class_type=class_type, **kwargs)350 self.torch_function_fn,