Searched refs:comp_values (Results 1 – 1 of 1) sorted by relevance
| /external/pytorch/torch/csrc/jit/tensorexpr/ |
| D | loopnest.cpp | 1245 std::vector<ExprPtr>* comp_values, in isConditionalFromCat() argument
1259 if (!isConditionalFromCat(true_ite, cond_var, comp_values, sub_exprs)) { in isConditionalFromCat()
1269 comp_values->push_back(comp_value); in isConditionalFromCat()
1276 bool areConstantsAndSorted(const std::vector<ExprPtr>& comp_values) { in areConstantsAndSorted() argument
1278 comp_consts.reserve(comp_values.size()); in areConstantsAndSorted()
1279 for (const auto& c : comp_values) { in areConstantsAndSorted()
1302 std::vector<ExprPtr> comp_values; in optimizeConditionals() local
1314 ifthenelse_exprs.front(), &cond_var, &comp_values, &sub_exprs)) { in optimizeConditionals()
1318 !comp_values.empty(), in optimizeConditionals()
1321 comp_values.insert(comp_values.begin(), immLike(comp_values[0], 0)); in optimizeConditionals()
[all …]
|