output, pack_0:0, TF_FLOAT, [5, 3, 2]