Commit c27eb2ae authored by Brian Pickrell's avatar Brian Pickrell
Browse files

formatting

parent eb6ac6a5
......@@ -30,15 +30,13 @@
* attribute.
*
* Inputs: (1) the shape of the set to be populated.
* (2) randomization seed (uint32). If not given at inference time, the attribute value,
* or auto seeding, will be used.
* Attributes:
* use_auto_seed bool Have hardware generate random seed at runtime, overriding the attribute seed
* seed uint32 Randomization seed
* (2) randomization seed (uint32). If not given at inference time, the attribute
* value, or auto seeding, will be used. Attributes: use_auto_seed bool Have hardware generate
* random seed at runtime, overriding the attribute seed seed uint32 Randomization seed
*
* Output: Same shape.
*
*/
*/
#ifndef MIGRAPHX_GUARD_OPERATORS_MULTINOMIAL_HPP
#define MIGRAPHX_GUARD_OPERATORS_MULTINOMIAL_HPP
......@@ -67,9 +65,8 @@ struct rand_uniform
template <class Self, class F>
static auto reflect(Self& self, F f)
{
return pack(f(self.dtype, "dtype"),
f(self.seed, "seed"),
f(self.use_auto_seed, "use_auto_seed"));
return pack(
f(self.dtype, "dtype"), f(self.seed, "seed"), f(self.use_auto_seed, "use_auto_seed"));
}
std::string name() const { return "rand_uniform"; }
......
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