TightPadding.md

runner.TightPadding

View source 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