Lines Matching full:zeros
44 Tensor min = tf_in.zeros({2, 3, 2}); in test_min_out_invalid_dimensions()
45 Tensor min_indices = tf_in.zeros({2, 3}); in test_min_out_invalid_dimensions()
53 min = tf_in.zeros({2, 3, 2}); in test_min_out_invalid_dimensions()
54 min_indices = tf_in.zeros({2, 3, 2}); in test_min_out_invalid_dimensions()
60 min = tf_in.zeros({2, 3, 1}); in test_min_out_invalid_dimensions()
61 min_indices = tf_in.zeros({2, 3, 1}); in test_min_out_invalid_dimensions()
67 min = tf_in.zeros({2, 3, 1}); in test_min_out_invalid_dimensions()
68 min_indices = tf_in.zeros({2, 3, 1}); in test_min_out_invalid_dimensions()
100 Tensor min = tf.zeros(out_shape, dynamism); in test_dynamic_shape()
101 Tensor min_indices = tfl.zeros(out_shape, dynamism); in test_dynamic_shape()
126 Tensor min = tf_in.zeros({2, 4}); in test_min_out_dtype()
127 Tensor min_indices = tf_long.zeros({2, 4}); in test_min_out_dtype()
163 min = tf_in.zeros({2, 3, 1}); in test_min_out_dtype()
164 min_indices = tf_long.zeros({2, 3, 1}); in test_min_out_dtype()
189 Tensor min = tf_bool.zeros({2, 3, 1}); in test_min_out_dtype()
190 Tensor min_indices = tf_long.zeros({2, 3, 1}); in test_min_out_dtype()
232 Tensor out = tf_out.zeros({}); in test_min_unary_out_dtype()
242 Tensor out = tf_in.zeros({}); in test_min_unary_out_empty_integer()
252 Tensor out = tf_in.zeros({}); in test_min_unary_out_empty_floating()
297 Tensor min = tf_long.zeros({2, 3, 1}); in TEST_F()
298 Tensor min_indices = tf_long.zeros({2, 3, 1}); in TEST_F()
306 min = tf_float.zeros({2, 3, 1}); in TEST_F()
307 min_indices = tf_float.zeros({2, 3, 1}); in TEST_F()
336 Tensor min = tf_float.zeros({2, 3, 1}); in TEST_F()
337 Tensor min_indices = tf_long.zeros({2, 3, 1}); in TEST_F()