Renumber the indices of _Arg nodes in a graph, according to 'permutation' that maps old indices to new indices.

Renumber the indices of _Arg nodes in a graph, according to 'permutation' that maps old indices to new indices.

Recomputes rank assignments to make them compatible with the edges (producer has smaller rank than its consumer)

ReplaceDiscClass replace the disc node with prim::CallMethod

Reserve enough random samples in the generator for the given output count.

Returns a fusion plan if success, otherwise none.

Currently, we use a TF function to represent a compilable subgraph. TF function can be arbitrarily nested both using explict way (e.g. a call node) an

Call ParseArgvFromString(..., a) on a string derived from the setting of an environment variable kEnvVar, or a file it points to.

Return in 'sorted' a topological sort of clusters according to the dependencies encoded in ancestors. clusters is the list of all clusters including c

Python wrapper for the Graph Transform Tool. Gives access to all graph transforms available through the command line tool. See docume

The `src` cluster will be cleared if the merge is succeed.

The `src` cluster will be cleared if the merge is succeed.

Helper function to convert an 32-bit integer to a float between [0..1).

Return true if the input mlir graph is valid, return false otherwise.

Returns a mapping from module name to VLOG level, derived from the TF_CPP_VMODULE environment variable; ownership is transferred to the caller.

Maps `n` to the XlaResourceOpKind corresponding to its operation. If `n` is not a resource operation recognized by XLA then sets `out_resource_op_kin

Create a logger with color.

Create a logger with color.

Create a logger with color.

Args: c_in: Input channel dimension c_hidden: Overall hidden channel dimension (not p