| 86 | MGB_DEFINE_OPR_CLASS( |
| 87 | WorkspaceAllocTestOpr, cg::SingleCNOutshapePureByInshapeOprBase) // { |
| 88 | void get_output_var_shape( |
| 89 | const TensorShapeArray& inp_shape, |
| 90 | TensorShapeArray& out_shape) const override { |
| 91 | MGB_MARK_USED_VAR(inp_shape); |
| 92 | out_shape.at(0) = {2}; |
| 93 | out_shape.at(1) = {3}; |
| 94 | } |
| 95 | |
| 96 | void scn_do_execute() override { |
| 97 | ASSERT_EQ(TensorShape{2}, output(0)->dev_tensor().shape()); |
no test coverage detected