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 ...@@ -496,6 +496,11 @@ struct cpu_pooling
struct cpu_op struct cpu_op
{ {
operation 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(); } std::string name() const { return "cpu::" + op.name(); }
shape compute_shape(const std::vector<shape>& inputs) const { return op.compute_shape(inputs); } 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 argument compute(context&, const shape& output_shape, const std::vector<argument>& args) const
......
...@@ -15,6 +15,12 @@ struct hip_rnn_var_sl_shift_sequence ...@@ -15,6 +15,12 @@ struct hip_rnn_var_sl_shift_sequence
{ {
op::rnn_var_sl_shift_sequence op; 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"; } std::string name() const { return "gpu::rnn_var_sl_shift_sequence"; }
shape compute_shape(std::vector<shape> inputs) const; shape compute_shape(std::vector<shape> inputs) const;
argument 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