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

additional change for the logsoftmax operator.

parent 07214d76
......@@ -632,8 +632,6 @@ struct cpu_logsoftmax
auto lens = output_shape.lens();
std::vector<std::size_t> batch_lens(lens.begin(), lens.begin() + op.axis);
shape batch_shape{migraphx::shape::uint32_type, batch_lens};
// use float for now, need to change later
visit_all(result, args[0])([&](auto output, auto input) {
using value_type = typename decltype(input)::value_type;
std::vector<value_type> batch_max(batch_shape.elements(),
......
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