DistRegBuilder.add_groups()

DistRegBuilder.add_groups(*groups, to_float32=True)

Adds groups to the graph.

Parameters:

• *groups (Group) – The groups to add to the graph.

• to_float32 (bool) – Whether to convert the dtype of the values of the added nodes from float64 to float32. (default: True)

Return type:

GraphBuilder

Returns:

The graph builder.