Home
last modified time | relevance | path

Searched refs:functional_impl (Results 1 – 4 of 4) sorted by relevance

/external/pytorch/torch/_custom_op/
Dfunctional.py52 functional_impl = construct_functional_impl(mutable_op)
53 lib.impl(new_op_name, functional_impl, 'CompositeExplicitAutograd')
70 def functional_impl(*args): function
90 return functional_impl
/external/pytorch/torch/csrc/inductor/
Dresize_storage_bytes.cpp43 auto functional_impl = in resize_storage_bytes__functionalize() local
47 op.call(functional_impl->value(), new_size); in resize_storage_bytes__functionalize()
/external/pytorch/aten/src/ATen/
DFunctionalTensorWrapper.h293 auto functional_impl = in unsafeGetFunctionalWrapper() local
295 TORCH_INTERNAL_ASSERT_DEBUG_ONLY(functional_impl != nullptr); in unsafeGetFunctionalWrapper()
296 return functional_impl; in unsafeGetFunctionalWrapper()
DFunctionalTensorWrapper.cpp477 auto functional_impl = in shallow_copy_from() local
480 /*src_impl=*/functional_impl, in shallow_copy_from()
613 auto functional_impl = at::functionalization::impl::unsafeGetFunctionalWrapper(t); in sync() local
614 functional_impl->sync_(); in sync()