Searched defs:FunctionalTensorWrapper (Results 1 – 2 of 2) sorted by relevance
50 struct TORCH_API FunctionalTensorWrapper : public c10::TensorImpl { struct63 const Tensor& value() const { in value()68 int64_t level() const { in level()71 void set_level(int64_t level) { in set_level()74 bool has_metadata_mutation() const { in has_metadata_mutation()78 void mark_mutation() { in mark_mutation()83 void mark_mutation_hidden_from_autograd() { in mark_mutation_hidden_from_autograd()86 void mark_mutation_during_no_grad_or_inference_mode() { in mark_mutation_during_no_grad_or_inference_mode()90 bool are_all_mutations_hidden_from_autograd() const { in are_all_mutations_hidden_from_autograd()95 bool are_all_mutations_under_no_grad_or_inference_mode() const { in are_all_mutations_under_no_grad_or_inference_mode()[all …]
51 FunctionalTensorWrapper::FunctionalTensorWrapper(const Tensor& value) in FunctionalTensorWrapper() function in at::FunctionalTensorWrapper132 FunctionalTensorWrapper::FunctionalTensorWrapper(const Tensor& view_value, const FunctionalTensorWr… in FunctionalTensorWrapper() function in at::FunctionalTensorWrapper