mirror of https://github.com/hpcaitech/ColossalAI
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
Zirui Zhu
1c29b173c9
|
2 years ago | |
---|---|---|
.. | ||
experimental | ||
strategy | 2 years ago | |
__init__.py | 2 years ago | |
addmm_handler.py | ||
batch_norm_handler.py | 2 years ago | |
binary_elementwise_handler.py | 2 years ago | |
bmm_handler.py | ||
conv_handler.py | 2 years ago | |
embedding_handler.py | ||
getattr_handler.py | 2 years ago | |
getitem_handler.py | 2 years ago | |
layer_norm_handler.py | ||
linear_handler.py | 2 years ago | |
matmul_handler.py | ||
node_handler.py | 2 years ago | |
normal_pooling_handler.py | 2 years ago | |
output_handler.py | 2 years ago | |
placeholder_handler.py | 2 years ago | |
registry.py | ||
reshape_handler.py | 2 years ago | |
softmax_handler.py | ||
sum_handler.py | ||
tensor_constructor_handler.py | ||
unary_elementwise_handler.py | 2 years ago | |
where_handler.py |