We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 8800830 commit 56f7e43Copy full SHA for 56f7e43
onnxruntime/core/graph/contrib_ops/contrib_defs.cc
@@ -3371,7 +3371,8 @@ void RegisterContribSchemas() {
3371
"tensor(uint64)",
3372
"tensor(float16)",
3373
"tensor(float)",
3374
- "tensor(double)"},
+ "tensor(double)",
3375
+ "tensor(bfloat16)"},
3376
"Constrain input and output types.");
3377
3378
static const char* BitmaskDropout_ver1_doc = R"DOC(
0 commit comments