Commit 1fea52ab authored by Shucai Xiao's avatar Shucai Xiao
Browse files

fix review comments.

parent 20d25efb
...@@ -30,15 +30,8 @@ struct convert : unary<convert> ...@@ -30,15 +30,8 @@ struct convert : unary<convert>
shape compute_shape(std::vector<shape> inputs) const shape compute_shape(std::vector<shape> inputs) const
{ {
check_shapes{inputs, *this}.has(1); check_shapes{inputs, *this}.has(1);
if(inputs.at(0).packed())
{
return {target_type, inputs.at(0).lens(), inputs.at(0).strides()}; return {target_type, inputs.at(0).lens(), inputs.at(0).strides()};
} }
else
{
return {target_type, inputs.at(0).lens()};
}
}
auto apply() const auto apply() const
{ {
......
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