wehub-resource-syncCustom Ops Examples
The following subdirectories have examples of Custom Ops.
- multiplex_1: starting example, similar to np.where
- multiplex_2: GPU (and CPU) example, similar to np.where
- multiplex_3: dispatch to special case kernel (and sparse tensors)
- multiplex_4: C++ backward compatibility example: lists of tensors
- simple_hash_table: internal state using a Resource using ref-counting
- sleep: Asynchronous (non-blocking) sleep op using AsyncOpKernel