| 62 | } |
| 63 | |
| 64 | size_t CondTakeImpl::get_workspace_in_bytes( |
| 65 | const TensorLayout& data, const TensorLayout& mask) { |
| 66 | return make_bundle(data, mask).total_size_in_bytes(); |
| 67 | } |
| 68 | |
| 69 | CondTakeImpl::Output CondTakeImpl::exec( |
| 70 | _megdnn_tensor_in data, _megdnn_tensor_in mask, _megdnn_workspace workspace, |
nothing calls this directly
no test coverage detected