| /aosp_15_r20/external/pytorch/c10/core/ |
| H A D | SymBool.h | 16 /*implicit*/ SymBool(bool b) : data_(b){}; in SymBool() function
17 SymBool(SymNode ptr) : data_(false), ptr_(std::move(ptr)) { in SymBool() function
20 SymBool() : data_(false) {} in SymBool() function
|
| H A D | SymBool.cpp | 39 DEFINE_BINARY(sym_and, std::logical_and<>(), sym_and, SymBool) in DEFINE_BINARY() argument
|
| /aosp_15_r20/external/pytorch/torch/onnx/_internal/ |
| H A D | onnxruntime.py | 320 torch.Tensor, torch.SymInt, int, torch.SymFloat, float, torch.SymBool, bool
332 torch.Tensor, torch.SymInt, int, torch.SymFloat, float, torch.SymBool, bool
425 torch.SymBool,
457 torch.Tensor, torch.SymInt, int, torch.SymFloat, float, torch.SymBool, bool
538 torch.Tensor, torch.SymInt, int, torch.SymFloat, float, torch.SymBool, bool
|
| /aosp_15_r20/external/executorch/exir/passes/ |
| H A D | sym_to_tensor_pass.py | 40 symbol: Union[SymInt, SymFloat, SymBool] argument
|
| /aosp_15_r20/external/pytorch/torch/fx/experimental/ |
| H A D | symbolic_shapes.py | 243 def guard_size_oblivious(expr: Union[torch.SymBool, bool]) -> bool:
451 def is_concrete_bool(a: Union[bool, SymBool]) -> bool: argument
780 def statically_known_true(x: Union[bool, SymBool]) -> bool: argument
1389 def is_symbolic(val: Union[int, SymInt, float, SymFloat, bool, SymBool]) -> bool: argument
1450 def cast_symbool_to_symint_guardless(symbool: torch.SymBool) -> torch.SymInt:
|
| H A D | sym_node.py | 1370 def get_constant(x: Union[SymInt, int, SymFloat, float, SymBool, bool]): argument
|
| /aosp_15_r20/external/pytorch/torch/ |
| H A D | types.py | 26 SymBool as SymBool, unknown
|
| H A D | __init__.py | 663 class SymBool: class
1519 cond: _Union[builtins.bool, SymBool], argument
|
| /aosp_15_r20/external/pytorch/torch/onnx/_internal/fx/ |
| H A D | diagnostics.py | 90 def _torch_fx_symbolic_bool(obj: torch.SymBool) -> str:
|
| /aosp_15_r20/external/pytorch/torch/_export/serde/ |
| H A D | schema.py | 81 class SymBool(_Union): class
|
| H A D | serialize.py | 241 def serialize_sym_bool(s: Union[bool, torch.SymBool]) -> SymBool:
1550 def deserialize_sym_bool(self, s: SymBool) -> Union[bool, torch.SymBool]:
|
| /aosp_15_r20/external/executorch/exir/serde/ |
| H A D | schema.py | 89 class SymBool(_Union): class
|
| H A D | export_serialize.py | 249 def serialize_sym_bool(s: Union[bool, torch.SymBool]) -> SymBool:
1532 def deserialize_sym_bool(self, s: SymBool) -> Union[bool, torch.SymBool]:
|
| /aosp_15_r20/external/pytorch/torch/_prims_common/ |
| H A D | __init__.py | 1354 x: Union[NumberType, torch.SymInt, torch.SymFloat, torch.SymBool]
|
| /aosp_15_r20/external/pytorch/torchgen/ |
| H A D | model.py | 1895 SymBool = auto() variable in BaseTy
|