Commit 2220bd25 authored by Paul's avatar Paul
Browse files

Formatting

parent c9e391fe
...@@ -50,7 +50,7 @@ argument softmax(hipStream_t stream, argument result, argument arg, int axis) ...@@ -50,7 +50,7 @@ argument softmax(hipStream_t stream, argument result, argument arg, int axis)
for(std::size_t j = 0; j < n_dims; ++j) for(std::size_t j = 0; j < n_dims; ++j)
{ {
data_idx[axis] = j; data_idx[axis] = j;
output[data_idx] = output[data_idx] / batch_sum; output[data_idx] = output[data_idx] / batch_sum;
} }
}); });
......
...@@ -37,9 +37,7 @@ shape hip_softmax::compute_shape(const std::vector<shape>& inputs) const ...@@ -37,9 +37,7 @@ shape hip_softmax::compute_shape(const std::vector<shape>& inputs) const
return op.compute_shape({inputs.at(0)}); return op.compute_shape({inputs.at(0)});
} }
argument hip_softmax::compute(context& ctx, argument hip_softmax::compute(context& ctx, const shape&, const std::vector<argument>& args) const
const shape&,
const std::vector<argument>& args) const
{ {
return device::softmax(ctx.get_stream().get(), args[1], args[0], op.axis); return device::softmax(ctx.get_stream().get(), args[1], args[0], op.axis);
} }
......
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