Searched refs:ConcreteModuleType (Results 1 – 8 of 8) sorted by relevance
| /external/pytorch/torch/csrc/jit/frontend/ |
| D | concrete_module_type.cpp | 48 std::shared_ptr<ConcreteModuleType> ConcreteModuleType::fromJitType( in fromJitType()
66 builder.addModule(attrName, ConcreteModuleType::fromJitType(attrType)); in fromJitType()
83 auto ret = std::shared_ptr<ConcreteModuleType>(new ConcreteModuleType()); in fromJitType()
90 ConcreteModuleType::ConcreteModuleType(ConcreteModuleTypeBuilder data) in ConcreteModuleType() function in torch::jit::ConcreteModuleType
148 TypePtr ConcreteModuleType::getJitType() const { in getJitType()
152 std::optional<py::object> ConcreteModuleType::getPyClass() const { in getPyClass()
159 std::optional<std::vector<std::string>> ConcreteModuleType::findOverloads( in findOverloads()
168 std::optional<Function*> ConcreteModuleType::findFunctionAttribute( in findFunctionAttribute()
177 std::optional<c10::Symbol> ConcreteModuleType::findBuiltinFunction( in findBuiltinFunction()
186 std::optional<std::string> ConcreteModuleType::findFailedAttribute( in findFailedAttribute()
[all …]
|
| D | concrete_module_type.h | 13 class ConcreteModuleType; variable
76 void addModule(std::string name, std::shared_ptr<ConcreteModuleType> meta);
93 std::shared_ptr<ConcreteModuleType> build() const { in build()
94 return std::make_shared<ConcreteModuleType>(*this); in build()
129 ModuleInfo(std::string name, std::shared_ptr<ConcreteModuleType> meta) in ModuleInfo()
135 std::shared_ptr<ConcreteModuleType> meta_;
185 friend ConcreteModuleType; variable
190 class VISIBILITY_HIDDEN ConcreteModuleType {
192 explicit ConcreteModuleType(ConcreteModuleTypeBuilder data);
194 static std::shared_ptr<ConcreteModuleType> fromJitType(TypePtr type);
[all …]
|
| /external/pytorch/torch/csrc/jit/python/ |
| D | script_init.cpp | 434 ModuleSelf(std::shared_ptr<ConcreteModuleType> concreteType) in ModuleSelf()
447 std::shared_ptr<ConcreteModuleType> concreteType_;
1131 std::shared_ptr<ConcreteModuleType> concreteType, in initJitScriptBindings()
2205 py::class_<ConcreteModuleType, std::shared_ptr<ConcreteModuleType>>( in initJitScriptBindings()
2207 .def_property_readonly("py_class", &ConcreteModuleType::getPyClass) in initJitScriptBindings()
2208 .def_property_readonly("jit_type", &ConcreteModuleType::getJitType) in initJitScriptBindings()
2209 .def_static("from_jit_type", &ConcreteModuleType::fromJitType) in initJitScriptBindings()
2210 .def("get_constants", &ConcreteModuleType::getConstantsPy) in initJitScriptBindings()
2211 .def("get_attributes", &ConcreteModuleType::getAttributesPy) in initJitScriptBindings()
2212 .def("get_modules", &ConcreteModuleType::getModulesPy) in initJitScriptBindings()
[all …]
|
| D | python_sugared_value.h | 164 ModuleValue(Value* self, std::shared_ptr<ConcreteModuleType> concreteType) in ModuleValue()
242 std::shared_ptr<ConcreteModuleType> concreteType_;
|
| D | python_sugared_value.cpp | 640 ConcreteModuleType::fromJitType(selfType->getAttribute(field))); in tryGetAttr()
|
| /external/pytorch/torch/jit/ |
| D | _recursive.py | 280 sub_concrete_type = torch._C.ConcreteModuleType.from_jit_type(
424 type_store: Dict[Type[Module], List[torch._C.ConcreteModuleType]]
425 methods_compiled: Set[torch._C.ConcreteModuleType]
1016 script_module._concrete_type = torch._C.ConcreteModuleType.from_jit_type(
|
| D | _script.py | 680 self._concrete_type = torch._C.ConcreteModuleType.from_jit_type(
|
| /external/pytorch/torch/_C/ |
| D | __init__.pyi.in | 967 def add_module(self, name: str, meta: ConcreteModuleType): ...
983 class ConcreteModuleType:
985 def equals(self, other: ConcreteModuleType) -> _bool: ...
987 def from_jit_type(ty: JitType) -> ConcreteModuleType: ...
|