tree: c9cc8fd0c2a0d8dc63ad01f238b0c892be82cc5e [path history] [tgz]
  1. multiplex_1/
  2. multiplex_2/
  3. multiplex_3/
  4. multiplex_4/
  5. simple_hash_table/
  6. sleep/
  7. README.md
tensorflow/examples/custom_ops_doc/README.md

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