Searched defs:DeviceTypePropagationPass (Results 1 – 1 of 1) sorted by relevance
149 struct DeviceTypePropagationPass : public PropertyPropBase { struct150 explicit DeviceTypePropagationPass(std::shared_ptr<Graph> graph) in DeviceTypePropagationPass() argument156 bool run() { in run()162 void propagateNode(Node* n, bool _ = true) override { in propagateNode()201 void processAtenOps(Node* n) { in processAtenOps()218 void buildRuleRegistry() { in buildRuleRegistry()243 static std::unique_ptr<OperatorMap<PropRule>> device_prop_registry_;244 bool changed_ = false;