Skip to main content
Back to top
Ctrl
+
K
Search
Ctrl
+
K
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
.rst
.pdf
register_general_batching
Contents
register_general_batching
register_general_batching
#
class
braintaichi.
register_general_batching
(
prim
)
[source]
#
Contents
register_general_batching
