Commit 0525939c authored by Shucai Xiao's avatar Shucai Xiao
Browse files

rename a structure

parent 8ff6431c
......@@ -28,7 +28,7 @@ struct id
}
};
struct scale
struct mean
{
size_t item_num = 1;
template <class T>
......
......@@ -9,7 +9,7 @@ namespace device {
void reduce_mean(hipStream_t stream, const argument& result, const argument& arg)
{
std::size_t item_num = arg.get_shape().elements() / result.get_shape().elements();
reduce(stream, result, arg, sum{}, 0, id{}, scale{item_num});
reduce(stream, result, arg, sum{}, 0, id{}, mean{item_num});
}
} // namespace device
......
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