Skip to content

Files

Latest commit

 

History

History

custom_ops_doc

Folders and files

NameName
Last commit message
Last commit date

parent directory

..
 
 
 
 
 
 
 
 
 
 
 
 
 
 

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