Lines Matching full:zeros
42 Tensor out = tf_out.zeros(out_sizes_1); in test_glu_out()
49 out = tf_out.zeros(out_sizes_2); in test_glu_out()
63 Tensor in = tf_in.zeros(/*sizes=*/{4, 4, 4}); in test_glu_out_mismatched_shape()
64 Tensor out = tf_in.zeros(/*sizes=*/{2, 4, 2}); in test_glu_out_mismatched_shape()
68 out = tf_in.zeros(/*sizes=*/{4, 4, 4}); in test_glu_out_mismatched_shape()
76 Tensor in = tf_in.zeros(/*sizes=*/{2, 2}); in test_glu_out_invalid_dim()
78 Tensor out = tf_in.zeros(out_sizes); in test_glu_out_invalid_dim()
84 in = tf_in.zeros(/*sizes=*/{3, 2}); in test_glu_out_invalid_dim()
97 Tensor out = tf_float.zeros(out_sizes); in test_div_invalid_input_dtype_dies()
111 Tensor out = tf_out.zeros(out_sizes); in test_div_invalid_output_dtype_dies()
137 Tensor out = tf.zeros(out_sizes); in TEST_F()
197 tf.zeros({4, 1}, torch::executor::TensorShapeDynamism::DYNAMIC_BOUND); in TEST_F()
223 tf.zeros({10, 10}, torch::executor::TensorShapeDynamism::DYNAMIC_BOUND); in TEST_F()
250 tf.zeros({1, 1}, torch::executor::TensorShapeDynamism::DYNAMIC_UNBOUND); in TEST_F()