| 54 | } |
| 55 | |
| 56 | void IndexingOneHotForward::check_exec( |
| 57 | const TensorLayout& src, const TensorLayout& index, const TensorLayout& dst, |
| 58 | size_t workspace_in_bytes) { |
| 59 | check_layout_fwd(src, index, dst); |
| 60 | auto required_workspace_in_bytes = get_workspace_in_bytes(src, index, dst); |
| 61 | megdnn_assert(workspace_in_bytes >= required_workspace_in_bytes); |
| 62 | } |
| 63 | |
| 64 | void IndexingSetOneHotForward::check_exec( |
| 65 | const TensorLayout& data, const TensorLayout& index, const TensorLayout& sub, |
nothing calls this directly
no test coverage detected