From 57d87a79c5c9ac27d9afed7157de1fd46558b682 Mon Sep 17 00:00:00 2001 From: xiangyangkan Date: Mon, 13 Feb 2023 10:47:14 +0800 Subject: [PATCH] Typo in README.md Signed-off-by: xiangyangkan --- plugin/embLayerNormPlugin/README.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/plugin/embLayerNormPlugin/README.md b/plugin/embLayerNormPlugin/README.md index bee408dd..1956b368 100644 --- a/plugin/embLayerNormPlugin/README.md +++ b/plugin/embLayerNormPlugin/README.md @@ -27,7 +27,7 @@ Assuming contiguous input masks, encodes the masks as a single number denoting t ### Structure -The `embLayerNormPlugin` takes three inputs; `token_id`, `segmend_id`, and `input_mask`. +The `embLayerNormPlugin` takes three inputs; `token_id`, `segment_id`, and `input_mask`. `token_id` An input sequence containing token ids. token_id is an `int32` tensor with shape `[S, B]` where `S` is the sequence length and `B` is the batch size.