Skip to content

GitLab

Commit 48187e79 authored by turneram's avatar turneram
Browse files

Formatting

parent 6202ea15
Show whitespace changes
Inline Side-by-side
Showing with 9 additions and 5 deletions
test/onnx/gen_onnx.py
View file @ 48187e79
...@@ -190,11 +190,15 @@ def atanh_test(): ...@@ -190,11 +190,15 @@ def atanh_test():
@onnx_test @onnx_test
def attention_test(): def attention_test():
input = helper.make_tensor_value_info('input', TensorProto.FLOAT, [2, 384, 768]) input = helper.make_tensor_value_info('input', TensorProto.FLOAT,
weights = helper.make_tensor_value_info('weights', TensorProto.FLOAT, [768, 2304]) [2, 384, 768])
weights = helper.make_tensor_value_info('weights', TensorProto.FLOAT,
[768, 2304])
bias = helper.make_tensor_value_info('bias', TensorProto.FLOAT, [2304]) bias = helper.make_tensor_value_info('bias', TensorProto.FLOAT, [2304])
mask_index = helper.make_tensor_value_info('mask_index', TensorProto.INT64, [2, 384]) mask_index = helper.make_tensor_value_info('mask_index', TensorProto.INT64,
result = helper.make_tensor_value_info('result', TensorProto.FLOAT, [2, 384, 768]) [2, 384])
result = helper.make_tensor_value_info('result', TensorProto.FLOAT,
[2, 384, 768])
node = helper.make_node('Attention', node = helper.make_node('Attention',
inputs=['input', 'weights', 'bias', 'mask_index'], inputs=['input', 'weights', 'bias', 'mask_index'],
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment