| 51 | } |
| 52 | |
| 53 | size_t TopKImpl::get_workspace_in_bytes( |
| 54 | int k, const TensorLayout& data, const TensorLayout& values, |
| 55 | const TensorLayout& indices) { |
| 56 | MEGDNN_MARK_USED_VAR(values); |
| 57 | MEGDNN_MARK_USED_VAR(indices); |
| 58 | return make_bundle(k, data, values, indices).total_size_in_bytes(); |
| 59 | } |
| 60 | |
| 61 | template <typename ctype> |
| 62 | void TopKImpl::dispatch_with_ctype( |
nothing calls this directly
no test coverage detected