| /external/pytorch/torch/_dynamo/ |
| D | guards.py | 451 source, base_guard_manager, base_example_value, example_value, guard_manager_enum argument
473 example_value=source.index,
482 example_value=example_value,
582 def guard_on_dict_keys_and_ignore_order(self, example_value, guard): argument
592 for key in example_value.keys():
593 value = example_value[key]
596 value_source, example_value
601 example_value=value,
621 example_value=key,
644 example_value, argument
[all …]
|
| D | output_graph.py | 487 example_value = fn(*args)
501 result = VariableBuilder(self.root_tx, source)(example_value)
810 example_value=target,
1221 example_value = node.meta.get("example_value", None)
1222 if isinstance(example_value, torch._subclasses.FakeTensor):
1223 size = example_value.size()
1231 example_value = node.meta.get("example_value", None)
1232 if isinstance(example_value, torch._subclasses.FakeTensor):
1233 size = example_value.size()
|
| D | utils.py | 1180 def set_example_value(node, example_value): argument
1187 node.meta["example_value"] = example_value
1190 shape_env, example_value
|
| D | symbolic_convert.py | 462 example_value=get_fake_value(scalar_to_tensor_proxy.node, self),
|
| /external/pytorch/torch/_C/_dynamo/ |
| D | guards.pyi | 22 example_value,
29 example_value,
36 example_value,
42 example_value,
49 example_value,
56 example_value,
86 example_value,
93 example_value,
|
| /external/pytorch/torch/_dynamo/variables/ |
| D | builder.py | 1237 example_value=value,
1596 example_value=value,
1689 example_value=tensor_value,
1694 example_value = numpy_ndarray_variable.proxy.node.meta["example_value"]
1702 fake_tensor=example_value,
1835 example_value = unspec_var.proxy.node.meta["example_value"]
1900 example_value=wrapped_value,
1911 example_value = unspec_var.proxy.node.meta["example_value"]
1912 assert is_fake(example_value)
1914 fake_tensor_value = example_value
[all …]
|
| D | higher_order_ops.py | 108 example_value=flat_example_value,
182 example_value = (
191 example_value=example_value,
212 example_value = (
219 example_value=example_value,
1030 example_value=out_meta,
1155 example_value = lowered_module.original_module.module()(*real_sub_args)
1160 _assert_tensors_nonaliasing(real_sub_args, example_value)
1173 example_value=example_value,
1234 example_value = pytree.tree_map_only(
[all …]
|
| D | tensor.py | 231 example_value = getattr(fake_val, name)
234 assert isinstance(example_value, torch.Tensor)
237 return wrap_fx_proxy(tx=tx, proxy=proxy, example_value=example_value)
240 elif not callable(example_value):
243 return SourcelessBuilder.create(tx, example_value)
1171 example_value = self.as_proxy().node.meta["example_value"]
1172 example_ndarray = tnp.ndarray(example_value)
1341 example_value: torch.UntypedStorage,
1347 self.example_value = example_value
1359 result = self.example_value.size()
|
| D | torch.py | 520 example_value=None,
1111 example_value = torch.nn.Parameter(
1114 result = VariableBuilder(tx, source)(example_value)
|
| D | user_defined.py | 603 example_value = fn(*args, **kwargs)
609 return VariableBuilder(tx, source).wrap_unspecialized_primitive(example_value)
|
| /external/pytorch/torch/csrc/dynamo/ |
| D | guards.cpp | 882 bool is_immutable_object(py::handle example_value) { in is_immutable_object() argument
883 if (PyTuple_Check(example_value.ptr())) { in is_immutable_object()
885 for (Py_ssize_t i = 0; i < PyTuple_Size(example_value.ptr()); ++i) { in is_immutable_object()
887 py::handle(PyTuple_GetItem(example_value.ptr(), i)))) { in is_immutable_object()
893 return PyLong_Check(example_value.ptr()) || in is_immutable_object()
894 PyFloat_Check(example_value.ptr()) || PyBool_Check(example_value.ptr()) || in is_immutable_object()
895 PyUnicode_Check(example_value.ptr()) || in is_immutable_object()
896 THPVariable_Check(example_value.ptr()); in is_immutable_object()
1502 py::handle example_value,
1527 py::handle example_value, in GuardAccessor() argument
[all …]
|
| /external/pytorch/torch/distributed/pipelining/ |
| D | stage.py | 884 example_value = placeholder.meta["val"]
888 return _RootArgPlaceholder(example_value)
905 example_value.shape,
906 example_value.dtype,
908 buffer = _make_tensor_from_meta(example_value, self.device)
1021 example_value = output.meta["val"]
1033 _make_tensor_from_meta(example_value, self.device),
|
| /external/pytorch/torch/fx/passes/ |
| D | runtime_assert.py | 273 and (example_value := _get_example_value(node)) is not None
287 match_symbol(example_value, lambda: node)
288 if isinstance(t := example_value, torch.Tensor):
|
| /external/pytorch/torch/_inductor/fx_passes/ |
| D | group_batch_fusion.py | 80 example_value = torch.stack(metadata, dim=dim)
82 example_value = torch.unbind(metadata, dim=dim) # type: ignore[assignment]
85 node.meta["example_value"] = example_value
94 example_value = torch.add(input, other)
96 example_value = torch.mul(input, other)
99 pointwise_node.meta["example_value"] = example_value
|
| /external/cronet/stable/third_party/rust/chromium_crates_io/vendor/bytemuck_derive-1.8.0/src/ |
| D | lib.rs | 536 example_value: &str, invalid_value_msg: &str, in find_and_parse_helper_attributes()
|
| /external/cronet/tot/third_party/rust/chromium_crates_io/vendor/bytemuck_derive-1.8.0/src/ |
| D | lib.rs | 536 example_value: &str, invalid_value_msg: &str, in find_and_parse_helper_attributes()
|
| /external/rust/android-crates-io/crates/bytemuck_derive/src/ |
| D | lib.rs | 536 example_value: &str, invalid_value_msg: &str, in find_and_parse_helper_attributes()
|
| /external/pytorch/torch/fx/experimental/ |
| D | symbolic_shapes.py | 584 def compute_unbacked_bindings(shape_env, example_value, old_example_value=None, peek=False): argument
694 symbol_to_path = free_unbacked_symbols_with_path(example_value, ())
697 repr((example_value.stride(), example_value.storage_offset()))
698 if isinstance(example_value, torch.Tensor)
724 new_sym = pytree.key_get(example_value, keypath)
|
| /external/pytorch/test/dynamo/ |
| D | test_guard_manager.py | 425 example_value=None,
|