"ts/vscode:/vscode.git/clone" did not exist on "761732ab5f8e2606d6b5a5c6e0bf025d3a085b60"
Commit 8fbb7af9 authored by Khalique Ahmed's avatar Khalique Ahmed
Browse files

some mlir changes

parent 94cf1bf3
......@@ -179,7 +179,7 @@ struct find_mlir_op
}
top_inputs.push_back(input);
instruction_ref prev_input =
mm->add_parameter("y" + std::to_string(input_cnt++), input->get_shape());
mm->add_parameter("y" + std::to_string(input_cnt++), {input->get_shape().type(), input->get_shape().lens()});
for(const auto& op : reverse(op_stream))
{
prev_input = mm->add_instruction(op, {prev_input});
......
......@@ -767,9 +767,9 @@ void adjust_param_shapes(module& m, const std::vector<instruction_ref>& inputs)
{
auto perm = find_permutation(input);
auto iperm = invert_permutation(perm);
lens = reorder_dims(lens, iperm);
strides = reorder_dims(strides, iperm);
ops.push_back(make_op("transpose", {{"permutation", perm}}));
lens = reorder_dims(lens, perm);
strides = reorder_dims(strides, perm);
ops.push_back(make_op("transpose", {{"permutation", iperm}}));
}
if(input.broadcasted())
{
......
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