Home
last modified time | relevance | path

Searched defs:reduce_vars (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/jit/tensorexpr/
H A Dtensor.h153 std::vector<VarHandle> reduce_vars = create_index_vars(reduce_dims); in Reduce() local
/aosp_15_r20/external/pytorch/torch/_inductor/
H A Dloop_body.py80 def __init__(self, fn, args, var_ranges, iter_vars, reduce_vars): argument