runner.TightPadding

December 14, 2023 · View on GitHub

Calculates tight SizeConstraints for GraphTensor padding.

Inherits From: GraphTensorPadding

runner.TightPadding(
    gtspec: tfgnn.GraphTensorSpec,
    dataset_provider: runner.DatasetProvider,
    min_nodes_per_component: Optional[Mapping[str, int]] = None
)

See: tfgnn.find_tight_size_constraints.

Methods

`get_filter_fn`

View source

get_filter_fn(
    size_constraints: SizeConstraints
) -> Callable[..., bool]

`get_size_constraints`

View source

get_size_constraints(
    target_batch_size: int
) -> SizeConstraints