Searched defs:keep_dim (Results 1 – 4 of 4) sorted by relevance
151 auto keep_dim = std::get<bool>(inputs[2]); in computeMax() local
164 reduction_dims: List[int], input_ndim: int, keep_dim=False argument
1316 bool keep_dim = extra_args[1]; in nnc_aten_max_red() local1341 bool keep_dim = extra_args[1]; in nnc_aten_max_red_out() local
445 bool keep_dim = false; in TEST() local