| 66 | } |
| 67 | |
| 68 | void MatrixMul::add_input_layout_constraint() { |
| 69 | auto check = [](const TensorLayout& ly) { |
| 70 | return check_layout(ly, 0) || check_layout(ly, 1); |
| 71 | }; |
| 72 | input(0)->add_layout_constraint(check); |
| 73 | input(1)->add_layout_constraint(check); |
| 74 | } |
| 75 | |
| 76 | size_t MatrixMul::get_workspace_size_bytes( |
| 77 | const TensorShapeArray& input_shapes, |
nothing calls this directly
no test coverage detected