Searched refs:new_loop_vars (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/python/autograph/operators/ |
| D | control_flow.py | 704 new_loop_vars = aug_get_state()
709 init_vars, loop_vars, new_loop_vars, symbol_names, opts)
710 return new_loop_vars
792 new_loop_vars = get_state()
794 init_vars, loop_vars, new_loop_vars, symbol_names, opts,
796 return new_loop_vars
800 new_loop_vars = control_flow_ops.cond(
805 new_loop_vars = main_path()
807 scan_outputs = new_loop_vars, extra_cond
808 new_scan_state = new_loop_vars
[all …]
|
| /external/pytorch/torch/csrc/jit/tensorexpr/ |
| D | loopnest.cpp | 2864 std::vector<VarPtr> new_loop_vars; in cacheAccesses() local
2874 new_loop_vars.push_back( in cacheAccesses()
2876 new_loop_vars_expr.push_back(new_loop_vars[i]); in cacheAccesses()
2886 for (size_t i = 0; i < new_loop_vars.size(); ++i) { in cacheAccesses()
2887 tmp_params.push_back(alloc<Add>(new_loop_vars[i], info.start[i])); in cacheAccesses()
2925 for (int64_t i = new_loop_vars.size() - 1; i >= 0; --i) { in cacheAccesses()
2927 new_loop_vars[i], immLike(tmp_dims[i], 0), tmp_dims[i], tmp_init); in cacheAccesses()
2946 for (int64_t i = new_loop_vars.size() - 1; i >= 0; --i) { in cacheAccesses()
2948 new_loop_vars[i], immLike(tmp_dims[i], 0), tmp_dims[i], tmp_store); in cacheAccesses()
2965 for (int64_t i = new_loop_vars.size() - 1; i >= 0; --i) { in cacheAccesses()
[all …]
|