Searched defs:FunctionalStorageImpl (Results 1 – 2 of 2) sorted by relevance
90 struct TORCH_API FunctionalStorageImpl : public c10::StorageImpl { struct99 explicit FunctionalStorageImpl(const Tensor& value); argument123 ~FunctionalStorageImpl() override = default; argument125 void mark_mutation() { in mark_mutation()128 void mark_mutation_during_no_grad_or_inference_mode() { in mark_mutation_during_no_grad_or_inference_mode()131 void mark_mutation_hidden_from_autograd() { in mark_mutation_hidden_from_autograd()135 bool are_all_mutations_under_no_grad_or_inference_mode() const { in are_all_mutations_under_no_grad_or_inference_mode()144 bool are_all_mutations_hidden_from_autograd() const { in are_all_mutations_hidden_from_autograd()150 void mark_inductor_storage_resize(c10::SymInt new_size) { in mark_inductor_storage_resize()155 bool was_inductor_storage_resized() { in was_inductor_storage_resized()[all …]
92 FunctionalStorageImpl::FunctionalStorageImpl(const Tensor& base) in FunctionalStorageImpl() function in at::functionalization::FunctionalStorageImpl