Searched defs:DtypePropagationPass (Results 1 – 1 of 1) sorted by relevance
200 struct DtypePropagationPass { struct201 explicit DtypePropagationPass(std::shared_ptr<Graph> graph) in DtypePropagationPass() function207 bool run() { in run()212 bool processBlocks(at::ArrayRef<Block*> blocks) { in processBlocks()220 bool processBlock(Block* block) { in processBlock()229 bool processNode(Node* n) { in processNode()275 bool mergeTensorProperties( in mergeTensorProperties()284 bool processIf(Node* node) { in processIf()301 bool processAtenOps(Node* n) { in processAtenOps()312 void buildDtypeRuleRegistry() { in buildDtypeRuleRegistry()[all …]