DistRegBuilder.add_groups()

DistRegBuilder.add_groups()#

DistRegBuilder.add_groups(*groups, to_float32=None)#

Adds groups to the graph.

Parameters:

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

  • to_float32 (bool | None) – Whether to convert the dtype of the values of the added nodes from float64 to float32. If None (default), the GraphBuilder’s attribute GraphBuilder.to_float32, which is set during initialization will be used instead. (default: None)

Return type:

GraphBuilder

Returns:

The graph builder.