348#define MIM_tensor_NORMALIZER_IMPL \
349 void register_normalizers(Normalizers& normalizers) {\
350 normalizers[flags_t(Annex::Base<get>)] = &normalize_get; \
351 normalizers[flags_t(Annex::Base<set>)] = &normalize_set; \
352 normalizers[flags_t(Annex::Base<shape>)] = &normalize_shape; \
353 normalizers[flags_t(Annex::Base<map_reduce>)] = &normalize_map_reduce; \
354 normalizers[flags_t(Annex::Base<repeat>)] = &normalize_repeat; \
355 normalizers[flags_t(Annex::Base<reshape>)] = &normalize_reshape; \
356 normalizers[flags_t(Annex::Base<slice>)] = &normalize_slice; \
357 normalizers[flags_t(Annex::Base<flip>)] = &normalize_flip; \
358 normalizers[flags_t(Annex::Base<pad>)] = &normalize_pad; \
359 normalizers[flags_t(Annex::Base<concat>)] = &normalize_concat; \
360 normalizers[flags_t(Annex::Base<broadcast>)] = &normalize_broadcast; \
361 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_slice(const Def *, const Def *c, const Def *arg)
const Def * normalize_repeat(const Def *, const Def *c, const Def *arg)
const Def * normalize_pad(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)
const Def * normalize_concat(const Def *, const Def *, const Def *)
const Def * normalize_reshape(const Def *, const Def *c, const Def *arg)
const Def * normalize_shape(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_flip(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