Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/functorch/
H A DDynamicLayer.cpp34 std::optional<bool> functionalize_add_back_views) in DynamicLayer() argument
53 interpreter_ = Interpreter::Functionalize(layerId, functionalize_add_back_views.value()); in DynamicLayer()
257 std::optional<bool> functionalize_add_back_views) { in initAndPushDynamicLayer() argument
260 …d::move(batch_size), randomness, prev_grad_mode, prev_fwd_grad_mode, functionalize_add_back_views); in initAndPushDynamicLayer()
H A DDynamicLayer.h50 std::optional<bool> functionalize_add_back_views = std::nullopt);
72 std::optional<bool> functionalize_add_back_views = std::nullopt);
/aosp_15_r20/external/pytorch/torch/_subclasses/
H A Dfunctional_tensor.py775 args, reapply_views=self.interpreter.functionalize_add_back_views()
783 if self.interpreter.functionalize_add_back_views()
/aosp_15_r20/external/pytorch/torch/_functorch/
H A Dpyfunctorch.py235 def functionalize_add_back_views(self): member in FunctionalizeInterpreter