(class_name, cu, shouldMangle)
| 16537 | return this._graph; |
| 16538 | } |
| 16539 | static create_module_object(class_name, cu, shouldMangle) { |
| 16540 | shouldMangle = shouldMangle || false; |
| 16541 | if (!class_name.prefix()) { |
| 16542 | class_name = new torch._C.QualifiedName('__torch__', class_name.name()); |
| 16543 | } |
| 16544 | if (shouldMangle && cu.get_class(class_name)) { |
| 16545 | class_name = cu.mangle(class_name); |
| 16546 | } |
| 16547 | const cls = torch.ClassType.create(class_name, cu, true); |
| 16548 | cu.register_type(cls); |
| 16549 | return [cls, cu]; |
| 16550 | } |
| 16551 | register_module(name, module) { |
| 16552 | this.type().addOrCheckAttribute(name, module.type()); |
| 16553 | this.__setattr__(name, module); // _ivalue()->setAttr(name, module._ivalue()); |
no test coverage detected