saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-05-29 23:06:41 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Enable tests for EyeLike and enable datatypes present in tests (#424)

Browse source 
* Enable tests for EyeLike and enable datatypes present in tests

* fix failure
This commit is contained in:
Ashwini Khade 2019-02-01 00:21:08 -08:00 committed by Pranav Sharma
parent 9f0298261d
commit 60cdb79204
2 changed files with 18 additions and 13 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

26
onnxruntime/core/providers/cpu/tensor/eye_like.cc
View file

@ -15,15 +15,19 @@ ONNX_CPU_OPERATOR_KERNEL(
KernelDefBuilder().TypeConstraint("T1",
std::vector<MLDataType>{
DataTypeImpl::GetTensorType<float>(),
DataTypeImpl::GetTensorType<int64_t>(),
DataTypeImpl::GetTensorType<double>(),
DataTypeImpl::GetTensorType<uint64_t>(),
DataTypeImpl::GetTensorType<int64_t>(),
DataTypeImpl::GetTensorType<int32_t>()
})
.TypeConstraint("T2",
std::vector<MLDataType>{
DataTypeImpl::GetTensorType<float>(),
DataTypeImpl::GetTensorType<uint64_t>(),
DataTypeImpl::GetTensorType<int64_t>(),
}),
.TypeConstraint("T2",
std::vector<MLDataType>{
DataTypeImpl::GetTensorType<float>(),
DataTypeImpl::GetTensorType<double>(),
DataTypeImpl::GetTensorType<uint64_t>(),
DataTypeImpl::GetTensorType<int64_t>(),
DataTypeImpl::GetTensorType<int32_t>()
}),
EyeLike);
Status EyeLike::Compute(OpKernelContext* context) const {
@ -34,10 +38,14 @@ Status EyeLike::Compute(OpKernelContext* context) const {
switch (output_tensor_dtype) {
case onnx::TensorProto_DataType_FLOAT:
return ComputeImpl<float>(context, T1);
case onnx::TensorProto_DataType_INT64:
return ComputeImpl<int64_t>(context, T1);
case onnx::TensorProto_DataType_DOUBLE:
return ComputeImpl<double>(context, T1);
case onnx::TensorProto_DataType_INT32:
return ComputeImpl<int32_t>(context, T1);
case onnx::TensorProto_DataType_UINT64:
return ComputeImpl<uint64_t>(context, T1);
case onnx::TensorProto_DataType_INT64:
return ComputeImpl<int64_t>(context, T1);
default:
ORT_THROW("Unsupported 'dtype' value: ", output_tensor_dtype);
}

5
onnxruntime/test/onnx/main.cc
View file

@ -274,10 +274,7 @@ int real_main(int argc, char* argv[]) {
{"PoissonNLLLLoss_no_reduce", "disable reason"},
{"Softsign", "disable reason"},
{"convtranspose_1d", "disable reason"},
{"convtranspose_3d", "disable reason"},
{"eyelike_populate_off_main_diagonal", "disable reason"},
{"eyelike_with_dtype", "disable reason"},
{"eyelike_without_dtype", "disable reason"},
{"convtranspose_3d", "disable reason"},
{"flatten_axis0", "disable reason"},
{"flatten_axis1", "disable reason"},
{"flatten_axis2", "disable reason"},