Home
last modified time | relevance | path

Searched refs:FakeTensor (Results 1 – 25 of 99) sorted by relevance

1234

/external/pytorch/torch/_subclasses/
Dfake_tensor.py173 if isinstance(x, FakeTensor):
193 if isinstance(t, FakeTensor):
279 def add_constant_storage_mapping(self, fake_tensor: FakeTensor) -> None: argument
283 assert isinstance(fake_tensor, FakeTensor) and fake_tensor.constant is not None
294 assert not isinstance(tensor, FakeTensor)
308 def _get_memo(self, t: Tensor) -> Optional[FakeTensor]:
314 def set_tensor_memo(self, t: Tensor, v: FakeTensor) -> None: argument
334 ) -> FakeTensor:
357 def mk_fake_tensor(make_meta_t: Callable[[], object]) -> FakeTensor:
366 return FakeTensor(
[all …]
Dfake_impls.py24 FakeTensor,
177 return FakeTensor(fake_mode, r, out_device)
229 if isinstance(x, FakeTensor):
509 return FakeTensor(fake_mode, out, out_device)
549 device, _ = FakeTensor._find_common_device(func, [tl[i] for tl in tensor_lists])
706 return FakeTensor(fake_mode, t, device)
907 return FakeTensor(
920 return FakeTensor(
941 return FakeTensor(fake_mode, out, x.device)
D__init__.py4 FakeTensor,
/external/executorch/backends/cadence/aot/
Dgraph_builder.py10 from torch._subclasses import FakeTensor, FakeTensorMode
51 self, target: str, fake_tensor: Union[FakeTensor, torch.Tensor] argument
53 if not isinstance(fake_tensor, FakeTensor):
82 placeholders: Sequence[Union[torch.Tensor, FakeTensor]], argument
103 (torch.Tensor, FakeTensor), lambda x: op_to_placeholder_dict[x], (args, kwargs)
/external/pytorch/torch/fx/passes/
Dreinplace.py5 from torch._subclasses.fake_tensor import FakeTensorMode, FakeTensor
103 assert isinstance(node.meta['fake_result'], FakeTensor)
104 assert isinstance(node.meta['view_of'].meta['fake_result'], FakeTensor)
180 if isinstance(x, FakeTensor):
221 assert isinstance(base.meta['fake_result'], FakeTensor)
223 assert isinstance(mutated_view.meta['fake_result'], FakeTensor)
484 if isinstance(x, FakeTensor):
634 ) for x in old_flattened_res if isinstance(x, FakeTensor)}
638 ) for x in node_flattened_res if isinstance(x, FakeTensor)}
655 ) for x in new_flattened_res if isinstance(x, FakeTensor)}
Dfake_tensor_prop.py8 from torch._subclasses.fake_tensor import FakeTensorMode, FakeTensor
42 if isinstance(obj, FakeTensor):
/external/executorch/backends/vulkan/
Dutils.py21 from torch._subclasses.fake_tensor import FakeTensor
74 if isinstance(node.meta["val"], FakeTensor):
78 return all(isinstance(x, FakeTensor) for x in node.meta["val"])
125 if isinstance(node.meta["val"], FakeTensor):
183 if isinstance(node.meta["val"], FakeTensor):
/external/executorch/backends/arm/_passes/
Darm_pass_utils.py27 from torch._subclasses.fake_tensor import FakeTensor
120 def get_first_fake_tensor(node: torch.fx.Node) -> FakeTensor:
133 fake_tensor, FakeTensor
/external/executorch/backends/vulkan/_passes/
Dtag_memory_meta_pass.py26 from torch._subclasses.fake_tensor import FakeTensor
142 arg.meta["val"], FakeTensor
182 arg.meta["val"], FakeTensor
205 if not isinstance(node.meta["val"], FakeTensor):
Dremove_local_scalar_dense_ops.py13 from torch._subclasses.fake_tensor import FakeTensor
46 if not isinstance(tensor_val, FakeTensor):
/external/pytorch/torch/_export/
Dpass_base.py14 from torch._subclasses import FakeTensor, UnsupportedFakeTensorException
80 elif isinstance(a, FakeTensor):
100 … ) -> Union[FakeTensor, torch.SymInt, torch.SymFloat, torch.SymBool, int, float, bool, str, None]:
101 if isinstance(x, FakeTensor):
136 if not isinstance(x, FakeTensor) and isinstance(x, torch.Tensor):
286 def extract_input(node: torch.fx.Node) -> Optional[FakeTensor]:
294 return FakeTensor(
423 if isinstance(i, FakeTensor):
/external/executorch/exir/passes/
Dconst_prop_pass.py13 from torch._subclasses.fake_tensor import FakeTensor
35 if isinstance(arg, FakeTensor):
D__init__.py59 from torch._subclasses import FakeTensor
263 Optional[FakeTensor], List[Optional[FakeTensor]], Tuple[Optional[FakeTensor]] argument
283 elif isinstance(val, FakeTensor):
419 assert isinstance(val, FakeTensor)
/external/executorch/devtools/debug_format/
Det_schema.py32 from torch._subclasses import FakeTensor
397 if isinstance(val, (FakeTensor, torch.Tensor)):
401 isinstance(tensor, (FakeTensor, torch.Tensor)) for tensor in val
411 if isinstance(val, (FakeTensor, torch.Tensor)):
415 isinstance(tensor, (FakeTensor, torch.Tensor)) for tensor in val
/external/executorch/backends/qualcomm/_passes/
Di64_to_i32.py10 from torch._subclasses.fake_tensor import FakeTensor
41 return isinstance(node_val, FakeTensor) and node_val.dtype == dtype
/external/executorch/exir/dialects/backend/test/
Dtest_backend_ops.py16 from torch._subclasses.fake_tensor import FakeTensor, FakeTensorMode
59 out = FakeTensor.from_tensor(x, mode)
/external/pytorch/test/
Dtest_fake_tensor.py29 FakeTensor,
84 self.assertTrue(isinstance(t, FakeTensor))
106 self.assertTrue(isinstance(z, FakeTensor))
145 self.assertIsInstance(param, FakeTensor)
171 self.assertTrue(isinstance(out, FakeTensor))
188 self.assertTrue(isinstance(out, FakeTensor))
215 self.assertTrue(isinstance(out, FakeTensor))
269 self.assertIsInstance(fake_tensor, FakeTensor)
275 self.assertTrue(isinstance(x, FakeTensor))
283 self.assertTrue(isinstance(out, FakeTensor))
[all …]
Dtest_content_store.py8 from torch._subclasses.fake_tensor import FakeTensor, FakeTensorMode
115 self.assertIsInstance(x4, FakeTensor)
/external/pytorch/torch/_functorch/_aot_autograd/
Djit_compile_runtime_wrappers.py24 from torch._subclasses import FakeTensor
256 fw_ins: List[Optional[FakeTensor]], argument
257 user_fw_outs: List[Optional[FakeTensor]], argument
258 bw_outs: List[Optional[FakeTensor]], argument
259 saved_tensors: List[FakeTensor], argument
268 if isinstance(t, FakeTensor):
/external/pytorch/docs/source/
Dexport.ir_spec.rst237 of type `<symint>`, `<FakeTensor>`, a
238 ``List[Union[FakeTensor, SymInt]]``, or ``None``.
355 FakeTensor section in References
358 A FakeTensor is an object that contains the metadata of a tensor. It can be
363 class FakeTensor:
369 The size field of FakeTensor is a list of integers or SymInts. If SymInts are
374 are no strides in FakeTensor.
379 FakeTensor describing that tensor.
389 - ``aten::add`` returns a Tensor; so its spec will be a FakeTensor with dtype
394 will also be a 2-tuple of FakeTensor objects, the first TensorMeta describes
[all …]
/external/executorch/exir/
Dpass_base.py37 from torch._subclasses.fake_tensor import FakeTensor
189 elif isinstance(a, FakeTensor):
212 FakeTensor,
222 if isinstance(x, FakeTensor):
268 if not isinstance(x, FakeTensor) and isinstance(x, torch.Tensor):
423 def extract_input(node: torch.fx.Node) -> Optional[FakeTensor]:
431 return FakeTensor(
560 if isinstance(i, FakeTensor):
/external/executorch/backends/transforms/
Dconvert_dtype_pass.py15 from torch._subclasses.fake_tensor import FakeTensor
38 return isinstance(node_val, FakeTensor) and node_val.dtype == dtype
/external/executorch/backends/arm/quantizer/
Darm_quantizer_utils.py19 from torch._subclasses import FakeTensor
129 if "val" not in node.meta or not isinstance(node.meta["val"], FakeTensor):
/external/executorch/exir/verification/
Dverifier.py27 from torch._subclasses import FakeTensor
141 def _get_inputs(graph_module: GraphModule) -> List[Optional[FakeTensor]]:
142 def extract_input(node: torch.fx.Node) -> Optional[FakeTensor]:
/external/pytorch/torch/
D_tensor_str.py454 from torch._subclasses.fake_tensor import FakeTensor
456 is_meta = self.is_meta or isinstance(self, FakeTensor)
493 from torch._subclasses.fake_tensor import FakeTensor
496 is_meta = self.is_meta or isinstance(self, FakeTensor)
592 from torch._subclasses.fake_tensor import FakeTensor
594 if self.is_meta or isinstance(self, FakeTensor):

1234