Mojo trait

MHAPartitionScheme

Implemented traits

AnyType, UnknownDestructibility

Aliases

accum_dtype

alias accum_dtype

do_partition

alias do_partition

Methods

num_partitions

num_partitions(self: _Self) -> SIMD[uint32, 1]

Returns:

SIMD

get_exp_sum_qk_max_pointer

get_exp_sum_qk_max_pointer(self: _Self) -> UnsafePointer[SIMD[get_witness(_Self, nn::mha_utils::MHAPartitionScheme, accum_dtype), 1]]

Returns:

UnsafePointer

Was this page helpful?

Thank you! We'll create more content like this.

Thank you for helping us improve!

😔 What went wrong?

Some code doesn’t work

It includes inaccurate information

It's missing information I need

It was difficult to understand

Other

Submit