Announcing the TensorFlow Dev Summit 2020 Learn more

Module: tfp.experimental.auto_batching.dsl

View source on GitHub

Python-embedded DSL frontend for authoring autobatchable IR programs.

Aliases: tfp.experimental.auto_batching.frontend.dsl

This domain-specific language frontend serves two purposes:

  • Being easier and more pleasant to author by humans than writing IR directly
  • Being close enough to the structure of natural Python programs that the remaining gap can be bridged by a source-to-source transformation

Classes

class ProgramBuilder: An auto-batching DSL context.