Commit 6e51e8b2 by xuchen

add the annotation and modify the indent

parent b8d2319d
...@@ -341,7 +341,6 @@ void MatrixMul(const XTensor &a, MATRIX_TRANS_TYPE transposedA, ...@@ -341,7 +341,6 @@ void MatrixMul(const XTensor &a, MATRIX_TRANS_TYPE transposedA,
/* destroy variables */ /* destroy variables */
delete[] dimSize; delete[] dimSize;
} }
/* call _MatrixMul function */ /* call _MatrixMul function */
......
...@@ -127,6 +127,14 @@ XTensor ConvertDataType(const XTensor & input, TENSOR_DATA_TYPE dataType) ...@@ -127,6 +127,14 @@ XTensor ConvertDataType(const XTensor & input, TENSOR_DATA_TYPE dataType)
return output; return output;
} }
/*
convert data type (return an XTensor structure)
make a new tensor to keep the result and return it
>> input - the input tensor
>> output - the output tensor
<< return - the output tensor with the specified data type
*/
void ConvertDataType(const XTensor & input, XTensor & output, TENSOR_DATA_TYPE dataType) void ConvertDataType(const XTensor & input, XTensor & output, TENSOR_DATA_TYPE dataType)
{ {
if (!output.isInit || input.dataType != output.dataType) { if (!output.isInit || input.dataType != output.dataType) {
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论