Unverified Commit c02c4dc3 authored by kahmed10's avatar kahmed10 Committed by GitHub
Browse files

Additional reflect functions (#585)



* add reflect method

* add reflect to cpu_op
Co-authored-by: default avatarmvermeulen <5479696+mvermeulen@users.noreply.github.com>
parent 9169fbb3
......@@ -496,6 +496,11 @@ struct cpu_pooling
struct cpu_op
{
operation op;
template <class Self, class F>
static auto reflect(Self& self, F f)
{
return migraphx::reflect(self.op, f);
}
std::string name() const { return "cpu::" + op.name(); }
shape compute_shape(const std::vector<shape>& inputs) const { return op.compute_shape(inputs); }
argument compute(context&, const shape& output_shape, const std::vector<argument>& args) const
......
......@@ -15,6 +15,12 @@ struct hip_rnn_var_sl_shift_sequence
{
op::rnn_var_sl_shift_sequence op;
template <class Self, class F>
static auto reflect(Self& self, F f)
{
return migraphx::reflect(self.op, f);
}
std::string name() const { return "gpu::rnn_var_sl_shift_sequence"; }
shape compute_shape(std::vector<shape> inputs) const;
argument
......
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