Searched refs:orig_method (Results 1 – 5 of 5) sorted by relevance
| /external/pytorch/torch/fx/ |
| D | proxy.py | 499 def __torch_function__(cls, orig_method, types, args=None, kwargs=None): argument
516 if isinstance(orig_method, torch._C.ScriptMethod):
517 args = (orig_method.owner,) + args
518 return tracer.create_proxy('call_method', orig_method.name, args, kwargs)
519 if torch.overrides.is_tensor_method_or_property(orig_method):
520 return tracer.create_proxy('call_method', orig_method.__name__, args, kwargs)
522 if isinstance(orig_method, torch._ops.HigherOrderOperator):
525 return tracer.create_proxy('call_function', orig_method, args, kwargs,
526 name=tracer.graph._target_to_str(orig_method.__name__))
|
| D | node.py | 69 def _find_module_of_method(orig_method: Callable[..., Any]) -> str:
70 name = orig_method.__name__
71 module = orig_method.__module__
75 if getattr(guess, name, None) is orig_method:
|
| /external/pytorch/torch/distributed/_composable/fsdp/ |
| D | fully_shard.py | 426 orig_method = getattr(module, method_name)
428 @functools.wraps(orig_method)
432 out = orig_method(*args, **kwargs)
|
| /external/cronet/tot/base/allocator/partition_allocator/src/partition_alloc/shim/ |
| D | allocator_interception_apple.mm | 529 Method orig_method =
532 reinterpret_cast<allocWithZone_t>(method_getImplementation(orig_method));
535 method_setImplementation(orig_method,
|
| /external/cronet/stable/base/allocator/partition_allocator/src/partition_alloc/shim/ |
| D | allocator_interception_apple.mm | 529 Method orig_method =
532 reinterpret_cast<allocWithZone_t>(method_getImplementation(orig_method));
535 method_setImplementation(orig_method,
|