Lines Matching full:zeros
51 Tensor in = tf_data.zeros(sizes); in test_scatter_src_out()
52 Tensor out = tf_data.zeros(sizes); in test_scatter_src_out()
101 out = tf_data.zeros(/*sizes=*/{2, 3, 3}); in test_scatter_src_out()
131 out = tf_data.zeros(/*sizes=*/{2, 3, 3}); in test_scatter_src_out()
170 Tensor self = tf_data.zeros(sizes); in test_scatter_src_out_invalid_dim()
171 Tensor out = tf_data.zeros(sizes); in test_scatter_src_out_invalid_dim()
180 src = tf_data.zeros(/*sizes=*/{2, 2, 2}); in test_scatter_src_out_invalid_dim()
184 src = tf_data.zeros(/*sizes=*/{5, 5}); in test_scatter_src_out_invalid_dim()
185 index = tf_index.zeros(/*sizes=*/{2, 2, 2}); in test_scatter_src_out_invalid_dim()
191 index = tf_index.zeros(/*sizes=*/{4, 6}); in test_scatter_src_out_invalid_dim()
197 index = tf_index.zeros(/*sizes=*/{4, 5}); in test_scatter_src_out_invalid_dim()
229 Tensor self = tf_data.zeros(sizes); in test_scatter_value_out()
230 Tensor out = tf_data.zeros(sizes); in test_scatter_value_out()
257 out = tf_data.zeros(/*sizes=*/{2, 3, 3}); in test_scatter_value_out()
287 out = tf_data.zeros(/*sizes=*/{2, 3, 3}); in test_scatter_value_out()
327 Tensor out = tf_data.zeros(sizes); in test_scatter_value_out_invalid_dim()
336 index = tf_index.zeros(/*sizes=*/{2, 2, 2}); in test_scatter_value_out_invalid_dim()
342 index = tf_index.zeros(/*sizes=*/{3, 5}); in test_scatter_value_out_invalid_dim()
359 Tensor index = tf_index.zeros({2, 3, 4}); in test_dynamic_shape()
362 Tensor out = tf.zeros(out_shape, dynamism); in test_dynamic_shape()
401 Tensor out = tf_data.zeros({2, 5}); in TEST_F()
432 Tensor out = tf_char.zeros(sizes); in TEST_F()
441 out = tf_char.zeros(sizes); in TEST_F()
469 Tensor index = tf_index.zeros({2, 0, 3}); in TEST_F()
471 Tensor out = tf_data.zeros({2, 5}); in TEST_F()
481 Tensor index = tf_index.zeros({}); in TEST_F()
483 Tensor out = tf_data.zeros({}); in TEST_F()
495 Tensor out = tf_data.zeros({}); in TEST_F()
507 Tensor out = tf_data.zeros({2, 3}); in TEST_F()
565 Tensor index = tf_index.zeros({2, 0, 3}); in TEST_F()
567 Tensor out = tf_data.zeros({2, 5}); in TEST_F()
577 Tensor index = tf_index.zeros({}); in TEST_F()
579 Tensor out = tf_data.zeros({}); in TEST_F()
591 Tensor out = tf_data.zeros({}); in TEST_F()
603 Tensor out = tf_data.zeros({2, 3}); in TEST_F()