diff --git a/onnxruntime/core/providers/cpu/sequence/sequence_ops.cc b/onnxruntime/core/providers/cpu/sequence/sequence_ops.cc index c9ab58540d..88130dc22c 100644 --- a/onnxruntime/core/providers/cpu/sequence/sequence_ops.cc +++ b/onnxruntime/core/providers/cpu/sequence/sequence_ops.cc @@ -327,6 +327,7 @@ ONNX_CPU_OPERATOR_KERNEL( KernelDefBuilder().TypeConstraint("T", std::vector{ DataTypeImpl::GetTensorType(), + DataTypeImpl::GetTensorType(), DataTypeImpl::GetTensorType(), DataTypeImpl::GetTensorType()}) .TypeConstraint("S", DataTypeImpl::AllSequenceTensorTypes()) @@ -349,6 +350,8 @@ Status SplitToSequence::Compute(OpKernelContext* context) const { if (data_type == DataTypeImpl::GetType()) status = ComputeImpl(*context, input, p_split_input); + else if (data_type == DataTypeImpl::GetType()) + status = ComputeImpl(*context, input, p_split_input); else if (data_type == DataTypeImpl::GetType()) status = ComputeImpl(*context, input, p_split_input); else if (data_type == DataTypeImpl::GetType())