2. OpBuilders

Warning

1. OpBuilders Python API 当前为 Beta 版本，仅用于适配 OpenSora 模型（提供 codegen pass 功能）;

2. 当前仅提供 binary/unary elementwise 类型和 reduce 类型 op。

Note

Python Op API 语义与 C++ Op API 完全一致，只在接口命名上有区别，算子语义信息请参考 《TopsGraphBuilder C++ API参考》。

2.1. abs

abs(input: builder::Op)

2.2. acos

acos(input: builder::Op)

2.3. add

add(lhs: builder::Op, rhs: builder::Op)

2.4. asin

asin(input: builder::Op)

2.5. asinh

asinh(input: builder::Op)

2.6. atan

atan(input: builder::Op)

2.7. atanh

atanh(input: builder::Op)

2.8. ceil

ceil(input: builder::Op)

2.9. const

const(builder: shared_ptr<builder::Builder>, data: numpy.array)

参数:

• builder: 全局构图句柄

• data: numpy 封装的 array

2.10. convert

convert(lhs: builder::Op, out_type: builder::Type)

2.11. cos

cos(input: builder::Op)

2.12. div

div(lhs: builder::Op, rhs: builder::Op)

2.13. equal

equal(lhs: builder::Op, rhs: builder::Op)

2.14. erf

erf(input: builder::Op)

2.15. exp

exp(input: builder::Op)

2.16. floor

floor(input: builder::Op)

2.17. greater

greater(lhs: builder::Op, rhs: builder::Op)

2.18. greater_equal

greater_equal(lhs: builder::Op, rhs: builder::Op)

2.19. less

less(lhs: builder::Op, rhs: builder::Op)

2.20. less_equal

less_equal(lhs: builder::Op, rhs: builder::Op)

2.21. log

log(input: builder::Op)

2.22. log1p

log1p(input: builder::Op)

2.23. logical_and

logical_and(lhs: builder::Op, rhs: builder::Op)

2.24. logical_not

logical_not(input: builder::Op)

2.25. logical_or

logical_or(lhs: builder::Op, rhs: builder::Op)

2.26. logical_xor

logical_xor(lhs: builder::Op, rhs: builder::Op)

2.27. max

max(lhs: builder::Op, rhs: builder::Op)

2.28. min

min(lhs: builder::Op, rhs: builder::Op)

2.29. mul

mul(lhs: builder::Op, rhs: builder::Op)

2.30. neg

neg(input: builder::Op)

2.31. not_equal

not_equal(lhs: builder::Op, rhs: builder::Op)

2.32. pow

pow(lhs: builder::Op, rhs: builder::Op)

2.33. reduce_max

reduce_max(input: builder::Op, keepdim: bool, dims: list[int])

2.34. reduce_mean

reduce_mean(input: builder::Op, keepdim: bool, dims: list[int])

2.35. reduce_prod

reduce_prod(input: builder::Op, keepdim: bool, dims: list[int])

2.36. reduce_sum

reduce_sum(input: builder::Op, keepdim: bool, dims: list[int])

2.37. relu

relu(input: builder::Op)

2.38. rsqrt

rsqrt(input: builder::Op)

2.39. sign

sign(input: builder::Op)

2.40. silu

silu(input: builder::Op)

2.41. softmax

softmax(input: builder::Op, axis: int[-1], accurate: bool[False], logarithmic: bool[False], eps: float[0.0])

2.42. sqrt

sqrt(input: builder::Op)

2.43. sub

sub(lhs: builder::Op, rhs: builder::Op)

2.44. tan

tan(input: builder::Op)

2.45. tanh

tanh(input: builder::Op)

2.46. where

where(pred: builder::Op, on_true: builder::Op, on_false: builder::Op)