diff options
Diffstat (limited to 'res/TensorFlowTests/UNIT_Pack_003/test.pbtxt')
-rw-r--r-- | res/TensorFlowTests/UNIT_Pack_003/test.pbtxt | 78 |
1 files changed, 78 insertions, 0 deletions
diff --git a/res/TensorFlowTests/UNIT_Pack_003/test.pbtxt b/res/TensorFlowTests/UNIT_Pack_003/test.pbtxt new file mode 100644 index 000000000..73b278c02 --- /dev/null +++ b/res/TensorFlowTests/UNIT_Pack_003/test.pbtxt @@ -0,0 +1,78 @@ +node { + name: "val_1" + op: "Const" + attr { + key: "dtype" + value { + type: DT_FLOAT + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_FLOAT + tensor_shape { + dim { + size: 5 + } + dim { + size: 3 + } + } + float_val: 0 + } + } + } +} +node { + name: "val_2" + op: "Const" + attr { + key: "dtype" + value { + type: DT_FLOAT + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_FLOAT + tensor_shape { + dim { + size: 5 + } + dim { + size: 3 + } + } + float_val: 1 + } + } + } +} +node { + name: "pack_0" + op: "Pack" + input: "val_1" + input: "val_2" + attr { + key: "N" + value { + i: 2 + } + } + attr { + key: "T" + value { + type: DT_FLOAT + } + } + attr { + key: "axis" + value { + i: -2 + } + } +} |