Home
last modified time | relevance | path

Searched refs:getDynamicLayerStack (Results 1 – 7 of 7) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/functorch/
H A Dinit.cpp395 std::cout << getDynamicLayerStack() << std::endl; in dump_dls()
408 while (at::functorch::getDynamicLayerStack().size() > depth) { in popDynamicLayerStackToDepth()
529 const auto& stack = getDynamicLayerStack(); in initFuncTorchBindings()
541 const auto& stack = getDynamicLayerStack(); in initFuncTorchBindings()
549 return getDynamicLayerStack().size(); in initFuncTorchBindings()
/aosp_15_r20/external/pytorch/aten/src/ATen/functorch/
H A DVmapInterpreter.cpp18 if (getDynamicLayerStack().empty()) { in sendToNextInterpreterImpl()
H A DFunctionalizeInterpreter.cpp58 if (getDynamicLayerStack().empty()) { in sendToNextInterpreterImpl()
H A DDynamicLayer.h75 TORCH_API const std::vector<DynamicLayer>& getDynamicLayerStack();
H A DADInterpreters.cpp177 if (getDynamicLayerStack().empty()) { in autogradBasedTransformSendToNext()
H A DPyTorchOperatorHacks.cpp62 const auto& stack = getDynamicLayerStack(); in linear_hack()
H A DDynamicLayer.cpp207 const std::vector<DynamicLayer>& getDynamicLayerStack() { in getDynamicLayerStack() function