Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/
H A Dinteger_value_refinement.cpp22 struct IntegerValueRefiner { struct
23 IntegerValueRefiner(std::shared_ptr<Graph> graph) in IntegerValueRefiner() function
26 bool run() { in run()
35 bool blockHasIntComparisons(Block* b) { in blockHasIntComparisons()
56 void removeIfNodeOutputsWithRefinements( in removeIfNodeOutputsWithRefinements()
120 IntegerRefinement RefineIntegerValues( in RefineIntegerValues()
206 std::optional<int64_t> tryFindRefinement(Value* v) { in tryFindRefinement()
216 std::shared_ptr<Graph> graph_;
218 std::vector<IntegerRefinement*> active_refinements_;
220 std::unordered_map<Value*, BooleanRefinementMapping> info_;
[all …]