File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -193,17 +193,17 @@ ContextProperties ContextImplOrt::GetContextProperties() {
193193 /* sigmoid_input=*/ {DataTypeConstraint::kFloat16To32 , kMaxRank },
194194 /* slice_input=*/
195195 {DataTypeConstraint::kAllDataTypesAtLeast8bits , kMaxRank },
196- /* softmax_input=*/ {DataTypeConstraint::kFloat16To32 , kMaxRank },
196+ /* softmax_input=*/ {DataTypeConstraint::kFloat16To32 , kNonScalarMaxRank },
197197 /* softplus_input=*/ {},
198198 /* softsign_input=*/ {},
199199 /* split_input=*/
200- {DataTypeConstraint::kAllDataTypesAtLeast8bits , kMaxRank },
200+ {DataTypeConstraint::kAllDataTypesAtLeast8bits , kNonScalarMaxRank },
201201 /* tanh_input=*/ {},
202202 /* tile_input=*/ {},
203203 /* transpose_input=*/
204204 {DataTypeConstraint::kAllDataTypesAtLeast8bits , kMaxRank },
205205 /* triangular_input=*/
206- {DataTypeConstraint::kAllDataTypesAtLeast8bits , kMaxRank },
206+ {DataTypeConstraint::kAllDataTypesAtLeast8bits , { 2 , 8 } },
207207 /* where_condition=*/ {DataTypeConstraint::kUint8 , kMaxRank },
208208 /* where_value=*/
209209 {DataTypeConstraint::kAllDataTypesAtLeast8bits , kMaxRank }});
You can’t perform that action at this time.
0 commit comments