| 105 | } |
| 106 | |
| 107 | virtual migraphx::shape compute_shape(migraphx::shapes inputs) const override |
| 108 | { |
| 109 | if(not inputs[0].standard() or not inputs[1].standard()) |
| 110 | { |
| 111 | throw std::runtime_error("Input args must be standard shaped"); |
| 112 | } |
| 113 | if(inputs.size() != 2) |
| 114 | { |
| 115 | throw std::runtime_error("number of inputs must be 2"); |
| 116 | } |
| 117 | return inputs.back(); |
| 118 | } |
| 119 | }; |
| 120 | |
| 121 | // overwrites input buffer |