Back to Tensorflow

Custom Ops Examples

tensorflow/examples/custom_ops_doc/README.md

2.21.0466 B
Original Source

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