From 60a0ed6b4e276dcb9cc23f4b39d3fbe1f6dfdd0f Mon Sep 17 00:00:00 2001
From: Yin Li <liyin@xiaomi.com>
Date: Fri, 1 Dec 2017 11:14:12 +0800
Subject: [PATCH] Add output & memory related info

---
 mace/proto/mace.proto | 18 ++++++++++++++++++
 1 file changed, 18 insertions(+)

diff --git a/mace/proto/mace.proto b/mace/proto/mace.proto
index 8e680821..7a0ad721 100644
--- a/mace/proto/mace.proto
+++ b/mace/proto/mace.proto
@@ -67,12 +67,20 @@ message NodeInput {
   optional int32 output_port = 2;
 }
 
+message OutputShape {
+  repeated int64 dims = 1;
+}
+
 message OperatorDef {
   repeated string input = 1;
   repeated string output = 2;
   optional string name = 3;
   optional string type = 4;
   repeated Argument arg = 5;
+  optional OutputShape output_shape = 6;
+
+  // Memory optimization: only support one single output op
+  optional int32 mem_id = 10 [default = -1];
 
   // for hexagon mace-nnlib
   optional uint32 node_id = 100;
@@ -82,6 +90,16 @@ message OperatorDef {
   repeated int32 out_max_byte_size = 104; // only support 32-bit len
 }
 
+// for memory optimization
+message MemoryBlock {
+  optional int32 mem_id = 1;
+  optional uint32 x = 2;
+  optional uint32 y = 3;
+}
+message MemoryArena {
+  repeated MemoryBlock mem_block = 1;
+}
+
 // for hexagon mace-nnlib
 message InputInfo {
   optional string name = 1;
-- 
GitLab