beanmachine.ppl.compiler.devectorizer_transformer module

class beanmachine.ppl.compiler.devectorizer_transformer.CopyContext: Bases: object

class beanmachine.ppl.compiler.devectorizer_transformer.DevectorizeTransformation(value)

Bases: enum.Enum

An enumeration.

NO = 3

YES = 1

YES_WITH_MERGE = 2

class beanmachine.ppl.compiler.devectorizer_transformer.Devectorizer(cloner: beanmachine.ppl.compiler.copy_and_replace.Cloner, sizer: beanmachine.ppl.compiler.sizer.Sizer)

Bases: beanmachine.ppl.compiler.copy_and_replace.NodeTransformer

assess_node(node: beanmachine.ppl.compiler.bmg_nodes.BMGNode, original: beanmachine.ppl.compiler.bm_graph_builder.BMGraphBuilder) → beanmachine.ppl.compiler.copy_and_replace.TransformAssessment

transform_node(node: beanmachine.ppl.compiler.bmg_nodes.BMGNode, new_inputs: List[beanmachine.ppl.compiler.bmg_nodes.BMGNode]) → Optional[Union[beanmachine.ppl.compiler.bmg_nodes.BMGNode, List[beanmachine.ppl.compiler.bmg_nodes.BMGNode]]]

class beanmachine.ppl.compiler.devectorizer_transformer.ElementType(value)

Bases: enum.Enum

An enumeration.

ANY = 3

SCALAR = 2

TENSOR = 1

beanmachine.ppl.compiler.devectorizer_transformer.vectorized_graph_fixer() → Callable[[beanmachine.ppl.compiler.bm_graph_builder.BMGraphBuilder], Tuple[beanmachine.ppl.compiler.bm_graph_builder.BMGraphBuilder, bool, beanmachine.ppl.compiler.error_report.ErrorReport]]