Home
last modified time | relevance | path

Searched refs:create_proxy (Results 1 – 25 of 56) sorted by relevance

123

/external/pytorch/torch/fx/experimental/
Dmeta_tracer.py64 return proxy.tracer.create_proxy('call_function', target, args, kwargs)
76 return self.tracer.create_proxy('call_method', 'size', (self, dim) if dim else (self,), {})
81 return self.tracer.create_proxy('call_method', 'dim', (self,), {})
87 return self.tracer.create_proxy('call_function', builtins.getattr, (self, 'shape'), {})
93 return self.tracer.create_proxy('call_function', builtins.getattr, (self, 'dtype'), {})
121 … self._node = self.tracer.create_proxy('call_function', getattr, (self.root, self.attr), {}).node
125 return self.tracer.create_proxy('call_method', self.attr, (self.root,) + args, kwargs)
144 …def create_proxy(self, kind, target, args, kwargs, name=None, type_expr=None, proxy_factory_fn=Non… member in MetaTracer
145 rv = super().create_proxy(kind, target, args, kwargs, name, type_expr, proxy_factory_fn)
/external/pytorch/torch/distributed/fsdp/
D_trace_utils.py98 orig_create_proxy = tracer.create_proxy
103 tracer.create_proxy = functools.partial( # type: ignore[method-assign]
113 tracer.create_proxy = orig_create_proxy # type: ignore[method-assign]
161 create_proxy: Callable,
207 proxy = create_proxy(
/external/pytorch/torch/fx/
Dproxy.py193 def create_proxy(self, kind: str, target: Target, args: Tuple[Any, ...], kwargs: Dict[str, Any], member in TracerBase
439 return self.tracer.create_proxy('call_method', '__call__', (self,) + args, kwargs)
459 return self.tracer.create_proxy('call_function', operator.abs, (self,), {})
484 self.tracer.create_proxy('call_function', assert_fn, (self,), {})
518 return tracer.create_proxy('call_method', orig_method.name, args, kwargs)
520 return tracer.create_proxy('call_method', orig_method.__name__, args, kwargs)
525 return tracer.create_proxy('call_function', orig_method, args, kwargs,
543 … self._node = self.tracer.create_proxy('call_function', getattr, (self.root, self.attr), {}).node
547 return self.tracer.create_proxy('call_method', self.attr, (self.root,) + args, kwargs)
592 return tracer.create_proxy('call_function', target, args, kwargs)
[all …]
D_symbolic_trace.py120 return tracer.create_proxy("call_function", cls, args, kwargs)
521 ret_val = self.create_proxy("call_module", module_qualified_name, args, kwargs)
559 in inspect.signature(self.create_proxy).parameters
568 … val_proxy = self.create_proxy("get_attr", n, (), {}, **kwargs) # type: ignore[arg-type]
629 out = self.create_proxy("placeholder", f"input_{name_idx}", (), {})
860 out = self.create_proxy(
890 self.create_proxy("call_function", _assert_is_none, args, {})
906 return self.create_proxy(
961 return_proxy = proxy.tracer.create_proxy(
984 return proxy.tracer.create_proxy("call_method", name, args, kwargs)
Dinterpreter.py487 return self.tracer.create_proxy("get_attr", target, args, kwargs)
499 return self.tracer.create_proxy('call_function', target, args, kwargs)
/external/skia/tests/
DTextureOpTest.cpp69 static sk_sp<GrSurfaceProxy> create_proxy(GrRecordingContext* rContext) { in create_proxy() function
123 GrSurfaceProxyView proxyViewA(create_proxy(dContext), in DEF_GANESH_TEST_FOR_RENDERING_CONTEXTS()
126 GrSurfaceProxyView proxyViewB(create_proxy(dContext), in DEF_GANESH_TEST_FOR_RENDERING_CONTEXTS()
129 GrSurfaceProxyView proxyViewC(create_proxy(dContext), in DEF_GANESH_TEST_FOR_RENDERING_CONTEXTS()
DBulkRectTest.cpp68 static sk_sp<GrSurfaceProxy> create_proxy(GrRecordingContext* rContext) { in create_proxy() function
167 sk_sp<GrSurfaceProxy> proxyA = create_proxy(dContext); in textureop_creation_test()
168 sk_sp<GrSurfaceProxy> proxyB = create_proxy(dContext); in textureop_creation_test()
186 sk_sp<GrSurfaceProxy> proxyA = create_proxy(dContext); in textureop_creation_test()
/external/pytorch/torch/_dynamo/variables/
Dtensor.py537 tx.output.create_proxy(
656 tx.output.create_proxy(
711 proxy = tx.output.create_proxy("call_method", "cpu", (t.as_proxy(),), {})
714 proxy = tx.output.create_proxy(
833 tx.output.create_proxy(
921 proxy=tx.output.create_proxy(
947 proxy=tx.output.create_proxy(
1011 tx.output.create_proxy(
1136 tx.output.create_proxy(
1177 tx.output.create_proxy(
[all …]
Dctx_manager.py830 tx.output.create_proxy(
855 tx.output.create_proxy(
864 tx.output.create_proxy(
874 tx.output.create_proxy(
1012 tx.output.create_proxy(
1020 proxy=tx.output.create_proxy(
1028 proxy=tx.output.create_proxy(
1072 tx.output.create_proxy(
1080 proxy=tx.output.create_proxy(
Dhigher_order_ops.py102 proxy=tx.output.create_proxy(
495 node = tx.output.create_proxy(
772 proxy=tx.output.create_proxy(
1027 proxy=tx.output.create_proxy(
1167 proxy=tx.output.create_proxy(
1286 proxy=tx.output.create_proxy(
1404 proxy=tx.output.create_proxy(
1433 proxy=tx.output.create_proxy(
1559 proxy=tx.output.create_proxy(
1832 proxy=tx.output.create_proxy(
Dbuiltin.py345 tx.output.create_proxy(
898 proxy = tx.output.create_proxy(
906 proxy = tx.output.create_proxy(
1010 proxy=tx.output.create_proxy(
1099 proxy=tx.output.create_proxy(
1163 proxy = tx.output.create_proxy(
1224 tx.output.create_proxy(
1807 tx.output.create_proxy(
1825 tx.output.create_proxy(
2012 proxy = tx.output.create_proxy(
[all …]
Dtorch.py261 tx.output.create_proxy(
515 proxy=tx.output.create_proxy(
615 proxy=tx.output.create_proxy(
693 proxy=tx.output.create_proxy(
899 proxy=tx.output.create_proxy(
1064 tx.output.create_proxy(
Ddistributed.py351 proxy = tx.output.create_proxy(
393 tx.output.create_proxy(
/external/pytorch/torch/_inductor/
Dloop_body.py419 return tracer.create_proxy(
492 return tracer.create_proxy(
506 result = tracer.create_proxy(
536 tracer.create_proxy(
546 tracer.create_proxy("output", "output", (result,), {})
550 proxy_ops = tracer.create_proxy("placeholder", "ops", (), {})
Dsubgraph_lowering.py86 parent = tracer.create_proxy("placeholder", "ops", (), {})
91 self.tracer.create_proxy("placeholder", f"input{i}", (), {})
/external/pytorch/test/distributed/fsdp/
Dtest_fsdp_fx.py49 orig_create_proxy = tracer.create_proxy
56 self.assertEqual(orig_create_proxy, tracer.create_proxy)
/external/pytorch/torch/_dynamo/
Dcompiled_autograd.py98 self.fx_tracer.create_proxy("placeholder", name, (), {})
159 proxies = self.fx_tracer.create_proxy(
186 return self.fx_tracer.create_proxy(
296 self.fx_tracer.create_proxy(
D_trace_wrapped_higher_order_op.py89 out_proxy = mode.tracer.create_proxy(
/external/pytorch/torch/_higher_order_ops/
Drun_const_graph.py30 proxy = mode.tracer.create_proxy("call_function", run_const_graph, p_args, {})
Deffects.py182 out_proxy = mode.tracer.create_proxy(
250 token_proxy = proxy_tensor_mode.tracer.create_proxy(
Dstrict_mode.py72 out_proxy = mode.tracer.create_proxy(
Dtorchbind.py83 out_proxy = mode.tracer.create_proxy(
/external/pytorch/test/expect/
DTestFXAPIBackwardCompatibility.test_class_member_back_compat-fx_backcompat_class_members.expect18 torch.fx.proxy.TracerBase ['check_mutable_operations', 'create_arg', 'create_node', 'create_proxy',…
/external/executorch/exir/
Dtracer.py427 proxy = self.create_proxy("get_attr", n, (), {})
509 placeholder = self.create_proxy("placeholder", f"ph_{i}", (), {})
545 self.create_proxy(
/external/pytorch/torch/_prims/
Drng_prims.py210 out_proxy = mode.tracer.create_proxy(
274 out_proxy = mode.tracer.create_proxy(

123