Skip to main content
Ctrl+K
braintaichi - Home braintaichi - Home

Tutorial

  • BrainTaichi Introduction
  • Complete Example

API Documentation

  • Operator Registration Routine
    • XLACustomOp
    • defjvp
    • register_general_batching
  • Sparse Computation Operators
    • coomv
    • csrmv
    • csrmm
  • Event-driven Computation Operators
    • event_csrmv
    • event_csrmm
  • JIT Connectivity Operators
    • jitc_mv_prob_homo
    • jitc_mv_prob_uniform
    • jitc_mv_prob_normal
    • jitc_event_mv_prob_homo
    • jitc_event_mv_prob_uniform
    • jitc_event_mv_prob_normal

Index

B | C | D | E | J | M | R | X

B

  • braintaichi
    • module, [1], [2], [3]

C

  • coomv (class in braintaichi)
  • csrmm (class in braintaichi)
  • csrmv (class in braintaichi)

D

  • def_abstract_eval() (braintaichi.XLACustomOp method)
  • def_batching_rule() (braintaichi.XLACustomOp method)
  • def_jvp_rule() (braintaichi.XLACustomOp method)
  • def_mlir_lowering() (braintaichi.XLACustomOp method)
  • def_transpose_rule() (braintaichi.XLACustomOp method)
  • def_xla_translation() (braintaichi.XLACustomOp method)
  • defjvp (class in braintaichi)
  • defjvp() (braintaichi.XLACustomOp method)

E

  • event_csrmm (class in braintaichi)
  • event_csrmv (class in braintaichi)

J

  • jitc_event_mv_prob_homo (class in braintaichi)
  • jitc_event_mv_prob_normal (class in braintaichi)
  • jitc_event_mv_prob_uniform (class in braintaichi)
  • jitc_mv_prob_homo (class in braintaichi)
  • jitc_mv_prob_normal (class in braintaichi)
  • jitc_mv_prob_uniform (class in braintaichi)

M

  • module
    • braintaichi, [1], [2], [3]

R

  • register_general_batching (class in braintaichi)

X

  • XLACustomOp (class in braintaichi)

By BDP Ecosystem

© Copyright 2024, braintaichi.

Last updated on Dec 13, 2024.