mlir.dialects.linalg.opdsl.lang.yaml_helper

YAML serialization is routed through here to centralize common logic.

Classes

YAMLObject

An object that can dump itself to a YAML stream

Functions

yaml_dump(data[, sort_keys])
yaml_dump_all(data[, sort_keys, explicit_start])

Module Contents

class mlir.dialects.linalg.opdsl.lang.yaml_helper.YAMLObject

Bases: yaml.YAMLObject

An object that can dump itself to a YAML stream and load itself from a YAML stream.

classmethod to_yaml(dumper, self)

Default to a custom dictionary mapping.

abstract to_yaml_custom_dict()

as_linalg_yaml()

mlir.dialects.linalg.opdsl.lang.yaml_helper.yaml_dump(data, sort_keys=False, **kwargs)

mlir.dialects.linalg.opdsl.lang.yaml_helper.yaml_dump_all(data, sort_keys=False, explicit_start=True, **kwargs)