Searched refs:dynamicLayerStack (Results 1 – 2 of 2) sorted by relevance
89 result->dynamicLayerStack = dynamicLayerStack; in deepcopy()94 TORCH_INTERNAL_ASSERT(dynamicLayerStack.empty() || getSingleLevelAutogradFunctionAllowed(), in checkSupportsSingleLevelAutogradFunction()103 dynamicLayerStack.empty(), in checkSupportsCppAutogradFunction()108 TORCH_CHECK(dynamicLayerStack.empty() || allow_inplace_requires_grad_, in checkSupportsInplaceRequiresGrad()117 TORCH_CHECK(dynamicLayerStack.empty(), in checkSupportsRetainGrad()126 std::vector<DynamicLayer> dynamicLayerStack; member in at::functorch::FuncTorchTLS163 return getRawFunctorchTLS()->dynamicLayerStack; in dynamicLayerStackAccessor()167 auto& dynamicLayerStack = dynamicLayerStackAccessor(); in getLifeHandleForLevel() local169 (int64_t)dynamicLayerStack.size() >= level && level >= 1, in getLifeHandleForLevel()174 auto& dynamic_layer = dynamicLayerStack[level - 1]; in getLifeHandleForLevel()[all …]
105 …API std::ostream& operator<<(std::ostream& os, const std::vector<DynamicLayer>& dynamicLayerStack);