Searched refs:tensor_source (Results 1 – 7 of 7) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/_dynamo/variables/ |
| H A D | builder.py | 2403 tensor_source=outer_context.tensor_source,
2415 tensor_source=source,
2433 tensor_source=source,
2667 tensor_source=source,
2685 tensor_source=source,
|
| /aosp_15_r20/external/pytorch/torch/_dynamo/ |
| H A D | guards.py | 1989 value, is_tensor=True, tensor_source=guard.originating_source
2662 for tensor_source in guard_manager.no_tensor_aliasing_sources: # type: ignore[attr-defined]
2663 global_scope["__compile_source__"] = tensor_source
2664 tensor_id = id(eval(tensor_source, global_scope, scope))
2665 ids_to_source[tensor_id].append(tensor_source)
|
| H A D | utils.py | 2642 source = symbolic_context.tensor_source
|
| /aosp_15_r20/external/pytorch/torch/_functorch/ |
| H A D | aot_autograd.py | 478 source = symbolic_context.tensor_source
|
| /aosp_15_r20/external/pytorch/torch/_subclasses/ |
| H A D | meta_utils.py | 912 tensor_source=source,
|
| H A D | fake_tensor.py | 345 source = symbolic_context.tensor_source
|
| /aosp_15_r20/external/pytorch/torch/fx/experimental/ |
| H A D | symbolic_shapes.py | 1354 tensor_source: Source = None variable in StatefulSymbolicContext
1369 assert self.tensor_source is not None
|