[res/TensorFlowTests] Modify Custom Op user data (#5941)

This commit adds `dtype` of custom op, which will be accessed for TypeInference.

Signed-off-by: Hyun Sik Yoon <hyunsik.yoon@samsung.com>

diff --git a/res/TensorFlowTests/CustomOp_000/model.meta.pbtxt b/res/TensorFlowTests/CustomOp_000/compileInfo.pbtxt
similarity index 80%
rename from res/TensorFlowTests/CustomOp_000/model.meta.pbtxt
rename to res/TensorFlowTests/CustomOp_000/compileInfo.pbtxt
index a0d74ac..08ba0f0 100644 (file)
--- a/res/TensorFlowTests/CustomOp_000/model.meta.pbtxt
+++ b/res/TensorFlowTests/CustomOp_000/compileInfo.pbtxt
@@ -3,6 +3,12 @@ custom_op {
   name: "my/customOp/000"
   op: "new_custom_op"
   attr {
+    key: "dtype"
+    value {
+      type: DT_FLOAT
+    }
+  }
+  attr {
     key: "output_shape"
     value {
       shape {

diff --git a/res/TensorFlowTests/CustomOp_001/model.meta.pbtxt b/res/TensorFlowTests/CustomOp_001/compileInfo.pbtxt
similarity index 74%
rename from res/TensorFlowTests/CustomOp_001/model.meta.pbtxt
rename to res/TensorFlowTests/CustomOp_001/compileInfo.pbtxt
index 64d2620..92c0dd2 100644 (file)
--- a/res/TensorFlowTests/CustomOp_001/model.meta.pbtxt
+++ b/res/TensorFlowTests/CustomOp_001/compileInfo.pbtxt
@@ -3,6 +3,12 @@ custom_op {
   name: "FillFrom"
   op: "FillFrom"
   attr {
+    key: "dtype"
+    value {
+      type: DT_FLOAT
+    }
+  }
+  attr {
     key: "output_shape"
     value {
       shape {