Skip to content

Commit 56f7e43

Browse files
author
mingyue
committed
EPContext node support bfloat16
1 parent 8800830 commit 56f7e43

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

onnxruntime/core/graph/contrib_ops/contrib_defs.cc

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3371,7 +3371,8 @@ void RegisterContribSchemas() {
33713371
"tensor(uint64)",
33723372
"tensor(float16)",
33733373
"tensor(float)",
3374-
"tensor(double)"},
3374+
"tensor(double)",
3375+
"tensor(bfloat16)"},
33753376
"Constrain input and output types.");
33763377

33773378
static const char* BitmaskDropout_ver1_doc = R"DOC(

0 commit comments

Comments
 (0)