For the complete documentation index, see llms.txt. Markdown versions of all pages are available by appending .md to any URL (e.g. /max/get-started.md).
Mojo struct
Learnable2DInterpPosEmb
struct Learnable2DInterpPosEmb
Registers the learnable_2d_interp_pos_emb graph op with the graph compiler.
Implemented traits
Methods
execute
static def execute[dtype: DType, //, target: StringSlice[ImmStaticOrigin]](output: ManagedTensorSlice[IOSpec[_, _].Output, static_spec=output.static_spec], x: ManagedTensorSlice[IOSpec[_, _].Input, static_spec=x.static_spec], weight: ManagedTensorSlice[IOSpec[_, _].Input, static_spec=weight.static_spec], grid_thws: ManagedTensorSlice[IOSpec[_, _].Input, static_spec=grid_thws.static_spec], time_weight: ManagedTensorSlice[IOSpec[_, _].Input, static_spec=time_weight.static_spec], ctx: DeviceContext)
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!