From 96b33f45976724eb0a92fa4eb39018f2f4f74315 Mon Sep 17 00:00:00 2001 From: Pranav Sharma Date: Sat, 19 Oct 2019 14:11:19 -0700 Subject: [PATCH] Add support for double to SplitToSequence op to allow test_sequence_model7 to pass. (#2173) --- onnxruntime/core/providers/cpu/sequence/sequence_ops.cc | 3 +++ 1 file changed, 3 insertions(+) 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())