Searched refs:max_by_dims (Results 1 – 1 of 1) sorted by relevance
79 Tensor<float, 1, DataLayout> max_by_dims = tensor.maximum(reduce_dims); in test_argmax_tuple_reducer() local82 VERIFY_IS_EQUAL(max_by_dims(l), reduced_by_dims(l).second); in test_argmax_tuple_reducer()