Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/csrc/jit/frontend/
H A Dconcrete_module_type.h12 enum class IterableModuleKind { NONE, LIST, DICT, PARAMLIST, PARAMDICT }; enum
87 void setIterableModuleKind(IterableModuleKind kind);
178 IterableModuleKind iterableModuleKind_ = IterableModuleKind::NONE;
198 IterableModuleKind getIterableModuleKind() const;
H A Dconcrete_module_type.cpp211 void ConcreteModuleTypeBuilder::setIterableModuleKind(IterableModuleKind kind) { in setIterableModuleKind()
215 IterableModuleKind ConcreteModuleType::getIterableModuleKind() const { in getIterableModuleKind()
/aosp_15_r20/external/pytorch/torch/csrc/jit/python/
H A Dpython_sugared_value.cpp268 if (concreteType_->getIterableModuleKind() == IterableModuleKind::LIST) { in asTupleValue()
306 if (concreteType_->getIterableModuleKind() == IterableModuleKind::LIST) { in getitem()
328 concreteType_->getIterableModuleKind() == IterableModuleKind::PARAMLIST) { in getitem()
332 concreteType_->getIterableModuleKind() == IterableModuleKind::DICT || in getitem()
333 concreteType_->getIterableModuleKind() == IterableModuleKind::PARAMDICT) { in getitem()
336 if (concreteType_->getIterableModuleKind() == IterableModuleKind::DICT) { in getitem()
340 IterableModuleKind::PARAMDICT) { in getitem()
652 if (concreteType_->getIterableModuleKind() == IterableModuleKind::DICT) { in tryGetAttr()
869 if (iterableModuleKind == IterableModuleKind::NONE) { in iter()
876 if (iterableModuleKind == IterableModuleKind::DICT) { in iter()
[all …]
H A Dscript_init.cpp2182 self.setIterableModuleKind(IterableModuleKind::DICT); in initJitScriptBindings()
2194 self.setIterableModuleKind(IterableModuleKind::LIST); in initJitScriptBindings()
2199 self.setIterableModuleKind(IterableModuleKind::PARAMLIST); in initJitScriptBindings()
2202 self.setIterableModuleKind(IterableModuleKind::PARAMDICT); in initJitScriptBindings()