Home
last modified time | relevance | path

Searched refs:version_counter (Results 1 – 23 of 23) sorted by relevance

/external/pytorch/aten/src/ATen/
DOpaqueTensorImpl.h74 const c10::VariableVersion& version_counter, in shallow_copy_and_detach()
85 /*version_counter=*/version_counter, in shallow_copy_and_detach()
98 c10::VariableVersion&& version_counter, in shallow_copy_and_detach()
109 /*version_counter=*/std::move(version_counter), in shallow_copy_and_detach()
128 /*version_counter=*/version_counter(), in shallow_copy_from()
152 const c10::VariableVersion& version_counter, in copy_tensor_metadata()
157 version_counter, in copy_tensor_metadata()
167 c10::VariableVersion&& version_counter, in copy_tensor_metadata()
172 std::move(version_counter), in copy_tensor_metadata()
DSparseCsrTensorImpl.h113 VariableVersion&& version_counter, in shallow_copy_and_detach_core()
133 /*version_counter=*/version_counter, in shallow_copy_and_detach_core()
139 r->set_version_counter(std::forward<VariableVersion>(version_counter)); in shallow_copy_and_detach_core()
151 const c10::VariableVersion& version_counter, in shallow_copy_and_detach()
154 version_counter, allow_tensor_metadata_change); in shallow_copy_and_detach()
164 c10::VariableVersion&& version_counter, in shallow_copy_and_detach()
167 std::move(version_counter), allow_tensor_metadata_change); in shallow_copy_and_detach()
191 c10::VariableVersion version_counter, in copy_tensor_metadata()
196 std::move(version_counter), in copy_tensor_metadata()
DSparseTensorImpl.h312 VariableVersion&& version_counter, in shallow_copy_and_detach_core()
331 /*version_counter=*/version_counter, in shallow_copy_and_detach_core()
337 r->set_version_counter(std::forward<VariableVersion>(version_counter)); in shallow_copy_and_detach_core()
349 const c10::VariableVersion& version_counter, in shallow_copy_and_detach()
352 version_counter, allow_tensor_metadata_change); in shallow_copy_and_detach()
362 c10::VariableVersion&& version_counter, in shallow_copy_and_detach()
365 std::move(version_counter), allow_tensor_metadata_change); in shallow_copy_and_detach()
380 /*version_counter=*/version_counter(), in shallow_copy_from()
402 c10::VariableVersion version_counter, in copy_tensor_metadata()
407 std::move(version_counter), in copy_tensor_metadata()
DNestedTensorImpl.cpp299 VariableVersion&& version_counter, in shallow_copy_and_detach_core() argument
305 r->set_version_counter(std::forward<VariableVersion>(version_counter)); in shallow_copy_and_detach_core()
323 /*version_counter=*/std::forward<VariableVersion>(version_counter), in shallow_copy_and_detach_core()
329 const c10::VariableVersion& version_counter, in shallow_copy_and_detach() argument
332 version_counter, allow_tensor_metadata_change); in shallow_copy_and_detach()
336 c10::VariableVersion&& version_counter, in shallow_copy_and_detach() argument
339 std::move(version_counter), allow_tensor_metadata_change); in shallow_copy_and_detach()
DFunctionalTensorWrapper.cpp48 version_counter_ = value_.unsafeGetTensorImpl()->version_counter(); in set_constructor_metadata()
408 const c10::VariableVersion& version_counter, in copy_tensor_metadata() argument
413 version_counter, in copy_tensor_metadata()
431 const c10::VariableVersion& version_counter, in copy_tensor_metadata_and_refresh() argument
433 copy_tensor_metadata(src_impl, dest_impl, version_counter, allow_tensor_metadata_change); in copy_tensor_metadata_and_refresh()
440 VariableVersion&& version_counter, in shallow_copy_and_detach_core() argument
446 r->set_version_counter(std::forward<VariableVersion>(version_counter)); in shallow_copy_and_detach_core()
456 /*version_counter=*/std::forward<VariableVersion>(version_counter), in shallow_copy_and_detach_core()
462 const c10::VariableVersion& version_counter, in shallow_copy_and_detach() argument
465 version_counter, allow_tensor_metadata_change); in shallow_copy_and_detach()
[all …]
DNestedTensorImpl.h134 const c10::VariableVersion& version_counter,
138 c10::VariableVersion&& version_counter,
145 /*version_counter=*/version_counter(), in shallow_copy_from()
178 VariableVersion&& version_counter,
DFunctionalTensorWrapper.h213 const c10::VariableVersion& version_counter,
217 c10::VariableVersion&& version_counter,
249 VariableVersion&& version_counter,
256 const c10::VariableVersion& version_counter,
282 const c10::VariableVersion& version_counter,
/external/pytorch/aten/src/ATen/quantized/
DQTensorImpl.h49 const c10::VariableVersion& version_counter, in shallow_copy_and_detach()
56 /*version_counter=*/version_counter, in shallow_copy_and_detach()
70 c10::VariableVersion&& version_counter, in shallow_copy_and_detach()
77 /*version_counter=*/std::move(version_counter), in shallow_copy_and_detach()
96 /*version_counter=*/version_counter(), in shallow_copy_from()
116 const c10::VariableVersion& version_counter, in copy_tensor_metadata()
118 …TensorImpl::copy_tensor_metadata(src_q_impl, dest_q_impl, version_counter, allow_tensor_metadata_c… in copy_tensor_metadata()
/external/pytorch/torch/csrc/lazy/core/
Dtensor_impl.cpp97 const c10::VariableVersion& version_counter, in shallow_copy_and_detach() argument
103 /*version_counter=*/version_counter, in shallow_copy_and_detach()
109 c10::VariableVersion&& version_counter, in shallow_copy_and_detach() argument
115 /*version_counter=*/std::move(version_counter), in shallow_copy_and_detach()
127 /*version_counter=*/version_counter(), in shallow_copy_from()
Dtensor_impl.h31 const c10::VariableVersion& version_counter,
35 c10::VariableVersion&& version_counter,
/external/pytorch/torch/csrc/autograd/
Dsaved_variable.cpp158 auto current_version = impl::version_counter(data_).current_version(); in unpack()
219 impl::set_version_counter(var, impl::version_counter(data)); in unpack()
239 const auto version = impl::version_counter(data).current_version(); in set_hooks_and_pack_data()
242 version == impl::version_counter(data).current_version(), in set_hooks_and_pack_data()
Dvariable.cpp56 impl::version_counter(backward_info_.value().base_)); in DifferentiableViewMeta()
57 attr_version_ = self_impl->version_counter().current_version(); in DifferentiableViewMeta()
355 const c10::VariableVersion& version_counter) { in set_version_counter() argument
358 self.unsafeGetTensorImpl()->set_version_counter(version_counter); in set_version_counter()
366 const c10::VariableVersion& version_counter(const Variable& self) { in version_counter() function
369 return self.unsafeGetTensorImpl()->version_counter(); in version_counter()
451 /*version_counter=*/self.unsafeGetTensorImpl()->version_counter(), in tensor_data()
526 return self.unsafeGetTensorImpl()->version_counter().current_version(); in _version()
Dvariable.h172 const c10::VariableVersion& version_counter);
175 TORCH_API const c10::VariableVersion& version_counter(const Variable&);
840 /*version_counter=*/impl::version_counter(base),
Dinit.cpp387 auto vc = torch::autograd::impl::version_counter(t); in THPAutograd_initExtension()
/external/pytorch/c10/core/
DTensorImpl.cpp496 VariableVersion&& version_counter, in shallow_copy_and_detach_core() argument
513 r->set_version_counter(std::forward<VariableVersion>(version_counter)); in shallow_copy_and_detach_core()
528 /*version_counter=*/std::forward<VariableVersion>(version_counter), in shallow_copy_and_detach_core()
534 const c10::VariableVersion& version_counter, in shallow_copy_and_detach() argument
537 version_counter, allow_tensor_metadata_change); in shallow_copy_and_detach()
541 c10::VariableVersion&& version_counter, in shallow_copy_and_detach() argument
544 std::move(version_counter), allow_tensor_metadata_change); in shallow_copy_and_detach()
622 const c10::VariableVersion& version_counter, in copy_tensor_metadata() argument
630 dest_impl->set_version_counter(version_counter); in copy_tensor_metadata()
637 c10::VariableVersion&& version_counter, in copy_tensor_metadata() argument
[all …]
DTensorImpl.h2064 VariableVersion&& version_counter,
2075 const c10::VariableVersion& version_counter,
2085 c10::VariableVersion&& version_counter,
2098 /*version_counter=*/version_counter(), in shallow_copy_from()
2104 void set_version_counter(const c10::VariableVersion& version_counter) { in set_version_counter()
2106 !(is_inference() && version_counter.enabled()), in set_version_counter()
2108 version_counter_ = version_counter; in set_version_counter()
2111 void set_version_counter(c10::VariableVersion&& version_counter) { in set_version_counter()
2113 !(is_inference() && version_counter.enabled()), in set_version_counter()
2115 version_counter_ = std::move(version_counter); in set_version_counter()
[all …]
/external/pytorch/aten/src/ATen/functorch/
DTensorWrapper.cpp115 const c10::VariableVersion& version_counter, in shallow_copy_and_detach() argument
118 dest_impl->set_version_counter(version_counter); in shallow_copy_and_detach()
126 c10::VariableVersion&& version_counter, in shallow_copy_and_detach() argument
129 dest_impl->set_version_counter(version_counter); in shallow_copy_and_detach()
DTensorWrapper.h62 const c10::VariableVersion& version_counter,
65 c10::VariableVersion&& version_counter,
DBatchedTensorImpl.h76 const c10::VariableVersion& version_counter,
79 c10::VariableVersion&& version_counter,
DBatchedTensorImpl.cpp156 const c10::VariableVersion& version_counter, in shallow_copy_and_detach() argument
164 c10::VariableVersion&& version_counter, in shallow_copy_and_detach() argument
/external/pytorch/torch/distributed/elastic/rendezvous/
Detcd_rendezvous.py479 version_counter = self.client.get(self.get_path("/rdzv/version_counter"))
480 version_counter.value = str(int(version_counter.value) + 1)
481 self.client.update(version_counter)
507 "version": version_counter.value,
/external/pytorch/test/cpp/api/
Dinference_mode.cpp537 t.unsafeGetTensorImpl()->version_counter().current_version(), in TEST()
542 t.unsafeGetTensorImpl()->version_counter().current_version(), in TEST()
549 uint32_t v = c.unsafeGetTensorImpl()->version_counter().current_version(); in TEST()
552 c.unsafeGetTensorImpl()->version_counter().current_version(), v + 1); in TEST()
/external/pytorch/aten/src/ATen/native/sparse/
DSparseTensor.cpp205 /*version_counter=*/indices.unsafeGetTensorImpl()->version_counter(), in new_with_dims_and_tensor_sparse_symint()
209 /*version_counter=*/values.unsafeGetTensorImpl()->version_counter(), in new_with_dims_and_tensor_sparse_symint()