tvm_tirx.tirx.tile.views.Sqrt

class tvm_tirx.tirx.tile.views.Sqrt

Bases: UnaryOp

Compute sqrt(src) and store to dst.

kind = 'sqrt'

get_private_tensors_trn(tensor_dict, sctx)

Parameters:

• op (TensorCall)

• tensor_dict (dict[Any, tuple[Bind, Stmt | None]])

• sctx (DispatchContext)

Return type:

dict[str, Any]