p31042957
/
InfiniTensor
forked from jiuyuan/InfiniTensor
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

fix: 移除中文注释 

Signed-off-by: YdrMaster <ydrml@hotmail.com>
This commit is contained in:
YdrMaster 2023-03-15 15:16:16 +08:00
parent 71a87c27d1
commit 62fd619987
2 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
pyinfinitensor/src/pyinfinitensor/onnx.py
View File

@ -351,7 +351,6 @@ def to_onnx(graph: backend.GraphHandler, name: str) -> ModelProto:
def push_output(self, name: str, tensor: backend.Tensor) -> str: def push_output(self, name: str, tensor: backend.Tensor) -> str:
self.names[tensor] = name self.names[tensor] = name
# TODO 需要判断全图输出并保存到 outputs
return name return name
def push_input(self, tensor: backend.Tensor) -> str: def push_input(self, tensor: backend.Tensor) -> str:

2
src/ffi/ffi_infinitensor.cc
View File

@ -29,7 +29,7 @@ void export_values(py::module &m) {
#define VALUE(TYPE, NAME) value(#NAME, TYPE::NAME) #define VALUE(TYPE, NAME) value(#NAME, TYPE::NAME)
py::enum_<ActType>(m, "ActType") py::enum_<ActType>(m, "ActType")
.value("Linear", ActType::None) // None 是 Python 关键字，不能用 .value("Linear", ActType::None) // `None` is Python keyword
.VALUE(ActType, Relu) .VALUE(ActType, Relu)
.VALUE(ActType, Sigmoid) .VALUE(ActType, Sigmoid)
.VALUE(ActType, Tanh) .VALUE(ActType, Tanh)