216#define MIM_tensor_NORMALIZER_IMPL \
217 void register_normalizers(Normalizers& normalizers) {\
218 normalizers[flags_t(Annex::Base<get>)] = &normalize_get; \
219 normalizers[flags_t(Annex::Base<set>)] = &normalize_set; \
220 normalizers[flags_t(Annex::Base<map_reduce>)] = &normalize_map_reduce; \
221 normalizers[flags_t(Annex::Base<map_reduce_aff>)] = &normalize_map_reduce_aff; \
222 normalizers[flags_t(Annex::Base<broadcast>)] = &normalize_broadcast; \
223 normalizers[flags_t(Annex::Base<broadcast_in_dim>)] = &normalize_broadcast_in_dim; \
const Def * normalize_broadcast(const Def *, const Def *c, const Def *arg)
const Def * normalize_get(const Def *, const Def *c, const Def *arg)
const Def * normalize_broadcast_in_dim(const Def *, const Def *, const Def *)
const Def * normalize_set(const Def *, const Def *c, const Def *arg)
static constexpr plugin_t Plugin_Id
void register_normalizers(Normalizers &normalizers)
const Def * normalize_map_reduce(const Def *, const Def *, const Def *)
const Def * normalize_map_reduce_aff(const Def *, const Def *, const Def *)
absl::flat_hash_map< flags_t, NormalizeFn > Normalizers
static constexpr size_t Num
Number of Axm::subtags.
static constexpr flags_t Base