1189 lines
55 KiB
C++
1189 lines
55 KiB
C++
// The profile-based compliance content below is auto-generated by the script
|
|
// `tools/genspec.py` in https://github.com/arm/tosa-specification
|
|
profileComplianceMap = {
|
|
{"tosa.argmax",
|
|
{{{Profile::pro_int}, {{{i8T, i32T}, SpecificationVersion::V_1_0}}},
|
|
{{Profile::pro_fp},
|
|
{{{fp16T, i32T}, SpecificationVersion::V_1_0},
|
|
{{fp32T, i32T}, SpecificationVersion::V_1_0}}}}},
|
|
{"tosa.avg_pool2d",
|
|
{{{Profile::pro_int},
|
|
{{{i8T, i8T, i8T, i32T, i8T}, SpecificationVersion::V_1_0}}},
|
|
{{Profile::pro_fp},
|
|
{{{fp16T, fp16T, fp16T, fp16T, fp16T}, SpecificationVersion::V_1_0},
|
|
{{fp16T, fp16T, fp16T, fp32T, fp16T}, SpecificationVersion::V_1_0},
|
|
{{fp32T, fp32T, fp32T, fp32T, fp32T}, SpecificationVersion::V_1_0}}}}},
|
|
{"tosa.avg_pool2d_adaptive",
|
|
{{{Profile::pro_int},
|
|
{{{i8T, i8T, i8T, i32T, i8T}, SpecificationVersion::V_1_1_DRAFT}}},
|
|
{{Profile::pro_fp},
|
|
{{{fp16T, fp16T, fp16T, fp16T, fp16T},
|
|
SpecificationVersion::V_1_1_DRAFT},
|
|
{{fp16T, fp16T, fp16T, fp32T, fp16T},
|
|
SpecificationVersion::V_1_1_DRAFT},
|
|
{{fp32T, fp32T, fp32T, fp32T, fp32T},
|
|
SpecificationVersion::V_1_1_DRAFT}}}}},
|
|
{"tosa.conv2d",
|
|
{{{Profile::pro_int},
|
|
{{{i8T, i8T, i32T, i8T, i8T, i32T, i32T}, SpecificationVersion::V_1_0}}},
|
|
{{Profile::pro_fp},
|
|
{{{fp16T, fp16T, fp16T, fp16T, fp16T, fp16T, fp16T},
|
|
SpecificationVersion::V_1_0},
|
|
{{fp16T, fp16T, fp16T, fp16T, fp16T, fp32T, fp16T},
|
|
SpecificationVersion::V_1_0},
|
|
{{fp32T, fp32T, fp32T, fp32T, fp32T, fp32T, fp32T},
|
|
SpecificationVersion::V_1_0}}}}},
|
|
{"tosa.conv3d",
|
|
{{{Profile::pro_int},
|
|
{{{i8T, i8T, i32T, i8T, i8T, i32T, i32T}, SpecificationVersion::V_1_0}}},
|
|
{{Profile::pro_fp},
|
|
{{{fp16T, fp16T, fp16T, fp16T, fp16T, fp16T, fp16T},
|
|
SpecificationVersion::V_1_0},
|
|
{{fp16T, fp16T, fp16T, fp16T, fp16T, fp32T, fp16T},
|
|
SpecificationVersion::V_1_0},
|
|
{{fp32T, fp32T, fp32T, fp32T, fp32T, fp32T, fp32T},
|
|
SpecificationVersion::V_1_0}}}}},
|
|
{"tosa.depthwise_conv2d",
|
|
{{{Profile::pro_int},
|
|
{{{i8T, i8T, i32T, i8T, i8T, i32T, i32T}, SpecificationVersion::V_1_0}}},
|
|
{{Profile::pro_fp},
|
|
{{{fp16T, fp16T, fp16T, fp16T, fp16T, fp16T, fp16T},
|
|
SpecificationVersion::V_1_0},
|
|
{{fp16T, fp16T, fp16T, fp16T, fp16T, fp32T, fp16T},
|
|
SpecificationVersion::V_1_0},
|
|
{{fp32T, fp32T, fp32T, fp32T, fp32T, fp32T, fp32T},
|
|
SpecificationVersion::V_1_0}}}}},
|
|
{"tosa.matmul",
|
|
{{{Profile::pro_int},
|
|
{{{i8T, i8T, i8T, i8T, i32T}, SpecificationVersion::V_1_0}}},
|
|
{{Profile::pro_fp},
|
|
{{{fp16T, fp16T, fp16T, fp16T, fp16T}, SpecificationVersion::V_1_0},
|
|
{{fp16T, fp16T, fp16T, fp16T, fp32T}, SpecificationVersion::V_1_0},
|
|
{{fp32T, fp32T, fp32T, fp32T, fp32T}, SpecificationVersion::V_1_0}}}}},
|
|
{"tosa.max_pool2d",
|
|
{{{Profile::pro_int}, {{{i8T, i8T}, SpecificationVersion::V_1_0}}},
|
|
{{Profile::pro_fp},
|
|
{{{fp16T, fp16T}, SpecificationVersion::V_1_0},
|
|
{{fp32T, fp32T}, SpecificationVersion::V_1_0}}}}},
|
|
{"tosa.max_pool2d_adaptive",
|
|
{{{Profile::pro_int}, {{{i8T, i8T}, SpecificationVersion::V_1_1_DRAFT}}},
|
|
{{Profile::pro_fp},
|
|
{{{fp16T, fp16T}, SpecificationVersion::V_1_1_DRAFT},
|
|
{{fp32T, fp32T}, SpecificationVersion::V_1_1_DRAFT}}}}},
|
|
{"tosa.transpose_conv2d",
|
|
{{{Profile::pro_int},
|
|
{{{i8T, i8T, i32T, i8T, i8T, i32T, i32T}, SpecificationVersion::V_1_0}}},
|
|
{{Profile::pro_fp},
|
|
{{{fp16T, fp16T, fp16T, fp16T, fp16T, fp16T, fp16T},
|
|
SpecificationVersion::V_1_0},
|
|
{{fp16T, fp16T, fp16T, fp16T, fp16T, fp32T, fp16T},
|
|
SpecificationVersion::V_1_0},
|
|
{{fp32T, fp32T, fp32T, fp32T, fp32T, fp32T, fp32T},
|
|
SpecificationVersion::V_1_0}}}}},
|
|
{"tosa.clamp",
|
|
{{{Profile::pro_int}, {{{i8T, i8T}, SpecificationVersion::V_1_0}}},
|
|
{{Profile::pro_fp},
|
|
{{{fp16T, fp16T}, SpecificationVersion::V_1_0},
|
|
{{fp32T, fp32T}, SpecificationVersion::V_1_0}}}}},
|
|
{"tosa.erf",
|
|
{{{Profile::pro_fp},
|
|
{{{fp16T, fp16T}, SpecificationVersion::V_1_0},
|
|
{{fp32T, fp32T}, SpecificationVersion::V_1_0}}}}},
|
|
{"tosa.sigmoid",
|
|
{{{Profile::pro_fp},
|
|
{{{fp16T, fp16T}, SpecificationVersion::V_1_0},
|
|
{{fp32T, fp32T}, SpecificationVersion::V_1_0}}}}},
|
|
{"tosa.tanh",
|
|
{{{Profile::pro_fp},
|
|
{{{fp16T, fp16T}, SpecificationVersion::V_1_0},
|
|
{{fp32T, fp32T}, SpecificationVersion::V_1_0}}}}},
|
|
{"tosa.add",
|
|
{{{Profile::pro_int, Profile::pro_fp},
|
|
{{{i32T, i32T, i32T}, SpecificationVersion::V_1_0}},
|
|
anyOf},
|
|
{{Profile::pro_fp},
|
|
{{{fp16T, fp16T, fp16T}, SpecificationVersion::V_1_0},
|
|
{{fp32T, fp32T, fp32T}, SpecificationVersion::V_1_0}}}}},
|
|
{"tosa.arithmetic_right_shift",
|
|
{{{Profile::pro_int},
|
|
{{{i8T, i8T, i8T}, SpecificationVersion::V_1_0},
|
|
{{i16T, i16T, i16T}, SpecificationVersion::V_1_0},
|
|
{{i32T, i32T, i32T}, SpecificationVersion::V_1_0}}}}},
|
|
{"tosa.bitwise_and",
|
|
{{{Profile::pro_int},
|
|
{{{i8T, i8T, i8T}, SpecificationVersion::V_1_0},
|
|
{{i16T, i16T, i16T}, SpecificationVersion::V_1_0},
|
|
{{i32T, i32T, i32T}, SpecificationVersion::V_1_0}}}}},
|
|
{"tosa.bitwise_or",
|
|
{{{Profile::pro_int},
|
|
{{{i8T, i8T, i8T}, SpecificationVersion::V_1_0},
|
|
{{i16T, i16T, i16T}, SpecificationVersion::V_1_0},
|
|
{{i32T, i32T, i32T}, SpecificationVersion::V_1_0}}}}},
|
|
{"tosa.bitwise_xor",
|
|
{{{Profile::pro_int},
|
|
{{{i8T, i8T, i8T}, SpecificationVersion::V_1_0},
|
|
{{i16T, i16T, i16T}, SpecificationVersion::V_1_0},
|
|
{{i32T, i32T, i32T}, SpecificationVersion::V_1_0}}}}},
|
|
{"tosa.intdiv",
|
|
{{{Profile::pro_int, Profile::pro_fp},
|
|
{{{i32T, i32T, i32T}, SpecificationVersion::V_1_0}},
|
|
anyOf}}},
|
|
{"tosa.logical_and",
|
|
{{{Profile::pro_int, Profile::pro_fp},
|
|
{{{boolT, boolT, boolT}, SpecificationVersion::V_1_0}},
|
|
anyOf}}},
|
|
{"tosa.logical_left_shift",
|
|
{{{Profile::pro_int, Profile::pro_fp},
|
|
{{{i8T, i8T, i8T}, SpecificationVersion::V_1_0},
|
|
{{i16T, i16T, i16T}, SpecificationVersion::V_1_0},
|
|
{{i32T, i32T, i32T}, SpecificationVersion::V_1_0}},
|
|
anyOf}}},
|
|
{"tosa.logical_right_shift",
|
|
{{{Profile::pro_int, Profile::pro_fp},
|
|
{{{i8T, i8T, i8T}, SpecificationVersion::V_1_0},
|
|
{{i16T, i16T, i16T}, SpecificationVersion::V_1_0},
|
|
{{i32T, i32T, i32T}, SpecificationVersion::V_1_0}},
|
|
anyOf}}},
|
|
{"tosa.logical_or",
|
|
{{{Profile::pro_int, Profile::pro_fp},
|
|
{{{boolT, boolT, boolT}, SpecificationVersion::V_1_0}},
|
|
anyOf}}},
|
|
{"tosa.logical_xor",
|
|
{{{Profile::pro_int, Profile::pro_fp},
|
|
{{{boolT, boolT, boolT}, SpecificationVersion::V_1_0}},
|
|
anyOf}}},
|
|
{"tosa.maximum",
|
|
{{{Profile::pro_int}, {{{i32T, i32T, i32T}, SpecificationVersion::V_1_0}}},
|
|
{{Profile::pro_fp},
|
|
{{{fp16T, fp16T, fp16T}, SpecificationVersion::V_1_0},
|
|
{{fp32T, fp32T, fp32T}, SpecificationVersion::V_1_0}}}}},
|
|
{"tosa.minimum",
|
|
{{{Profile::pro_int}, {{{i32T, i32T, i32T}, SpecificationVersion::V_1_0}}},
|
|
{{Profile::pro_fp},
|
|
{{{fp16T, fp16T, fp16T}, SpecificationVersion::V_1_0},
|
|
{{fp32T, fp32T, fp32T}, SpecificationVersion::V_1_0}}}}},
|
|
{"tosa.mul",
|
|
{{{Profile::pro_int},
|
|
{{{i8T, i8T, i32T}, SpecificationVersion::V_1_0},
|
|
{{i16T, i16T, i32T}, SpecificationVersion::V_1_0}}},
|
|
{{Profile::pro_int, Profile::pro_fp},
|
|
{{{i32T, i32T, i32T}, SpecificationVersion::V_1_0}},
|
|
anyOf},
|
|
{{Profile::pro_fp},
|
|
{{{fp16T, fp16T, fp16T}, SpecificationVersion::V_1_0},
|
|
{{fp32T, fp32T, fp32T}, SpecificationVersion::V_1_0}}}}},
|
|
{"tosa.pow",
|
|
{{{Profile::pro_fp},
|
|
{{{fp16T, fp16T, fp16T}, SpecificationVersion::V_1_0},
|
|
{{fp32T, fp32T, fp32T}, SpecificationVersion::V_1_0}}}}},
|
|
{"tosa.sub",
|
|
{{{Profile::pro_int, Profile::pro_fp},
|
|
{{{i32T, i32T, i32T}, SpecificationVersion::V_1_0}},
|
|
anyOf},
|
|
{{Profile::pro_fp},
|
|
{{{fp16T, fp16T, fp16T}, SpecificationVersion::V_1_0},
|
|
{{fp32T, fp32T, fp32T}, SpecificationVersion::V_1_0}}}}},
|
|
{"tosa.table",
|
|
{{{Profile::pro_int}, {{{i8T, i8T, i8T}, SpecificationVersion::V_1_0}}}}},
|
|
{"tosa.abs",
|
|
{{{Profile::pro_int}, {{{i32T, i32T}, SpecificationVersion::V_1_0}}},
|
|
{{Profile::pro_fp},
|
|
{{{fp16T, fp16T}, SpecificationVersion::V_1_0},
|
|
{{fp32T, fp32T}, SpecificationVersion::V_1_0}}}}},
|
|
{"tosa.bitwise_not",
|
|
{{{Profile::pro_int},
|
|
{{{i8T, i8T}, SpecificationVersion::V_1_0},
|
|
{{i16T, i16T}, SpecificationVersion::V_1_0},
|
|
{{i32T, i32T}, SpecificationVersion::V_1_0}}}}},
|
|
{"tosa.ceil",
|
|
{{{Profile::pro_fp},
|
|
{{{fp16T, fp16T}, SpecificationVersion::V_1_0},
|
|
{{fp32T, fp32T}, SpecificationVersion::V_1_0}}}}},
|
|
{"tosa.clz",
|
|
{{{Profile::pro_int}, {{{i32T, i32T}, SpecificationVersion::V_1_0}}}}},
|
|
{"tosa.cos",
|
|
{{{Profile::pro_fp},
|
|
{{{fp16T, fp16T}, SpecificationVersion::V_1_0},
|
|
{{fp32T, fp32T}, SpecificationVersion::V_1_0}}}}},
|
|
{"tosa.exp",
|
|
{{{Profile::pro_fp},
|
|
{{{fp16T, fp16T}, SpecificationVersion::V_1_0},
|
|
{{fp32T, fp32T}, SpecificationVersion::V_1_0}}}}},
|
|
{"tosa.floor",
|
|
{{{Profile::pro_fp},
|
|
{{{fp16T, fp16T}, SpecificationVersion::V_1_0},
|
|
{{fp32T, fp32T}, SpecificationVersion::V_1_0}}}}},
|
|
{"tosa.log",
|
|
{{{Profile::pro_fp},
|
|
{{{fp16T, fp16T}, SpecificationVersion::V_1_0},
|
|
{{fp32T, fp32T}, SpecificationVersion::V_1_0}}}}},
|
|
{"tosa.logical_not",
|
|
{{{Profile::pro_int, Profile::pro_fp},
|
|
{{{boolT, boolT}, SpecificationVersion::V_1_0}},
|
|
anyOf}}},
|
|
{"tosa.negate",
|
|
{{{Profile::pro_int},
|
|
{{{i8T, i8T, i8T, i8T}, SpecificationVersion::V_1_0},
|
|
{{i16T, i16T, i16T, i16T}, SpecificationVersion::V_1_0},
|
|
{{i32T, i32T, i32T, i32T}, SpecificationVersion::V_1_0}}},
|
|
{{Profile::pro_fp},
|
|
{{{fp16T, fp16T, fp16T, fp16T}, SpecificationVersion::V_1_0},
|
|
{{fp32T, fp32T, fp32T, fp32T}, SpecificationVersion::V_1_0}}}}},
|
|
{"tosa.reciprocal",
|
|
{{{Profile::pro_fp},
|
|
{{{fp16T, fp16T}, SpecificationVersion::V_1_0},
|
|
{{fp32T, fp32T}, SpecificationVersion::V_1_0}}}}},
|
|
{"tosa.rsqrt",
|
|
{{{Profile::pro_fp},
|
|
{{{fp16T, fp16T}, SpecificationVersion::V_1_0},
|
|
{{fp32T, fp32T}, SpecificationVersion::V_1_0}}}}},
|
|
{"tosa.sin",
|
|
{{{Profile::pro_fp},
|
|
{{{fp16T, fp16T}, SpecificationVersion::V_1_0},
|
|
{{fp32T, fp32T}, SpecificationVersion::V_1_0}}}}},
|
|
{"tosa.select",
|
|
{{{Profile::pro_int, Profile::pro_fp},
|
|
{{{boolT, boolT, boolT}, SpecificationVersion::V_1_0}},
|
|
anyOf},
|
|
{{Profile::pro_int},
|
|
{{{i8T, i8T, i8T}, SpecificationVersion::V_1_0},
|
|
{{i16T, i16T, i16T}, SpecificationVersion::V_1_0},
|
|
{{i32T, i32T, i32T}, SpecificationVersion::V_1_0}}},
|
|
{{Profile::pro_fp},
|
|
{{{fp16T, fp16T, fp16T}, SpecificationVersion::V_1_0},
|
|
{{fp32T, fp32T, fp32T}, SpecificationVersion::V_1_0}}}}},
|
|
{"tosa.equal",
|
|
{{{Profile::pro_int},
|
|
{{{i32T, i32T, boolT}, SpecificationVersion::V_1_0}}},
|
|
{{Profile::pro_fp},
|
|
{{{fp16T, fp16T, boolT}, SpecificationVersion::V_1_0},
|
|
{{fp32T, fp32T, boolT}, SpecificationVersion::V_1_0}}}}},
|
|
{"tosa.greater",
|
|
{{{Profile::pro_int},
|
|
{{{i32T, i32T, boolT}, SpecificationVersion::V_1_0}}},
|
|
{{Profile::pro_fp},
|
|
{{{fp16T, fp16T, boolT}, SpecificationVersion::V_1_0},
|
|
{{fp32T, fp32T, boolT}, SpecificationVersion::V_1_0}}}}},
|
|
{"tosa.greater_equal",
|
|
{{{Profile::pro_int},
|
|
{{{i32T, i32T, boolT}, SpecificationVersion::V_1_0}}},
|
|
{{Profile::pro_fp},
|
|
{{{fp16T, fp16T, boolT}, SpecificationVersion::V_1_0},
|
|
{{fp32T, fp32T, boolT}, SpecificationVersion::V_1_0}}}}},
|
|
{"tosa.reduce_all",
|
|
{{{Profile::pro_int, Profile::pro_fp},
|
|
{{{boolT, boolT}, SpecificationVersion::V_1_0}},
|
|
anyOf}}},
|
|
{"tosa.reduce_any",
|
|
{{{Profile::pro_int, Profile::pro_fp},
|
|
{{{boolT, boolT}, SpecificationVersion::V_1_0}},
|
|
anyOf}}},
|
|
{"tosa.reduce_max",
|
|
{{{Profile::pro_int},
|
|
{{{i8T, i8T}, SpecificationVersion::V_1_0},
|
|
{{i16T, i16T}, SpecificationVersion::V_1_0},
|
|
{{i32T, i32T}, SpecificationVersion::V_1_0}}},
|
|
{{Profile::pro_fp},
|
|
{{{fp16T, fp16T}, SpecificationVersion::V_1_0},
|
|
{{fp32T, fp32T}, SpecificationVersion::V_1_0}}}}},
|
|
{"tosa.reduce_min",
|
|
{{{Profile::pro_int},
|
|
{{{i8T, i8T}, SpecificationVersion::V_1_0},
|
|
{{i16T, i16T}, SpecificationVersion::V_1_0},
|
|
{{i32T, i32T}, SpecificationVersion::V_1_0}}},
|
|
{{Profile::pro_fp},
|
|
{{{fp16T, fp16T}, SpecificationVersion::V_1_0},
|
|
{{fp32T, fp32T}, SpecificationVersion::V_1_0}}}}},
|
|
{"tosa.reduce_product",
|
|
{{{Profile::pro_fp},
|
|
{{{fp16T, fp16T}, SpecificationVersion::V_1_0},
|
|
{{fp32T, fp32T}, SpecificationVersion::V_1_0}}}}},
|
|
{"tosa.reduce_sum",
|
|
{{{Profile::pro_int}, {{{i32T, i32T}, SpecificationVersion::V_1_0}}},
|
|
{{Profile::pro_fp},
|
|
{{{fp16T, fp16T}, SpecificationVersion::V_1_0},
|
|
{{fp32T, fp32T}, SpecificationVersion::V_1_0}}}}},
|
|
{"tosa.concat",
|
|
{{{Profile::pro_int, Profile::pro_fp},
|
|
{{{boolT, boolT}, SpecificationVersion::V_1_0}},
|
|
anyOf},
|
|
{{Profile::pro_int},
|
|
{{{i8T, i8T}, SpecificationVersion::V_1_0},
|
|
{{i32T, i32T}, SpecificationVersion::V_1_0}}},
|
|
{{Profile::pro_fp},
|
|
{{{fp16T, fp16T}, SpecificationVersion::V_1_0},
|
|
{{fp32T, fp32T}, SpecificationVersion::V_1_0}}}}},
|
|
{"tosa.pad",
|
|
{{{Profile::pro_int, Profile::pro_fp},
|
|
{{{boolT, boolT, boolT}, SpecificationVersion::V_1_0}},
|
|
anyOf},
|
|
{{Profile::pro_int},
|
|
{{{i8T, i8T, i8T}, SpecificationVersion::V_1_0},
|
|
{{i16T, i16T, i16T}, SpecificationVersion::V_1_0},
|
|
{{i32T, i32T, i32T}, SpecificationVersion::V_1_0}}},
|
|
{{Profile::pro_fp},
|
|
{{{fp16T, fp16T, fp16T}, SpecificationVersion::V_1_0},
|
|
{{fp32T, fp32T, fp32T}, SpecificationVersion::V_1_0}}}}},
|
|
{"tosa.reshape",
|
|
{{{Profile::pro_int, Profile::pro_fp},
|
|
{{{boolT, boolT}, SpecificationVersion::V_1_0}},
|
|
anyOf},
|
|
{{Profile::pro_int},
|
|
{{{i8T, i8T}, SpecificationVersion::V_1_0},
|
|
{{i16T, i16T}, SpecificationVersion::V_1_0},
|
|
{{i32T, i32T}, SpecificationVersion::V_1_0}}},
|
|
{{Profile::pro_fp},
|
|
{{{fp16T, fp16T}, SpecificationVersion::V_1_0},
|
|
{{fp32T, fp32T}, SpecificationVersion::V_1_0}}}}},
|
|
{"tosa.reverse",
|
|
{{{Profile::pro_int, Profile::pro_fp},
|
|
{{{boolT, boolT}, SpecificationVersion::V_1_0}},
|
|
anyOf},
|
|
{{Profile::pro_int},
|
|
{{{i8T, i8T}, SpecificationVersion::V_1_0},
|
|
{{i16T, i16T}, SpecificationVersion::V_1_0},
|
|
{{i32T, i32T}, SpecificationVersion::V_1_0}}},
|
|
{{Profile::pro_fp},
|
|
{{{fp16T, fp16T}, SpecificationVersion::V_1_0},
|
|
{{fp32T, fp32T}, SpecificationVersion::V_1_0}}}}},
|
|
{"tosa.slice",
|
|
{{{Profile::pro_int, Profile::pro_fp},
|
|
{{{boolT, boolT}, SpecificationVersion::V_1_0}},
|
|
anyOf},
|
|
{{Profile::pro_int},
|
|
{{{i8T, i8T}, SpecificationVersion::V_1_0},
|
|
{{i16T, i16T}, SpecificationVersion::V_1_0},
|
|
{{i32T, i32T}, SpecificationVersion::V_1_0}}},
|
|
{{Profile::pro_fp},
|
|
{{{fp16T, fp16T}, SpecificationVersion::V_1_0},
|
|
{{fp32T, fp32T}, SpecificationVersion::V_1_0}}}}},
|
|
{"tosa.tile",
|
|
{{{Profile::pro_int, Profile::pro_fp},
|
|
{{{boolT, boolT}, SpecificationVersion::V_1_0}},
|
|
anyOf},
|
|
{{Profile::pro_int},
|
|
{{{i8T, i8T}, SpecificationVersion::V_1_0},
|
|
{{i16T, i16T}, SpecificationVersion::V_1_0},
|
|
{{i32T, i32T}, SpecificationVersion::V_1_0}}},
|
|
{{Profile::pro_fp},
|
|
{{{fp16T, fp16T}, SpecificationVersion::V_1_0},
|
|
{{fp32T, fp32T}, SpecificationVersion::V_1_0}}}}},
|
|
{"tosa.transpose",
|
|
{{{Profile::pro_int, Profile::pro_fp},
|
|
{{{boolT, boolT}, SpecificationVersion::V_1_0}},
|
|
anyOf},
|
|
{{Profile::pro_int},
|
|
{{{i8T, i8T}, SpecificationVersion::V_1_0},
|
|
{{i16T, i16T}, SpecificationVersion::V_1_0},
|
|
{{i32T, i32T}, SpecificationVersion::V_1_0}}},
|
|
{{Profile::pro_fp},
|
|
{{{fp16T, fp16T}, SpecificationVersion::V_1_0},
|
|
{{fp32T, fp32T}, SpecificationVersion::V_1_0}}}}},
|
|
{"tosa.gather",
|
|
{{{Profile::pro_int},
|
|
{{{i8T, i32T, i8T}, SpecificationVersion::V_1_0},
|
|
{{i16T, i32T, i16T}, SpecificationVersion::V_1_0},
|
|
{{i32T, i32T, i32T}, SpecificationVersion::V_1_0}}},
|
|
{{Profile::pro_fp},
|
|
{{{i8T, i32T, i8T}, SpecificationVersion::V_1_1_DRAFT},
|
|
{{i16T, i32T, i16T}, SpecificationVersion::V_1_1_DRAFT},
|
|
{{i32T, i32T, i32T}, SpecificationVersion::V_1_1_DRAFT},
|
|
{{fp16T, i32T, fp16T}, SpecificationVersion::V_1_0},
|
|
{{fp32T, i32T, fp32T}, SpecificationVersion::V_1_0}}},
|
|
{{Profile::pro_fp, Profile::pro_int},
|
|
{{{boolT, i32T, boolT}, SpecificationVersion::V_1_1_DRAFT},
|
|
{{i8T, i64T, i8T}, SpecificationVersion::V_1_1_DRAFT},
|
|
{{i16T, i64T, i16T}, SpecificationVersion::V_1_1_DRAFT},
|
|
{{i32T, i64T, i32T}, SpecificationVersion::V_1_1_DRAFT}},
|
|
anyOf}}},
|
|
{"tosa.row_gather",
|
|
{{{Profile::pro_int},
|
|
{{{boolT, i32T, boolT}, SpecificationVersion::V_1_1_DRAFT},
|
|
{{i8T, i32T, i8T}, SpecificationVersion::V_1_1_DRAFT},
|
|
{{i16T, i32T, i16T}, SpecificationVersion::V_1_1_DRAFT},
|
|
{{i32T, i32T, i32T}, SpecificationVersion::V_1_1_DRAFT},
|
|
{{i64T, i32T, i64T}, SpecificationVersion::V_1_1_DRAFT},
|
|
{{boolT, i64T, boolT}, SpecificationVersion::V_1_1_DRAFT},
|
|
{{i8T, i64T, i8T}, SpecificationVersion::V_1_1_DRAFT},
|
|
{{i16T, i64T, i16T}, SpecificationVersion::V_1_1_DRAFT},
|
|
{{i32T, i64T, i32T}, SpecificationVersion::V_1_1_DRAFT},
|
|
{{i64T, i64T, i64T}, SpecificationVersion::V_1_1_DRAFT}}},
|
|
{{Profile::pro_fp},
|
|
{{{boolT, i32T, boolT}, SpecificationVersion::V_1_1_DRAFT},
|
|
{{i8T, i32T, i8T}, SpecificationVersion::V_1_1_DRAFT},
|
|
{{i16T, i32T, i16T}, SpecificationVersion::V_1_1_DRAFT},
|
|
{{i32T, i32T, i32T}, SpecificationVersion::V_1_1_DRAFT},
|
|
{{i64T, i32T, i64T}, SpecificationVersion::V_1_1_DRAFT},
|
|
{{fp16T, i32T, fp16T}, SpecificationVersion::V_1_1_DRAFT},
|
|
{{fp32T, i32T, fp32T}, SpecificationVersion::V_1_1_DRAFT},
|
|
{{boolT, i64T, boolT}, SpecificationVersion::V_1_1_DRAFT},
|
|
{{i8T, i64T, i8T}, SpecificationVersion::V_1_1_DRAFT},
|
|
{{i16T, i64T, i16T}, SpecificationVersion::V_1_1_DRAFT},
|
|
{{i32T, i64T, i32T}, SpecificationVersion::V_1_1_DRAFT},
|
|
{{i64T, i64T, i64T}, SpecificationVersion::V_1_1_DRAFT},
|
|
{{fp16T, i64T, fp16T}, SpecificationVersion::V_1_1_DRAFT},
|
|
{{fp32T, i64T, fp32T}, SpecificationVersion::V_1_1_DRAFT}}}}},
|
|
{"tosa.row_gather_block_scaled",
|
|
{{{Profile::pro_int},
|
|
{{{i8T, i32T, i32T, i8T}, SpecificationVersion::V_1_1_DRAFT},
|
|
{{i16T, i32T, i32T, i16T}, SpecificationVersion::V_1_1_DRAFT},
|
|
{{i32T, i32T, i32T, i32T}, SpecificationVersion::V_1_1_DRAFT}}},
|
|
{{Profile::pro_fp},
|
|
{{{i8T, i32T, i32T, i8T}, SpecificationVersion::V_1_1_DRAFT},
|
|
{{i16T, i32T, i32T, i16T}, SpecificationVersion::V_1_1_DRAFT},
|
|
{{i32T, i32T, i32T, i32T}, SpecificationVersion::V_1_1_DRAFT},
|
|
{{fp16T, i32T, i32T, fp16T}, SpecificationVersion::V_1_1_DRAFT},
|
|
{{fp32T, i32T, i32T, fp32T}, SpecificationVersion::V_1_1_DRAFT}}},
|
|
{{Profile::pro_fp, Profile::pro_int},
|
|
{{{boolT, i32T, i32T, boolT}, SpecificationVersion::V_1_1_DRAFT},
|
|
{{i8T, i64T, i32T, i8T}, SpecificationVersion::V_1_1_DRAFT},
|
|
{{i16T, i64T, i32T, i16T}, SpecificationVersion::V_1_1_DRAFT},
|
|
{{i32T, i64T, i32T, i32T}, SpecificationVersion::V_1_1_DRAFT},
|
|
{{i64T, i64T, i32T, i64T}, SpecificationVersion::V_1_1_DRAFT},
|
|
{{boolT, i64T, i32T, boolT}, SpecificationVersion::V_1_1_DRAFT}},
|
|
anyOf}}},
|
|
{"tosa.scatter",
|
|
{{{Profile::pro_int},
|
|
{{{i8T, i32T, i8T, i8T}, SpecificationVersion::V_1_0},
|
|
{{i16T, i32T, i16T, i16T}, SpecificationVersion::V_1_0},
|
|
{{i32T, i32T, i32T, i32T}, SpecificationVersion::V_1_0}}},
|
|
{{Profile::pro_fp},
|
|
{{{i8T, i32T, i8T, i8T}, SpecificationVersion::V_1_1_DRAFT},
|
|
{{i16T, i32T, i16T, i16T}, SpecificationVersion::V_1_1_DRAFT},
|
|
{{i32T, i32T, i32T, i32T}, SpecificationVersion::V_1_1_DRAFT},
|
|
{{fp16T, i32T, fp16T, fp16T}, SpecificationVersion::V_1_0},
|
|
{{fp32T, i32T, fp32T, fp32T}, SpecificationVersion::V_1_0}}},
|
|
{{Profile::pro_fp, Profile::pro_int},
|
|
{{{boolT, i32T, boolT, boolT}, SpecificationVersion::V_1_1_DRAFT},
|
|
{{i8T, i64T, i8T, i8T}, SpecificationVersion::V_1_1_DRAFT},
|
|
{{i16T, i64T, i16T, i16T}, SpecificationVersion::V_1_1_DRAFT},
|
|
{{i32T, i64T, i32T, i32T}, SpecificationVersion::V_1_1_DRAFT}},
|
|
anyOf}}},
|
|
{"tosa.resize",
|
|
{{{Profile::pro_int},
|
|
{{{i8T, i32T}, SpecificationVersion::V_1_0},
|
|
{{i8T, i8T}, SpecificationVersion::V_1_0}}},
|
|
{{Profile::pro_fp},
|
|
{{{fp16T, fp16T}, SpecificationVersion::V_1_0},
|
|
{{fp32T, fp32T}, SpecificationVersion::V_1_0}}}}},
|
|
{"tosa.cast",
|
|
{{{Profile::pro_int},
|
|
{{{boolT, i8T}, SpecificationVersion::V_1_0},
|
|
{{boolT, i16T}, SpecificationVersion::V_1_0},
|
|
{{boolT, i32T}, SpecificationVersion::V_1_0},
|
|
{{i8T, boolT}, SpecificationVersion::V_1_0},
|
|
{{i8T, i16T}, SpecificationVersion::V_1_0},
|
|
{{i8T, i32T}, SpecificationVersion::V_1_0},
|
|
{{i16T, boolT}, SpecificationVersion::V_1_0},
|
|
{{i16T, i8T}, SpecificationVersion::V_1_0},
|
|
{{i16T, i32T}, SpecificationVersion::V_1_0},
|
|
{{i32T, boolT}, SpecificationVersion::V_1_0},
|
|
{{i32T, i8T}, SpecificationVersion::V_1_0},
|
|
{{i32T, i16T}, SpecificationVersion::V_1_0}}},
|
|
{{Profile::pro_fp},
|
|
{{{boolT, fp32T}, SpecificationVersion::V_1_1_DRAFT},
|
|
{{i8T, fp16T}, SpecificationVersion::V_1_0},
|
|
{{i8T, fp32T}, SpecificationVersion::V_1_0},
|
|
{{i16T, fp16T}, SpecificationVersion::V_1_0},
|
|
{{i16T, fp32T}, SpecificationVersion::V_1_0},
|
|
{{i32T, fp16T}, SpecificationVersion::V_1_0},
|
|
{{i32T, fp32T}, SpecificationVersion::V_1_0},
|
|
{{fp16T, i8T}, SpecificationVersion::V_1_0},
|
|
{{fp16T, i16T}, SpecificationVersion::V_1_0},
|
|
{{fp16T, i32T}, SpecificationVersion::V_1_0},
|
|
{{fp16T, fp32T}, SpecificationVersion::V_1_0},
|
|
{{fp32T, i8T}, SpecificationVersion::V_1_0},
|
|
{{fp32T, i16T}, SpecificationVersion::V_1_0},
|
|
{{fp32T, i32T}, SpecificationVersion::V_1_0},
|
|
{{fp32T, fp16T}, SpecificationVersion::V_1_0},
|
|
{{fp32T, boolT}, SpecificationVersion::V_1_1_DRAFT}}}}},
|
|
{"tosa.rescale",
|
|
{{{Profile::pro_int},
|
|
{{{i8T, i8T, i8T, i8T}, SpecificationVersion::V_1_0},
|
|
{{i8T, i8T, i16T, i16T}, SpecificationVersion::V_1_0},
|
|
{{i8T, i8T, i32T, i32T}, SpecificationVersion::V_1_0},
|
|
{{i16T, i16T, i8T, i8T}, SpecificationVersion::V_1_0},
|
|
{{i16T, i16T, i16T, i16T}, SpecificationVersion::V_1_0},
|
|
{{i16T, i16T, i32T, i32T}, SpecificationVersion::V_1_0},
|
|
{{i32T, i32T, i8T, i8T}, SpecificationVersion::V_1_0},
|
|
{{i32T, i32T, i16T, i16T}, SpecificationVersion::V_1_0},
|
|
{{i32T, i32T, i32T, i32T}, SpecificationVersion::V_1_0}}}}},
|
|
{"tosa.const",
|
|
{{{Profile::pro_int, Profile::pro_fp},
|
|
{{{boolT}, SpecificationVersion::V_1_0},
|
|
{{i8T}, SpecificationVersion::V_1_0},
|
|
{{i16T}, SpecificationVersion::V_1_0},
|
|
{{i32T}, SpecificationVersion::V_1_0}},
|
|
anyOf},
|
|
{{Profile::pro_fp},
|
|
{{{fp16T}, SpecificationVersion::V_1_0},
|
|
{{fp32T}, SpecificationVersion::V_1_0}}}}},
|
|
{"tosa.identity",
|
|
{{{Profile::pro_int, Profile::pro_fp},
|
|
{{{boolT, boolT}, SpecificationVersion::V_1_0},
|
|
{{i8T, i8T}, SpecificationVersion::V_1_0},
|
|
{{i16T, i16T}, SpecificationVersion::V_1_0},
|
|
{{i32T, i32T}, SpecificationVersion::V_1_0}},
|
|
anyOf},
|
|
{{Profile::pro_fp},
|
|
{{{fp16T, fp16T}, SpecificationVersion::V_1_0},
|
|
{{fp32T, fp32T}, SpecificationVersion::V_1_0}}}}},
|
|
{"tosa.variable",
|
|
{{{Profile::pro_int}, {{{i8T}, SpecificationVersion::V_1_0}}},
|
|
{{Profile::pro_fp},
|
|
{{{fp16T}, SpecificationVersion::V_1_0},
|
|
{{fp32T}, SpecificationVersion::V_1_0}}}}},
|
|
{"tosa.variable_write",
|
|
{{{Profile::pro_int}, {{{i8T}, SpecificationVersion::V_1_0}}},
|
|
{{Profile::pro_fp},
|
|
{{{fp16T}, SpecificationVersion::V_1_0},
|
|
{{fp32T}, SpecificationVersion::V_1_0}}}}},
|
|
{"tosa.variable_read",
|
|
{{{Profile::pro_int}, {{{i8T}, SpecificationVersion::V_1_0}}},
|
|
{{Profile::pro_fp},
|
|
{{{fp16T}, SpecificationVersion::V_1_0},
|
|
{{fp32T}, SpecificationVersion::V_1_0}}}}},
|
|
};
|
|
|
|
extensionComplianceMap = {
|
|
{"tosa.argmax",
|
|
{{{Extension::int16}, {{{i16T, i32T}, SpecificationVersion::V_1_0}}},
|
|
{{Extension::int64},
|
|
{{{i8T, i64T}, SpecificationVersion::V_1_1_DRAFT},
|
|
{{i16T, i64T}, SpecificationVersion::V_1_1_DRAFT},
|
|
{{i32T, i64T}, SpecificationVersion::V_1_1_DRAFT},
|
|
{{i64T, i64T}, SpecificationVersion::V_1_1_DRAFT},
|
|
{{fp16T, i64T}, SpecificationVersion::V_1_1_DRAFT},
|
|
{{fp32T, i64T}, SpecificationVersion::V_1_1_DRAFT}}},
|
|
{{Extension::fp8e4m3}, {{{fp8e4m3T, i32T}, SpecificationVersion::V_1_0}}},
|
|
{{Extension::fp8e5m2}, {{{fp8e5m2T, i32T}, SpecificationVersion::V_1_0}}},
|
|
{{Extension::bf16}, {{{bf16T, i32T}, SpecificationVersion::V_1_0}}},
|
|
{{Extension::fp8e4m3, Extension::int64},
|
|
{{{fp8e4m3T, i64T}, SpecificationVersion::V_1_1_DRAFT}},
|
|
allOf},
|
|
{{Extension::fp8e5m2, Extension::int64},
|
|
{{{fp8e5m2T, i64T}, SpecificationVersion::V_1_1_DRAFT}},
|
|
allOf},
|
|
{{Extension::bf16, Extension::int64},
|
|
{{{bf16T, i64T}, SpecificationVersion::V_1_1_DRAFT}},
|
|
allOf}}},
|
|
{"tosa.avg_pool2d",
|
|
{{{Extension::int16},
|
|
{{{i16T, i16T, i16T, i32T, i16T}, SpecificationVersion::V_1_0}}},
|
|
{{Extension::fp8e4m3},
|
|
{{{fp8e4m3T, fp8e4m3T, fp8e4m3T, fp16T, fp8e4m3T},
|
|
SpecificationVersion::V_1_0}}},
|
|
{{Extension::fp8e5m2},
|
|
{{{fp8e5m2T, fp8e5m2T, fp8e5m2T, fp16T, fp8e5m2T},
|
|
SpecificationVersion::V_1_0}}},
|
|
{{Extension::bf16},
|
|
{{{bf16T, bf16T, bf16T, fp32T, bf16T}, SpecificationVersion::V_1_0}}}}},
|
|
{"tosa.avg_pool2d_adaptive",
|
|
{{{Extension::int16},
|
|
{{{i16T, i16T, i16T, i32T, i16T}, SpecificationVersion::V_1_1_DRAFT}}},
|
|
{{Extension::fp8e4m3},
|
|
{{{fp8e4m3T, fp8e4m3T, fp8e4m3T, fp16T, fp8e4m3T},
|
|
SpecificationVersion::V_1_1_DRAFT}}},
|
|
{{Extension::fp8e5m2},
|
|
{{{fp8e5m2T, fp8e5m2T, fp8e5m2T, fp16T, fp8e5m2T},
|
|
SpecificationVersion::V_1_1_DRAFT}}},
|
|
{{Extension::bf16},
|
|
{{{bf16T, bf16T, bf16T, fp32T, bf16T},
|
|
SpecificationVersion::V_1_1_DRAFT}}}}},
|
|
{"tosa.conv2d",
|
|
{{{Extension::int4},
|
|
{{{i8T, i4T, i32T, i8T, i4T, i32T, i32T}, SpecificationVersion::V_1_0}}},
|
|
{{Extension::int16},
|
|
{{{i16T, i8T, i48T, i16T, i8T, i48T, i48T},
|
|
SpecificationVersion::V_1_0}}},
|
|
{{Extension::fp8e4m3},
|
|
{{{fp8e4m3T, fp8e4m3T, fp16T, fp8e4m3T, fp8e4m3T, fp16T, fp16T},
|
|
SpecificationVersion::V_1_0},
|
|
{{fp8e4m3T, fp8e4m3T, fp32T, fp8e4m3T, fp8e4m3T, fp32T, fp32T},
|
|
SpecificationVersion::V_1_1_DRAFT}}},
|
|
{{Extension::fp8e5m2},
|
|
{{{fp8e5m2T, fp8e5m2T, fp16T, fp8e5m2T, fp8e5m2T, fp16T, fp16T},
|
|
SpecificationVersion::V_1_0},
|
|
{{fp8e5m2T, fp8e5m2T, fp32T, fp8e5m2T, fp8e5m2T, fp32T, fp32T},
|
|
SpecificationVersion::V_1_1_DRAFT}}},
|
|
{{Extension::bf16},
|
|
{{{bf16T, bf16T, bf16T, bf16T, bf16T, fp32T, bf16T},
|
|
SpecificationVersion::V_1_0}}}}},
|
|
{"tosa.conv2d_block_scaled",
|
|
{{{Extension::mxfp_conv},
|
|
{{{fp4e2m1T, fp8ue8m0T, fp4e2m1T, fp8ue8m0T, fp32T, fp32T},
|
|
SpecificationVersion::V_1_1_DRAFT},
|
|
{{fp6e2m3T, fp8ue8m0T, fp6e2m3T, fp8ue8m0T, fp32T, fp32T},
|
|
SpecificationVersion::V_1_1_DRAFT},
|
|
{{fp6e3m2T, fp8ue8m0T, fp6e3m2T, fp8ue8m0T, fp32T, fp32T},
|
|
SpecificationVersion::V_1_1_DRAFT},
|
|
{{fp8e4m3T, fp8ue8m0T, fp8e4m3T, fp8ue8m0T, fp32T, fp32T},
|
|
SpecificationVersion::V_1_1_DRAFT},
|
|
{{fp8e5m2T, fp8ue8m0T, fp8e5m2T, fp8ue8m0T, fp32T, fp32T},
|
|
SpecificationVersion::V_1_1_DRAFT}}}}},
|
|
{"tosa.conv3d",
|
|
{{{Extension::int4},
|
|
{{{i8T, i4T, i32T, i8T, i4T, i32T, i32T}, SpecificationVersion::V_1_0}}},
|
|
{{Extension::int16},
|
|
{{{i16T, i8T, i48T, i16T, i8T, i48T, i48T},
|
|
SpecificationVersion::V_1_0}}},
|
|
{{Extension::fp8e4m3},
|
|
{{{fp8e4m3T, fp8e4m3T, fp16T, fp8e4m3T, fp8e4m3T, fp16T, fp16T},
|
|
SpecificationVersion::V_1_0},
|
|
{{fp8e4m3T, fp8e4m3T, fp32T, fp8e4m3T, fp8e4m3T, fp32T, fp32T},
|
|
SpecificationVersion::V_1_1_DRAFT}}},
|
|
{{Extension::fp8e5m2},
|
|
{{{fp8e5m2T, fp8e5m2T, fp16T, fp8e5m2T, fp8e5m2T, fp16T, fp16T},
|
|
SpecificationVersion::V_1_0},
|
|
{{fp8e5m2T, fp8e5m2T, fp32T, fp8e5m2T, fp8e5m2T, fp32T, fp32T},
|
|
SpecificationVersion::V_1_1_DRAFT}}},
|
|
{{Extension::bf16},
|
|
{{{bf16T, bf16T, bf16T, bf16T, bf16T, fp32T, bf16T},
|
|
SpecificationVersion::V_1_0}}}}},
|
|
{"tosa.depthwise_conv2d",
|
|
{{{Extension::int4},
|
|
{{{i8T, i4T, i32T, i8T, i4T, i32T, i32T}, SpecificationVersion::V_1_0}}},
|
|
{{Extension::int16},
|
|
{{{i16T, i8T, i48T, i16T, i8T, i48T, i48T},
|
|
SpecificationVersion::V_1_0}}},
|
|
{{Extension::fp8e4m3},
|
|
{{{fp8e4m3T, fp8e4m3T, fp16T, fp8e4m3T, fp8e4m3T, fp16T, fp16T},
|
|
SpecificationVersion::V_1_0},
|
|
{{fp8e4m3T, fp8e4m3T, fp32T, fp8e4m3T, fp8e4m3T, fp32T, fp32T},
|
|
SpecificationVersion::V_1_1_DRAFT}}},
|
|
{{Extension::fp8e5m2},
|
|
{{{fp8e5m2T, fp8e5m2T, fp16T, fp8e5m2T, fp8e5m2T, fp16T, fp16T},
|
|
SpecificationVersion::V_1_0},
|
|
{{fp8e5m2T, fp8e5m2T, fp32T, fp8e5m2T, fp8e5m2T, fp32T, fp32T},
|
|
SpecificationVersion::V_1_1_DRAFT}}},
|
|
{{Extension::bf16},
|
|
{{{bf16T, bf16T, bf16T, bf16T, bf16T, fp32T, bf16T},
|
|
SpecificationVersion::V_1_0}}}}},
|
|
{"tosa.fft2d",
|
|
{{{Extension::fft},
|
|
{{{fp32T, fp32T, fp32T, fp32T}, SpecificationVersion::V_1_0}}}}},
|
|
{"tosa.matmul",
|
|
{{{Extension::int16},
|
|
{{{i16T, i16T, i16T, i16T, i48T}, SpecificationVersion::V_1_0}}},
|
|
{{Extension::fp8e4m3},
|
|
{{{fp8e4m3T, fp8e4m3T, fp8e4m3T, fp8e4m3T, fp16T},
|
|
SpecificationVersion::V_1_0},
|
|
{{fp8e4m3T, fp8e4m3T, fp8e4m3T, fp8e4m3T, fp32T},
|
|
SpecificationVersion::V_1_1_DRAFT}}},
|
|
{{Extension::fp8e5m2},
|
|
{{{fp8e5m2T, fp8e5m2T, fp8e5m2T, fp8e5m2T, fp16T},
|
|
SpecificationVersion::V_1_0},
|
|
{{fp8e5m2T, fp8e5m2T, fp8e5m2T, fp8e5m2T, fp32T},
|
|
SpecificationVersion::V_1_1_DRAFT}}},
|
|
{{Extension::fp8e4m3, Extension::fp8e5m2},
|
|
{{{fp8e4m3T, fp8e5m2T, fp8e4m3T, fp8e5m2T, fp16T},
|
|
SpecificationVersion::V_1_1_DRAFT},
|
|
{{fp8e4m3T, fp8e5m2T, fp8e4m3T, fp8e5m2T, fp32T},
|
|
SpecificationVersion::V_1_1_DRAFT},
|
|
{{fp8e5m2T, fp8e4m3T, fp8e5m2T, fp8e4m3T, fp16T},
|
|
SpecificationVersion::V_1_1_DRAFT},
|
|
{{fp8e5m2T, fp8e4m3T, fp8e5m2T, fp8e4m3T, fp32T},
|
|
SpecificationVersion::V_1_1_DRAFT}},
|
|
allOf},
|
|
{{Extension::bf16},
|
|
{{{bf16T, bf16T, bf16T, bf16T, fp32T}, SpecificationVersion::V_1_0}}}}},
|
|
{"tosa.matmul_t_block_scaled",
|
|
{{{Extension::mxfp},
|
|
{{{fp4e2m1T, fp8ue8m0T, fp4e2m1T, fp8ue8m0T, fp32T},
|
|
SpecificationVersion::V_1_1_DRAFT},
|
|
{{fp6e2m3T, fp8ue8m0T, fp6e2m3T, fp8ue8m0T, fp32T},
|
|
SpecificationVersion::V_1_1_DRAFT},
|
|
{{fp6e3m2T, fp8ue8m0T, fp6e3m2T, fp8ue8m0T, fp32T},
|
|
SpecificationVersion::V_1_1_DRAFT},
|
|
{{fp8e4m3T, fp8ue8m0T, fp8e4m3T, fp8ue8m0T, fp32T},
|
|
SpecificationVersion::V_1_1_DRAFT},
|
|
{{fp8e5m2T, fp8ue8m0T, fp8e5m2T, fp8ue8m0T, fp32T},
|
|
SpecificationVersion::V_1_1_DRAFT},
|
|
{{mxint8T, fp8ue8m0T, mxint8T, fp8ue8m0T, fp32T},
|
|
SpecificationVersion::V_1_1_DRAFT}}}}},
|
|
{"tosa.max_pool2d",
|
|
{{{Extension::int16}, {{{i16T, i16T}, SpecificationVersion::V_1_0}}},
|
|
{{Extension::fp8e4m3},
|
|
{{{fp8e4m3T, fp8e4m3T}, SpecificationVersion::V_1_0}}},
|
|
{{Extension::fp8e5m2},
|
|
{{{fp8e5m2T, fp8e5m2T}, SpecificationVersion::V_1_0}}},
|
|
{{Extension::bf16}, {{{bf16T, bf16T}, SpecificationVersion::V_1_0}}}}},
|
|
{"tosa.max_pool2d_adaptive",
|
|
{{{Extension::int16}, {{{i16T, i16T}, SpecificationVersion::V_1_1_DRAFT}}},
|
|
{{Extension::fp8e4m3},
|
|
{{{fp8e4m3T, fp8e4m3T}, SpecificationVersion::V_1_1_DRAFT}}},
|
|
{{Extension::fp8e5m2},
|
|
{{{fp8e5m2T, fp8e5m2T}, SpecificationVersion::V_1_1_DRAFT}}},
|
|
{{Extension::bf16},
|
|
{{{bf16T, bf16T}, SpecificationVersion::V_1_1_DRAFT}}}}},
|
|
{"tosa.rfft2d",
|
|
{{{Extension::fft},
|
|
{{{fp32T, fp32T, fp32T}, SpecificationVersion::V_1_0}}}}},
|
|
{"tosa.transpose_conv2d",
|
|
{{{Extension::int4},
|
|
{{{i8T, i4T, i32T, i8T, i4T, i32T, i32T}, SpecificationVersion::V_1_0}}},
|
|
{{Extension::int16},
|
|
{{{i16T, i8T, i48T, i16T, i8T, i48T, i48T},
|
|
SpecificationVersion::V_1_0}}},
|
|
{{Extension::fp8e4m3},
|
|
{{{fp8e4m3T, fp8e4m3T, fp16T, fp8e4m3T, fp8e4m3T, fp16T, fp16T},
|
|
SpecificationVersion::V_1_0},
|
|
{{fp8e4m3T, fp8e4m3T, fp32T, fp8e4m3T, fp8e4m3T, fp32T, fp32T},
|
|
SpecificationVersion::V_1_1_DRAFT}}},
|
|
{{Extension::fp8e5m2},
|
|
{{{fp8e5m2T, fp8e5m2T, fp16T, fp8e5m2T, fp8e5m2T, fp16T, fp16T},
|
|
SpecificationVersion::V_1_0},
|
|
{{fp8e5m2T, fp8e5m2T, fp32T, fp8e5m2T, fp8e5m2T, fp32T, fp32T},
|
|
SpecificationVersion::V_1_1_DRAFT}}},
|
|
{{Extension::bf16},
|
|
{{{bf16T, bf16T, bf16T, bf16T, bf16T, fp32T, bf16T},
|
|
SpecificationVersion::V_1_0}}}}},
|
|
{"tosa.clamp",
|
|
{{{Extension::int16}, {{{i16T, i16T}, SpecificationVersion::V_1_0}}},
|
|
{{Extension::bf16}, {{{bf16T, bf16T}, SpecificationVersion::V_1_0}}}}},
|
|
{"tosa.erf",
|
|
{{{Extension::bf16}, {{{bf16T, bf16T}, SpecificationVersion::V_1_0}}}}},
|
|
{"tosa.sigmoid",
|
|
{{{Extension::bf16}, {{{bf16T, bf16T}, SpecificationVersion::V_1_0}}}}},
|
|
{"tosa.tanh",
|
|
{{{Extension::bf16}, {{{bf16T, bf16T}, SpecificationVersion::V_1_0}}}}},
|
|
{"tosa.add",
|
|
{{{Extension::int64},
|
|
{{{i64T, i64T, i64T}, SpecificationVersion::V_1_1_DRAFT}}},
|
|
{{Extension::bf16},
|
|
{{{bf16T, bf16T, bf16T}, SpecificationVersion::V_1_0}}}}},
|
|
{"tosa.arithmetic_right_shift",
|
|
{{{Extension::int64},
|
|
{{{i64T, i64T, i64T}, SpecificationVersion::V_1_1_DRAFT}}}}},
|
|
{"tosa.bitwise_and",
|
|
{{{Extension::int64},
|
|
{{{i64T, i64T, i64T}, SpecificationVersion::V_1_1_DRAFT}}}}},
|
|
{"tosa.bitwise_or",
|
|
{{{Extension::int64},
|
|
{{{i64T, i64T, i64T}, SpecificationVersion::V_1_1_DRAFT}}}}},
|
|
{"tosa.bitwise_xor",
|
|
{{{Extension::int64},
|
|
{{{i64T, i64T, i64T}, SpecificationVersion::V_1_1_DRAFT}}}}},
|
|
{"tosa.intdiv",
|
|
{{{Extension::int64},
|
|
{{{i64T, i64T, i64T}, SpecificationVersion::V_1_1_DRAFT}}}}},
|
|
{"tosa.logical_left_shift",
|
|
{{{Extension::int64},
|
|
{{{i64T, i64T, i64T}, SpecificationVersion::V_1_1_DRAFT}}}}},
|
|
{"tosa.logical_right_shift",
|
|
{{{Extension::int64},
|
|
{{{i64T, i64T, i64T}, SpecificationVersion::V_1_1_DRAFT}}}}},
|
|
{"tosa.maximum",
|
|
{{{Extension::int64},
|
|
{{{i64T, i64T, i64T}, SpecificationVersion::V_1_1_DRAFT}}},
|
|
{{Extension::bf16},
|
|
{{{bf16T, bf16T, bf16T}, SpecificationVersion::V_1_0}}}}},
|
|
{"tosa.minimum",
|
|
{{{Extension::int64},
|
|
{{{i64T, i64T, i64T}, SpecificationVersion::V_1_1_DRAFT}}},
|
|
{{Extension::bf16},
|
|
{{{bf16T, bf16T, bf16T}, SpecificationVersion::V_1_0}}}}},
|
|
{"tosa.mul",
|
|
{{{Extension::int64},
|
|
{{{i64T, i64T, i64T}, SpecificationVersion::V_1_1_DRAFT}}},
|
|
{{Extension::bf16},
|
|
{{{bf16T, bf16T, bf16T}, SpecificationVersion::V_1_0}}}}},
|
|
{"tosa.pow",
|
|
{{{Extension::bf16},
|
|
{{{bf16T, bf16T, bf16T}, SpecificationVersion::V_1_0}}}}},
|
|
{"tosa.sub",
|
|
{{{Extension::int64},
|
|
{{{i64T, i64T, i64T}, SpecificationVersion::V_1_1_DRAFT}}},
|
|
{{Extension::bf16},
|
|
{{{bf16T, bf16T, bf16T}, SpecificationVersion::V_1_0}}}}},
|
|
{"tosa.table",
|
|
{{{Extension::int16},
|
|
{{{i16T, i16T, i32T}, SpecificationVersion::V_1_0}}}}},
|
|
{"tosa.abs",
|
|
{{{Extension::int64}, {{{i64T, i64T}, SpecificationVersion::V_1_1_DRAFT}}},
|
|
{{Extension::bf16}, {{{bf16T, bf16T}, SpecificationVersion::V_1_0}}}}},
|
|
{"tosa.bitwise_not",
|
|
{{{Extension::int64},
|
|
{{{i64T, i64T}, SpecificationVersion::V_1_1_DRAFT}}}}},
|
|
{"tosa.ceil",
|
|
{{{Extension::bf16}, {{{bf16T, bf16T}, SpecificationVersion::V_1_0}}}}},
|
|
{"tosa.clz",
|
|
{{{Extension::int64},
|
|
{{{i64T, i64T}, SpecificationVersion::V_1_1_DRAFT}}}}},
|
|
{"tosa.cos",
|
|
{{{Extension::bf16}, {{{bf16T, bf16T}, SpecificationVersion::V_1_0}}}}},
|
|
{"tosa.exp",
|
|
{{{Extension::bf16}, {{{bf16T, bf16T}, SpecificationVersion::V_1_0}}}}},
|
|
{"tosa.floor",
|
|
{{{Extension::bf16}, {{{bf16T, bf16T}, SpecificationVersion::V_1_0}}}}},
|
|
{"tosa.log",
|
|
{{{Extension::bf16}, {{{bf16T, bf16T}, SpecificationVersion::V_1_0}}}}},
|
|
{"tosa.negate",
|
|
{{{Extension::int64},
|
|
{{{i64T, i64T, i64T, i64T}, SpecificationVersion::V_1_1_DRAFT}}},
|
|
{{Extension::bf16},
|
|
{{{bf16T, bf16T, bf16T, bf16T}, SpecificationVersion::V_1_0}}}}},
|
|
{"tosa.reciprocal",
|
|
{{{Extension::bf16}, {{{bf16T, bf16T}, SpecificationVersion::V_1_0}}}}},
|
|
{"tosa.rsqrt",
|
|
{{{Extension::bf16}, {{{bf16T, bf16T}, SpecificationVersion::V_1_0}}}}},
|
|
{"tosa.sin",
|
|
{{{Extension::bf16}, {{{bf16T, bf16T}, SpecificationVersion::V_1_0}}}}},
|
|
{"tosa.select",
|
|
{{{Extension::int64},
|
|
{{{i64T, i64T, i64T}, SpecificationVersion::V_1_1_DRAFT}}},
|
|
{{Extension::bf16},
|
|
{{{bf16T, bf16T, bf16T}, SpecificationVersion::V_1_0}}}}},
|
|
{"tosa.equal",
|
|
{{{Extension::int64},
|
|
{{{i64T, i64T, boolT}, SpecificationVersion::V_1_1_DRAFT}}},
|
|
{{Extension::bf16},
|
|
{{{bf16T, bf16T, boolT}, SpecificationVersion::V_1_0}}}}},
|
|
{"tosa.greater",
|
|
{{{Extension::int64},
|
|
{{{i64T, i64T, boolT}, SpecificationVersion::V_1_1_DRAFT}}},
|
|
{{Extension::bf16},
|
|
{{{bf16T, bf16T, boolT}, SpecificationVersion::V_1_0}}}}},
|
|
{"tosa.greater_equal",
|
|
{{{Extension::int64},
|
|
{{{i64T, i64T, boolT}, SpecificationVersion::V_1_1_DRAFT}}},
|
|
{{Extension::bf16},
|
|
{{{bf16T, bf16T, boolT}, SpecificationVersion::V_1_0}}}}},
|
|
{"tosa.reduce_max",
|
|
{{{Extension::int64}, {{{i64T, i64T}, SpecificationVersion::V_1_1_DRAFT}}},
|
|
{{Extension::bf16}, {{{bf16T, bf16T}, SpecificationVersion::V_1_0}}}}},
|
|
{"tosa.reduce_min",
|
|
{{{Extension::int64}, {{{i64T, i64T}, SpecificationVersion::V_1_1_DRAFT}}},
|
|
{{Extension::bf16}, {{{bf16T, bf16T}, SpecificationVersion::V_1_0}}}}},
|
|
{"tosa.reduce_product",
|
|
{{{Extension::bf16}, {{{bf16T, bf16T}, SpecificationVersion::V_1_0}}}}},
|
|
{"tosa.reduce_sum",
|
|
{{{Extension::int64}, {{{i64T, i64T}, SpecificationVersion::V_1_1_DRAFT}}},
|
|
{{Extension::bf16}, {{{bf16T, bf16T}, SpecificationVersion::V_1_0}}}}},
|
|
{"tosa.concat",
|
|
{{{Extension::int16}, {{{i16T, i16T}, SpecificationVersion::V_1_0}}},
|
|
{{Extension::int64}, {{{i64T, i64T}, SpecificationVersion::V_1_1_DRAFT}}},
|
|
{{Extension::fp8e4m3},
|
|
{{{fp8e4m3T, fp8e4m3T}, SpecificationVersion::V_1_0}}},
|
|
{{Extension::fp8e5m2},
|
|
{{{fp8e5m2T, fp8e5m2T}, SpecificationVersion::V_1_0}}},
|
|
{{Extension::bf16}, {{{bf16T, bf16T}, SpecificationVersion::V_1_0}}}}},
|
|
{"tosa.pad",
|
|
{{{Extension::int64},
|
|
{{{i64T, i64T, i64T}, SpecificationVersion::V_1_1_DRAFT}}},
|
|
{{Extension::fp8e4m3},
|
|
{{{fp8e4m3T, fp8e4m3T, fp8e4m3T}, SpecificationVersion::V_1_0}}},
|
|
{{Extension::fp8e5m2},
|
|
{{{fp8e5m2T, fp8e5m2T, fp8e5m2T}, SpecificationVersion::V_1_0}}},
|
|
{{Extension::bf16},
|
|
{{{bf16T, bf16T, bf16T}, SpecificationVersion::V_1_0}}}}},
|
|
{"tosa.reshape",
|
|
{{{Extension::int64}, {{{i64T, i64T}, SpecificationVersion::V_1_1_DRAFT}}},
|
|
{{Extension::fp8e4m3},
|
|
{{{fp8e4m3T, fp8e4m3T}, SpecificationVersion::V_1_0}}},
|
|
{{Extension::fp8e5m2},
|
|
{{{fp8e5m2T, fp8e5m2T}, SpecificationVersion::V_1_0}}},
|
|
{{Extension::bf16}, {{{bf16T, bf16T}, SpecificationVersion::V_1_0}}}}},
|
|
{"tosa.reverse",
|
|
{{{Extension::int64}, {{{i64T, i64T}, SpecificationVersion::V_1_1_DRAFT}}},
|
|
{{Extension::fp8e4m3},
|
|
{{{fp8e4m3T, fp8e4m3T}, SpecificationVersion::V_1_0}}},
|
|
{{Extension::fp8e5m2},
|
|
{{{fp8e5m2T, fp8e5m2T}, SpecificationVersion::V_1_0}}},
|
|
{{Extension::bf16}, {{{bf16T, bf16T}, SpecificationVersion::V_1_0}}}}},
|
|
{"tosa.slice",
|
|
{{{Extension::int64}, {{{i64T, i64T}, SpecificationVersion::V_1_1_DRAFT}}},
|
|
{{Extension::fp8e4m3},
|
|
{{{fp8e4m3T, fp8e4m3T}, SpecificationVersion::V_1_0}}},
|
|
{{Extension::fp8e5m2},
|
|
{{{fp8e5m2T, fp8e5m2T}, SpecificationVersion::V_1_0}}},
|
|
{{Extension::bf16}, {{{bf16T, bf16T}, SpecificationVersion::V_1_0}}}}},
|
|
{"tosa.tile",
|
|
{{{Extension::int64}, {{{i64T, i64T}, SpecificationVersion::V_1_1_DRAFT}}},
|
|
{{Extension::fp8e4m3},
|
|
{{{fp8e4m3T, fp8e4m3T}, SpecificationVersion::V_1_0}}},
|
|
{{Extension::fp8e5m2},
|
|
{{{fp8e5m2T, fp8e5m2T}, SpecificationVersion::V_1_0}}},
|
|
{{Extension::bf16}, {{{bf16T, bf16T}, SpecificationVersion::V_1_0}}}}},
|
|
{"tosa.transpose",
|
|
{{{Extension::int64}, {{{i64T, i64T}, SpecificationVersion::V_1_1_DRAFT}}},
|
|
{{Extension::fp8e4m3},
|
|
{{{fp8e4m3T, fp8e4m3T}, SpecificationVersion::V_1_0}}},
|
|
{{Extension::fp8e5m2},
|
|
{{{fp8e5m2T, fp8e5m2T}, SpecificationVersion::V_1_0}}},
|
|
{{Extension::bf16}, {{{bf16T, bf16T}, SpecificationVersion::V_1_0}}}}},
|
|
{"tosa.gather",
|
|
{{{Extension::fp8e4m3},
|
|
{{{fp8e4m3T, i32T, fp8e4m3T}, SpecificationVersion::V_1_0}}},
|
|
{{Extension::fp8e5m2},
|
|
{{{fp8e5m2T, i32T, fp8e5m2T}, SpecificationVersion::V_1_0}}},
|
|
{{Extension::bf16},
|
|
{{{bf16T, i32T, bf16T}, SpecificationVersion::V_1_0}}},
|
|
{{Extension::int64},
|
|
{{{i8T, i64T, i8T}, SpecificationVersion::V_1_1_DRAFT},
|
|
{{i16T, i64T, i16T}, SpecificationVersion::V_1_1_DRAFT},
|
|
{{i32T, i64T, i32T}, SpecificationVersion::V_1_1_DRAFT},
|
|
{{i64T, i64T, i64T}, SpecificationVersion::V_1_1_DRAFT},
|
|
{{fp16T, i64T, fp16T}, SpecificationVersion::V_1_1_DRAFT},
|
|
{{fp32T, i64T, fp32T}, SpecificationVersion::V_1_1_DRAFT},
|
|
{{boolT, i64T, boolT}, SpecificationVersion::V_1_1_DRAFT}}},
|
|
{{Extension::fp8e4m3, Extension::int64},
|
|
{{{fp8e4m3T, i64T, fp8e4m3T}, SpecificationVersion::V_1_1_DRAFT}},
|
|
allOf},
|
|
{{Extension::fp8e5m2, Extension::int64},
|
|
{{{fp8e5m2T, i64T, fp8e5m2T}, SpecificationVersion::V_1_1_DRAFT}},
|
|
allOf},
|
|
{{Extension::bf16, Extension::int64},
|
|
{{{bf16T, i64T, bf16T}, SpecificationVersion::V_1_1_DRAFT}},
|
|
allOf}}},
|
|
{"tosa.row_gather",
|
|
{
|
|
{{Extension::int16},
|
|
{{{i16T, i32T, i16T}, SpecificationVersion::V_1_1_DRAFT}}},
|
|
{{Extension::int64},
|
|
{{{i64T, i32T, i64T}, SpecificationVersion::V_1_1_DRAFT},
|
|
{{boolT, i64T, boolT}, SpecificationVersion::V_1_1_DRAFT},
|
|
{{i8T, i64T, i8T}, SpecificationVersion::V_1_1_DRAFT},
|
|
{{i32T, i64T, i32T}, SpecificationVersion::V_1_1_DRAFT},
|
|
{{i64T, i64T, i64T}, SpecificationVersion::V_1_1_DRAFT},
|
|
{{i64T, i32T, i64T}, SpecificationVersion::V_1_1_DRAFT},
|
|
{{fp16T, i64T, fp16T}, SpecificationVersion::V_1_1_DRAFT},
|
|
{{fp32T, i64T, fp32T}, SpecificationVersion::V_1_1_DRAFT}}},
|
|
{{Extension::int16, Extension::int64},
|
|
{{{i16T, i64T, i16T}, SpecificationVersion::V_1_1_DRAFT}},
|
|
allOf},
|
|
{{Extension::fp8e4m3},
|
|
{{{fp8e4m3T, i32T, fp8e4m3T}, SpecificationVersion::V_1_1_DRAFT}}},
|
|
{{Extension::fp8e5m2},
|
|
{{{fp8e5m2T, i32T, fp8e5m2T}, SpecificationVersion::V_1_1_DRAFT}}},
|
|
{{Extension::bf16},
|
|
{{{bf16T, i32T, bf16T}, SpecificationVersion::V_1_1_DRAFT}}},
|
|
{{Extension::fp8e4m3, Extension::int64},
|
|
{{{fp8e4m3T, i64T, fp8e4m3T}, SpecificationVersion::V_1_1_DRAFT}},
|
|
allOf},
|
|
{{Extension::fp8e5m2, Extension::int64},
|
|
{{{fp8e5m2T, i64T, fp8e5m2T}, SpecificationVersion::V_1_1_DRAFT}},
|
|
allOf},
|
|
{{Extension::bf16, Extension::int64},
|
|
{{{bf16T, i64T, bf16T}, SpecificationVersion::V_1_1_DRAFT}},
|
|
allOf},
|
|
}},
|
|
{"tosa.row_gather_block_scaled",
|
|
{{{Extension::fp8e4m3},
|
|
{{{fp8e4m3T, i32T, i32T, fp8e4m3T}, SpecificationVersion::V_1_1_DRAFT}}},
|
|
{{Extension::fp8e5m2},
|
|
{{{fp8e5m2T, i32T, i32T, fp8e5m2T}, SpecificationVersion::V_1_1_DRAFT}}},
|
|
{{Extension::bf16},
|
|
{{{bf16T, i32T, i32T, bf16T}, SpecificationVersion::V_1_1_DRAFT}}},
|
|
{{Extension::int64},
|
|
{{{i8T, i64T, i32T, i8T}, SpecificationVersion::V_1_1_DRAFT},
|
|
{{i16T, i64T, i32T, i16T}, SpecificationVersion::V_1_1_DRAFT},
|
|
{{i32T, i64T, i32T, i32T}, SpecificationVersion::V_1_1_DRAFT},
|
|
{{i64T, i64T, i32T, i64T}, SpecificationVersion::V_1_1_DRAFT},
|
|
{{fp16T, i64T, i32T, fp16T}, SpecificationVersion::V_1_1_DRAFT},
|
|
{{fp32T, i64T, i32T, fp32T}, SpecificationVersion::V_1_1_DRAFT},
|
|
{{boolT, i64T, i32T, boolT}, SpecificationVersion::V_1_1_DRAFT}}},
|
|
{{Extension::fp8e4m3, Extension::int64},
|
|
{{{fp8e4m3T, i64T, i32T, fp8e4m3T}, SpecificationVersion::V_1_1_DRAFT}},
|
|
allOf},
|
|
{{Extension::fp8e5m2, Extension::int64},
|
|
{{{fp8e5m2T, i64T, i32T, fp8e5m2T}, SpecificationVersion::V_1_1_DRAFT}},
|
|
allOf},
|
|
{{Extension::bf16, Extension::int64},
|
|
{{{bf16T, i64T, i32T, bf16T}, SpecificationVersion::V_1_1_DRAFT}},
|
|
allOf},
|
|
{{Extension::mxfp},
|
|
{{{fp4e2m1T, fp8ue8m0T, i32T, i32T, fp4e2m1T, fp8ue8m0T},
|
|
SpecificationVersion::V_1_1_DRAFT},
|
|
{{fp6e2m3T, fp8ue8m0T, i32T, i32T, fp6e2m3T, fp8ue8m0T},
|
|
SpecificationVersion::V_1_1_DRAFT},
|
|
{{fp6e3m2T, fp8ue8m0T, i32T, i32T, fp6e3m2T, fp8ue8m0T},
|
|
SpecificationVersion::V_1_1_DRAFT},
|
|
{{fp8e4m3T, fp8ue8m0T, i32T, i32T, fp8e4m3T, fp8ue8m0T},
|
|
SpecificationVersion::V_1_1_DRAFT},
|
|
{{fp8e5m2T, fp8ue8m0T, i32T, i32T, fp8e5m2T, fp8ue8m0T},
|
|
SpecificationVersion::V_1_1_DRAFT},
|
|
{{mxint8T, fp8ue8m0T, i32T, i32T, mxint8T, fp8ue8m0T},
|
|
SpecificationVersion::V_1_1_DRAFT}}},
|
|
{{Extension::mxfp, Extension::int64},
|
|
{{{fp4e2m1T, fp8ue8m0T, i64T, i32T, fp4e2m1T, fp8ue8m0T},
|
|
SpecificationVersion::V_1_1_DRAFT},
|
|
{{fp6e2m3T, fp8ue8m0T, i64T, i32T, fp6e2m3T, fp8ue8m0T},
|
|
SpecificationVersion::V_1_1_DRAFT},
|
|
{{fp6e3m2T, fp8ue8m0T, i64T, i32T, fp6e3m2T, fp8ue8m0T},
|
|
SpecificationVersion::V_1_1_DRAFT},
|
|
{{fp8e4m3T, fp8ue8m0T, i64T, i32T, fp8e4m3T, fp8ue8m0T},
|
|
SpecificationVersion::V_1_1_DRAFT},
|
|
{{fp8e5m2T, fp8ue8m0T, i64T, i32T, fp8e5m2T, fp8ue8m0T},
|
|
SpecificationVersion::V_1_1_DRAFT},
|
|
{{mxint8T, fp8ue8m0T, i64T, i32T, mxint8T, fp8ue8m0T},
|
|
SpecificationVersion::V_1_1_DRAFT}},
|
|
allOf}}},
|
|
{"tosa.scatter",
|
|
{{{Extension::fp8e4m3},
|
|
{{{fp8e4m3T, i32T, fp8e4m3T, fp8e4m3T}, SpecificationVersion::V_1_0}}},
|
|
{{Extension::fp8e5m2},
|
|
{{{fp8e5m2T, i32T, fp8e5m2T, fp8e5m2T}, SpecificationVersion::V_1_0}}},
|
|
{{Extension::bf16},
|
|
{{{bf16T, i32T, bf16T, bf16T}, SpecificationVersion::V_1_0}}},
|
|
{{Extension::int64},
|
|
{{{i8T, i64T, i8T, i8T}, SpecificationVersion::V_1_1_DRAFT},
|
|
{{i16T, i64T, i16T, i16T}, SpecificationVersion::V_1_1_DRAFT},
|
|
{{i32T, i64T, i32T, i32T}, SpecificationVersion::V_1_1_DRAFT},
|
|
{{i64T, i64T, i64T, i64T}, SpecificationVersion::V_1_1_DRAFT},
|
|
{{fp16T, i64T, fp16T, fp16T}, SpecificationVersion::V_1_1_DRAFT},
|
|
{{fp32T, i64T, fp32T, fp32T}, SpecificationVersion::V_1_1_DRAFT},
|
|
{{boolT, i64T, boolT, boolT}, SpecificationVersion::V_1_1_DRAFT}}},
|
|
{{Extension::fp8e4m3, Extension::int64},
|
|
{{{fp8e4m3T, i64T, fp8e4m3T, fp8e4m3T},
|
|
SpecificationVersion::V_1_1_DRAFT}},
|
|
allOf},
|
|
{{Extension::fp8e5m2, Extension::int64},
|
|
{{{fp8e5m2T, i64T, fp8e5m2T, fp8e5m2T},
|
|
SpecificationVersion::V_1_1_DRAFT}},
|
|
allOf},
|
|
{{Extension::bf16, Extension::int64},
|
|
{{{bf16T, i64T, bf16T, bf16T}, SpecificationVersion::V_1_1_DRAFT}},
|
|
allOf}}},
|
|
{"tosa.resize",
|
|
{{{Extension::int16},
|
|
{{{i16T, i48T}, SpecificationVersion::V_1_0},
|
|
{{i16T, i16T}, SpecificationVersion::V_1_0}}},
|
|
{{Extension::bf16}, {{{bf16T, bf16T}, SpecificationVersion::V_1_0}}}}},
|
|
{"tosa.cast",
|
|
{{{Extension::bf16},
|
|
{{{i8T, bf16T}, SpecificationVersion::V_1_0},
|
|
{{i16T, bf16T}, SpecificationVersion::V_1_0},
|
|
{{i32T, bf16T}, SpecificationVersion::V_1_0},
|
|
{{bf16T, i8T}, SpecificationVersion::V_1_0},
|
|
{{bf16T, i16T}, SpecificationVersion::V_1_0},
|
|
{{bf16T, i32T}, SpecificationVersion::V_1_0},
|
|
{{bf16T, fp32T}, SpecificationVersion::V_1_0},
|
|
{{fp32T, bf16T}, SpecificationVersion::V_1_0}}},
|
|
{{Extension::int64},
|
|
{{{i32T, i64T}, SpecificationVersion::V_1_1_DRAFT},
|
|
{{i64T, i32T}, SpecificationVersion::V_1_1_DRAFT},
|
|
{{boolT, i64T}, SpecificationVersion::V_1_1_DRAFT},
|
|
{{i64T, boolT}, SpecificationVersion::V_1_1_DRAFT}}},
|
|
{{Extension::bf16, Extension::fp8e4m3},
|
|
{{{bf16T, fp8e4m3T}, SpecificationVersion::V_1_0},
|
|
{{fp8e4m3T, bf16T}, SpecificationVersion::V_1_0}},
|
|
allOf},
|
|
{{Extension::bf16, Extension::fp8e5m2},
|
|
{{{bf16T, fp8e5m2T}, SpecificationVersion::V_1_0},
|
|
{{fp8e5m2T, bf16T}, SpecificationVersion::V_1_0}},
|
|
allOf},
|
|
{{Extension::fp8e4m3},
|
|
{{{fp8e4m3T, fp16T}, SpecificationVersion::V_1_0},
|
|
{{fp8e4m3T, fp32T}, SpecificationVersion::V_1_0},
|
|
{{fp16T, fp8e4m3T}, SpecificationVersion::V_1_0},
|
|
{{fp32T, fp8e4m3T}, SpecificationVersion::V_1_0}}},
|
|
{{Extension::fp8e5m2},
|
|
{{{fp8e5m2T, fp16T}, SpecificationVersion::V_1_0},
|
|
{{fp8e5m2T, fp32T}, SpecificationVersion::V_1_0},
|
|
{{fp16T, fp8e5m2T}, SpecificationVersion::V_1_0},
|
|
{{fp32T, fp8e5m2T}, SpecificationVersion::V_1_0}}}}},
|
|
{"tosa.cast_from_block_scaled",
|
|
{{{Extension::bf16, Extension::mxfp},
|
|
{{{fp4e2m1T, fp8ue8m0T, bf16T}, SpecificationVersion::V_1_1_DRAFT},
|
|
{{fp6e2m3T, fp8ue8m0T, bf16T}, SpecificationVersion::V_1_1_DRAFT},
|
|
{{fp6e3m2T, fp8ue8m0T, bf16T}, SpecificationVersion::V_1_1_DRAFT},
|
|
{{fp8e4m3T, fp8ue8m0T, bf16T}, SpecificationVersion::V_1_1_DRAFT},
|
|
{{fp8e5m2T, fp8ue8m0T, bf16T}, SpecificationVersion::V_1_1_DRAFT},
|
|
{{mxint8T, fp8ue8m0T, bf16T}, SpecificationVersion::V_1_1_DRAFT}},
|
|
allOf},
|
|
{{Extension::mxfp},
|
|
{{{fp4e2m1T, fp8ue8m0T, fp32T}, SpecificationVersion::V_1_1_DRAFT},
|
|
{{fp6e2m3T, fp8ue8m0T, fp32T}, SpecificationVersion::V_1_1_DRAFT},
|
|
{{fp6e3m2T, fp8ue8m0T, fp32T}, SpecificationVersion::V_1_1_DRAFT},
|
|
{{fp8e4m3T, fp8ue8m0T, fp32T}, SpecificationVersion::V_1_1_DRAFT},
|
|
{{fp8e5m2T, fp8ue8m0T, fp32T}, SpecificationVersion::V_1_1_DRAFT},
|
|
{{mxint8T, fp8ue8m0T, fp32T}, SpecificationVersion::V_1_1_DRAFT}}}}},
|
|
{"tosa.cast_to_block_scaled",
|
|
{{{Extension::mxfp},
|
|
{{{bf16T, fp4e2m1T, fp8ue8m0T}, SpecificationVersion::V_1_1_DRAFT},
|
|
{{fp32T, fp4e2m1T, fp8ue8m0T}, SpecificationVersion::V_1_1_DRAFT},
|
|
{{fp32T, fp6e2m3T, fp8ue8m0T}, SpecificationVersion::V_1_1_DRAFT},
|
|
{{fp32T, fp6e3m2T, fp8ue8m0T}, SpecificationVersion::V_1_1_DRAFT},
|
|
{{fp32T, fp8e4m3T, fp8ue8m0T}, SpecificationVersion::V_1_1_DRAFT},
|
|
{{fp32T, fp8e5m2T, fp8ue8m0T}, SpecificationVersion::V_1_1_DRAFT},
|
|
{{fp32T, mxint8T, fp8ue8m0T}, SpecificationVersion::V_1_1_DRAFT}}},
|
|
{{Extension::bf16, Extension::mxfp},
|
|
{{{bf16T, fp6e2m3T, fp8ue8m0T}, SpecificationVersion::V_1_1_DRAFT},
|
|
{{bf16T, fp6e3m2T, fp8ue8m0T}, SpecificationVersion::V_1_1_DRAFT},
|
|
{{bf16T, fp8e4m3T, fp8ue8m0T}, SpecificationVersion::V_1_1_DRAFT},
|
|
{{bf16T, fp8e5m2T, fp8ue8m0T}, SpecificationVersion::V_1_1_DRAFT},
|
|
{{bf16T, mxint8T, fp8ue8m0T}, SpecificationVersion::V_1_1_DRAFT}},
|
|
allOf}}},
|
|
{"tosa.rescale",
|
|
{{{Extension::int16},
|
|
{{{i48T, i48T, i8T, i8T}, SpecificationVersion::V_1_0},
|
|
{{i48T, i48T, i16T, i16T}, SpecificationVersion::V_1_0},
|
|
{{i48T, i48T, i32T, i32T}, SpecificationVersion::V_1_0}}}}},
|
|
{"tosa.const",
|
|
{{{Extension::int4}, {{{i4T}, SpecificationVersion::V_1_0}}},
|
|
{{Extension::int16}, {{{i48T}, SpecificationVersion::V_1_0}}},
|
|
{{Extension::int64}, {{{i64T}, SpecificationVersion::V_1_1_DRAFT}}},
|
|
{{Extension::fp8e4m3}, {{{fp8e4m3T}, SpecificationVersion::V_1_0}}},
|
|
{{Extension::fp8e5m2}, {{{fp8e5m2T}, SpecificationVersion::V_1_0}}},
|
|
{{Extension::bf16}, {{{bf16T}, SpecificationVersion::V_1_0}}},
|
|
{{Extension::mxfp},
|
|
{{{fp8ue8m0T}, SpecificationVersion::V_1_1_DRAFT},
|
|
{{fp6e3m2T}, SpecificationVersion::V_1_1_DRAFT},
|
|
{{fp6e2m3T}, SpecificationVersion::V_1_1_DRAFT},
|
|
{{fp4e2m1T}, SpecificationVersion::V_1_1_DRAFT},
|
|
{{mxint8T}, SpecificationVersion::V_1_1_DRAFT}}}}},
|
|
{"tosa.identity",
|
|
{{{Extension::int4}, {{{i4T, i4T}, SpecificationVersion::V_1_0}}},
|
|
{{Extension::int16}, {{{i48T, i48T}, SpecificationVersion::V_1_0}}},
|
|
{{Extension::int64}, {{{i64T, i64T}, SpecificationVersion::V_1_1_DRAFT}}},
|
|
{{Extension::fp8e4m3},
|
|
{{{fp8e4m3T, fp8e4m3T}, SpecificationVersion::V_1_0}}},
|
|
{{Extension::fp8e5m2},
|
|
{{{fp8e5m2T, fp8e5m2T}, SpecificationVersion::V_1_0}}},
|
|
{{Extension::bf16}, {{{bf16T, bf16T}, SpecificationVersion::V_1_0}}}}},
|
|
{"tosa.variable",
|
|
{{{Extension::variable},
|
|
{{{i8T}, SpecificationVersion::V_1_0},
|
|
{{fp16T}, SpecificationVersion::V_1_0},
|
|
{{fp32T}, SpecificationVersion::V_1_0}}}}},
|
|
{"tosa.variable_write",
|
|
{{{Extension::variable},
|
|
{{{i8T}, SpecificationVersion::V_1_0},
|
|
{{fp16T}, SpecificationVersion::V_1_0},
|
|
{{fp32T}, SpecificationVersion::V_1_0}}}}},
|
|
{"tosa.variable_read",
|
|
{{{Extension::variable},
|
|
{{{i8T}, SpecificationVersion::V_1_0},
|
|
{{fp16T}, SpecificationVersion::V_1_0},
|
|
{{fp32T}, SpecificationVersion::V_1_0}}}}},
|
|
{"tosa.dim",
|
|
{{{Extension::shape},
|
|
{{{boolT}, SpecificationVersion::V_1_1_DRAFT},
|
|
{{i8T}, SpecificationVersion::V_1_1_DRAFT},
|
|
{{i16T}, SpecificationVersion::V_1_1_DRAFT},
|
|
{{i32T}, SpecificationVersion::V_1_1_DRAFT},
|
|
{{fp16T}, SpecificationVersion::V_1_1_DRAFT},
|
|
{{fp32T}, SpecificationVersion::V_1_1_DRAFT}}},
|
|
{{Extension::fp8e4m3, Extension::shape},
|
|
{{{fp8e4m3T}, SpecificationVersion::V_1_1_DRAFT}},
|
|
allOf},
|
|
{{Extension::fp8e5m2, Extension::shape},
|
|
{{{fp8e5m2T}, SpecificationVersion::V_1_1_DRAFT}},
|
|
allOf},
|
|
{{Extension::bf16, Extension::shape},
|
|
{{{bf16T}, SpecificationVersion::V_1_1_DRAFT}},
|
|
allOf},
|
|
{{Extension::mxfp, Extension::shape},
|
|
{{{fp6e3m2T}, SpecificationVersion::V_1_1_DRAFT},
|
|
{{fp6e2m3T}, SpecificationVersion::V_1_1_DRAFT},
|
|
{{fp4e2m1T}, SpecificationVersion::V_1_1_DRAFT},
|
|
{{mxint8T}, SpecificationVersion::V_1_1_DRAFT},
|
|
{{fp8ue8m0T}, SpecificationVersion::V_1_1_DRAFT}},
|
|
allOf},
|
|
{{Extension::int64, Extension::shape},
|
|
{{{i64T}, SpecificationVersion::V_1_1_DRAFT}},
|
|
allOf}}}};
|
|
|
|
// End of auto-generated metadata
|