https://www.tensorflow.org/probability/api_docs/python/tfp/experimental/auto_batching/instructions/PrimOp
tfp.experimental.auto_batching.instructions.PrimOp | TensorFlow Probability
An arbitrary already-batched computation, a 'primitive operation'.
tfpexperimentalautobatchinginstructions