Searched refs:_extract_tensor_metadata (Results 1 – 11 of 11) sorted by relevance
| /external/pytorch/torch/_inductor/fx_passes/ |
| D | replace_random.py | 7 from torch.fx.passes.shape_prop import _extract_tensor_metadata
67 combined.meta["tensor_meta"] = _extract_tensor_metadata(
|
| D | ddp_fusion.py | 26 from torch.fx.passes.shape_prop import _extract_tensor_metadata, TensorMetadata
68 _extract_tensor_metadata, (node.meta["val"],)
|
| /external/pytorch/torch/fx/passes/ |
| D | shape_prop.py | 32 def _extract_tensor_metadata(result : torch.Tensor, include_contiguity=True) -> TensorMetadata: function
170 return _extract_tensor_metadata(obj)
|
| /external/pytorch/torch/_higher_order_ops/ |
| D | cond.py | 36 from torch.fx.passes.shape_prop import _extract_tensor_metadata
437 true_meta = _extract_tensor_metadata(true_out)
438 false_meta = _extract_tensor_metadata(false_out)
|
| /external/pytorch/torch/distributed/tensor/experimental/ |
| D | _tp_transform.py | 25 from torch.fx.passes.shape_prop import _extract_tensor_metadata
483 local_tensor_meta = _extract_tensor_metadata(node.meta["val"])
|
| /external/pytorch/torch/_export/ |
| D | pass_base.py | 20 from torch.fx.passes.shape_prop import _extract_tensor_metadata, TensorMetadata
148 tensor_meta = _extract_tensor_metadata(x)
|
| /external/pytorch/torch/_dynamo/variables/ |
| D | higher_order_ops.py | 23 from torch.fx.passes.shape_prop import _extract_tensor_metadata
684 meta1 = _extract_tensor_metadata(
687 meta2 = _extract_tensor_metadata(
852 cond_r_meta = _extract_tensor_metadata(
|
| /external/pytorch/torch/_inductor/ |
| D | debug.py | 22 from torch.fx.passes.shape_prop import _extract_tensor_metadata, TensorMetadata
646 return TensorMetadataHolder(_extract_tensor_metadata(x), x.device)
|
| /external/pytorch/test/fx/ |
| D | test_fx_const_fold.py | 8 from torch.fx.passes.shape_prop import _extract_tensor_metadata, ShapeProp
703 self.assertEqual(_extract_tensor_metadata(attr), n.meta["tensor_meta"])
|
| /external/executorch/exir/ |
| D | pass_base.py | 43 from torch.fx.passes.shape_prop import _extract_tensor_metadata, TensorMetadata
280 tensor_meta = _extract_tensor_metadata(x)
|
| /external/pytorch/torch/fx/experimental/ |
| D | proxy_tensor.py | 61 from torch.fx.passes.shape_prop import _extract_tensor_metadata
491 proxy.node.meta["tensor_meta"] = _extract_tensor_metadata(val)
493 proxy.node.meta["tensor_meta"] = _extract_tensor_metadata(val)
|