Commit 2e04ea47 authored by Shucai Xiao's avatar Shucai Xiao
Browse files

fixed review comments.

parent 427bbe21
...@@ -610,7 +610,7 @@ struct cpu_unary ...@@ -610,7 +610,7 @@ struct cpu_unary
argument result{output_shape}; argument result{output_shape};
result.visit([&](auto output) { result.visit([&](auto output) {
args[0].visit([&](auto input) { args[0].visit([&](auto input) {
if(input.get_shape().packed()) if(input.get_shape().standard())
{ {
std::transform(input.begin(), input.end(), output.begin(), op.fcn()); std::transform(input.begin(), input.end(), output.begin(), op.fcn());
} }
...@@ -811,7 +811,7 @@ struct cpu_binary ...@@ -811,7 +811,7 @@ struct cpu_binary
visit_all(result, args[0], args[1])([&](auto output, auto input1, auto input2) { visit_all(result, args[0], args[1])([&](auto output, auto input1, auto input2) {
auto s1 = input1.get_shape(); auto s1 = input1.get_shape();
auto s2 = input2.get_shape(); auto s2 = input2.get_shape();
if(s1 == s2 and s1.packed()) if(s1 == s2 and s1.standard())
{ {
std::transform( std::transform(
input1.begin(), input1.end(), input2.begin(), output.begin(), op.fcn()); input1.begin(), input1.end(), input2.begin(), output.begin(), op.fcn());
......
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