Skip to content

Latest commit

 

History

History
10 lines (8 loc) · 467 Bytes

File metadata and controls

10 lines (8 loc) · 467 Bytes

Custom 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