Home
last modified time | relevance | path

Searched refs:unsafeGetFunctionalWrapper (Results 1 – 8 of 8) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/
H A DFunctionalTensorWrapper.cpp561 auto impl = unsafeGetFunctionalWrapper(tensor); in from_functional_tensor()
613 auto functional_impl = at::functionalization::impl::unsafeGetFunctionalWrapper(t); in sync()
634 unsafeGetFunctionalWrapper(functional_tensor)->replace_(other); in replace_()
649 … at::_propagate_xla_data(at::functionalization::impl::unsafeGetFunctionalWrapper(functional_tensor) in propagate_xla_data()
680 unsafeGetFunctionalWrapper(functional_tensor)->commit_update(); in commit_update()
691 unsafeGetFunctionalWrapper(functional_tensor)->_unsafe_reset_storage(); in unsafe_reset_storage()
696 unsafeGetFunctionalWrapper(functional_tensor)->mark_mutation_hidden_from_autograd(); in mark_mutation_hidden_from_autograd()
701 return unsafeGetFunctionalWrapper(functional_tensor)->are_all_mutations_hidden_from_autograd(); in are_all_mutations_hidden_from_autograd()
706 …return unsafeGetFunctionalWrapper(functional_tensor)->are_all_mutations_under_no_grad_or_inference… in are_all_mutations_under_no_grad_or_inference_mode()
715 return unsafeGetFunctionalWrapper(tensor)->isBaseTensor(); in isBaseTensor()
[all …]
H A DFunctionalizeFallbackKernel.cpp159 auto func_impl = at::functionalization::impl::unsafeGetFunctionalWrapper(self); in resize__functionalization()
340 auto self_impl = at::functionalization::impl::unsafeGetFunctionalWrapper(self); in set__functionalize()
341 auto src_impl = at::functionalization::impl::unsafeGetFunctionalWrapper(src); in set__functionalize()
H A DFunctionalTensorWrapper.h291 TORCH_API inline FunctionalTensorWrapper* unsafeGetFunctionalWrapper( in unsafeGetFunctionalWrapper() function
/aosp_15_r20/external/pytorch/torch/csrc/autograd/
H A Dpython_torch_functions_manual.cpp619 auto impl = at::functionalization::impl::unsafeGetFunctionalWrapper(t); in initTorchFunctions()
643 at::functionalization::impl::unsafeGetFunctionalWrapper(tensor); in initTorchFunctions()
648 auto impl = at::functionalization::impl::unsafeGetFunctionalWrapper(t); in initTorchFunctions()
673 auto t_impl = at::functionalization::impl::unsafeGetFunctionalWrapper(t); in initTorchFunctions()
691 at::functionalization::impl::unsafeGetFunctionalWrapper(t); in initTorchFunctions()
696 auto t_impl = at::functionalization::impl::unsafeGetFunctionalWrapper(t); in initTorchFunctions()
704 at::functionalization::impl::unsafeGetFunctionalWrapper(t); in initTorchFunctions()
710 auto wrapper = at::functionalization::impl::unsafeGetFunctionalWrapper(t); in initTorchFunctions()
715 auto wrapper = at::functionalization::impl::unsafeGetFunctionalWrapper(t); in initTorchFunctions()
/aosp_15_r20/external/pytorch/torch/csrc/functorch/
H A Dinit.cpp44 at::functionalization::impl::unsafeGetFunctionalWrapper(t)->set_level(level); in _wrap_functional_tensor()
56 at::functionalization::impl::unsafeGetFunctionalWrapper(wrapped); in _assert_wrapped_functional()
70 at::functionalization::impl::unsafeGetFunctionalWrapper(wrapped); in _propagate_functional_input_mutation()
176 at::functionalization::impl::unsafeGetFunctionalWrapper(self); in _unwrap_functional_tensor()
333 at::functionalization::impl::unsafeGetFunctionalWrapper(tensor); in get_unwrapped()
355 at::functionalization::impl::unsafeGetFunctionalWrapper(tensor); in maybe_get_level()
/aosp_15_r20/external/pytorch/torch/csrc/inductor/
H A Dresize_storage_bytes.cpp44 at::functionalization::impl::unsafeGetFunctionalWrapper(variable); in resize_storage_bytes__functionalize()
/aosp_15_r20/external/pytorch/aten/src/ATen/functorch/
H A DFunctionalizeInterpreter.cpp34 auto wrapper = at::functionalization::impl::unsafeGetFunctionalWrapper(tensor); in processImpl()
/aosp_15_r20/external/pytorch/torch/csrc/lazy/core/
H A Dtensor_util.h70 return at::functionalization::impl::unsafeGetFunctionalWrapper(tensor) in maybe_unwrap_functional()