saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-29 20:14:01 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Add support for double to SplitToSequence op to allow test_sequence_model7 to pass. (#2173)

Browse source
This commit is contained in:
Pranav Sharma 2019-10-19 14:11:19 -07:00 committed by GitHub
parent f8c30b8aa9
commit 96b33f4597
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
onnxruntime/core/providers/cpu/sequence/sequence_ops.cc
View file

@ -327,6 +327,7 @@ ONNX_CPU_OPERATOR_KERNEL(
KernelDefBuilder().TypeConstraint("T",
std::vector<MLDataType>{
DataTypeImpl::GetTensorType<float>(),
DataTypeImpl::GetTensorType<double>(),
DataTypeImpl::GetTensorType<int32_t>(),
DataTypeImpl::GetTensorType<std::string>()})
.TypeConstraint("S", DataTypeImpl::AllSequenceTensorTypes())
@ -349,6 +350,8 @@ Status SplitToSequence::Compute(OpKernelContext* context) const {
if (data_type == DataTypeImpl::GetType<float>())
status = ComputeImpl<float>(*context, input, p_split_input);
else if (data_type == DataTypeImpl::GetType<double>())
status = ComputeImpl<double>(*context, input, p_split_input);
else if (data_type == DataTypeImpl::GetType<int32_t>())
status = ComputeImpl<int32_t>(*context, input, p_split_input);
else if (data_type == DataTypeImpl::GetType<std::string>())