Home
last modified time | relevance | path

Searched refs:var_mapping (Results 1 – 3 of 3) sorted by relevance

/external/pytorch/torch/csrc/jit/tensorexpr/
Deval.h290 inline ExprPtr Substitute(const ExprPtr& expr, const VarMapping& var_mapping) { in Substitute() argument
291 VarSubMutator var_sub(var_mapping); in Substitute()
297 inline StmtPtr Substitute(const StmtPtr& stmt, const VarMapping& var_mapping) { in Substitute() argument
298 VarSubMutator var_sub(var_mapping); in Substitute()
308 const VarMapping& var_mapping) { in SubstituteInClone() argument
309 VarSubMutator var_sub(var_mapping); in SubstituteInClone()
319 const VarMapping& var_mapping) { in SubstituteInClone() argument
320 VarSubMutator var_sub(var_mapping); in SubstituteInClone()
Dvar_substitutor.h19 VarSubMutator(const VarMapping& var_mapping) { in VarSubMutator() argument
20 for (auto& entry : var_mapping) { in VarSubMutator()
Dloopnest.cpp2439 VarMapping var_mapping; in flatten() local
2446 var_mapping.emplace_back(curr_loop->var(), sub_expr); in flatten()
2450 Substitute(normalized_loops.back()->removeBody(), var_mapping); in flatten()