Runnable TopK model accesses input tensor as wrong type

[alexander-camuto]

Have a simple 1l TOPK model that analyzes fine:

(base) ➜  ✗ tract network.onnx
┏ 0 Source input
┃   ━━━ batch_size,6,I64
┣┳ 1 Topk /TopK
     ━━━ batch_size,4,I64
     ━━━ batch_size,4,I64

But once it is converted into a runnable graph to run on the following data:

data: [1,6,I64 2, 1, 5, 3, 3, 7]

It fails with:

[E] failed: Evaluating #2 "/TopK" Topk
Error: Evaluating #2 "/TopK" Topk

Caused by:
    Tensor datum type error: tensor is I64, accessed as F32

network.onnx.zip