Searched refs:toModule (Results 1 – 13 of 13) sorted by relevance
30 AT_ASSERT(module.attr("dropout").toModule().is_training()); in testEvalModeForLoadedModule()32 AT_ASSERT(!module.attr("dropout").toModule().is_training()); in testEvalModeForLoadedModule()34 AT_ASSERT(module.attr("dropout").toModule().is_training()); in testEvalModeForLoadedModule()
669 Module current_sm = submodule.toModule(); in TEST()764 Module current_sm = submodule.toModule(); in TEST()
66 preservedSubModule_.insert(value.toModule()._ivalue()); in AttributePropagator()247 attrModule = attrModule.attr(moduleName).toModule(); in getModuleFromPath()755 attrModule = attrModule.attr(name).toModule(); in applyToForkSubgraph()846 modules.insert(attr.toModule()._ivalue()); in recordReferencedAttrs()911 auto attrModule = attr.toModule(); in handleSharedClassType()944 auto mod = val.toModule(); in removeUnusedAttrs()951 auto mod = SharedTypeSubModules_[type].begin()->toModule(); in removeUnusedAttrs()
63 current = submodule.toModule(); in toBackendSelectiveImpl()
80 archive.module_ = iv.toModule(); in try_read()
46 attrModule = attrModule.attr(moduleName).toModule(); in findSubModuleAttr()
411 attrModule = attrModule.attr(moduleName).toModule(); in findSubModuleAttr()
634 m = m.attr(p).toModule(); in findChildModule()654 m = m.attr(p).toModule(); in getInvokedModuleOpt()
254 auto observer_module = module.attr(observer_name.value()).toModule(); in getObserverDtype()266 auto observer_module = module.attr(findObserverName(v).value()).toModule(); in getEmbeddingBagObsName()1069 auto observer_module = module.attr(observer_name.value()).toModule(); in getQSchemeAndQParamVector()1566 auto observer_module = module.attr(observer_name.value()).toModule(); in insertCalculateQParams()
481 cursors_.emplace_back(SlotCursor{cur().toModule(), 0}); in next()
636 torch::jit::Module IValue::toModule() const { in toModule() function in c10::IValue
184 push(stack, module_dict.toModule().attr(ind.toStringRef())); in __anon992a14c81102()
834 torch::jit::Module toModule() const;