Searched refs:new_loop_vars_expr (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/csrc/jit/tensorexpr/ |
| H A D | loopnest.cpp | 2865 std::vector<ExprPtr> new_loop_vars_expr; in cacheAccesses() local
2876 new_loop_vars_expr.push_back(new_loop_vars[i]); in cacheAccesses()
2923 tmp_buf, new_loop_vars_expr, getImmediateByType(tmp_buf->dtype(), 0)); in cacheAccesses()
2942 alloc<Load>(tmp_buf, new_loop_vars_expr), in cacheAccesses()
2963 tmp_buf, new_loop_vars_expr, alloc<Load>(producer, tmp_params)); in cacheAccesses()
2980 producer, tmp_params, alloc<Load>(tmp_buf, new_loop_vars_expr)); in cacheAccesses()
|