Home
last modified time | relevance | path

Searched defs:ConvertTracedAttrReferences (Results 1 – 1 of 1) sorted by relevance

/external/pytorch/torch/csrc/jit/passes/
Dfixup_trace_scope_blocks.cpp85 struct ConvertTracedAttrReferences { struct
86 void run(const std::shared_ptr<Graph>& graph) { in run()
101 void buildAttrMap(const std::shared_ptr<Graph>& graph) { in buildAttrMap()
109 void addSelfArgToTracedForwardNodes(Block* b) { in addSelfArgToTracedForwardNodes()
131 std::vector<Value*> convertAttrReferencesToLocalGetAttrs( in convertAttrReferencesToLocalGetAttrs()
185 void replaceTracedAttrInputOnNode( in replaceTracedAttrInputOnNode()
228 void destroyTracedAttrNodes(const std::shared_ptr<Graph>& graph) { in destroyTracedAttrNodes()
236 std::unordered_map<std::string, Value*> attr_qualname_to_value;