#include <mim/plug/affine/affine.h>#include <mim/plug/core/core.h>#include <mim/plug/direct/direct.h>#include <mim/plug/tuple/tuple.h>#include <mim/plug/vec/vec.h>#include "mim/def.h"#include "mim/plugin.h"#include "mim/world.h"#include "mim/util/sets.h"#include "mim/plug/tensor/tensor.h"Go to the source code of this file.
Namespaces | |
| namespace | mim |
| namespace | mim::plug |
| namespace | mim::plug::tensor |
| The tensor Plugin | |
Functions | |
| std::tuple< u64, const Def *, const Def * > | mim::plug::tensor::fold_shape_and_index (const Def *shape, const Def *index) |
%tensor.get | |
| const Def * | mim::plug::tensor::normalize_get (const Def *, const Def *c, const Def *arg) |
%tensor.set | |
| const Def * | mim::plug::tensor::normalize_set (const Def *, const Def *c, const Def *arg) |
%tensor.broadcast | |
| const Def * | mim::plug::tensor::normalize_broadcast (const Def *, const Def *c, const Def *arg) |
%tensor.broadcast_in_dim | |
| const Def * | mim::plug::tensor::normalize_broadcast_in_dim (const Def *, const Def *, const Def *) |
%tensor.map_reduce | |
| const Def * | mim::plug::tensor::normalize_map_reduce (const Def *, const Def *, const Def *) |
%tensor.map_reduce_aff | |
| const Def * | mim::plug::tensor::normalize_map_reduce_aff (const Def *, const Def *, const Def *) |