未验证 提交 80bcf9df 编写于 作者: B BowShotDS 提交者: GitHub

Update clip op test tensorrt (#982)

* Update pad sort order

* Update clip int8

* apply code-format changes

* Update clip op test tensorrt
Co-authored-by: NBowShotDS <BowShotDS@users.noreply.github.com>
上级 fb965cd6
......@@ -23,6 +23,7 @@
*/
#include "test_op.h"
#include "operator/prototype/clip_param.h"
int create_test_clip_node(graph_t graph, const char* input_name, const char* node_name, int data_type, int layout, int n, int c, int h, int w)
{
......@@ -50,6 +51,12 @@ int create_test_clip_node(graph_t graph, const char* input_name, const char* nod
tensor_t output_tensor = create_graph_tensor(graph, node_name, data_type);
set_node_output_tensor(test_node, 0, output_tensor, TENSOR_TYPE_VAR);
/* set params */
struct clip_param* clip_param = (struct clip_param*)(struct node*)test_node->op.param_mem;
clip_param->min = 0;
clip_param->max = 6;
return 0;
}
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册