Searched refs:functional_impl (Results 1 – 4 of 4) sorted by relevance
52 functional_impl = construct_functional_impl(mutable_op)53 lib.impl(new_op_name, functional_impl, 'CompositeExplicitAutograd')70 def functional_impl(*args): function90 return functional_impl
43 auto functional_impl = in resize_storage_bytes__functionalize() local47 op.call(functional_impl->value(), new_size); in resize_storage_bytes__functionalize()
293 auto functional_impl = in unsafeGetFunctionalWrapper() local295 TORCH_INTERNAL_ASSERT_DEBUG_ONLY(functional_impl != nullptr); in unsafeGetFunctionalWrapper()296 return functional_impl; in unsafeGetFunctionalWrapper()
477 auto functional_impl = in shallow_copy_from() local480 /*src_impl=*/functional_impl, in shallow_copy_from()613 auto functional_impl = at::functionalization::impl::unsafeGetFunctionalWrapper(t); in sync() local614 functional_impl->sync_(); in sync()