Searched refs:inline_mapping_ (Results 1 – 1 of 1) sorted by relevance
693 auto iter = inline_mapping_.find(func_callee_arg); in mutate_loads()694 if (iter != inline_mapping_.end()) { in mutate_loads()700 inline_mapping_[func_callee_arg] = func_caller_param; in mutate_loads()720 ExprPtr inlined = inline_mapping_[v]; in mutate_loads()727 inline_mapping_.erase(v); in mutate_loads()760 auto iter = inline_mapping_.find(v); in mutate()761 if (iter == inline_mapping_.end()) { in mutate()868 std::unordered_map<VarPtr, ExprPtr> inline_mapping_; member in torch::jit::tensorexpr::FunctionInliner