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

remove required standard input for transpose

parent 58711bbc
...@@ -29,7 +29,7 @@ struct transpose ...@@ -29,7 +29,7 @@ struct transpose
std::string name() const { return "transpose"; } std::string name() const { return "transpose"; }
shape compute_shape(std::vector<shape> inputs) const shape compute_shape(std::vector<shape> inputs) const
{ {
check_shapes{inputs, *this}.has(1).standard(); check_shapes{inputs, *this}.has(1);
auto input = inputs.at(0); auto input = inputs.at(0);
auto input_lens = input.lens(); auto input_lens = input.lens();
auto input_strides = input.strides(); auto input_strides = input.strides();
......
...@@ -846,10 +846,6 @@ struct onnx_parser ...@@ -846,10 +846,6 @@ struct onnx_parser
auto&& perm_vals = attributes["perm"].ints(); auto&& perm_vals = attributes["perm"].ints();
perm = std::vector<int64_t>(perm_vals.begin(), perm_vals.end()); perm = std::vector<int64_t>(perm_vals.begin(), perm_vals.end());
} }
if(!args.front()->get_shape().standard())
{
args.front() = prog.add_instruction(migraphx::op::contiguous{}, args.front());
}
return prog.add_instruction(migraphx::op::transpose{perm}, args.front()); return prog.add_instruction(migraphx::op::transpose{perm}, args.front());
} }
......
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