Skip to Content

conj_op

View the code on GitHub

Structs

Struct: Conj

Fields

Methods

fwd(arg0: Array) -> Array
more details
Args
  • arg0: Array
Returns
  • Array
jvp(primals: List[Array], tangents: List[Array]) -> Array
more details
Args
  • primals: List[Array]

  • tangents: List[Array]

Returns
  • Array
vjp(primals: List[Array], grad: Array, out: Array) -> List[Array]
more details
Args
  • primals: List[Array]

  • grad: Array

  • out: Array

Returns
  • List[Array]
unary_simd_op(arg0_real: SIMD[float32, nelts[::DType]().__mul__(2).__floordiv__(2)], arg0_imag: SIMD[float32, nelts[::DType]().__mul__(2).__floordiv__(2)]) -> Tuple[SIMD[float32, nelts[::DType]().__mul__(2).__floordiv__(2)], SIMD[float32, nelts[::DType]().__mul__(2).__floordiv__(2)]]
more details
Args
  • arg0_real: SIMD[float32, nelts[::DType]().__mul__(2).__floordiv__(2)]

  • arg0_imag: SIMD[float32, nelts[::DType]().__mul__(2).__floordiv__(2)]

Returns
  • Tuple[SIMD[float32, nelts[::DType]().__mul__(2).__floordiv__(2)], SIMD[float32, nelts[::DType]().__mul__(2).__floordiv__(2)]]
__call__(mut curr: Array, args: List[Array])
more details
Args
  • curr: Array

  • args: List[Array]

Functions

conj

conj(arg0: Array) -> Array
more details
Args
  • arg0: Array
Returns
  • Array
Last updated on