packages/core/src/architectures/owlv2.js (180 lines of code) (raw):
// NOTE: This file has been auto-generated. Do not edit directly.
export default {
model_type: "owlv2",
models: [
{
model_id: "Xenova/owlv2-base-patch16",
dtype: "quantized",
architectures: ["Owlv2ForObjectDetection"],
ops: [
"Abs",
"Add",
"ArgMax",
"Cast",
"Concat",
"Constant",
"ConstantOfShape",
"ConvInteger",
"DequantizeLinear",
"Div",
"DynamicQuantizeLinear",
"Einsum",
"Elu",
"Equal",
"Erf",
"Expand",
"Flatten",
"Gather",
"Greater",
"Less",
"Log",
"MatMul",
"MatMulInteger",
"Mul",
"Neg",
"Pow",
"Range",
"Reciprocal",
"ReduceMean",
"ReduceSum",
"Reshape",
"Shape",
"Sigmoid",
"Slice",
"Softmax",
"Sqrt",
"Squeeze",
"Sub",
"Transpose",
"Unsqueeze",
"Where",
],
},
{
model_id: "Xenova/owlv2-base-patch16",
dtype: "fp32",
architectures: ["Owlv2ForObjectDetection"],
ops: [
"Abs",
"Add",
"ArgMax",
"Cast",
"Concat",
"Constant",
"ConstantOfShape",
"Conv",
"Div",
"Einsum",
"Elu",
"Equal",
"Erf",
"Expand",
"Flatten",
"Gather",
"Greater",
"Less",
"Log",
"MatMul",
"Mul",
"Neg",
"Pow",
"Range",
"Reciprocal",
"ReduceMean",
"ReduceSum",
"Reshape",
"Shape",
"Sigmoid",
"Slice",
"Softmax",
"Sqrt",
"Squeeze",
"Sub",
"Transpose",
"Unsqueeze",
"Where",
],
},
{
model_id:
"onnx-internal-testing/tiny-random-Owlv2ForObjectDetection-ONNX",
dtype: "fp32",
architectures: ["Owlv2ForObjectDetection"],
ops: [
"Add",
"ArgMax",
"Cast",
"Concat",
"Constant",
"ConstantOfShape",
"Conv",
"Div",
"Einsum",
"Elu",
"Equal",
"Erf",
"Expand",
"Flatten",
"Gather",
"Greater",
"Identity",
"Less",
"MatMul",
"Mul",
"Pow",
"Range",
"ReduceL2",
"ReduceMean",
"Reshape",
"Shape",
"Sigmoid",
"Slice",
"Softmax",
"Sqrt",
"Squeeze",
"Sub",
"Transpose",
"Unsqueeze",
"Where",
],
},
{
model_id: "onnx-internal-testing/tiny-random-Owlv2Model-ONNX",
dtype: "fp32",
architectures: ["Owlv2Model"],
ops: [
"Add",
"ArgMax",
"Cast",
"Concat",
"Constant",
"ConstantOfShape",
"Conv",
"Div",
"Equal",
"Exp",
"Expand",
"Flatten",
"Gather",
"Identity",
"Less",
"MatMul",
"Mul",
"Pow",
"Range",
"ReduceL2",
"ReduceMean",
"Reshape",
"Shape",
"Sigmoid",
"Slice",
"Softmax",
"Sqrt",
"Squeeze",
"Sub",
"Transpose",
"Unsqueeze",
"Where",
],
},
],
};