DistRegBuilder.add_groups()#

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

Adds groups to the graph.

Parameters:

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

Return type:

GraphBuilder