Go to the source code of this file.
Namespaces | |
| namespace | mim::plug::btensor |
| The btensor Plugin | |
| namespace | mim |
| namespace | mim::plug |
Macros | |
| #define | MIM_btensor_NORMALIZER_IMPL |
Enumerations | |
%btensor.Ring | |
| enum class | mim::plug::btensor::Ring : flags_t |
%btensor.id | |
| enum class | mim::plug::btensor::id : flags_t |
%btensor.map_reduce_post | |
| enum class | mim::plug::btensor::map_reduce_post : flags_t |
%btensor.mk_sched | |
| enum class | mim::plug::btensor::mk_sched : flags_t |
%btensor.proj_map | |
| enum class | mim::plug::btensor::proj_map : flags_t |
%btensor.broadcast | |
| enum class | mim::plug::btensor::broadcast : flags_t |
%btensor.concat | |
| enum class | mim::plug::btensor::concat : flags_t |
%btensor.pad | |
| enum class | mim::plug::btensor::pad : flags_t |
%btensor.transpose_2d | |
| enum class | mim::plug::btensor::transpose_2d : flags_t |
%btensor.product_2d | |
| enum class | mim::plug::btensor::product_2d : flags_t |
%btensor.gather | |
| enum class | mim::plug::btensor::gather : flags_t |
%btensor.scatter | |
| enum class | mim::plug::btensor::scatter : flags_t |
%btensor.NestT | |
| enum class | mim::plug::btensor::NestT : flags_t |
%btensor.mr_nest | |
| enum class | mim::plug::btensor::mr_nest : flags_t |
%btensor.lower_map_reduce | |
| enum class | mim::plug::btensor::lower_map_reduce : flags_t |
Functions | |
| void | mim::plug::btensor::register_normalizers (Normalizers &normalizers) |
Variables | |
| static constexpr plugin_t | mim::plug::btensor::Plugin_Id = 0xd518f5104c00000 |
%btensor.map_reduce | |
| enum class | mim::plug::btensor::map_reduce : flags_t |
| const Def * | mim::plug::btensor::normalize_map_reduce (const Def *, const Def *c, const Def *arg) |
| btensor.map_reduce is btensor.map_reduce_post without the epilogue and with the neutral schedule: delegate with post = btensor.id (so Tp = To), no epilogue inputs (nps = 0), no vector dim (vdim = Rn, the out-of-range sentinel), and no unrolling. | |