Skip to content
Nicholas Smith edited this page Jun 24, 2024 · 1 revision

ttnn.add (tt::ttnn::AddOp)

Eltwise add.

Eltwise add operation.

Traits: AttrSizedOperandSegments, Elementwise

Interfaces: DestinationStyleOpInterface

Operands:

Operand Description
inputs variadic of ranked tensor of any type values
outputs variadic of ranked tensor of any type values

Results:

Result Description
results variadic of ranked tensor of any type values

ttnn.alloc (tt::ttnn::AllocOp)

Alloc op.

Tensor Alloc operation

Attributes:

Attribute MLIR Type Description
address ::mlir::IntegerAttr 64-bit signless integer attribute
size ::mlir::IntegerAttr 64-bit signless integer attribute
memory_space ::mlir::tt::MemorySpaceAttr
TT MemorySpace{{% markdown %}}Enum cases: * system (`System`) * mmio (`SystemMMIO`) * dram (`DeviceDRAM`) * l1 (`DeviceL1`){{% /markdown %}}

Results:

Result Description
result ranked tensor of any type values

ttnn.close_device (tt::ttnn::CloseDeviceOp)

Close Device op.

Operands:

Operand Description
device TT device

ttnn.dealloc (tt::ttnn::DeallocOp)

Dealloc op.

Tensor Dealloc operation

Operands:

Operand Description
input ranked tensor of any type values

ttnn.full (tt::ttnn::FullOp)

Full op.

Tensor full operation

Attributes:

Attribute MLIR Type Description
fillValue ::mlir::FloatAttr 32-bit float attribute

Operands:

Operand Description
device TT device

Results:

Result Description
result ranked tensor of any type values

ttnn.multiply (tt::ttnn::MultiplyOp)

Eltwise multiply.

Eltwise multiply operation.

Traits: AttrSizedOperandSegments, Elementwise

Interfaces: DestinationStyleOpInterface

Operands:

Operand Description
inputs variadic of ranked tensor of any type values
outputs variadic of ranked tensor of any type values

Results:

Result Description
results variadic of ranked tensor of any type values

ttnn.open_device (tt::ttnn::OpenDeviceOp)

Open Device op.

Results:

Result Description
device TT device

ttnn.to_memory_config (tt::ttnn::ToMemoryConfigOp)

ToMemoryConfig op.

Interfaces: DestinationStyleOpInterface

Operands:

Operand Description
input ranked tensor of any type values
output ranked tensor of any type values

Results:

Result Description
result ranked tensor of any type values