| 440 | } |
| 441 | |
| 442 | std::vector<Device> CustomOp::infer_output_device( |
| 443 | const std::vector<Device>& inputs, const Param& param) const { |
| 444 | assert_inputs_size_right(inputs); |
| 445 | |
| 446 | std::vector<Device> outputs(output_num()); |
| 447 | OpImplRef(m_impl.get())->infer_output_device_func(inputs, param, outputs); |
| 448 | |
| 449 | assert_outputs_size_right(outputs); |
| 450 | return outputs; |
| 451 | } |
| 452 | |
| 453 | std::vector<Shape> CustomOp::infer_output_shape( |
| 454 | const std::vector<Shape>& inputs, const Param& param) const { |