add name of the layer

source/sample/transformer/T2TDecoder.cpp

@@ -159,6 +159,8 @@ XTensor AttDecoder::Make(XTensor &inputDec, XTensor &outputEnc, XTensor &mask, X
         /* layer normalization */
         x = fnnLayerNorms[i].Make(res);
     }
+    
+    x.SetName(DECODING_NAME);
 
     return x;
 }

source/sample/transformer/T2TDecoder.h

@@ -26,6 +26,9 @@
 
 namespace transformer
 {
+    
+#define DECODING_NAME "decoding"
+#define DECODING_INPUT_NAME "decoding_input"
 
 class AttDecoder
 {
@@ -100,4 +103,4 @@ public:
 
 }
 
-#endif
\ No newline at end of file
+#endif

source/sample/transformer/T2TEncoder.cpp

@@ -139,6 +139,9 @@ XTensor AttEncoder::Make(XTensor &input, XTensor &mask, XTensor &maskEncDec, boo
         /* layer normalization */
         x = fnnLayerNorms[i].Make(res);
     }
+    
+    x.SetName(ENCODING_NAME);
+    input.SetName(ENCODING_INPUT_NAME);
 
     return x;
 }

source/sample/transformer/T2TEncoder.h

@@ -32,6 +32,9 @@ using namespace nts;
 
 namespace transformer
 {
+    
+#define ENCODING_NAME "encoding"
+#define ENCODING_INPUT_NAME "encoding_input"
 
 /* 
 base class of the encoder 

source/sample/transformer/T2TOutput.cpp

@@ -95,6 +95,8 @@ void T2TOutput::Make(XTensor &input, XTensor &output)
 
     output = LogSoftmax(MMul(x, w), -1);
     //output = Softmax(MMul(x, w), -1);
+    
+    output.SetName(OUTPUT_NAME);
 }
 
 }

source/sample/transformer/T2TOutput.h

@@ -28,6 +28,8 @@ using namespace nts;
 
 namespace transformer
 {
+    
+#define OUTPUT_NAME "output"
 
 /* output layer */
 class T2TOutput
@@ -71,4 +73,4 @@ public:
 
 }
 
-#endif
\ No newline at end of file
+#endif