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
std::string name() const { return "transpose"; }
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_lens = input.lens();
auto input_strides = input.strides();
......
......@@ -846,10 +846,6 @@ struct onnx_parser
auto&& perm_vals = attributes["perm"].ints();
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());
}
......
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