Searched defs:values_indices (Results 1 – 1 of 1) sorted by relevance
1755 auto values_indices = torch::max(input, dim, /*keepdim=*/keepdim); in TEST_F() local1773 auto values_indices = torch::min(input, dim, /*keepdim=*/keepdim); in TEST_F() local