Skip to content

GitLab

Commit 61f3895c authored by Brian Pickrell's avatar Brian Pickrell
Browse files

Merge branch 'multinomial_parse' of... 

Merge branch 'multinomial_parse' of github.com:ROCmSoftwarePlatform/AMDMIGraphX into multinomial_parse
parents 38920ca7 da4fa01f
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 0 deletions
src/include/migraphx/op/multinomial.hpp
View file @ 61f3895c
...@@ -54,7 +54,11 @@ struct multinomial ...@@ -54,7 +54,11 @@ struct multinomial
MIGRAPHX_THROW( MIGRAPHX_THROW(
"Multinomial: Invalid output type. Valid types are int32_type and int64_type."); "Multinomial: Invalid output type. Valid types are int32_type and int64_type.");
<<<<<<< HEAD
return inputs.front().normalize_standard(); return inputs.front().normalize_standard();
=======
return {dtype, {inputs.front().max_lens().front(), sample_size}};
>>>>>>> da4fa01ff583c14d23b9e10c5fc178a4e3fe3bc2
} }
argument compute(const dyn_output& dyn_out, std::vector<argument> args) const argument compute(const dyn_output& dyn_out, std::vector<argument> args) const
......
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