未验证 提交 dd410963 编写于 作者: Z Zhang Zheng 提交者: GitHub

Add large dim test of log_softmax (#53957)

* Add large dim test of log_softmax

* fix
上级 9f54be49
......@@ -158,6 +158,12 @@ class TestLogSoftmaxBF16Op(OpTest):
)
class TestLogSoftmaxLargeDimFP16OP(TestLogSoftmaxOp):
def set_attrs(self):
self.dtype = np.float16
self.shape = [16, 100000]
class TestNNLogSoftmaxAPI(unittest.TestCase):
def setUp(self):
self.x_shape = [2, 3, 4, 5]
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册