Skip to content

Commit 30a4457

Browse files
sxjsciencesxjscience
committed
remove float16
1 parent 1f805d9 commit 30a4457

File tree

1 file changed

+0
-2
lines changed

1 file changed

+0
-2
lines changed

‎src/operator/tensor/ordering_op-inl.h‎

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -83,7 +83,6 @@ struct TopKParam : public dmlc::Parameter<TopKParam> {
8383
DMLC_DECLARE_FIELD(dtype)
8484
.add_enum("uint8", mshadow::kUint8)
8585
.add_enum("int32", mshadow::kInt32)
86-
.add_enum("float16", mshadow::kFloat16)
8786
.add_enum("float32", mshadow::kFloat32)
8887
.add_enum("float64", mshadow::kFloat64)
8988
.set_default(mshadow::kFloat32)
@@ -118,7 +117,6 @@ struct ArgSortParam : public dmlc::Parameter<ArgSortParam> {
118117
DMLC_DECLARE_FIELD(dtype)
119118
.add_enum("uint8", mshadow::kUint8)
120119
.add_enum("int32", mshadow::kInt32)
121-
.add_enum("float16", mshadow::kFloat16)
122120
.add_enum("float32", mshadow::kFloat32)
123121
.add_enum("float64", mshadow::kFloat64)
124122
.set_default(mshadow::kFloat32)

0 commit comments

Comments
 (0)