Skip to content

Commit 8b4b5be

Browse files
committed
update template
1 parent 7950d54 commit 8b4b5be

1 file changed

Lines changed: 6 additions & 3 deletions

File tree

  • utensor_cgen/backend/utensor/snippets/templates/snippets/rearch
Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,12 +1,12 @@
11
/*
22
FIXME: {{op_type}} currently not supported, you have to fill up this section or it won't compile
33
4-
Input Tensors:
4+
1. Declare the operators of type {{op_type}}
5+
2. Declare input tensors:
56
{%for tensor, var_name in zip(input_tensors, input_var_names)%}
67
- {{tensor.name}}, of type {{tensor.dtype}} and is named as {{var_name}} in this file
78
{%endfor%}
8-
9-
Output Tensors:
9+
3. Declare output Tensors:
1010
{%for tensor, var_name in zip(output_tensors, out_var_names)%}
1111
- {{tensor.name}} is of type {{tensor.dtype}}, shape {{tensor.shape}} and should be named as {{var_name}}
1212
{%if quant_params_map[tensor.name]%}
@@ -16,6 +16,9 @@
1616
- is per tensor quantization: {{quant_params_map[tensor.name]["is_per_tensor"]}}
1717
{%endif%}
1818
{%endfor%}
19+
4. invoke `eval()`
20+
21+
NOTE: normally, operator is implemented as template with type parameters as the data types of input/output tensors
1922
*/
2023
// Don't forget to comment this out when you fill up the missing op and tensors
2124
static_assert(false, "{{op_type}} is not currently supported. Read information above please.");

0 commit comments

Comments
 (0)