add name of the layer

source/sample/transformer/T2TDecoder.cpp

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

source/sample/transformer/T2TDecoder.h

@@ -27,6 +27,9 @@
 namespace transformer
 {
     
+#define DECODING_NAME "decoding"
+#define DECODING_INPUT_NAME "decoding_input"
+
 class AttDecoder
 {
 public:

source/sample/transformer/T2TEncoder.cpp

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

source/sample/transformer/T2TEncoder.h

@@ -33,6 +33,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

@@ -29,6 +29,8 @@ using namespace nts;
 namespace transformer
 {
     
+#define OUTPUT_NAME "output"
+
 /* output layer */
 class T2TOutput
 {