Skip to content

GitLab

Commit 71276f4d authored by Khalique's avatar Khalique
Browse files

initial progress

parent c58ce6f5
Show whitespace changes
Inline Side-by-side
Showing with 25 additions and 11 deletions
src/onnx/onnx.cpp
View file @ 71276f4d
...@@ -137,12 +137,24 @@ struct onnx_parser ...@@ -137,12 +137,24 @@ struct onnx_parser
const std::vector<std::size_t>* s0 = &args[0]->get_shape().lens(); const std::vector<std::size_t>* s0 = &args[0]->get_shape().lens();
const std::vector<std::size_t>* s1 = &args[1]->get_shape().lens(); const std::vector<std::size_t>* s1 = &args[1]->get_shape().lens();
bool swapped = false;
// Make sure s0 is the smaller size // Make sure s0 is the smaller size
if(s0->size() > s1->size()) if(s0->size() > s1->size())
{
std::swap(s0, s1); std::swap(s0, s1);
swapped = true;
}
// Copy the larger vector to output_lens
std::vector<std::size_t> output_lens(s1->size()); std::vector<std::size_t> output_lens(s1->size());
// if (s0->size() == 0)
// {
// shape s = swapped ? args[0]->get_shape() : args[1]->get_shape();
// auto l0 = prog.add_instruction(migraphx::op::scalar{s}, 1.0f);
// return prog.add_instruction(x, l0, args[1]);
// }
// else
// {
// Copy the larger vector to output_lens
auto offset = s1->size() - s0->size(); auto offset = s1->size() - s0->size();
std::transform(s0->begin(), std::transform(s0->begin(),
s0->end(), s0->end(),
...@@ -153,6 +165,8 @@ struct onnx_parser ...@@ -153,6 +165,8 @@ struct onnx_parser
auto l0 = prog.add_instruction(op::multibroadcast{output_lens}, args[0]); auto l0 = prog.add_instruction(op::multibroadcast{output_lens}, args[0]);
auto l1 = prog.add_instruction(op::multibroadcast{output_lens}, args[1]); auto l1 = prog.add_instruction(op::multibroadcast{output_lens}, args[1]);
return prog.add_instruction(x, l0, l1); return prog.add_instruction(x, l0, l1);
// }
} }
else else
{ {
......
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