Commit 0e9eb89f authored by Shucai Xiao's avatar Shucai Xiao
Browse files

clang format

parent 4231388a
...@@ -9,7 +9,7 @@ namespace device { ...@@ -9,7 +9,7 @@ namespace device {
void reduce_mean(hipStream_t stream, const argument& result, const argument& arg) void reduce_mean(hipStream_t stream, const argument& result, const argument& arg)
{ {
std::size_t batch_item_num = arg.get_shape().elements() / result.get_shape().elements(); std::size_t batch_item_num = arg.get_shape().elements() / result.get_shape().elements();
float factor = 1.0f / batch_item_num; float factor = 1.0f / batch_item_num;
reduce(stream, result, arg, sum{}, 0, id{}, scale{factor}); reduce(stream, result, arg, sum{}, 0, id{}, scale{factor});
} }
......
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