"ts/webui/src/components/common/LogPathChild.tsx" did not exist on "f9580cd140c6caa9426fb58e3dff93e81301b0c4"
Commit d4613133 authored by Paul's avatar Paul
Browse files

Fix mean

parent 616cbd16
...@@ -227,7 +227,7 @@ struct reduce_op ...@@ -227,7 +227,7 @@ struct reduce_op
auto reduce_elements = s.elements() / ins->get_shape().elements(); auto reduce_elements = s.elements() / ins->get_shape().elements();
auto reduce_type = s.type(); auto reduce_type = s.type();
r.reduction = "op::sum{}"; r.reduction = "op::sum{}";
std::string mean = "op::mean{" + std::to_string(reduce_elements) + "}"; std::string mean = "op::mean<" + std::to_string(reduce_elements) + ">{}";
// Use float accumulator when reduction size is too large for half // Use float accumulator when reduction size is too large for half
if(reduce_type == shape::half_type and reduce_elements > 16384) if(reduce_type == shape::half_type and reduce_elements > 16384)
r.read = "compose(" + mean + ", op::convert_to<float>{})"; r.read = "compose(" + mean + ", op::convert_to<float>{})";
......
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