Home
last modified time | relevance | path

Searched defs:wrong_out (Results 1 – 8 of 8) sorted by relevance

/aosp_15_r20/external/executorch/kernels/test/
H A Dop_mm_test.cpp150 Tensor wrong_out = tf.ones({2, 2, 3}); in TEST_F() local
167 Tensor wrong_out = tf.ones({7, 5}); in TEST_F() local
H A Dop_linear_test.cpp150 Tensor wrong_out = tf.ones({2, 2, 3}); in TEST_F() local
167 Tensor wrong_out = tf.ones({7, 5}); in TEST_F() local
H A Dop_bmm_test.cpp192 Tensor wrong_out = tf.ones({7, 5}); in TEST_F() local
210 Tensor wrong_out = tf.ones({3, 7, 5}); in TEST_F() local
H A Dop_embedding_test.cpp243 for (auto wrong_out: wrong_outs) { in TEST_F() local
271 Tensor wrong_out = tfl.zeros({2, 2, 2}); in TEST_F() local
H A Dop_addmm_test.cpp206 Tensor wrong_out = tf.ones({2, 2, 3}); in TEST_F() local
229 Tensor wrong_out = tf.ones({7, 5}); in TEST_F() local
H A Dop_cumsum_test.cpp123 Tensor wrong_out = tff.zeros({2, 10, 4}); in TEST_F() local
H A Dop_softmax_test.cpp133 Tensor wrong_out = tf.zeros({2, 10, 4}); in TEST_F() local
H A Dop_log_softmax_test.cpp126 Tensor wrong_out = tf.zeros({2, 10, 4}); in TEST_F() local