Commit de433392 authored by turneram's avatar turneram
Browse files

Formatting

parent f14e2a44
......@@ -204,7 +204,6 @@ def attention_test():
inputs=['input', 'weights', 'bias', 'mask_index'],
outputs=['result'],
num_heads=12,
name="Attention_0")
return ([node], [input, weights, bias, mask_index], [result])
......
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