irx.builder.lowering.tensor

irx.builder.lowering.tensor

Source: packages/irx/src/irx/builder/lowering/tensor.py

title: Tensor visitor mixin for llvmliteir.

Classes

TensorVisitorMixin(VisitorMixinBase)

title: Tensor visitor mixin.

Methods

visit(self, node: astx.TensorLiteral) -> None

title: Visit TensorLiteral nodes.
parameters:
  node:
    type: astx.TensorLiteral

visit(self, node: astx.TensorView) -> None

title: Visit TensorView nodes.
parameters:
  node:
    type: astx.TensorView

visit(self, node: astx.TensorIndex) -> None

title: Visit TensorIndex nodes.
parameters:
  node:
    type: astx.TensorIndex

visit(self, node: astx.TensorStore) -> None

title: Visit TensorStore nodes.
parameters:
  node:
    type: astx.TensorStore

visit(self, node: astx.TensorNDim) -> None

title: Visit TensorNDim nodes.
parameters:
  node:
    type: astx.TensorNDim

visit(self, node: astx.TensorShape) -> None

title: Visit TensorShape nodes.
parameters:
  node:
    type: astx.TensorShape

visit(self, node: astx.TensorStride) -> None

title: Visit TensorStride nodes.
parameters:
  node:
    type: astx.TensorStride

visit(self, node: astx.TensorElementCount) -> None

title: Visit TensorElementCount nodes.
parameters:
  node:
    type: astx.TensorElementCount

visit(self, node: astx.TensorByteOffset) -> None

title: Visit TensorByteOffset nodes.
parameters:
  node:
    type: astx.TensorByteOffset

visit(self, node: astx.TensorRetain) -> None

title: Visit TensorRetain nodes.
parameters:
  node:
    type: astx.TensorRetain

visit(self, node: astx.TensorRelease) -> None

title: Visit TensorRelease nodes.
parameters:
  node:
    type: astx.TensorRelease