Skip to content

GitLab

Commit e33f42ed authored by Shucai Xiao's avatar Shucai Xiao
Browse files

use a separate PR to parsing softmax using a new implementation

parent ade4bccd
Hide whitespace changes
Inline Side-by-side
Showing with 0 additions and 11 deletions
src/onnx/onnx.cpp
View file @ e33f42ed
...@@ -261,17 +261,6 @@ struct onnx_parser ...@@ -261,17 +261,6 @@ struct onnx_parser
return prog.add_instruction(op, std::move(args)); return prog.add_instruction(op, std::move(args));
} }
// instruction_ref
// parse_softmax(const std::string&, const attribute_map&, std::vector<instruction_ref> args)
// {
// auto dims = args.front()->get_shape().lens();
// auto r =
// prog.add_instruction(op::reshape{{long(dims[0]), long(dims[1]), 1, 1}},
// args.front());
// auto s = prog.add_instruction(op::softmax{}, r);
// return prog.add_instruction(op::reshape{{long(dims[0]), long(dims[1])}}, s);
// }
instruction_ref parse_softmax(const std::string&, instruction_ref parse_softmax(const std::string&,
const attribute_map& attributes, const attribute_map& attributes,
std::vector<instruction_ref> args) std::vector<instruction_ref> args)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment