44     Tensor max = tf_in.zeros({2, 3, 2});  in test_max_out_invalid_dimensions()
45     Tensor max_indices = tf_in.zeros({2, 3});  in test_max_out_invalid_dimensions()
53     max = tf_in.zeros({2, 3, 2});  in test_max_out_invalid_dimensions()
54     max_indices = tf_in.zeros({2, 3, 2});  in test_max_out_invalid_dimensions()
60     max = tf_in.zeros({2, 3, 1});  in test_max_out_invalid_dimensions()
61     max_indices = tf_in.zeros({2, 3, 1});  in test_max_out_invalid_dimensions()
67     max = tf_in.zeros({2, 3, 1});  in test_max_out_invalid_dimensions()
68     max_indices = tf_in.zeros({2, 3, 1});  in test_max_out_invalid_dimensions()
104     Tensor max = tf.zeros(out_shape, dynamism);  in test_dynamic_shape()
105     Tensor max_indices = tfl.zeros(out_shape, dynamism);  in test_dynamic_shape()
130     Tensor max = tf_in.zeros({2, 4});  in test_max_out_dtype()
131     Tensor max_indices = tf_long.zeros({2, 4});  in test_max_out_dtype()
167     max = tf_in.zeros({2, 3, 1});  in test_max_out_dtype()
168     max_indices = tf_long.zeros({2, 3, 1});  in test_max_out_dtype()
193   Tensor max = tf_bool.zeros({2, 3, 1});  in test_max_out_dtype()
194   Tensor max_indices = tf_long.zeros({2, 3, 1});  in test_max_out_dtype()
236     Tensor out = tf_out.zeros({});  in test_max_unary_out_dtype()
246     Tensor out = tf_in.zeros({});  in test_max_unary_out_empty_integer()
256     Tensor out = tf_in.zeros({});  in test_max_unary_out_empty_floating()
301   Tensor max = tf_long.zeros({2, 3, 1});  in TEST_F()
302   Tensor max_indices = tf_long.zeros({2, 3, 1});  in TEST_F()
310   max = tf_float.zeros({2, 3, 1});  in TEST_F()
311   max_indices = tf_float.zeros({2, 3, 1});  in TEST_F()
340   Tensor max = tf_float.zeros({2, 3, 1});  in TEST_F()
341   Tensor max_indices = tf_long.zeros({2, 3, 1});  in TEST_F()