Skip to content

GitLab

Unverified Commit 913ae362 authored by Chris Austen's avatar Chris Austen Committed by GitHub
Browse files

Merge branch 'develop' into optimize

parents f1e16656 b8c8d09b
Hide whitespace changes
Inline Side-by-side
Showing with 894 additions and 397 deletions
src/targets/ref/lowering.cpp
View file @ 913ae362
...@@ -383,9 +383,9 @@ struct ref_gemm ...@@ -383,9 +383,9 @@ struct ref_gemm
std::string name() const { return "ref::dot"; } std::string name() const { return "ref::dot"; }
shape compute_shape(const std::vector<shape>& inputs) const { return op.compute_shape(inputs); } shape compute_shape(const std::vector<shape>& inputs) const { return op.compute_shape(inputs); }
argument compute(context&, const shape& output_shape, std::vector<argument> args) const argument compute(context&, const dyn_output& dyn_out, std::vector<argument> args) const
{ {
argument result{output_shape}; argument result{dyn_out.computed_shape};
migemm(result, args[0], args[1], 1.0f, 0.0f); migemm(result, args[0], args[1], 1.0f, 0.0f);
return result; return result;
...@@ -449,10 +449,10 @@ struct ref_softmax : auto_register_op<ref_softmax<Op>> ...@@ -449,10 +449,10 @@ struct ref_softmax : auto_register_op<ref_softmax<Op>>
{ {
return op.normalize_compute_shape(inputs); return op.normalize_compute_shape(inputs);
} }
argument compute(context&, const shape& output_shape, std::vector<argument> args) const argument compute(context&, const dyn_output& dyn_out, std::vector<argument> args) const
{ {
argument result{output_shape}; argument result{dyn_out.computed_shape};
auto batch_lens = output_shape.lens(); auto batch_lens = dyn_out.computed_shape.lens();
int64_t tuned_axis = tune_axis(args[0].get_shape().lens().size(), op.axis, op.name()); int64_t tuned_axis = tune_axis(args[0].get_shape().lens().size(), op.axis, op.name());
std::size_t n_dims = batch_lens[tuned_axis]; std::size_t n_dims = batch_lens[tuned_axis];
batch_lens[tuned_axis] = 1; batch_lens[tuned_axis] = 1;
...@@ -475,7 +475,7 @@ struct ref_softmax : auto_register_op<ref_softmax<Op>> ...@@ -475,7 +475,7 @@ struct ref_softmax : auto_register_op<ref_softmax<Op>>
for(std::size_t j = 0; j < n_dims; ++j) for(std::size_t j = 0; j < n_dims; ++j)
{ {
idx[tuned_axis] = j; idx[tuned_axis] = j;
std::size_t index = output_shape.index(idx); std::size_t index = dyn_out.computed_shape.index(idx);
output[index] = std::exp(input[index] - batch_max[i]); output[index] = std::exp(input[index] - batch_max[i]);
} }
......
test/gpu/hip.cpp 0 → 100644
View file @ 913ae362
/*
* The MIT License (MIT)
*
* Copyright (c) 2015-2022 Advanced Micro Devices, Inc. All rights reserved.
*
* Permission is hereby granted, free of charge, to any person obtaining a copy
* of this software and associated documentation files (the "Software"), to deal
* in the Software without restriction, including without limitation the rights
* to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
* copies of the Software, and to permit persons to whom the Software is
* furnished to do so, subject to the following conditions:
*
* The above copyright notice and this permission notice shall be included in
* all copies or substantial portions of the Software.
*
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
* AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
* OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
* THE SOFTWARE.
*/
#include <test.hpp>
#include <migraphx/argument.hpp>
#include <migraphx/gpu/hip.hpp>
#include <migraphx/gpu/target.hpp>
TEST_CASE(tuple_to_from_gpu)
{
migraphx::shape s1{migraphx::shape::float_type, {2, 3}};
migraphx::shape s2{migraphx::shape::int32_type, {2, 4}};
std::vector<float> p1_data = {1.1, 2.2, 3.3, 4.4, 5.5, 6.6};
std::vector<int> p2_data = {1, 2, 3, 4, 5, 6, 7, 8};
auto p1 = migraphx::argument{s1, p1_data.data()};
auto p2 = migraphx::argument{s2, p2_data.data()};
auto p1_gpu = migraphx::gpu::to_gpu(p1);
auto p2_gpu = migraphx::gpu::to_gpu(p2);
auto p_tuple = migraphx::gpu::from_gpu(migraphx::argument({p1_gpu, p2_gpu}));
std::vector<migraphx::argument> results = p_tuple.get_sub_objects();
std::vector<float> result1;
results[0].visit([&](auto output) { result1.assign(output.begin(), output.end()); });
std::vector<int> result2;
results[1].visit([&](auto output) { result2.assign(output.begin(), output.end()); });
EXPECT(result1 == p1_data);
EXPECT(result2 == p2_data);
}
int main(int argc, const char* argv[]) { test::run(argc, argv); }
test/gpu/mlir.cpp
View file @ 913ae362
...@@ -140,7 +140,7 @@ TEST_CASE(conv) ...@@ -140,7 +140,7 @@ TEST_CASE(conv)
{ {
const std::string mlir_output = R"__migraphx__( const std::string mlir_output = R"__migraphx__(
module { module {
func.func @main(%arg0: tensor<2x8x3x3xf32>, %arg1: tensor<1x8x4x4xf32>) -> tensor<1x2x2x2xf32> attributes {kernel = "mixr"} { func.func @main(%arg0: tensor<2x8x3x3xf32>, %arg1: tensor<1x8x4x4xf32>) -> tensor<1x2x2x2xf32> attributes {arch = "", kernel = "mixr"} {
%0 = migraphx.convolution(%arg1, %arg0) {dilation = [1, 1], group = 1 : i64, padding = [0, 0, 0, 0], padding_mode = 0 : i64, stride = [1, 1]} : (tensor<1x8x4x4xf32>, tensor<2x8x3x3xf32>) -> tensor<1x2x2x2xf32> %0 = migraphx.convolution(%arg1, %arg0) {dilation = [1, 1], group = 1 : i64, padding = [0, 0, 0, 0], padding_mode = 0 : i64, stride = [1, 1]} : (tensor<1x8x4x4xf32>, tensor<2x8x3x3xf32>) -> tensor<1x2x2x2xf32>
return %0 : tensor<1x2x2x2xf32> return %0 : tensor<1x2x2x2xf32>
} }
...@@ -163,7 +163,7 @@ TEST_CASE(conv_add_relu) ...@@ -163,7 +163,7 @@ TEST_CASE(conv_add_relu)
{ {
const std::string mlir_output = R"__migraphx__( const std::string mlir_output = R"__migraphx__(
module { module {
func.func @main(%arg0: tensor<1x2x2x2xf32>, %arg1: tensor<2x8x3x3xf32>, %arg2: tensor<1x8x4x4xf32>) -> tensor<1x2x2x2xf32> attributes {kernel = "mixr"} { func.func @main(%arg0: tensor<1x2x2x2xf32>, %arg1: tensor<2x8x3x3xf32>, %arg2: tensor<1x8x4x4xf32>) -> tensor<1x2x2x2xf32> attributes {arch = "", kernel = "mixr"} {
%0 = migraphx.convolution(%arg2, %arg1) {dilation = [1, 1], group = 1 : i64, padding = [0, 0, 0, 0], padding_mode = 0 : i64, stride = [1, 1]} : (tensor<1x8x4x4xf32>, tensor<2x8x3x3xf32>) -> tensor<1x2x2x2xf32> %0 = migraphx.convolution(%arg2, %arg1) {dilation = [1, 1], group = 1 : i64, padding = [0, 0, 0, 0], padding_mode = 0 : i64, stride = [1, 1]} : (tensor<1x8x4x4xf32>, tensor<2x8x3x3xf32>) -> tensor<1x2x2x2xf32>
%1 = migraphx.add(%0, %arg0) : (tensor<1x2x2x2xf32>, tensor<1x2x2x2xf32>) -> tensor<1x2x2x2xf32> %1 = migraphx.add(%0, %arg0) : (tensor<1x2x2x2xf32>, tensor<1x2x2x2xf32>) -> tensor<1x2x2x2xf32>
%2 = migraphx.relu(%1) : (tensor<1x2x2x2xf32>) -> tensor<1x2x2x2xf32> %2 = migraphx.relu(%1) : (tensor<1x2x2x2xf32>) -> tensor<1x2x2x2xf32>
......
src/include/migraphx/int_divide.hpp → test/instruction.cpp
View file @ 913ae362
...@@ -21,28 +21,31 @@ ...@@ -21,28 +21,31 @@
* OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
* THE SOFTWARE. * THE SOFTWARE.
*/ */
#ifndef MIGRAPHX_GUARD_RTGLIB_INT_DIVIDE_HPP
#define MIGRAPHX_GUARD_RTGLIB_INT_DIVIDE_HPP
#include <migraphx/config.hpp> #include <migraphx/instruction.hpp>
#include <cmath> #include <migraphx/program.hpp>
#include <migraphx/make_op.hpp>
#include "test.hpp"
namespace migraphx { TEST_CASE(check_undefined)
inline namespace MIGRAPHX_INLINE_NS {
template <class R, class T, class U>
R floor_divide(T x, U y)
{ {
return R(std::floor(double(x) / double(y))); migraphx::module m;
} auto und = m.add_instruction(migraphx::make_op("undefined"));
auto cov = m.add_instruction(
migraphx::make_op("convert", {{"target_type", migraphx::shape::half_type}}), und);
auto abs = m.add_instruction(migraphx::make_op("abs"), cov);
template <class R, class T, class U> migraphx::shape xs{migraphx::shape::float_type, {2, 3}};
R ceil_divide(T x, U y) std::vector<float> datax = {1, 2, 3, 4, 5, 6};
{
return R(std::ceil(double(x) / double(y))); auto lit = m.add_literal(migraphx::literal(xs, datax));
} auto mul = m.add_instruction(migraphx::make_op("mul"), lit, lit);
} // namespace MIGRAPHX_INLINE_NS EXPECT(und->is_undefined());
} // namespace migraphx EXPECT(cov->is_undefined());
EXPECT(abs->is_undefined());
EXPECT(not lit->is_undefined());
EXPECT(not mul->is_undefined());
}
#endif int main(int argc, const char* argv[]) { test::run(argc, argv); }
test/literal_test.cpp
View file @ 913ae362
...@@ -49,6 +49,25 @@ TEST_CASE(literal_test) ...@@ -49,6 +49,25 @@ TEST_CASE(literal_test)
EXPECT(l4.empty()); EXPECT(l4.empty());
} }
TEST_CASE(literal_nstd_shape_vector)
{
migraphx::shape nstd_shape{migraphx::shape::float_type, {1, 3, 2, 2}, {12, 1, 6, 3}};
std::vector<float> data(12);
std::iota(data.begin(), data.end(), 0);
auto l0 = migraphx::literal{nstd_shape, data};
// check data buffer is read in correctly
std::vector<float> expected_buffer = {0, 4, 8, 1, 5, 9, 2, 6, 10, 3, 7, 11};
const auto* start = reinterpret_cast<const float*>(l0.data());
std::vector<float> l0_data{start, start + 12};
EXPECT(l0_data == expected_buffer);
// check that using visit() (that uses a tensor view) gives data in correct order
std::vector<float> results_vector(12);
l0.visit([&](auto output) { results_vector.assign(output.begin(), output.end()); });
EXPECT(results_vector == data);
}
TEST_CASE(literal_os1) TEST_CASE(literal_os1)
{ {
migraphx::literal l{1}; migraphx::literal l{1};
......
test/onnx/external_constant_test.onnx 0 → 100644
View file @ 913ae362
external_constant_test:¡
v0"Constant*g
value*[B const_tensorj)
locationexternal_constant_test.weightj
offset48j
length24p external_constant_testb
0

B
\ No newline at end of file
test/onnx/gen_onnx.py
View file @ 913ae362
...@@ -28,28 +28,37 @@ import numpy as np ...@@ -28,28 +28,37 @@ import numpy as np
import onnx import onnx
from onnx import helper from onnx import helper
from onnx import TensorProto from onnx import TensorProto
from onnx.numpy_helper import from_array
def onnx_test(op_test):
def run_test(): def onnx_test(external_data=False):
op_info = op_test() def create_onnx_test(op_test):
if len(op_info) > 3: def run_test():
graph_def = helper.make_graph(op_info[0], op_info = op_test()
op_test.__name__, if len(op_info) > 3:
op_info[1], graph_def = helper.make_graph(op_info[0],
op_info[2], op_test.__name__,
initializer=op_info[3]) op_info[1],
else: op_info[2],
graph_def = helper.make_graph(op_info[0], op_test.__name__, initializer=op_info[3])
op_info[1], op_info[2]) else:
model_def = helper.make_model(graph_def, graph_def = helper.make_graph(op_info[0], op_test.__name__,
producer_name=op_test.__name__) op_info[1], op_info[2])
onnx.save(model_def, '{}.onnx'.format(op_test.__name__)) model_def = helper.make_model(graph_def,
producer_name=op_test.__name__)
return run_test onnx.save_model(model_def,
'{}.onnx'.format(op_test.__name__),
save_as_external_data=external_data,
@onnx_test location='{}.weight'.format(op_test.__name__),
size_threshold=0,
convert_attribute=True)
return run_test
return create_onnx_test
@onnx_test()
def acos_test(): def acos_test():
x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [10]) x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [10])
y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [10]) y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [10])
...@@ -63,7 +72,7 @@ def acos_test(): ...@@ -63,7 +72,7 @@ def acos_test():
return ([node], [x], [y]) return ([node], [x], [y])
@onnx_test @onnx_test()
def acosh_test(): def acosh_test():
x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [10]) x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [10])
y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [10]) y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [10])
...@@ -77,7 +86,7 @@ def acosh_test(): ...@@ -77,7 +86,7 @@ def acosh_test():
return ([node], [x], [y]) return ([node], [x], [y])
@onnx_test @onnx_test()
def add_bcast_test(): def add_bcast_test():
x = helper.make_tensor_value_info('0', TensorProto.FLOAT, [2, 3, 4, 5]) x = helper.make_tensor_value_info('0', TensorProto.FLOAT, [2, 3, 4, 5])
y = helper.make_tensor_value_info('1', TensorProto.FLOAT, [3, 4]) y = helper.make_tensor_value_info('1', TensorProto.FLOAT, [3, 4])
...@@ -92,7 +101,7 @@ def add_bcast_test(): ...@@ -92,7 +101,7 @@ def add_bcast_test():
return ([node], [x, y], [z]) return ([node], [x, y], [z])
@onnx_test @onnx_test()
def add_fp16_test(): def add_fp16_test():
x = helper.make_tensor_value_info('0', TensorProto.FLOAT16, [1]) x = helper.make_tensor_value_info('0', TensorProto.FLOAT16, [1])
y = helper.make_tensor_value_info('1', TensorProto.FLOAT16, [1]) y = helper.make_tensor_value_info('1', TensorProto.FLOAT16, [1])
...@@ -115,7 +124,7 @@ def add_fp16_test(): ...@@ -115,7 +124,7 @@ def add_fp16_test():
]) ])
@onnx_test @onnx_test()
def add_scalar_test(): def add_scalar_test():
x = helper.make_tensor_value_info('0', TensorProto.UINT8, [2, 3, 4, 5]) x = helper.make_tensor_value_info('0', TensorProto.UINT8, [2, 3, 4, 5])
y = helper.make_tensor_value_info('1', TensorProto.UINT8, []) y = helper.make_tensor_value_info('1', TensorProto.UINT8, [])
...@@ -126,7 +135,7 @@ def add_scalar_test(): ...@@ -126,7 +135,7 @@ def add_scalar_test():
return ([node], [x, y], [z]) return ([node], [x, y], [z])
@onnx_test @onnx_test()
def argmax_test(): def argmax_test():
x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [3, 4, 5, 6]) x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [3, 4, 5, 6])
y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [3, 4, 6]) y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [3, 4, 6])
...@@ -140,7 +149,21 @@ def argmax_test(): ...@@ -140,7 +149,21 @@ def argmax_test():
return ([node], [x], [y]) return ([node], [x], [y])
@onnx_test @onnx_test()
def argmax_dyn_test():
x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [None, 4, 5, 6])
y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [None, 4, 6])
node = onnx.helper.make_node('ArgMax',
inputs=['x'],
outputs=['y'],
axis=2,
keepdims=0)
return ([node], [x], [y])
@onnx_test()
def argmin_test(): def argmin_test():
x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [3, 4, 5, 6]) x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [3, 4, 5, 6])
y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [3, 4, 5]) y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [3, 4, 5])
...@@ -154,7 +177,7 @@ def argmin_test(): ...@@ -154,7 +177,7 @@ def argmin_test():
return ([node], [x], [y]) return ([node], [x], [y])
@onnx_test @onnx_test()
def asin_test(): def asin_test():
x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [10]) x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [10])
y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [10]) y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [10])
...@@ -168,7 +191,7 @@ def asin_test(): ...@@ -168,7 +191,7 @@ def asin_test():
return ([node], [x], [y]) return ([node], [x], [y])
@onnx_test @onnx_test()
def asinh_test(): def asinh_test():
x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [10]) x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [10])
y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [10]) y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [10])
...@@ -182,7 +205,7 @@ def asinh_test(): ...@@ -182,7 +205,7 @@ def asinh_test():
return ([node], [x], [y]) return ([node], [x], [y])
@onnx_test @onnx_test()
def atan_test(): def atan_test():
x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [10]) x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [10])
y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [10]) y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [10])
...@@ -196,7 +219,7 @@ def atan_test(): ...@@ -196,7 +219,7 @@ def atan_test():
return ([node], [x], [y]) return ([node], [x], [y])
@onnx_test @onnx_test()
def atanh_test(): def atanh_test():
x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [10]) x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [10])
y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [10]) y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [10])
...@@ -210,7 +233,7 @@ def atanh_test(): ...@@ -210,7 +233,7 @@ def atanh_test():
return ([node], [x], [y]) return ([node], [x], [y])
@onnx_test @onnx_test()
def averagepool_1d_test(): def averagepool_1d_test():
x = helper.make_tensor_value_info('0', TensorProto.FLOAT, [1, 3, 5]) x = helper.make_tensor_value_info('0', TensorProto.FLOAT, [1, 3, 5])
out = helper.make_tensor_value_info('1', TensorProto.FLOAT, [1, 3, 3]) out = helper.make_tensor_value_info('1', TensorProto.FLOAT, [1, 3, 3])
...@@ -223,7 +246,7 @@ def averagepool_1d_test(): ...@@ -223,7 +246,7 @@ def averagepool_1d_test():
return ([node], [x], [out]) return ([node], [x], [out])
@onnx_test @onnx_test()
def averagepool_3d_test(): def averagepool_3d_test():
x = helper.make_tensor_value_info('0', TensorProto.FLOAT, [1, 3, 5, 5, 5]) x = helper.make_tensor_value_info('0', TensorProto.FLOAT, [1, 3, 5, 5, 5])
out = helper.make_tensor_value_info('1', TensorProto.FLOAT, out = helper.make_tensor_value_info('1', TensorProto.FLOAT,
...@@ -237,7 +260,65 @@ def averagepool_3d_test(): ...@@ -237,7 +260,65 @@ def averagepool_3d_test():
return ([node], [x], [out]) return ([node], [x], [out])
@onnx_test @onnx_test()
def averagepool_dyn_test():
x = helper.make_tensor_value_info('0', TensorProto.FLOAT,
[None, 3, 5, 5, 5])
out = helper.make_tensor_value_info('1', TensorProto.FLOAT,
[None, 3, 3, 3, 3])
node = onnx.helper.make_node('AveragePool',
inputs=['0'],
outputs=['1'],
kernel_shape=[3, 3, 3])
return ([node], [x], [out])
@onnx_test()
def averagepool_dyn_autopad_error_test():
x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [None, 1, 5, 5])
y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [None, 1, 5, 5])
node = onnx.helper.make_node('AveragePool',
inputs=['x'],
outputs=['y'],
kernel_shape=[2, 2],
auto_pad='SAME_LOWER')
return ([node], [x], [y])
@onnx_test()
def averagepool_dyn_asym_padding_error_test():
x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [None, 1, 5, 5])
y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [None, 1, 3, 3])
node = onnx.helper.make_node('AveragePool',
inputs=['x'],
outputs=['y'],
kernel_shape=[2, 2],
strides=[2, 2],
pads=[0, 0, 1, 1])
return ([node], [x], [y])
@onnx_test()
def averagepool_dyn_cip_error_test():
x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [None, 1, 5, 5])
y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [None, 1, 1, 1])
node = onnx.helper.make_node('AveragePool',
inputs=['x'],
outputs=['y'],
kernel_shape=[2, 2],
count_include_pad=1)
return ([node], [x], [y])
@onnx_test()
def averagepool_notset_test(): def averagepool_notset_test():
x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [1, 1, 5, 5]) x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [1, 1, 5, 5])
y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [1, 1, 1, 1]) y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [1, 1, 1, 1])
...@@ -253,7 +334,7 @@ def averagepool_notset_test(): ...@@ -253,7 +334,7 @@ def averagepool_notset_test():
return ([node], [x], [y]) return ([node], [x], [y])
@onnx_test @onnx_test()
def averagepool_nt_cip_test(): def averagepool_nt_cip_test():
x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [1, 1, 5, 5]) x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [1, 1, 5, 5])
y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [1, 1, 1, 1]) y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [1, 1, 1, 1])
...@@ -270,7 +351,7 @@ def averagepool_nt_cip_test(): ...@@ -270,7 +351,7 @@ def averagepool_nt_cip_test():
return ([node], [x], [y]) return ([node], [x], [y])
@onnx_test @onnx_test()
def averagepool_same_lower_test(): def averagepool_same_lower_test():
x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [1, 1, 5, 5]) x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [1, 1, 5, 5])
y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [1, 1, 5, 5]) y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [1, 1, 5, 5])
...@@ -284,7 +365,7 @@ def averagepool_same_lower_test(): ...@@ -284,7 +365,7 @@ def averagepool_same_lower_test():
return ([node], [x], [y]) return ([node], [x], [y])
@onnx_test @onnx_test()
def averagepool_sl_cip_test(): def averagepool_sl_cip_test():
x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [1, 1, 5, 5]) x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [1, 1, 5, 5])
y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [1, 1, 5, 5]) y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [1, 1, 5, 5])
...@@ -299,7 +380,7 @@ def averagepool_sl_cip_test(): ...@@ -299,7 +380,7 @@ def averagepool_sl_cip_test():
return ([node], [x], [y]) return ([node], [x], [y])
@onnx_test @onnx_test()
def averagepool_same_upper_test(): def averagepool_same_upper_test():
x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [1, 1, 5, 5]) x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [1, 1, 5, 5])
y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [1, 1, 5, 5]) y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [1, 1, 5, 5])
...@@ -313,7 +394,7 @@ def averagepool_same_upper_test(): ...@@ -313,7 +394,7 @@ def averagepool_same_upper_test():
return ([node], [x], [y]) return ([node], [x], [y])
@onnx_test @onnx_test()
def batch_norm_flat_test(): def batch_norm_flat_test():
x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [10]) x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [10])
scale = helper.make_tensor_value_info('scale', TensorProto.FLOAT, [1]) scale = helper.make_tensor_value_info('scale', TensorProto.FLOAT, [1])
...@@ -331,7 +412,7 @@ def batch_norm_flat_test(): ...@@ -331,7 +412,7 @@ def batch_norm_flat_test():
return ([node], [x, scale, bias, mean, var], [out]) return ([node], [x, scale, bias, mean, var], [out])
@onnx_test @onnx_test()
def batch_norm_rank_2_test(): def batch_norm_rank_2_test():
x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [2, 5]) x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [2, 5])
scale = helper.make_tensor_value_info('scale', TensorProto.FLOAT, [5]) scale = helper.make_tensor_value_info('scale', TensorProto.FLOAT, [5])
...@@ -349,7 +430,7 @@ def batch_norm_rank_2_test(): ...@@ -349,7 +430,7 @@ def batch_norm_rank_2_test():
return ([node], [x, scale, bias, mean, var], [out]) return ([node], [x, scale, bias, mean, var], [out])
@onnx_test @onnx_test()
def batch_norm_1d_test(): def batch_norm_1d_test():
x = helper.make_tensor_value_info('x', TensorProto.FLOAT16, [2, 3, 4]) x = helper.make_tensor_value_info('x', TensorProto.FLOAT16, [2, 3, 4])
scale = helper.make_tensor_value_info('scale', TensorProto.FLOAT, [3]) scale = helper.make_tensor_value_info('scale', TensorProto.FLOAT, [3])
...@@ -366,7 +447,7 @@ def batch_norm_1d_test(): ...@@ -366,7 +447,7 @@ def batch_norm_1d_test():
return ([node], [x, scale, bias, mean, var], [out]) return ([node], [x, scale, bias, mean, var], [out])
@onnx_test @onnx_test()
def batch_norm_2d_test(): def batch_norm_2d_test():
x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [2, 3, 4, 4]) x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [2, 3, 4, 4])
scale = helper.make_tensor_value_info('scale', TensorProto.FLOAT, [3]) scale = helper.make_tensor_value_info('scale', TensorProto.FLOAT, [3])
...@@ -383,7 +464,7 @@ def batch_norm_2d_test(): ...@@ -383,7 +464,7 @@ def batch_norm_2d_test():
return ([node], [x, scale, bias, mean, var], [out]) return ([node], [x, scale, bias, mean, var], [out])
@onnx_test @onnx_test()
def batch_norm_3d_test(): def batch_norm_3d_test():
x = helper.make_tensor_value_info('x', TensorProto.FLOAT16, x = helper.make_tensor_value_info('x', TensorProto.FLOAT16,
[2, 2, 2, 2, 2]) [2, 2, 2, 2, 2])
...@@ -403,7 +484,7 @@ def batch_norm_3d_test(): ...@@ -403,7 +484,7 @@ def batch_norm_3d_test():
return ([node], [x, scale, bias, mean, var], [out]) return ([node], [x, scale, bias, mean, var], [out])
@onnx_test @onnx_test()
def batch_norm_invalid_bias_rank_test(): def batch_norm_invalid_bias_rank_test():
x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [2, 3, 4, 4]) x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [2, 3, 4, 4])
scale = helper.make_tensor_value_info('scale', TensorProto.FLOAT, [3]) scale = helper.make_tensor_value_info('scale', TensorProto.FLOAT, [3])
...@@ -420,7 +501,7 @@ def batch_norm_invalid_bias_rank_test(): ...@@ -420,7 +501,7 @@ def batch_norm_invalid_bias_rank_test():
return ([node], [x, scale, bias, mean, var], [out]) return ([node], [x, scale, bias, mean, var], [out])
@onnx_test @onnx_test()
def binary_dyn_brcst_prelu_test(): def binary_dyn_brcst_prelu_test():
arg0 = helper.make_tensor_value_info('0', TensorProto.FLOAT, arg0 = helper.make_tensor_value_info('0', TensorProto.FLOAT,
[None, 3, 4, 5]) [None, 3, 4, 5])
...@@ -437,7 +518,7 @@ def binary_dyn_brcst_prelu_test(): ...@@ -437,7 +518,7 @@ def binary_dyn_brcst_prelu_test():
return ([node], [arg0, arg1], [arg_out]) return ([node], [arg0, arg1], [arg_out])
@onnx_test @onnx_test()
def binary_dyn_brcst_add_test(): def binary_dyn_brcst_add_test():
arg0 = helper.make_tensor_value_info('0', TensorProto.FLOAT16, [4, 5]) arg0 = helper.make_tensor_value_info('0', TensorProto.FLOAT16, [4, 5])
arg1 = helper.make_tensor_value_info('1', TensorProto.FLOAT, arg1 = helper.make_tensor_value_info('1', TensorProto.FLOAT,
...@@ -454,7 +535,7 @@ def binary_dyn_brcst_add_test(): ...@@ -454,7 +535,7 @@ def binary_dyn_brcst_add_test():
return ([node], [arg0, arg1], [arg_out]) return ([node], [arg0, arg1], [arg_out])
@onnx_test @onnx_test()
def binary_dyn_brcst_attr_error_test(): def binary_dyn_brcst_attr_error_test():
arg0 = helper.make_tensor_value_info('0', TensorProto.FLOAT16, [4, 5]) arg0 = helper.make_tensor_value_info('0', TensorProto.FLOAT16, [4, 5])
arg1 = helper.make_tensor_value_info('1', TensorProto.FLOAT, arg1 = helper.make_tensor_value_info('1', TensorProto.FLOAT,
...@@ -471,7 +552,7 @@ def binary_dyn_brcst_attr_error_test(): ...@@ -471,7 +552,7 @@ def binary_dyn_brcst_attr_error_test():
return ([node], [arg0, arg1], [arg_out]) return ([node], [arg0, arg1], [arg_out])
@onnx_test @onnx_test()
def binary_dyn_brcst_mul_test(): def binary_dyn_brcst_mul_test():
arg0 = helper.make_tensor_value_info('0', TensorProto.FLOAT, arg0 = helper.make_tensor_value_info('0', TensorProto.FLOAT,
[None, 3, 4, 5]) [None, 3, 4, 5])
...@@ -488,7 +569,7 @@ def binary_dyn_brcst_mul_test(): ...@@ -488,7 +569,7 @@ def binary_dyn_brcst_mul_test():
return ([node], [arg0, arg1], [arg_out]) return ([node], [arg0, arg1], [arg_out])
@onnx_test @onnx_test()
def cast_test(): def cast_test():
x = helper.make_tensor_value_info('x', TensorProto.FLOAT16, [10]) x = helper.make_tensor_value_info('x', TensorProto.FLOAT16, [10])
y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [10]) y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [10])
...@@ -498,7 +579,7 @@ def cast_test(): ...@@ -498,7 +579,7 @@ def cast_test():
return ([node], [x], [y]) return ([node], [x], [y])
@onnx_test @onnx_test()
def ceil_test(): def ceil_test():
x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [10]) x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [10])
y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [10]) y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [10])
...@@ -512,7 +593,7 @@ def ceil_test(): ...@@ -512,7 +593,7 @@ def ceil_test():
return ([node], [x], [y]) return ([node], [x], [y])
@onnx_test @onnx_test()
def celu_alpha_test(): def celu_alpha_test():
x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [3]) x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [3])
y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [3]) y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [3])
...@@ -525,7 +606,7 @@ def celu_alpha_test(): ...@@ -525,7 +606,7 @@ def celu_alpha_test():
return ([node], [x], [y]) return ([node], [x], [y])
@onnx_test @onnx_test()
def celu_default_test(): def celu_default_test():
x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [2, 3]) x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [2, 3])
y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [2, 3]) y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [2, 3])
...@@ -535,7 +616,7 @@ def celu_default_test(): ...@@ -535,7 +616,7 @@ def celu_default_test():
return ([node], [x], [y]) return ([node], [x], [y])
@onnx_test @onnx_test()
def celu_verify_test(): def celu_verify_test():
x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [2, 3]) x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [2, 3])
y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [2, 3]) y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [2, 3])
...@@ -548,7 +629,7 @@ def celu_verify_test(): ...@@ -548,7 +629,7 @@ def celu_verify_test():
return ([node], [x], [y]) return ([node], [x], [y])
@onnx_test @onnx_test()
def celu_wrong_type_test(): def celu_wrong_type_test():
x = helper.make_tensor_value_info('x', TensorProto.FLOAT16, [2, 3]) x = helper.make_tensor_value_info('x', TensorProto.FLOAT16, [2, 3])
y = helper.make_tensor_value_info('y', TensorProto.FLOAT16, [2, 3]) y = helper.make_tensor_value_info('y', TensorProto.FLOAT16, [2, 3])
...@@ -558,7 +639,7 @@ def celu_wrong_type_test(): ...@@ -558,7 +639,7 @@ def celu_wrong_type_test():
return ([node], [x], [y]) return ([node], [x], [y])
@onnx_test @onnx_test()
def celu_zero_alpha_test(): def celu_zero_alpha_test():
x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [2, 3]) x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [2, 3])
y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [2, 3]) y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [2, 3])
...@@ -571,7 +652,7 @@ def celu_zero_alpha_test(): ...@@ -571,7 +652,7 @@ def celu_zero_alpha_test():
return ([node], [x], [y]) return ([node], [x], [y])
@onnx_test @onnx_test()
def clip_test(): def clip_test():
x = helper.make_tensor_value_info('0', TensorProto.FLOAT, [3]) x = helper.make_tensor_value_info('0', TensorProto.FLOAT, [3])
y = helper.make_tensor_value_info('1', TensorProto.FLOAT, [3]) y = helper.make_tensor_value_info('1', TensorProto.FLOAT, [3])
...@@ -585,7 +666,7 @@ def clip_test(): ...@@ -585,7 +666,7 @@ def clip_test():
return ([node], [x], [y]) return ([node], [x], [y])
@onnx_test @onnx_test()
def clip_test_op11(): def clip_test_op11():
x = helper.make_tensor_value_info('0', TensorProto.FLOAT, [3]) x = helper.make_tensor_value_info('0', TensorProto.FLOAT, [3])
y = helper.make_tensor_value_info('1', TensorProto.FLOAT, [3]) y = helper.make_tensor_value_info('1', TensorProto.FLOAT, [3])
...@@ -600,7 +681,7 @@ def clip_test_op11(): ...@@ -600,7 +681,7 @@ def clip_test_op11():
return ([node], [x], [y], [min_val, max_val]) return ([node], [x], [y], [min_val, max_val])
@onnx_test @onnx_test()
def clip_test_op11_max_only(): def clip_test_op11_max_only():
x = helper.make_tensor_value_info('0', TensorProto.FLOAT, [3]) x = helper.make_tensor_value_info('0', TensorProto.FLOAT, [3])
y = helper.make_tensor_value_info('1', TensorProto.FLOAT, [3]) y = helper.make_tensor_value_info('1', TensorProto.FLOAT, [3])
...@@ -614,7 +695,7 @@ def clip_test_op11_max_only(): ...@@ -614,7 +695,7 @@ def clip_test_op11_max_only():
return ([node], [x], [y], [max_val]) return ([node], [x], [y], [max_val])
@onnx_test @onnx_test()
def clip_test_op11_min_only(): def clip_test_op11_min_only():
x = helper.make_tensor_value_info('0', TensorProto.FLOAT, [3]) x = helper.make_tensor_value_info('0', TensorProto.FLOAT, [3])
y = helper.make_tensor_value_info('1', TensorProto.FLOAT, [3]) y = helper.make_tensor_value_info('1', TensorProto.FLOAT, [3])
...@@ -626,7 +707,7 @@ def clip_test_op11_min_only(): ...@@ -626,7 +707,7 @@ def clip_test_op11_min_only():
return ([node], [x], [y], [min_val]) return ([node], [x], [y], [min_val])
@onnx_test @onnx_test()
def clip_test_op11_no_args(): def clip_test_op11_no_args():
x = helper.make_tensor_value_info('0', TensorProto.FLOAT, [3]) x = helper.make_tensor_value_info('0', TensorProto.FLOAT, [3])
y = helper.make_tensor_value_info('1', TensorProto.FLOAT, [3]) y = helper.make_tensor_value_info('1', TensorProto.FLOAT, [3])
...@@ -636,7 +717,7 @@ def clip_test_op11_no_args(): ...@@ -636,7 +717,7 @@ def clip_test_op11_no_args():
return ([node], [x], [y]) return ([node], [x], [y])
@onnx_test @onnx_test()
def clip_test_op11_no_args1(): def clip_test_op11_no_args1():
x = helper.make_tensor_value_info('0', TensorProto.FLOAT, [3]) x = helper.make_tensor_value_info('0', TensorProto.FLOAT, [3])
y = helper.make_tensor_value_info('1', TensorProto.FLOAT, [3]) y = helper.make_tensor_value_info('1', TensorProto.FLOAT, [3])
...@@ -646,7 +727,7 @@ def clip_test_op11_no_args1(): ...@@ -646,7 +727,7 @@ def clip_test_op11_no_args1():
return ([node], [x], [y]) return ([node], [x], [y])
@onnx_test @onnx_test()
def clip_test_args_type_mismatch(): def clip_test_args_type_mismatch():
x = helper.make_tensor_value_info('0', TensorProto.FLOAT, [3, 3]) x = helper.make_tensor_value_info('0', TensorProto.FLOAT, [3, 3])
y = helper.make_tensor_value_info('1', TensorProto.FLOAT, [3, 3]) y = helper.make_tensor_value_info('1', TensorProto.FLOAT, [3, 3])
...@@ -662,7 +743,7 @@ def clip_test_args_type_mismatch(): ...@@ -662,7 +743,7 @@ def clip_test_args_type_mismatch():
return ([node], [x], [y], [min_val, max_val]) return ([node], [x], [y], [min_val, max_val])
@onnx_test @onnx_test()
def concat_test(): def concat_test():
x = helper.make_tensor_value_info('0', TensorProto.FLOAT, [2, 4, 3]) x = helper.make_tensor_value_info('0', TensorProto.FLOAT, [2, 4, 3])
y = helper.make_tensor_value_info('1', TensorProto.FLOAT, [7, 4, 3]) y = helper.make_tensor_value_info('1', TensorProto.FLOAT, [7, 4, 3])
...@@ -678,7 +759,7 @@ def concat_test(): ...@@ -678,7 +759,7 @@ def concat_test():
return ([node], [x, y], [z]) return ([node], [x, y], [z])
@onnx_test @onnx_test()
def constant_test(): def constant_test():
x = np.array([0, 1, 2]) x = np.array([0, 1, 2])
y = helper.make_tensor_value_info('0', TensorProto.FLOAT, [3]) y = helper.make_tensor_value_info('0', TensorProto.FLOAT, [3])
...@@ -698,7 +779,7 @@ def constant_test(): ...@@ -698,7 +779,7 @@ def constant_test():
return ([node], [], [y]) return ([node], [], [y])
@onnx_test @onnx_test()
def constant_fill_test(): def constant_fill_test():
value = helper.make_tensor_value_info('value', TensorProto.FLOAT, [2, 3]) value = helper.make_tensor_value_info('value', TensorProto.FLOAT, [2, 3])
...@@ -715,7 +796,7 @@ def constant_fill_test(): ...@@ -715,7 +796,7 @@ def constant_fill_test():
return ([node], [], [value]) return ([node], [], [value])
@onnx_test @onnx_test()
def constant_fill_input_as_shape_test(): def constant_fill_input_as_shape_test():
np_shape = np.array([2, 3]) np_shape = np.array([2, 3])
value = helper.make_tensor_value_info('value', TensorProto.FLOAT, [2, 3]) value = helper.make_tensor_value_info('value', TensorProto.FLOAT, [2, 3])
...@@ -744,7 +825,7 @@ def constant_fill_input_as_shape_test(): ...@@ -744,7 +825,7 @@ def constant_fill_input_as_shape_test():
return ([const_shape_node, node], [], [value]) return ([const_shape_node, node], [], [value])
@onnx_test @onnx_test()
def constant_scalar_test(): def constant_scalar_test():
x = np.array([1]) x = np.array([1])
y = helper.make_tensor_value_info('0', TensorProto.FLOAT, [1]) y = helper.make_tensor_value_info('0', TensorProto.FLOAT, [1])
...@@ -764,7 +845,7 @@ def constant_scalar_test(): ...@@ -764,7 +845,7 @@ def constant_scalar_test():
return ([node], [], [y]) return ([node], [], [y])
@onnx_test @onnx_test()
def constant_empty_scalar_int64_test(): def constant_empty_scalar_int64_test():
x = np.array([]).astype(np.int64) x = np.array([]).astype(np.int64)
y = helper.make_tensor_value_info('0', TensorProto.INT64, [0]) y = helper.make_tensor_value_info('0', TensorProto.INT64, [0])
...@@ -784,7 +865,7 @@ def constant_empty_scalar_int64_test(): ...@@ -784,7 +865,7 @@ def constant_empty_scalar_int64_test():
return ([node], [], [y]) return ([node], [], [y])
@onnx_test @onnx_test()
def constant_one_val_int64_test(): def constant_one_val_int64_test():
x = np.array([1]).astype(np.int64) x = np.array([1]).astype(np.int64)
y = helper.make_tensor_value_info('0', TensorProto.INT64, [0]) y = helper.make_tensor_value_info('0', TensorProto.INT64, [0])
...@@ -804,7 +885,7 @@ def constant_one_val_int64_test(): ...@@ -804,7 +885,7 @@ def constant_one_val_int64_test():
return ([node], [], [y]) return ([node], [], [y])
@onnx_test @onnx_test()
def const_of_shape_empty_input_test(): def const_of_shape_empty_input_test():
tensor_val = onnx.helper.make_tensor('value', onnx.TensorProto.INT64, [1], tensor_val = onnx.helper.make_tensor('value', onnx.TensorProto.INT64, [1],
[10]) [10])
...@@ -831,7 +912,7 @@ def const_of_shape_empty_input_test(): ...@@ -831,7 +912,7 @@ def const_of_shape_empty_input_test():
return ([shape_const, node], [], [y]) return ([shape_const, node], [], [y])
@onnx_test @onnx_test()
def const_of_shape_float_test(): def const_of_shape_float_test():
tensor_val = onnx.helper.make_tensor('value', onnx.TensorProto.FLOAT, [1], tensor_val = onnx.helper.make_tensor('value', onnx.TensorProto.FLOAT, [1],
[10]) [10])
...@@ -858,7 +939,7 @@ def const_of_shape_float_test(): ...@@ -858,7 +939,7 @@ def const_of_shape_float_test():
return ([shape_const, node], [], [y]) return ([shape_const, node], [], [y])
@onnx_test @onnx_test()
def const_of_shape_int64_test(): def const_of_shape_int64_test():
tensor_val = onnx.helper.make_tensor('value', onnx.TensorProto.INT64, [1], tensor_val = onnx.helper.make_tensor('value', onnx.TensorProto.INT64, [1],
[10]) [10])
...@@ -883,7 +964,7 @@ def const_of_shape_int64_test(): ...@@ -883,7 +964,7 @@ def const_of_shape_int64_test():
return ([shape_const, node], [], [y]) return ([shape_const, node], [], [y])
@onnx_test @onnx_test()
def const_of_shape_no_value_attr_test(): def const_of_shape_no_value_attr_test():
shape_val = np.array([2, 3, 4]).astype(np.int64) shape_val = np.array([2, 3, 4]).astype(np.int64)
shape_ts = helper.make_tensor(name='shape_tensor', shape_ts = helper.make_tensor(name='shape_tensor',
...@@ -907,7 +988,7 @@ def const_of_shape_no_value_attr_test(): ...@@ -907,7 +988,7 @@ def const_of_shape_no_value_attr_test():
return ([shape_const, node], [], [y]) return ([shape_const, node], [], [y])
@onnx_test @onnx_test()
def conv_1d_test(): def conv_1d_test():
x = helper.make_tensor_value_info('0', TensorProto.FLOAT, [1, 3, 5]) x = helper.make_tensor_value_info('0', TensorProto.FLOAT, [1, 3, 5])
y = helper.make_tensor_value_info('1', TensorProto.FLOAT, [1, 3, 3]) y = helper.make_tensor_value_info('1', TensorProto.FLOAT, [1, 3, 3])
...@@ -918,7 +999,7 @@ def conv_1d_test(): ...@@ -918,7 +999,7 @@ def conv_1d_test():
return ([node], [x, y], [out]) return ([node], [x, y], [out])
@onnx_test @onnx_test()
def conv_3d_test(): def conv_3d_test():
x = helper.make_tensor_value_info('0', TensorProto.FLOAT, [1, 3, 5, 5, 5]) x = helper.make_tensor_value_info('0', TensorProto.FLOAT, [1, 3, 5, 5, 5])
y = helper.make_tensor_value_info('1', TensorProto.FLOAT, [1, 3, 3, 3, 3]) y = helper.make_tensor_value_info('1', TensorProto.FLOAT, [1, 3, 3, 3, 3])
...@@ -930,7 +1011,7 @@ def conv_3d_test(): ...@@ -930,7 +1011,7 @@ def conv_3d_test():
return ([node], [x, y], [out]) return ([node], [x, y], [out])
@onnx_test @onnx_test()
def conv_attr_fail_test(): def conv_attr_fail_test():
x = helper.make_tensor_value_info('0', TensorProto.FLOAT, [1, 3, 5]) x = helper.make_tensor_value_info('0', TensorProto.FLOAT, [1, 3, 5])
y = helper.make_tensor_value_info('1', TensorProto.FLOAT, [1, 3, 3]) y = helper.make_tensor_value_info('1', TensorProto.FLOAT, [1, 3, 3])
...@@ -944,7 +1025,7 @@ def conv_attr_fail_test(): ...@@ -944,7 +1025,7 @@ def conv_attr_fail_test():
return ([node], [x, y], [out]) return ([node], [x, y], [out])
@onnx_test @onnx_test()
def conv_autopad_fail_test(): def conv_autopad_fail_test():
x = helper.make_tensor_value_info('0', TensorProto.FLOAT, [1, 3, 32, 32]) x = helper.make_tensor_value_info('0', TensorProto.FLOAT, [1, 3, 32, 32])
y = helper.make_tensor_value_info('1', TensorProto.FLOAT, [1, 3, 1, 1]) y = helper.make_tensor_value_info('1', TensorProto.FLOAT, [1, 3, 1, 1])
...@@ -961,7 +1042,7 @@ def conv_autopad_fail_test(): ...@@ -961,7 +1042,7 @@ def conv_autopad_fail_test():
return ([node], [x, y], [out]) return ([node], [x, y], [out])
@onnx_test @onnx_test()
def conv_autopad_same_test(): def conv_autopad_same_test():
x = helper.make_tensor_value_info('0', TensorProto.FLOAT, [1, 3, 32, 32]) x = helper.make_tensor_value_info('0', TensorProto.FLOAT, [1, 3, 32, 32])
y = helper.make_tensor_value_info('1', TensorProto.FLOAT, [1, 3, 3, 3]) y = helper.make_tensor_value_info('1', TensorProto.FLOAT, [1, 3, 3, 3])
...@@ -977,7 +1058,7 @@ def conv_autopad_same_test(): ...@@ -977,7 +1058,7 @@ def conv_autopad_same_test():
return ([node], [x, y], [out]) return ([node], [x, y], [out])
@onnx_test @onnx_test()
def conv_bias_test(): def conv_bias_test():
x = helper.make_tensor_value_info('0', TensorProto.FLOAT, [1, 3, 32, 32]) x = helper.make_tensor_value_info('0', TensorProto.FLOAT, [1, 3, 32, 32])
y = helper.make_tensor_value_info('1', TensorProto.FLOAT, [1, 3, 5, 5]) y = helper.make_tensor_value_info('1', TensorProto.FLOAT, [1, 3, 5, 5])
...@@ -993,7 +1074,7 @@ def conv_bias_test(): ...@@ -993,7 +1074,7 @@ def conv_bias_test():
return ([node], [x, y, z], [out]) return ([node], [x, y, z], [out])
@onnx_test @onnx_test()
def conv_bn_relu_maxpool_test(): def conv_bn_relu_maxpool_test():
x = helper.make_tensor_value_info('0', TensorProto.FLOAT, [1, 3, 32, 32]) x = helper.make_tensor_value_info('0', TensorProto.FLOAT, [1, 3, 32, 32])
y = helper.make_tensor_value_info('1', TensorProto.FLOAT, [1, 3, 5, 5]) y = helper.make_tensor_value_info('1', TensorProto.FLOAT, [1, 3, 5, 5])
...@@ -1029,7 +1110,7 @@ def conv_bn_relu_maxpool_test(): ...@@ -1029,7 +1110,7 @@ def conv_bn_relu_maxpool_test():
return ([node0, node1, node2, node3], [x, y, z, m, n, k, l], [out]) return ([node0, node1, node2, node3], [x, y, z, m, n, k, l], [out])
@onnx_test @onnx_test()
def conv_dynamic_batch_test(): def conv_dynamic_batch_test():
x = helper.make_tensor_value_info('0', TensorProto.FLOAT, [None, 3, 5, 5]) x = helper.make_tensor_value_info('0', TensorProto.FLOAT, [None, 3, 5, 5])
y = helper.make_tensor_value_info('1', TensorProto.FLOAT, [1, 3, 3, 3]) y = helper.make_tensor_value_info('1', TensorProto.FLOAT, [1, 3, 3, 3])
...@@ -1040,7 +1121,7 @@ def conv_dynamic_batch_test(): ...@@ -1040,7 +1121,7 @@ def conv_dynamic_batch_test():
return ([node], [x, y], [out]) return ([node], [x, y], [out])
@onnx_test @onnx_test()
def conv_dynamic_img_test(): def conv_dynamic_img_test():
x = helper.make_tensor_value_info('0', TensorProto.FLOAT, x = helper.make_tensor_value_info('0', TensorProto.FLOAT,
[1, 3, None, None]) [1, 3, None, None])
...@@ -1052,7 +1133,7 @@ def conv_dynamic_img_test(): ...@@ -1052,7 +1133,7 @@ def conv_dynamic_img_test():
return ([node], [x, y], [out]) return ([node], [x, y], [out])
@onnx_test @onnx_test()
def conv_dynamic_weights_test(): def conv_dynamic_weights_test():
x = helper.make_tensor_value_info('0', TensorProto.FLOAT, [1, 3, 5, 5]) x = helper.make_tensor_value_info('0', TensorProto.FLOAT, [1, 3, 5, 5])
y = helper.make_tensor_value_info('1', TensorProto.FLOAT, y = helper.make_tensor_value_info('1', TensorProto.FLOAT,
...@@ -1064,7 +1145,7 @@ def conv_dynamic_weights_test(): ...@@ -1064,7 +1145,7 @@ def conv_dynamic_weights_test():
return ([node], [x, y], [out]) return ([node], [x, y], [out])
@onnx_test @onnx_test()
def conv_dynamic_img_and_weights_test(): def conv_dynamic_img_and_weights_test():
x = helper.make_tensor_value_info('0', TensorProto.FLOAT, x = helper.make_tensor_value_info('0', TensorProto.FLOAT,
[1, 3, None, None]) [1, 3, None, None])
...@@ -1077,7 +1158,7 @@ def conv_dynamic_img_and_weights_test(): ...@@ -1077,7 +1158,7 @@ def conv_dynamic_img_and_weights_test():
return ([node], [x, y], [out]) return ([node], [x, y], [out])
@onnx_test @onnx_test()
def conv_dynamic_batch_same_upper_test(): def conv_dynamic_batch_same_upper_test():
x = helper.make_tensor_value_info('0', TensorProto.FLOAT, [None, 3, 5, 5]) x = helper.make_tensor_value_info('0', TensorProto.FLOAT, [None, 3, 5, 5])
y = helper.make_tensor_value_info('1', TensorProto.FLOAT, [1, 3, 3, 3]) y = helper.make_tensor_value_info('1', TensorProto.FLOAT, [1, 3, 3, 3])
...@@ -1090,7 +1171,7 @@ def conv_dynamic_batch_same_upper_test(): ...@@ -1090,7 +1171,7 @@ def conv_dynamic_batch_same_upper_test():
return ([node], [x, y], [out]) return ([node], [x, y], [out])
@onnx_test @onnx_test()
def conv_dynamic_img_same_upper_test(): def conv_dynamic_img_same_upper_test():
x = helper.make_tensor_value_info('0', TensorProto.FLOAT, x = helper.make_tensor_value_info('0', TensorProto.FLOAT,
[1, 3, None, None]) [1, 3, None, None])
...@@ -1105,7 +1186,7 @@ def conv_dynamic_img_same_upper_test(): ...@@ -1105,7 +1186,7 @@ def conv_dynamic_img_same_upper_test():
return ([node], [x, y], [out]) return ([node], [x, y], [out])
@onnx_test @onnx_test()
def conv_dynamic_kernel_same_lower_test(): def conv_dynamic_kernel_same_lower_test():
x = helper.make_tensor_value_info('0', TensorProto.FLOAT, [1, 3, 5, 5]) x = helper.make_tensor_value_info('0', TensorProto.FLOAT, [1, 3, 5, 5])
y = helper.make_tensor_value_info('1', TensorProto.FLOAT, y = helper.make_tensor_value_info('1', TensorProto.FLOAT,
...@@ -1119,7 +1200,7 @@ def conv_dynamic_kernel_same_lower_test(): ...@@ -1119,7 +1200,7 @@ def conv_dynamic_kernel_same_lower_test():
return ([node], [x, y], [out]) return ([node], [x, y], [out])
@onnx_test @onnx_test()
def conv_relu_maxpool_test(): def conv_relu_maxpool_test():
x = helper.make_tensor_value_info('0', TensorProto.FLOAT, [1, 3, 32, 32]) x = helper.make_tensor_value_info('0', TensorProto.FLOAT, [1, 3, 32, 32])
y = helper.make_tensor_value_info('1', TensorProto.FLOAT, [1, 3, 5, 5]) y = helper.make_tensor_value_info('1', TensorProto.FLOAT, [1, 3, 5, 5])
...@@ -1145,7 +1226,7 @@ def conv_relu_maxpool_test(): ...@@ -1145,7 +1226,7 @@ def conv_relu_maxpool_test():
return ([node1, node2, node3], [x, y, z], [out]) return ([node1, node2, node3], [x, y, z], [out])
@onnx_test @onnx_test()
def conv_relu_maxpool_x2_test(): def conv_relu_maxpool_x2_test():
x = helper.make_tensor_value_info('0', TensorProto.FLOAT, [1, 3, 32, 32]) x = helper.make_tensor_value_info('0', TensorProto.FLOAT, [1, 3, 32, 32])
y = helper.make_tensor_value_info('1', TensorProto.FLOAT, [5, 3, 5, 5]) y = helper.make_tensor_value_info('1', TensorProto.FLOAT, [5, 3, 5, 5])
...@@ -1189,7 +1270,7 @@ def conv_relu_maxpool_x2_test(): ...@@ -1189,7 +1270,7 @@ def conv_relu_maxpool_x2_test():
return ([node1, node2, node3, node4, node5, node6], [x, y, z, m, n], [out]) return ([node1, node2, node3, node4, node5, node6], [x, y, z, m, n], [out])
@onnx_test @onnx_test()
def convinteger_bias_test(): def convinteger_bias_test():
x = helper.make_tensor_value_info('0', TensorProto.INT8, [1, 3, 32, 32]) x = helper.make_tensor_value_info('0', TensorProto.INT8, [1, 3, 32, 32])
y = helper.make_tensor_value_info('1', TensorProto.INT8, [1, 3, 5, 5]) y = helper.make_tensor_value_info('1', TensorProto.INT8, [1, 3, 5, 5])
...@@ -1205,7 +1286,7 @@ def convinteger_bias_test(): ...@@ -1205,7 +1286,7 @@ def convinteger_bias_test():
return ([node], [x, y, z], [out]) return ([node], [x, y, z], [out])
@onnx_test @onnx_test()
def cos_test(): def cos_test():
x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [10]) x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [10])
y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [10]) y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [10])
...@@ -1219,7 +1300,7 @@ def cos_test(): ...@@ -1219,7 +1300,7 @@ def cos_test():
return ([node], [x], [y]) return ([node], [x], [y])
@onnx_test @onnx_test()
def cosh_test(): def cosh_test():
x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [1]) x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [1])
y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [1]) y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [1])
...@@ -1233,7 +1314,7 @@ def cosh_test(): ...@@ -1233,7 +1314,7 @@ def cosh_test():
return ([node], [x], [y]) return ([node], [x], [y])
@onnx_test @onnx_test()
def deconv_test(): def deconv_test():
x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [1, 1, 3, 3]) x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [1, 1, 3, 3])
w = helper.make_tensor_value_info('w', TensorProto.FLOAT, [1, 1, 3, 3]) w = helper.make_tensor_value_info('w', TensorProto.FLOAT, [1, 1, 3, 3])
...@@ -1247,7 +1328,7 @@ def deconv_test(): ...@@ -1247,7 +1328,7 @@ def deconv_test():
return ([node], [x, w], [y]) return ([node], [x, w], [y])
@onnx_test @onnx_test()
def deconv_bias_test(): def deconv_bias_test():
x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [1, 1, 3, 3]) x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [1, 1, 3, 3])
w = helper.make_tensor_value_info('w', TensorProto.FLOAT, [1, 1, 3, 3]) w = helper.make_tensor_value_info('w', TensorProto.FLOAT, [1, 1, 3, 3])
...@@ -1262,7 +1343,7 @@ def deconv_bias_test(): ...@@ -1262,7 +1343,7 @@ def deconv_bias_test():
return ([node], [x, w, b], [y]) return ([node], [x, w, b], [y])
@onnx_test @onnx_test()
def deconv_input_pads_strides_test(): def deconv_input_pads_strides_test():
x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [1, 1, 3, 3]) x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [1, 1, 3, 3])
w = helper.make_tensor_value_info('w', TensorProto.FLOAT, [1, 2, 3, 3]) w = helper.make_tensor_value_info('w', TensorProto.FLOAT, [1, 2, 3, 3])
...@@ -1277,7 +1358,7 @@ def deconv_input_pads_strides_test(): ...@@ -1277,7 +1358,7 @@ def deconv_input_pads_strides_test():
return ([node], [x, w], [y]) return ([node], [x, w], [y])
@onnx_test @onnx_test()
def deconv_input_pads_asymm_test(): def deconv_input_pads_asymm_test():
x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [1, 1, 3, 3]) x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [1, 1, 3, 3])
w = helper.make_tensor_value_info('w', TensorProto.FLOAT, [1, 2, 3, 3]) w = helper.make_tensor_value_info('w', TensorProto.FLOAT, [1, 2, 3, 3])
...@@ -1292,7 +1373,7 @@ def deconv_input_pads_asymm_test(): ...@@ -1292,7 +1373,7 @@ def deconv_input_pads_asymm_test():
return ([node], [x, w], [y]) return ([node], [x, w], [y])
@onnx_test @onnx_test()
def deconv_input_pads_asymm_1d_test(): def deconv_input_pads_asymm_1d_test():
x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [1, 1, 3]) x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [1, 1, 3])
w = helper.make_tensor_value_info('w', TensorProto.FLOAT, [1, 2, 3]) w = helper.make_tensor_value_info('w', TensorProto.FLOAT, [1, 2, 3])
...@@ -1308,7 +1389,7 @@ def deconv_input_pads_asymm_1d_test(): ...@@ -1308,7 +1389,7 @@ def deconv_input_pads_asymm_1d_test():
return ([node], [x, w], [y]) return ([node], [x, w], [y])
@onnx_test @onnx_test()
def deconv_output_padding_test(): def deconv_output_padding_test():
x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [1, 1, 3, 3]) x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [1, 1, 3, 3])
w = helper.make_tensor_value_info('w', TensorProto.FLOAT, [1, 2, 3, 3]) w = helper.make_tensor_value_info('w', TensorProto.FLOAT, [1, 2, 3, 3])
...@@ -1323,7 +1404,7 @@ def deconv_output_padding_test(): ...@@ -1323,7 +1404,7 @@ def deconv_output_padding_test():
return ([node], [x, w], [y]) return ([node], [x, w], [y])
@onnx_test @onnx_test()
def deconv_output_padding_3d_test(): def deconv_output_padding_3d_test():
x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [1, 1, 3, 3, 3]) x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [1, 1, 3, 3, 3])
w = helper.make_tensor_value_info('w', TensorProto.FLOAT, [1, 2, 3, 3, 3]) w = helper.make_tensor_value_info('w', TensorProto.FLOAT, [1, 2, 3, 3, 3])
...@@ -1338,7 +1419,7 @@ def deconv_output_padding_3d_test(): ...@@ -1338,7 +1419,7 @@ def deconv_output_padding_3d_test():
return ([node], [x, w], [y]) return ([node], [x, w], [y])
@onnx_test @onnx_test()
def deconv_output_shape_test(): def deconv_output_shape_test():
x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [1, 1, 3, 3]) x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [1, 1, 3, 3])
w = helper.make_tensor_value_info('w', TensorProto.FLOAT, [1, 2, 3, 3]) w = helper.make_tensor_value_info('w', TensorProto.FLOAT, [1, 2, 3, 3])
...@@ -1353,7 +1434,7 @@ def deconv_output_shape_test(): ...@@ -1353,7 +1434,7 @@ def deconv_output_shape_test():
return ([node], [x, w], [y]) return ([node], [x, w], [y])
@onnx_test @onnx_test()
def deconv_output_shape_3d_test(): def deconv_output_shape_3d_test():
x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [1, 1, 3, 3, 3]) x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [1, 1, 3, 3, 3])
w = helper.make_tensor_value_info('w', TensorProto.FLOAT, [1, 2, 3, 3, 3]) w = helper.make_tensor_value_info('w', TensorProto.FLOAT, [1, 2, 3, 3, 3])
...@@ -1368,7 +1449,7 @@ def deconv_output_shape_3d_test(): ...@@ -1368,7 +1449,7 @@ def deconv_output_shape_3d_test():
return ([node], [x, w], [y]) return ([node], [x, w], [y])
@onnx_test @onnx_test()
def deconv_stride_test(): def deconv_stride_test():
x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [1, 1, 3, 3]) x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [1, 1, 3, 3])
w = helper.make_tensor_value_info('w', TensorProto.FLOAT, [1, 2, 3, 3]) w = helper.make_tensor_value_info('w', TensorProto.FLOAT, [1, 2, 3, 3])
...@@ -1382,7 +1463,7 @@ def deconv_stride_test(): ...@@ -1382,7 +1463,7 @@ def deconv_stride_test():
return ([node], [x, w], [y]) return ([node], [x, w], [y])
@onnx_test @onnx_test()
def depthtospace_test(): def depthtospace_test():
x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [2, 8, 5, 5]) x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [2, 8, 5, 5])
...@@ -1397,7 +1478,7 @@ def depthtospace_test(): ...@@ -1397,7 +1478,7 @@ def depthtospace_test():
return ([node], [x], [y]) return ([node], [x], [y])
@onnx_test @onnx_test()
def depthtospace_simple_test(): def depthtospace_simple_test():
x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [1, 8, 2, 3]) x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [1, 8, 2, 3])
...@@ -1412,7 +1493,7 @@ def depthtospace_simple_test(): ...@@ -1412,7 +1493,7 @@ def depthtospace_simple_test():
return ([node], [x], [y]) return ([node], [x], [y])
@onnx_test @onnx_test()
def depthtospace_crd_test(): def depthtospace_crd_test():
x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [2, 8, 5, 5]) x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [2, 8, 5, 5])
...@@ -1427,7 +1508,7 @@ def depthtospace_crd_test(): ...@@ -1427,7 +1508,7 @@ def depthtospace_crd_test():
return ([node], [x], [y]) return ([node], [x], [y])
@onnx_test @onnx_test()
def spacetodepth_test(): def spacetodepth_test():
x = helper.make_tensor_value_info('x', TensorProto.float, [2, 2, 10, 10]) x = helper.make_tensor_value_info('x', TensorProto.float, [2, 2, 10, 10])
...@@ -1441,7 +1522,7 @@ def spacetodepth_test(): ...@@ -1441,7 +1522,7 @@ def spacetodepth_test():
return ([node], [x], [y]) return ([node], [x], [y])
@onnx_test @onnx_test()
def spacetodepth_simple_test(): def spacetodepth_simple_test():
x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [1, 2, 4, 6]) x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [1, 2, 4, 6])
...@@ -1455,7 +1536,7 @@ def spacetodepth_simple_test(): ...@@ -1455,7 +1536,7 @@ def spacetodepth_simple_test():
return ([node], [x], [y]) return ([node], [x], [y])
@onnx_test @onnx_test()
def spacetodepth_invalid_blocksize_test(): def spacetodepth_invalid_blocksize_test():
x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [1, 2, 4, 6]) x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [1, 2, 4, 6])
...@@ -1469,7 +1550,7 @@ def spacetodepth_invalid_blocksize_test(): ...@@ -1469,7 +1550,7 @@ def spacetodepth_invalid_blocksize_test():
return ([node], [x], [y]) return ([node], [x], [y])
@onnx_test @onnx_test()
def spacetodepth_nondivisibility_test(): def spacetodepth_nondivisibility_test():
x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [1, 2, 5, 5]) x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [1, 2, 5, 5])
...@@ -1483,7 +1564,7 @@ def spacetodepth_nondivisibility_test(): ...@@ -1483,7 +1564,7 @@ def spacetodepth_nondivisibility_test():
return ([node], [x], [y]) return ([node], [x], [y])
@onnx_test @onnx_test()
def dequantizelinear_test(): def dequantizelinear_test():
arg0 = helper.make_tensor_value_info('0', TensorProto.INT8, [5]) arg0 = helper.make_tensor_value_info('0', TensorProto.INT8, [5])
arg1 = helper.make_tensor_value_info('1', TensorProto.FLOAT, [1]) arg1 = helper.make_tensor_value_info('1', TensorProto.FLOAT, [1])
...@@ -1498,7 +1579,7 @@ def dequantizelinear_test(): ...@@ -1498,7 +1579,7 @@ def dequantizelinear_test():
return ([node], [arg0, arg1], [arg_out]) return ([node], [arg0, arg1], [arg_out])
@onnx_test @onnx_test()
def dequantizelinear_zero_point_test(): def dequantizelinear_zero_point_test():
arg0 = helper.make_tensor_value_info('0', TensorProto.INT8, [5]) arg0 = helper.make_tensor_value_info('0', TensorProto.INT8, [5])
arg1 = helper.make_tensor_value_info('1', TensorProto.FLOAT, [1]) arg1 = helper.make_tensor_value_info('1', TensorProto.FLOAT, [1])
...@@ -1529,17 +1610,17 @@ def make_dequantizelinear_axis_graph(axis): ...@@ -1529,17 +1610,17 @@ def make_dequantizelinear_axis_graph(axis):
return ([node], [arg0, arg1, arg2], [arg_out]) return ([node], [arg0, arg1, arg2], [arg_out])
@onnx_test @onnx_test()
def dequantizelinear_axis_test(): def dequantizelinear_axis_test():
return make_dequantizelinear_axis_graph(2) return make_dequantizelinear_axis_graph(2)
@onnx_test @onnx_test()
def dequantizelinear_neg_axis_test(): def dequantizelinear_neg_axis_test():
return make_dequantizelinear_axis_graph(-2) return make_dequantizelinear_axis_graph(-2)
@onnx_test @onnx_test()
def dropout_test(): def dropout_test():
x = helper.make_tensor_value_info('0', TensorProto.FLOAT, [1, 3, 2, 2]) x = helper.make_tensor_value_info('0', TensorProto.FLOAT, [1, 3, 2, 2])
y = helper.make_tensor_value_info('1', TensorProto.FLOAT, [1, 3, 2, 2]) y = helper.make_tensor_value_info('1', TensorProto.FLOAT, [1, 3, 2, 2])
...@@ -1553,7 +1634,7 @@ def dropout_test(): ...@@ -1553,7 +1634,7 @@ def dropout_test():
return ([node], [x], [y]) return ([node], [x], [y])
@onnx_test @onnx_test()
def elu_test(): def elu_test():
x = helper.make_tensor_value_info('0', TensorProto.FLOAT, [3]) x = helper.make_tensor_value_info('0', TensorProto.FLOAT, [3])
y = helper.make_tensor_value_info('1', TensorProto.FLOAT, [3]) y = helper.make_tensor_value_info('1', TensorProto.FLOAT, [3])
...@@ -1566,7 +1647,7 @@ def elu_test(): ...@@ -1566,7 +1647,7 @@ def elu_test():
return ([node], [x], [y]) return ([node], [x], [y])
@onnx_test @onnx_test()
def embedding_bag_test(): def embedding_bag_test():
index_val = np.array([1, 0, 2]) index_val = np.array([1, 0, 2])
...@@ -1619,7 +1700,7 @@ def embedding_bag_test(): ...@@ -1619,7 +1700,7 @@ def embedding_bag_test():
return ([index, offset, node1, node2, node3], [weight], [y1, y2, y3]) return ([index, offset, node1, node2, node3], [weight], [y1, y2, y3])
@onnx_test @onnx_test()
def embedding_bag_offset_test(): def embedding_bag_offset_test():
index_val = np.array([1, 0]) index_val = np.array([1, 0])
...@@ -1658,7 +1739,7 @@ def embedding_bag_offset_test(): ...@@ -1658,7 +1739,7 @@ def embedding_bag_offset_test():
return ([index, offset, node], [weight], [y]) return ([index, offset, node], [weight], [y])
@onnx_test @onnx_test()
def equal_test(): def equal_test():
ax1 = np.array([1.0, 2.0, 3.0, 4.0, 5.0, 6.0]) ax1 = np.array([1.0, 2.0, 3.0, 4.0, 5.0, 6.0])
x1 = helper.make_tensor("x1", x1 = helper.make_tensor("x1",
...@@ -1678,7 +1759,7 @@ def equal_test(): ...@@ -1678,7 +1759,7 @@ def equal_test():
return ([node], [x2], [y], [x1]) return ([node], [x2], [y], [x1])
@onnx_test @onnx_test()
def equal_bool_test(): def equal_bool_test():
x1 = helper.make_tensor_value_info('x1', TensorProto.FLOAT, [2, 3]) x1 = helper.make_tensor_value_info('x1', TensorProto.FLOAT, [2, 3])
...@@ -1696,7 +1777,7 @@ def equal_bool_test(): ...@@ -1696,7 +1777,7 @@ def equal_bool_test():
return ([node1, node2], [x1, x2], [y]) return ([node1, node2], [x1, x2], [y])
@onnx_test @onnx_test()
def erf_test(): def erf_test():
x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [10, 15]) x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [10, 15])
y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [10, 15]) y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [10, 15])
...@@ -1710,7 +1791,7 @@ def erf_test(): ...@@ -1710,7 +1791,7 @@ def erf_test():
return ([node], [x], [y]) return ([node], [x], [y])
@onnx_test @onnx_test()
def exp_test(): def exp_test():
x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [10]) x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [10])
y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [10]) y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [10])
...@@ -1724,7 +1805,7 @@ def exp_test(): ...@@ -1724,7 +1805,7 @@ def exp_test():
return ([node], [x], [y]) return ([node], [x], [y])
@onnx_test @onnx_test()
def expand_test(): def expand_test():
shape_val = np.array([2, 3, 4, 5]).astype(np.int64) shape_val = np.array([2, 3, 4, 5]).astype(np.int64)
shape_ts = helper.make_tensor(name='shape_tensor', shape_ts = helper.make_tensor(name='shape_tensor',
...@@ -1747,7 +1828,23 @@ def expand_test(): ...@@ -1747,7 +1828,23 @@ def expand_test():
return ([shape_const, node], [x], [y]) return ([shape_const, node], [x], [y])
@onnx_test @onnx_test(True)
def external_constant_test():
x = np.array([0, 1, 2])
y = helper.make_tensor_value_info('0', TensorProto.FLOAT, [3])
tensor = from_array(x)
tensor.name = 'const_tensor'
node = onnx.helper.make_node('Constant',
inputs=[],
outputs=['0'],
value=tensor)
return ([node], [], [y])
@onnx_test()
def eyelike_default_test(): def eyelike_default_test():
T1 = helper.make_tensor_value_info('T1', TensorProto.FLOAT, [3, 4]) T1 = helper.make_tensor_value_info('T1', TensorProto.FLOAT, [3, 4])
T2 = helper.make_tensor_value_info('T2', TensorProto.FLOAT, [3, 4]) T2 = helper.make_tensor_value_info('T2', TensorProto.FLOAT, [3, 4])
...@@ -1760,7 +1857,7 @@ def eyelike_default_test(): ...@@ -1760,7 +1857,7 @@ def eyelike_default_test():
return ([node], [T1], [T2]) return ([node], [T1], [T2])
@onnx_test @onnx_test()
def eyelike_double_test(): def eyelike_double_test():
T1 = helper.make_tensor_value_info('T1', TensorProto.DOUBLE, [6, 15]) T1 = helper.make_tensor_value_info('T1', TensorProto.DOUBLE, [6, 15])
T2 = helper.make_tensor_value_info('T2', TensorProto.DOUBLE, [6, 15]) T2 = helper.make_tensor_value_info('T2', TensorProto.DOUBLE, [6, 15])
...@@ -1773,7 +1870,7 @@ def eyelike_double_test(): ...@@ -1773,7 +1870,7 @@ def eyelike_double_test():
return ([node], [T1], [T2]) return ([node], [T1], [T2])
@onnx_test @onnx_test()
def eyelike_half_test(): def eyelike_half_test():
T1 = helper.make_tensor_value_info('T1', TensorProto.FLOAT16, [8, 8]) T1 = helper.make_tensor_value_info('T1', TensorProto.FLOAT16, [8, 8])
T2 = helper.make_tensor_value_info('T2', TensorProto.FLOAT16, [8, 8]) T2 = helper.make_tensor_value_info('T2', TensorProto.FLOAT16, [8, 8])
...@@ -1786,7 +1883,7 @@ def eyelike_half_test(): ...@@ -1786,7 +1883,7 @@ def eyelike_half_test():
return ([node], [T1], [T2]) return ([node], [T1], [T2])
@onnx_test @onnx_test()
def eyelike_k_test(): def eyelike_k_test():
T1 = helper.make_tensor_value_info('T1', TensorProto.FLOAT, [3, 4]) T1 = helper.make_tensor_value_info('T1', TensorProto.FLOAT, [3, 4])
T2 = helper.make_tensor_value_info('T2', TensorProto.FLOAT, [3, 4]) T2 = helper.make_tensor_value_info('T2', TensorProto.FLOAT, [3, 4])
...@@ -1794,7 +1891,7 @@ def eyelike_k_test(): ...@@ -1794,7 +1891,7 @@ def eyelike_k_test():
return ([node], [T1], [T2]) return ([node], [T1], [T2])
@onnx_test @onnx_test()
def eyelike_k_outofbounds_neg_test(): def eyelike_k_outofbounds_neg_test():
T1 = helper.make_tensor_value_info('T1', TensorProto.FLOAT, [2, 4]) T1 = helper.make_tensor_value_info('T1', TensorProto.FLOAT, [2, 4])
T2 = helper.make_tensor_value_info('T2', TensorProto.FLOAT, [2, 4]) T2 = helper.make_tensor_value_info('T2', TensorProto.FLOAT, [2, 4])
...@@ -1805,7 +1902,7 @@ def eyelike_k_outofbounds_neg_test(): ...@@ -1805,7 +1902,7 @@ def eyelike_k_outofbounds_neg_test():
return ([node], [T1], [T2]) return ([node], [T1], [T2])
@onnx_test @onnx_test()
def eyelike_k_outofbounds_pos_test(): def eyelike_k_outofbounds_pos_test():
T1 = helper.make_tensor_value_info('T1', TensorProto.FLOAT, [3, 4]) T1 = helper.make_tensor_value_info('T1', TensorProto.FLOAT, [3, 4])
T2 = helper.make_tensor_value_info('T2', TensorProto.FLOAT, [3, 4]) T2 = helper.make_tensor_value_info('T2', TensorProto.FLOAT, [3, 4])
...@@ -1813,7 +1910,7 @@ def eyelike_k_outofbounds_pos_test(): ...@@ -1813,7 +1910,7 @@ def eyelike_k_outofbounds_pos_test():
return ([node], [T1], [T2]) return ([node], [T1], [T2])
@onnx_test @onnx_test()
def eyelike_not_rank2_test(): def eyelike_not_rank2_test():
T1 = helper.make_tensor_value_info('T1', TensorProto.FLOAT, [3, 4, 2]) T1 = helper.make_tensor_value_info('T1', TensorProto.FLOAT, [3, 4, 2])
T2 = helper.make_tensor_value_info('T2', TensorProto.FLOAT, [3, 4]) T2 = helper.make_tensor_value_info('T2', TensorProto.FLOAT, [3, 4])
...@@ -1825,7 +1922,7 @@ def eyelike_not_rank2_test(): ...@@ -1825,7 +1922,7 @@ def eyelike_not_rank2_test():
return ([node], [T1], [T2]) return ([node], [T1], [T2])
@onnx_test @onnx_test()
def eyelike_verify_test(): def eyelike_verify_test():
T1 = helper.make_tensor_value_info('T1', TensorProto.FLOAT, [3, 4]) T1 = helper.make_tensor_value_info('T1', TensorProto.FLOAT, [3, 4])
T2 = helper.make_tensor_value_info('T2', TensorProto.FLOAT, [3, 4]) T2 = helper.make_tensor_value_info('T2', TensorProto.FLOAT, [3, 4])
...@@ -1833,7 +1930,7 @@ def eyelike_verify_test(): ...@@ -1833,7 +1930,7 @@ def eyelike_verify_test():
return ([node], [T1], [T2]) return ([node], [T1], [T2])
@onnx_test @onnx_test()
def eyelike_verify_negk_test(): def eyelike_verify_negk_test():
T1 = helper.make_tensor_value_info('T1', TensorProto.FLOAT, [3, 4]) T1 = helper.make_tensor_value_info('T1', TensorProto.FLOAT, [3, 4])
T2 = helper.make_tensor_value_info('T2', TensorProto.FLOAT, [3, 4]) T2 = helper.make_tensor_value_info('T2', TensorProto.FLOAT, [3, 4])
...@@ -1844,7 +1941,7 @@ def eyelike_verify_negk_test(): ...@@ -1844,7 +1941,7 @@ def eyelike_verify_negk_test():
return ([node], [T1], [T2]) return ([node], [T1], [T2])
@onnx_test @onnx_test()
def eyelike_set_dtype_test(): def eyelike_set_dtype_test():
T1 = helper.make_tensor_value_info('T1', TensorProto.FLOAT, [3, 4]) T1 = helper.make_tensor_value_info('T1', TensorProto.FLOAT, [3, 4])
T2 = helper.make_tensor_value_info('T2', TensorProto.DOUBLE, [3, 4]) T2 = helper.make_tensor_value_info('T2', TensorProto.DOUBLE, [3, 4])
...@@ -1855,7 +1952,7 @@ def eyelike_set_dtype_test(): ...@@ -1855,7 +1952,7 @@ def eyelike_set_dtype_test():
return ([node], [T1], [T2]) return ([node], [T1], [T2])
@onnx_test @onnx_test()
def flatten_test(): def flatten_test():
x = helper.make_tensor_value_info('0', TensorProto.FLOAT, [2, 3, 4, 5]) x = helper.make_tensor_value_info('0', TensorProto.FLOAT, [2, 3, 4, 5])
y = helper.make_tensor_value_info('2', TensorProto.FLOAT, [6, 20]) y = helper.make_tensor_value_info('2', TensorProto.FLOAT, [6, 20])
...@@ -1871,7 +1968,7 @@ def flatten_test(): ...@@ -1871,7 +1968,7 @@ def flatten_test():
return ([node, node2], [x], [y, y2]) return ([node, node2], [x], [y, y2])
@onnx_test @onnx_test()
def flatten_nonstd_test(): def flatten_nonstd_test():
x = helper.make_tensor_value_info('0', TensorProto.FLOAT, [2, 3, 5, 4]) x = helper.make_tensor_value_info('0', TensorProto.FLOAT, [2, 3, 5, 4])
y = helper.make_tensor_value_info('2', TensorProto.FLOAT, [6, 20]) y = helper.make_tensor_value_info('2', TensorProto.FLOAT, [6, 20])
...@@ -1894,7 +1991,20 @@ def flatten_nonstd_test(): ...@@ -1894,7 +1991,20 @@ def flatten_nonstd_test():
return ([trans, node, node2], [x], [y, y2]) return ([trans, node, node2], [x], [y, y2])
@onnx_test @onnx_test()
def flatten_dyn_test():
x = helper.make_tensor_value_info('0', TensorProto.FLOAT, [None, 3, 4, 5])
y = helper.make_tensor_value_info('2', TensorProto.FLOAT, [None, 20])
node = onnx.helper.make_node('Flatten',
inputs=['0'],
axis=2,
outputs=['2'])
return ([node], [x], [y])
@onnx_test()
def floor_test(): def floor_test():
x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [10]) x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [10])
y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [10]) y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [10])
...@@ -1908,7 +2018,7 @@ def floor_test(): ...@@ -1908,7 +2018,7 @@ def floor_test():
return ([node], [x], [y]) return ([node], [x], [y])
@onnx_test @onnx_test()
def gather_test(): def gather_test():
x = helper.make_tensor_value_info('data', TensorProto.FLOAT, [3, 4, 5, 6]) x = helper.make_tensor_value_info('data', TensorProto.FLOAT, [3, 4, 5, 6])
i = helper.make_tensor_value_info('indices', TensorProto.INT32, i = helper.make_tensor_value_info('indices', TensorProto.INT32,
...@@ -1925,7 +2035,7 @@ def gather_test(): ...@@ -1925,7 +2035,7 @@ def gather_test():
return ([node], [x, i], [y]) return ([node], [x, i], [y])
@onnx_test @onnx_test()
def gather_elements_axis0_test(): def gather_elements_axis0_test():
x = helper.make_tensor_value_info('data', TensorProto.FLOAT, [3, 4]) x = helper.make_tensor_value_info('data', TensorProto.FLOAT, [3, 4])
i = helper.make_tensor_value_info('indices', TensorProto.INT32, [2, 3]) i = helper.make_tensor_value_info('indices', TensorProto.INT32, [2, 3])
...@@ -1941,7 +2051,7 @@ def gather_elements_axis0_test(): ...@@ -1941,7 +2051,7 @@ def gather_elements_axis0_test():
return ([node], [x, i], [y]) return ([node], [x, i], [y])
@onnx_test @onnx_test()
def gather_elements_axis1_test(): def gather_elements_axis1_test():
x = helper.make_tensor_value_info('data', TensorProto.FLOAT, [3, 4]) x = helper.make_tensor_value_info('data', TensorProto.FLOAT, [3, 4])
i = helper.make_tensor_value_info('indices', TensorProto.INT32, [2, 3]) i = helper.make_tensor_value_info('indices', TensorProto.INT32, [2, 3])
...@@ -1957,7 +2067,7 @@ def gather_elements_axis1_test(): ...@@ -1957,7 +2067,7 @@ def gather_elements_axis1_test():
return ([node], [x, i], [y]) return ([node], [x, i], [y])
@onnx_test @onnx_test()
def gathernd_test(): def gathernd_test():
x = helper.make_tensor_value_info('data', TensorProto.FLOAT, [2, 2]) x = helper.make_tensor_value_info('data', TensorProto.FLOAT, [2, 2])
i = helper.make_tensor_value_info('indices', TensorProto.INT64, [2, 2]) i = helper.make_tensor_value_info('indices', TensorProto.INT64, [2, 2])
...@@ -1970,7 +2080,7 @@ def gathernd_test(): ...@@ -1970,7 +2080,7 @@ def gathernd_test():
return ([node], [x, i], [y]) return ([node], [x, i], [y])
@onnx_test @onnx_test()
def gathernd_batch_dims_test(): def gathernd_batch_dims_test():
x = helper.make_tensor_value_info('data', TensorProto.FLOAT, [2, 2, 2]) x = helper.make_tensor_value_info('data', TensorProto.FLOAT, [2, 2, 2])
i = helper.make_tensor_value_info('indices', TensorProto.INT64, [2, 1]) i = helper.make_tensor_value_info('indices', TensorProto.INT64, [2, 1])
...@@ -1986,7 +2096,7 @@ def gathernd_batch_dims_test(): ...@@ -1986,7 +2096,7 @@ def gathernd_batch_dims_test():
return ([node], [x, i], [y]) return ([node], [x, i], [y])
@onnx_test @onnx_test()
def gemm_test(): def gemm_test():
x = helper.make_tensor_value_info('0', TensorProto.FLOAT, [5, 7]) x = helper.make_tensor_value_info('0', TensorProto.FLOAT, [5, 7])
y = helper.make_tensor_value_info('1', TensorProto.FLOAT, [11, 5]) y = helper.make_tensor_value_info('1', TensorProto.FLOAT, [11, 5])
...@@ -2004,7 +2114,7 @@ def gemm_test(): ...@@ -2004,7 +2114,7 @@ def gemm_test():
return ([node], [x, y, z], [a]) return ([node], [x, y, z], [a])
@onnx_test @onnx_test()
def gemm_ex_test(): def gemm_ex_test():
m1 = helper.make_tensor_value_info('1', TensorProto.FLOAT, [1, 1, 8, 6]) m1 = helper.make_tensor_value_info('1', TensorProto.FLOAT, [1, 1, 8, 6])
m2 = helper.make_tensor_value_info('2', TensorProto.FLOAT, [1, 1, 8, 7]) m2 = helper.make_tensor_value_info('2', TensorProto.FLOAT, [1, 1, 8, 7])
...@@ -2021,7 +2131,7 @@ def gemm_ex_test(): ...@@ -2021,7 +2131,7 @@ def gemm_ex_test():
return ([node], [m1, m2, m3], [y]) return ([node], [m1, m2, m3], [y])
@onnx_test @onnx_test()
def gemm_ex_brcst_test(): def gemm_ex_brcst_test():
m1 = helper.make_tensor_value_info('1', TensorProto.FLOAT, [1, 1, 5, 6]) m1 = helper.make_tensor_value_info('1', TensorProto.FLOAT, [1, 1, 5, 6])
m2 = helper.make_tensor_value_info('2', TensorProto.FLOAT, [1, 1, 5, 7]) m2 = helper.make_tensor_value_info('2', TensorProto.FLOAT, [1, 1, 5, 7])
...@@ -2038,7 +2148,7 @@ def gemm_ex_brcst_test(): ...@@ -2038,7 +2148,7 @@ def gemm_ex_brcst_test():
return ([node], [m1, m2, m3], [y]) return ([node], [m1, m2, m3], [y])
@onnx_test @onnx_test()
def gemm_half_test(): def gemm_half_test():
m1 = helper.make_tensor_value_info('1', TensorProto.FLOAT16, [1, 1, 8, 6]) m1 = helper.make_tensor_value_info('1', TensorProto.FLOAT16, [1, 1, 8, 6])
m2 = helper.make_tensor_value_info('2', TensorProto.FLOAT16, [1, 1, 8, 7]) m2 = helper.make_tensor_value_info('2', TensorProto.FLOAT16, [1, 1, 8, 7])
...@@ -2055,7 +2165,7 @@ def gemm_half_test(): ...@@ -2055,7 +2165,7 @@ def gemm_half_test():
return ([node], [m1, m2, m3], [y]) return ([node], [m1, m2, m3], [y])
@onnx_test @onnx_test()
def globalavgpool_test(): def globalavgpool_test():
x = helper.make_tensor_value_info('0', TensorProto.FLOAT, [1, 3, 16, 16]) x = helper.make_tensor_value_info('0', TensorProto.FLOAT, [1, 3, 16, 16])
y = helper.make_tensor_value_info('1', TensorProto.FLOAT, [1, 3, 1, 1]) y = helper.make_tensor_value_info('1', TensorProto.FLOAT, [1, 3, 1, 1])
...@@ -2069,7 +2179,22 @@ def globalavgpool_test(): ...@@ -2069,7 +2179,22 @@ def globalavgpool_test():
return ([node], [x], [y]) return ([node], [x], [y])
@onnx_test @onnx_test()
def globalavgpool_dyn_test():
x = helper.make_tensor_value_info('0', TensorProto.FLOAT,
[None, 3, 16, 16])
y = helper.make_tensor_value_info('1', TensorProto.FLOAT, [None, 3, 1, 1])
node = onnx.helper.make_node(
'GlobalAveragePool',
inputs=['0'],
outputs=['1'],
)
return ([node], [x], [y])
@onnx_test()
def globallppool_test(): def globallppool_test():
x = helper.make_tensor_value_info('0', TensorProto.FLOAT, [1, 3, 16, 16]) x = helper.make_tensor_value_info('0', TensorProto.FLOAT, [1, 3, 16, 16])
y = helper.make_tensor_value_info('1', TensorProto.FLOAT, [1, 3, 1, 1]) y = helper.make_tensor_value_info('1', TensorProto.FLOAT, [1, 3, 1, 1])
...@@ -2083,7 +2208,22 @@ def globallppool_test(): ...@@ -2083,7 +2208,22 @@ def globallppool_test():
return ([node], [x], [y]) return ([node], [x], [y])
@onnx_test @onnx_test()
def globallppool_dyn_test():
x = helper.make_tensor_value_info('0', TensorProto.FLOAT,
[1, 3, None, None])
y = helper.make_tensor_value_info('1', TensorProto.FLOAT, [1, 3, 1, 1])
node = onnx.helper.make_node(
'GlobalLpPool',
inputs=['0'],
outputs=['1'],
)
return ([node], [x], [y])
@onnx_test()
def globalmaxpool_test(): def globalmaxpool_test():
x = helper.make_tensor_value_info('0', TensorProto.FLOAT, [1, 3, 16, 16]) x = helper.make_tensor_value_info('0', TensorProto.FLOAT, [1, 3, 16, 16])
y = helper.make_tensor_value_info('1', TensorProto.FLOAT, [1, 3, 1, 1]) y = helper.make_tensor_value_info('1', TensorProto.FLOAT, [1, 3, 1, 1])
...@@ -2097,7 +2237,22 @@ def globalmaxpool_test(): ...@@ -2097,7 +2237,22 @@ def globalmaxpool_test():
return ([node], [x], [y]) return ([node], [x], [y])
@onnx_test @onnx_test()
def globalmaxpool_dyn_test():
x = helper.make_tensor_value_info('0', TensorProto.FLOAT,
[None, 3, 32, 32])
y = helper.make_tensor_value_info('1', TensorProto.FLOAT, [None, 3, 1, 1])
node = onnx.helper.make_node(
'GlobalMaxPool',
inputs=['0'],
outputs=['1'],
)
return ([node], [x], [y])
@onnx_test()
def greater_test(): def greater_test():
ax1 = np.array([1.0, 2.0, 3.0, 4.0, 5.0, 6.0]) ax1 = np.array([1.0, 2.0, 3.0, 4.0, 5.0, 6.0])
x1 = helper.make_tensor("x1", x1 = helper.make_tensor("x1",
...@@ -2117,7 +2272,7 @@ def greater_test(): ...@@ -2117,7 +2272,7 @@ def greater_test():
return ([node], [x2], [y], [x1]) return ([node], [x2], [y], [x1])
@onnx_test @onnx_test()
def greater_bool_test(): def greater_bool_test():
x1 = helper.make_tensor_value_info('x1', TensorProto.FLOAT, [2, 3]) x1 = helper.make_tensor_value_info('x1', TensorProto.FLOAT, [2, 3])
...@@ -2135,7 +2290,7 @@ def greater_bool_test(): ...@@ -2135,7 +2290,7 @@ def greater_bool_test():
return ([node1, node2], [x1, x2], [y]) return ([node1, node2], [x1, x2], [y])
@onnx_test @onnx_test()
def greaterorequal_test(): def greaterorequal_test():
x1 = helper.make_tensor_value_info('x1', TensorProto.FLOAT, [3]) x1 = helper.make_tensor_value_info('x1', TensorProto.FLOAT, [3])
...@@ -2151,7 +2306,7 @@ def greaterorequal_test(): ...@@ -2151,7 +2306,7 @@ def greaterorequal_test():
return ([node], [x1, x2], [y]) return ([node], [x1, x2], [y])
@onnx_test @onnx_test()
def group_conv_test(): def group_conv_test():
x = helper.make_tensor_value_info('0', TensorProto.FLOAT, [1, 4, 16, 16]) x = helper.make_tensor_value_info('0', TensorProto.FLOAT, [1, 4, 16, 16])
y = helper.make_tensor_value_info('1', TensorProto.FLOAT, [4, 1, 3, 3]) y = helper.make_tensor_value_info('1', TensorProto.FLOAT, [4, 1, 3, 3])
...@@ -2167,7 +2322,7 @@ def group_conv_test(): ...@@ -2167,7 +2322,7 @@ def group_conv_test():
return ([node], [x, y], [z]) return ([node], [x, y], [z])
@onnx_test @onnx_test()
def hardsigmoid_default_test(): def hardsigmoid_default_test():
x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [1, 3, 4, 5]) x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [1, 3, 4, 5])
y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [1, 3, 4, 5]) y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [1, 3, 4, 5])
...@@ -2177,7 +2332,7 @@ def hardsigmoid_default_test(): ...@@ -2177,7 +2332,7 @@ def hardsigmoid_default_test():
return ([node], [x], [y]) return ([node], [x], [y])
@onnx_test @onnx_test()
def hardsigmoid_double_test(): def hardsigmoid_double_test():
x = helper.make_tensor_value_info('x', TensorProto.DOUBLE, [1, 3, 4, 5]) x = helper.make_tensor_value_info('x', TensorProto.DOUBLE, [1, 3, 4, 5])
y = helper.make_tensor_value_info('y', TensorProto.DOUBLE, [1, 3, 4, 5]) y = helper.make_tensor_value_info('y', TensorProto.DOUBLE, [1, 3, 4, 5])
...@@ -2191,7 +2346,7 @@ def hardsigmoid_double_test(): ...@@ -2191,7 +2346,7 @@ def hardsigmoid_double_test():
return ([node], [x], [y]) return ([node], [x], [y])
@onnx_test @onnx_test()
def hardsigmoid_half_test(): def hardsigmoid_half_test():
x = helper.make_tensor_value_info('x', TensorProto.FLOAT16, [1, 3, 4, 5]) x = helper.make_tensor_value_info('x', TensorProto.FLOAT16, [1, 3, 4, 5])
y = helper.make_tensor_value_info('y', TensorProto.FLOAT16, [1, 3, 4, 5]) y = helper.make_tensor_value_info('y', TensorProto.FLOAT16, [1, 3, 4, 5])
...@@ -2201,7 +2356,7 @@ def hardsigmoid_half_test(): ...@@ -2201,7 +2356,7 @@ def hardsigmoid_half_test():
return ([node], [x], [y]) return ([node], [x], [y])
@onnx_test @onnx_test()
def hardsigmoid_verify_test(): def hardsigmoid_verify_test():
x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [2, 5]) x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [2, 5])
y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [2, 5]) y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [2, 5])
...@@ -2211,7 +2366,7 @@ def hardsigmoid_verify_test(): ...@@ -2211,7 +2366,7 @@ def hardsigmoid_verify_test():
return ([node], [x], [y]) return ([node], [x], [y])
@onnx_test @onnx_test()
def hardswish_test(): def hardswish_test():
x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [2, 5]) x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [2, 5])
y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [2, 5]) y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [2, 5])
...@@ -2221,7 +2376,7 @@ def hardswish_test(): ...@@ -2221,7 +2376,7 @@ def hardswish_test():
return ([node], [x], [y]) return ([node], [x], [y])
@onnx_test @onnx_test()
def if_else_test(): def if_else_test():
x = onnx.helper.make_tensor_value_info('x', onnx.TensorProto.FLOAT, [2, 3]) x = onnx.helper.make_tensor_value_info('x', onnx.TensorProto.FLOAT, [2, 3])
y = onnx.helper.make_tensor_value_info('y', onnx.TensorProto.FLOAT, [2, 3]) y = onnx.helper.make_tensor_value_info('y', onnx.TensorProto.FLOAT, [2, 3])
...@@ -2275,7 +2430,7 @@ def if_else_test(): ...@@ -2275,7 +2430,7 @@ def if_else_test():
return ([node], [x, y], [res], [cond_tensor, xt_tensor, yt_tensor]) return ([node], [x, y], [res], [cond_tensor, xt_tensor, yt_tensor])
@onnx_test @onnx_test()
def if_literal_test(): def if_literal_test():
then_out = onnx.helper.make_tensor_value_info('then_out', then_out = onnx.helper.make_tensor_value_info('then_out',
onnx.TensorProto.FLOAT, [5]) onnx.TensorProto.FLOAT, [5])
...@@ -2323,7 +2478,7 @@ def if_literal_test(): ...@@ -2323,7 +2478,7 @@ def if_literal_test():
return ([node], [cond_input], [ret]) return ([node], [cond_input], [ret])
@onnx_test @onnx_test()
def if_param_excp_test(): def if_param_excp_test():
then_out = onnx.helper.make_tensor_value_info('then_out', then_out = onnx.helper.make_tensor_value_info('then_out',
onnx.TensorProto.FLOAT, onnx.TensorProto.FLOAT,
...@@ -2375,7 +2530,7 @@ def if_param_excp_test(): ...@@ -2375,7 +2530,7 @@ def if_param_excp_test():
return ([node], [cond_input, x, y], [ret]) return ([node], [cond_input, x, y], [ret])
@onnx_test @onnx_test()
def if_param_excp1_test(): def if_param_excp1_test():
then_out = onnx.helper.make_tensor_value_info('sub_out', then_out = onnx.helper.make_tensor_value_info('sub_out',
onnx.TensorProto.FLOAT, onnx.TensorProto.FLOAT,
...@@ -2410,7 +2565,7 @@ def if_param_excp1_test(): ...@@ -2410,7 +2565,7 @@ def if_param_excp1_test():
return ([node], [cond_input, x], [ret]) return ([node], [cond_input, x], [ret])
@onnx_test @onnx_test()
def if_param_test(): def if_param_test():
then_out = onnx.helper.make_tensor_value_info('then_out', then_out = onnx.helper.make_tensor_value_info('then_out',
onnx.TensorProto.FLOAT, onnx.TensorProto.FLOAT,
...@@ -2462,7 +2617,7 @@ def if_param_test(): ...@@ -2462,7 +2617,7 @@ def if_param_test():
return ([node], [cond_input, x, y], [ret]) return ([node], [cond_input, x, y], [ret])
@onnx_test @onnx_test()
def if_pl_test(): def if_pl_test():
out_x = onnx.helper.make_tensor_value_info('out_x', onnx.TensorProto.FLOAT, out_x = onnx.helper.make_tensor_value_info('out_x', onnx.TensorProto.FLOAT,
[2, 3]) [2, 3])
...@@ -2530,7 +2685,7 @@ def if_pl_test(): ...@@ -2530,7 +2685,7 @@ def if_pl_test():
return ([node], [cond_input, x, y], [ret], [xt_tensor, yt_tensor]) return ([node], [cond_input, x, y], [ret], [xt_tensor, yt_tensor])
@onnx_test @onnx_test()
def if_then_test(): def if_then_test():
x = onnx.helper.make_tensor_value_info('x', onnx.TensorProto.FLOAT, [2, 3]) x = onnx.helper.make_tensor_value_info('x', onnx.TensorProto.FLOAT, [2, 3])
y = onnx.helper.make_tensor_value_info('y', onnx.TensorProto.FLOAT, [2, 3]) y = onnx.helper.make_tensor_value_info('y', onnx.TensorProto.FLOAT, [2, 3])
...@@ -2584,7 +2739,7 @@ def if_then_test(): ...@@ -2584,7 +2739,7 @@ def if_then_test():
return ([node], [x, y], [res], [cond_tensor, xt_tensor, yt_tensor]) return ([node], [x, y], [res], [cond_tensor, xt_tensor, yt_tensor])
@onnx_test @onnx_test()
def if_tuple_test(): def if_tuple_test():
x = onnx.helper.make_tensor_value_info('x', onnx.TensorProto.FLOAT, [1, 4]) x = onnx.helper.make_tensor_value_info('x', onnx.TensorProto.FLOAT, [1, 4])
y = onnx.helper.make_tensor_value_info('y', onnx.TensorProto.FLOAT, [3, 4]) y = onnx.helper.make_tensor_value_info('y', onnx.TensorProto.FLOAT, [3, 4])
...@@ -2655,7 +2810,7 @@ def if_tuple_test(): ...@@ -2655,7 +2810,7 @@ def if_tuple_test():
y], [res0, res1], [one_tensor, two_tensor, three_tensor]) y], [res0, res1], [one_tensor, two_tensor, three_tensor])
@onnx_test @onnx_test()
def imagescaler_test(): def imagescaler_test():
x = helper.make_tensor_value_info('0', TensorProto.FLOAT, [1, 3, 16, 16]) x = helper.make_tensor_value_info('0', TensorProto.FLOAT, [1, 3, 16, 16])
y = helper.make_tensor_value_info('1', TensorProto.FLOAT, [1, 3, 16, 16]) y = helper.make_tensor_value_info('1', TensorProto.FLOAT, [1, 3, 16, 16])
...@@ -2669,7 +2824,7 @@ def imagescaler_test(): ...@@ -2669,7 +2824,7 @@ def imagescaler_test():
return ([node], [x], [y]) return ([node], [x], [y])
@onnx_test @onnx_test()
def imagescaler_half_test(): def imagescaler_half_test():
x = helper.make_tensor_value_info('0', TensorProto.FLOAT16, [1, 3, 16, 16]) x = helper.make_tensor_value_info('0', TensorProto.FLOAT16, [1, 3, 16, 16])
y = helper.make_tensor_value_info('1', TensorProto.FLOAT16, [1, 3, 16, 16]) y = helper.make_tensor_value_info('1', TensorProto.FLOAT16, [1, 3, 16, 16])
...@@ -2683,7 +2838,7 @@ def imagescaler_half_test(): ...@@ -2683,7 +2838,7 @@ def imagescaler_half_test():
return ([node], [x], [y]) return ([node], [x], [y])
@onnx_test @onnx_test()
def implicit_add_bcast_test(): def implicit_add_bcast_test():
x = helper.make_tensor_value_info('0', TensorProto.FLOAT, [2, 3, 4, 5]) x = helper.make_tensor_value_info('0', TensorProto.FLOAT, [2, 3, 4, 5])
y = helper.make_tensor_value_info('1', TensorProto.FLOAT, [3, 4, 1]) y = helper.make_tensor_value_info('1', TensorProto.FLOAT, [3, 4, 1])
...@@ -2698,7 +2853,7 @@ def implicit_add_bcast_test(): ...@@ -2698,7 +2853,7 @@ def implicit_add_bcast_test():
return ([node], [x, y], [z]) return ([node], [x, y], [z])
@onnx_test @onnx_test()
def implicit_pow_bcast_test(): def implicit_pow_bcast_test():
arg0 = helper.make_tensor_value_info('0', TensorProto.FLOAT, [2, 3, 4, 5]) arg0 = helper.make_tensor_value_info('0', TensorProto.FLOAT, [2, 3, 4, 5])
arg1 = helper.make_tensor_value_info('1', TensorProto.FLOAT, [3, 4, 1]) arg1 = helper.make_tensor_value_info('1', TensorProto.FLOAT, [3, 4, 1])
...@@ -2714,7 +2869,7 @@ def implicit_pow_bcast_test(): ...@@ -2714,7 +2869,7 @@ def implicit_pow_bcast_test():
return ([node], [arg0, arg1], [arg_out]) return ([node], [arg0, arg1], [arg_out])
@onnx_test @onnx_test()
def implicit_sub_bcast_test(): def implicit_sub_bcast_test():
arg0 = helper.make_tensor_value_info('0', TensorProto.UINT64, [2, 3, 4, 5]) arg0 = helper.make_tensor_value_info('0', TensorProto.UINT64, [2, 3, 4, 5])
arg1 = helper.make_tensor_value_info('1', TensorProto.UINT64, [4, 5]) arg1 = helper.make_tensor_value_info('1', TensorProto.UINT64, [4, 5])
...@@ -2730,7 +2885,7 @@ def implicit_sub_bcast_test(): ...@@ -2730,7 +2885,7 @@ def implicit_sub_bcast_test():
return ([node], [arg0, arg1], [arg_out]) return ([node], [arg0, arg1], [arg_out])
@onnx_test @onnx_test()
def initializer_not_an_input(): def initializer_not_an_input():
values = np.array([[1, 2, 3, 4], [5, 6, 7, 8]]) values = np.array([[1, 2, 3, 4], [5, 6, 7, 8]])
w = helper.make_tensor(name='w', w = helper.make_tensor(name='w',
...@@ -2750,7 +2905,7 @@ def initializer_not_an_input(): ...@@ -2750,7 +2905,7 @@ def initializer_not_an_input():
return ([node], [x], [y], [w]) return ([node], [x], [y], [w])
@onnx_test @onnx_test()
def instance_norm_test(): def instance_norm_test():
x = helper.make_tensor_value_info('0', TensorProto.FLOAT, [1, 2, 3, 3]) x = helper.make_tensor_value_info('0', TensorProto.FLOAT, [1, 2, 3, 3])
scale = helper.make_tensor_value_info('1', TensorProto.FLOAT, [2]) scale = helper.make_tensor_value_info('1', TensorProto.FLOAT, [2])
...@@ -2764,7 +2919,7 @@ def instance_norm_test(): ...@@ -2764,7 +2919,7 @@ def instance_norm_test():
return ([node], [x, scale, bias], [y]) return ([node], [x, scale, bias], [y])
@onnx_test @onnx_test()
def instance_norm_half_test(): def instance_norm_half_test():
x = helper.make_tensor_value_info('0', TensorProto.FLOAT16, [1, 2, 3, 3]) x = helper.make_tensor_value_info('0', TensorProto.FLOAT16, [1, 2, 3, 3])
scale = helper.make_tensor_value_info('1', TensorProto.FLOAT16, [2]) scale = helper.make_tensor_value_info('1', TensorProto.FLOAT16, [2])
...@@ -2778,7 +2933,7 @@ def instance_norm_half_test(): ...@@ -2778,7 +2933,7 @@ def instance_norm_half_test():
return ([node], [x, scale, bias], [y]) return ([node], [x, scale, bias], [y])
@onnx_test @onnx_test()
def instance_norm_type_mismatch_test(): def instance_norm_type_mismatch_test():
x = helper.make_tensor_value_info('0', TensorProto.FLOAT, [1, 2, 3, 3]) x = helper.make_tensor_value_info('0', TensorProto.FLOAT, [1, 2, 3, 3])
scale = helper.make_tensor_value_info('1', TensorProto.FLOAT16, [2]) scale = helper.make_tensor_value_info('1', TensorProto.FLOAT16, [2])
...@@ -2792,7 +2947,7 @@ def instance_norm_type_mismatch_test(): ...@@ -2792,7 +2947,7 @@ def instance_norm_type_mismatch_test():
return ([node], [x, scale, bias], [y]) return ([node], [x, scale, bias], [y])
@onnx_test @onnx_test()
def instance_norm_invalid_type_test(): def instance_norm_invalid_type_test():
x = helper.make_tensor_value_info('0', TensorProto.INT32, [1, 2, 3, 3]) x = helper.make_tensor_value_info('0', TensorProto.INT32, [1, 2, 3, 3])
scale = helper.make_tensor_value_info('1', TensorProto.FLOAT, [2]) scale = helper.make_tensor_value_info('1', TensorProto.FLOAT, [2])
...@@ -2806,7 +2961,7 @@ def instance_norm_invalid_type_test(): ...@@ -2806,7 +2961,7 @@ def instance_norm_invalid_type_test():
return ([node], [x, scale, bias], [y]) return ([node], [x, scale, bias], [y])
@onnx_test @onnx_test()
def instance_norm_nonbroadcastable_test(): def instance_norm_nonbroadcastable_test():
x = helper.make_tensor_value_info('0', TensorProto.FLOAT, [1, 2, 3, 3]) x = helper.make_tensor_value_info('0', TensorProto.FLOAT, [1, 2, 3, 3])
scale = helper.make_tensor_value_info('1', TensorProto.FLOAT, [4]) scale = helper.make_tensor_value_info('1', TensorProto.FLOAT, [4])
...@@ -2820,7 +2975,7 @@ def instance_norm_nonbroadcastable_test(): ...@@ -2820,7 +2975,7 @@ def instance_norm_nonbroadcastable_test():
return ([node], [x, scale, bias], [y]) return ([node], [x, scale, bias], [y])
@onnx_test @onnx_test()
def instance_norm_val_test(): def instance_norm_val_test():
x = np.array([[[[0, 1, 2], [3, 4, 5], [6, 7, 8]], x = np.array([[[[0, 1, 2], [3, 4, 5], [6, 7, 8]],
[[0, 1, 2], [3, 4, 5], [6, 7, 8]]]]) [[0, 1, 2], [3, 4, 5], [6, 7, 8]]]])
...@@ -2850,7 +3005,7 @@ def instance_norm_val_test(): ...@@ -2850,7 +3005,7 @@ def instance_norm_val_test():
return ([node], [], [y], [x_tensor, scale_tensor, bias_tensor]) return ([node], [], [y], [x_tensor, scale_tensor, bias_tensor])
@onnx_test @onnx_test()
def instance_norm_val_3d_test(): def instance_norm_val_3d_test():
x = np.array([[[[[0, 1], [2, 3]], [[4, 5], [6, 7]]], x = np.array([[[[[0, 1], [2, 3]], [[4, 5], [6, 7]]],
[[[0, 1], [2, 3]], [[4, 5], [6, 7]]]]]) [[[0, 1], [2, 3]], [[4, 5], [6, 7]]]]])
...@@ -2880,7 +3035,7 @@ def instance_norm_val_3d_test(): ...@@ -2880,7 +3035,7 @@ def instance_norm_val_3d_test():
return ([node], [], [y], [x_tensor, scale_tensor, bias_tensor]) return ([node], [], [y], [x_tensor, scale_tensor, bias_tensor])
@onnx_test @onnx_test()
def isnan_float_test(): def isnan_float_test():
t1 = helper.make_tensor_value_info('t1', TensorProto.FLOAT, [2, 3]) t1 = helper.make_tensor_value_info('t1', TensorProto.FLOAT, [2, 3])
t2 = helper.make_tensor_value_info('t2', TensorProto.FLOAT, [2, 3]) t2 = helper.make_tensor_value_info('t2', TensorProto.FLOAT, [2, 3])
...@@ -2893,7 +3048,7 @@ def isnan_float_test(): ...@@ -2893,7 +3048,7 @@ def isnan_float_test():
return ([node], [t1], [t2]) return ([node], [t1], [t2])
@onnx_test @onnx_test()
def isnan_half_test(): def isnan_half_test():
t1 = helper.make_tensor_value_info('t1', TensorProto.FLOAT16, [2, 3]) t1 = helper.make_tensor_value_info('t1', TensorProto.FLOAT16, [2, 3])
t2 = helper.make_tensor_value_info('t2', TensorProto.FLOAT16, [2, 3]) t2 = helper.make_tensor_value_info('t2', TensorProto.FLOAT16, [2, 3])
...@@ -2906,7 +3061,7 @@ def isnan_half_test(): ...@@ -2906,7 +3061,7 @@ def isnan_half_test():
return ([node], [t1], [t2]) return ([node], [t1], [t2])
@onnx_test @onnx_test()
def layernorm_test(): def layernorm_test():
x = helper.make_tensor_value_info('0', TensorProto.FLOAT, [1, 1, 5]) x = helper.make_tensor_value_info('0', TensorProto.FLOAT, [1, 1, 5])
y = helper.make_tensor_value_info('1', TensorProto.FLOAT, [1, 1, 5]) y = helper.make_tensor_value_info('1', TensorProto.FLOAT, [1, 1, 5])
...@@ -2969,7 +3124,7 @@ def layernorm_test(): ...@@ -2969,7 +3124,7 @@ def layernorm_test():
bias_add], [x, scale, bias], [y], [pow_tensor, epsilon_tensor]) bias_add], [x, scale, bias], [y], [pow_tensor, epsilon_tensor])
@onnx_test @onnx_test()
def leaky_relu_test(): def leaky_relu_test():
x = helper.make_tensor_value_info('0', TensorProto.FLOAT, [3]) x = helper.make_tensor_value_info('0', TensorProto.FLOAT, [3])
y = helper.make_tensor_value_info('1', TensorProto.FLOAT, [3]) y = helper.make_tensor_value_info('1', TensorProto.FLOAT, [3])
...@@ -2982,7 +3137,7 @@ def leaky_relu_test(): ...@@ -2982,7 +3137,7 @@ def leaky_relu_test():
return ([node], [x], [y]) return ([node], [x], [y])
@onnx_test @onnx_test()
def less_test(): def less_test():
ax1 = np.array([1.0, 2.0, 3.0, 4.0, 5.0, 6.0]) ax1 = np.array([1.0, 2.0, 3.0, 4.0, 5.0, 6.0])
x1 = helper.make_tensor("x1", x1 = helper.make_tensor("x1",
...@@ -3002,7 +3157,7 @@ def less_test(): ...@@ -3002,7 +3157,7 @@ def less_test():
return ([node], [x2], [y], [x1]) return ([node], [x2], [y], [x1])
@onnx_test @onnx_test()
def less_bool_test(): def less_bool_test():
x1 = helper.make_tensor_value_info('x1', TensorProto.FLOAT, [2, 3]) x1 = helper.make_tensor_value_info('x1', TensorProto.FLOAT, [2, 3])
...@@ -3020,7 +3175,7 @@ def less_bool_test(): ...@@ -3020,7 +3175,7 @@ def less_bool_test():
return ([node1, node2], [x1, x2], [y]) return ([node1, node2], [x1, x2], [y])
@onnx_test @onnx_test()
def lessorequal_test(): def lessorequal_test():
x1 = helper.make_tensor_value_info('x1', TensorProto.FLOAT, [3]) x1 = helper.make_tensor_value_info('x1', TensorProto.FLOAT, [3])
...@@ -3036,7 +3191,7 @@ def lessorequal_test(): ...@@ -3036,7 +3191,7 @@ def lessorequal_test():
return ([node], [x1, x2], [y]) return ([node], [x1, x2], [y])
@onnx_test @onnx_test()
def log_test(): def log_test():
x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [10]) x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [10])
y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [10]) y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [10])
...@@ -3050,7 +3205,7 @@ def log_test(): ...@@ -3050,7 +3205,7 @@ def log_test():
return ([node], [x], [y]) return ([node], [x], [y])
@onnx_test @onnx_test()
def logical_and_bcast_test(): def logical_and_bcast_test():
x = helper.make_tensor_value_info('0', TensorProto.BOOL, [2, 3, 4, 5]) x = helper.make_tensor_value_info('0', TensorProto.BOOL, [2, 3, 4, 5])
y = helper.make_tensor_value_info('1', TensorProto.BOOL, [4, 5]) y = helper.make_tensor_value_info('1', TensorProto.BOOL, [4, 5])
...@@ -3061,7 +3216,7 @@ def logical_and_bcast_test(): ...@@ -3061,7 +3216,7 @@ def logical_and_bcast_test():
return ([node], [x, y], [z]) return ([node], [x, y], [z])
@onnx_test @onnx_test()
def logical_or_test(): def logical_or_test():
x = helper.make_tensor_value_info('0', TensorProto.BOOL, [2, 3, 4, 5]) x = helper.make_tensor_value_info('0', TensorProto.BOOL, [2, 3, 4, 5])
y = helper.make_tensor_value_info('1', TensorProto.BOOL, [2, 3, 4, 5]) y = helper.make_tensor_value_info('1', TensorProto.BOOL, [2, 3, 4, 5])
...@@ -3072,7 +3227,7 @@ def logical_or_test(): ...@@ -3072,7 +3227,7 @@ def logical_or_test():
return ([node], [x, y], [z]) return ([node], [x, y], [z])
@onnx_test @onnx_test()
def logical_xor_bcast_test(): def logical_xor_bcast_test():
x = helper.make_tensor_value_info('0', TensorProto.BOOL, [2, 3, 4, 5]) x = helper.make_tensor_value_info('0', TensorProto.BOOL, [2, 3, 4, 5])
y = helper.make_tensor_value_info('1', TensorProto.BOOL, [4, 1]) y = helper.make_tensor_value_info('1', TensorProto.BOOL, [4, 1])
...@@ -3083,7 +3238,7 @@ def logical_xor_bcast_test(): ...@@ -3083,7 +3238,7 @@ def logical_xor_bcast_test():
return ([node], [x, y], [z]) return ([node], [x, y], [z])
@onnx_test @onnx_test()
def logsoftmax_test(): def logsoftmax_test():
x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [3, 4, 5, 6]) x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [3, 4, 5, 6])
y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [3, 4, 5, 6]) y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [3, 4, 5, 6])
...@@ -3096,7 +3251,7 @@ def logsoftmax_test(): ...@@ -3096,7 +3251,7 @@ def logsoftmax_test():
return ([node], [x], [y]) return ([node], [x], [y])
@onnx_test @onnx_test()
def logsoftmax_nonstd_input_test(): def logsoftmax_nonstd_input_test():
x = helper.make_tensor_value_info('0', TensorProto.FLOAT, [6, 9]) x = helper.make_tensor_value_info('0', TensorProto.FLOAT, [6, 9])
y = helper.make_tensor_value_info('2', TensorProto.FLOAT, [3, 4]) y = helper.make_tensor_value_info('2', TensorProto.FLOAT, [3, 4])
...@@ -3116,7 +3271,7 @@ def logsoftmax_nonstd_input_test(): ...@@ -3116,7 +3271,7 @@ def logsoftmax_nonstd_input_test():
return ([node0, node1], [x], [y]) return ([node0, node1], [x], [y])
@onnx_test @onnx_test()
def loop_default_test(): def loop_default_test():
body = helper.make_graph([ body = helper.make_graph([
helper.make_node("Add", ["a", "b_in"], ["my_local"]), helper.make_node("Add", ["a", "b_in"], ["my_local"]),
...@@ -3153,7 +3308,7 @@ def loop_default_test(): ...@@ -3153,7 +3308,7 @@ def loop_default_test():
return ([node], [a, b], [b_loop, uout]) return ([node], [a, b], [b_loop, uout])
@onnx_test @onnx_test()
def loop_test(): def loop_test():
body = helper.make_graph([ body = helper.make_graph([
helper.make_node("Add", ["a", "b_in"], ["my_local"]), helper.make_node("Add", ["a", "b_in"], ["my_local"]),
...@@ -3194,7 +3349,7 @@ def loop_test(): ...@@ -3194,7 +3349,7 @@ def loop_test():
return ([node], [iter, cond, a, b], [b_loop, uout]) return ([node], [iter, cond, a, b], [b_loop, uout])
@onnx_test @onnx_test()
def lpnormalization_axis_error_test(): def lpnormalization_axis_error_test():
x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [2, 3]) x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [2, 3])
y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [2, 3]) y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [2, 3])
...@@ -3206,7 +3361,7 @@ def lpnormalization_axis_error_test(): ...@@ -3206,7 +3361,7 @@ def lpnormalization_axis_error_test():
return ([node], [x], [y]) return ([node], [x], [y])
@onnx_test @onnx_test()
def lpnormalization_default_test(): def lpnormalization_default_test():
x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [3, 4]) x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [3, 4])
y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [3, 4]) y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [3, 4])
...@@ -3220,7 +3375,7 @@ def lpnormalization_default_test(): ...@@ -3220,7 +3375,7 @@ def lpnormalization_default_test():
return ([node], [x], [y]) return ([node], [x], [y])
@onnx_test @onnx_test()
def lpnormalization_l1_test(): def lpnormalization_l1_test():
x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [3, 4]) x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [3, 4])
y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [3, 4]) y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [3, 4])
...@@ -3234,7 +3389,7 @@ def lpnormalization_l1_test(): ...@@ -3234,7 +3389,7 @@ def lpnormalization_l1_test():
return ([node], [x], [y]) return ([node], [x], [y])
@onnx_test @onnx_test()
def lpnormalization_l2_test(): def lpnormalization_l2_test():
x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [3, 4]) x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [3, 4])
y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [3, 4]) y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [3, 4])
...@@ -3246,7 +3401,7 @@ def lpnormalization_l2_test(): ...@@ -3246,7 +3401,7 @@ def lpnormalization_l2_test():
return ([node], [x], [y]) return ([node], [x], [y])
@onnx_test @onnx_test()
def lpnormalization_p_error_test(): def lpnormalization_p_error_test():
x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [2, 3]) x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [2, 3])
y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [2, 3]) y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [2, 3])
...@@ -3258,7 +3413,7 @@ def lpnormalization_p_error_test(): ...@@ -3258,7 +3413,7 @@ def lpnormalization_p_error_test():
return ([node], [x], [y]) return ([node], [x], [y])
@onnx_test @onnx_test()
def lppool_l1_test(): def lppool_l1_test():
x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [1, 3, 5]) x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [1, 3, 5])
y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [1, 3, 3]) y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [1, 3, 3])
...@@ -3271,7 +3426,7 @@ def lppool_l1_test(): ...@@ -3271,7 +3426,7 @@ def lppool_l1_test():
return ([node], [x], [y]) return ([node], [x], [y])
@onnx_test @onnx_test()
def lppool_l2_test(): def lppool_l2_test():
x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [1, 3, 5]) x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [1, 3, 5])
y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [1, 3, 3]) y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [1, 3, 3])
...@@ -3284,7 +3439,7 @@ def lppool_l2_test(): ...@@ -3284,7 +3439,7 @@ def lppool_l2_test():
return ([node], [x], [y]) return ([node], [x], [y])
@onnx_test @onnx_test()
def lrn_test(): def lrn_test():
x = helper.make_tensor_value_info('0', TensorProto.FLOAT, [1, 28, 24, 24]) x = helper.make_tensor_value_info('0', TensorProto.FLOAT, [1, 28, 24, 24])
y = helper.make_tensor_value_info('1', TensorProto.FLOAT, [1, 28, 24, 24]) y = helper.make_tensor_value_info('1', TensorProto.FLOAT, [1, 28, 24, 24])
...@@ -3300,7 +3455,7 @@ def lrn_test(): ...@@ -3300,7 +3455,7 @@ def lrn_test():
return ([node], [x], [y]) return ([node], [x], [y])
@onnx_test @onnx_test()
def matmul_bmbm_test(): def matmul_bmbm_test():
m1 = helper.make_tensor_value_info('1', TensorProto.FLOAT, [3, 6, 7]) m1 = helper.make_tensor_value_info('1', TensorProto.FLOAT, [3, 6, 7])
m2 = helper.make_tensor_value_info('2', TensorProto.FLOAT, [5, 2, 1, 7, 8]) m2 = helper.make_tensor_value_info('2', TensorProto.FLOAT, [5, 2, 1, 7, 8])
...@@ -3315,7 +3470,7 @@ def matmul_bmbm_test(): ...@@ -3315,7 +3470,7 @@ def matmul_bmbm_test():
return ([node], [m1, m2], [y]) return ([node], [m1, m2], [y])
@onnx_test @onnx_test()
def matmul_bmv_test(): def matmul_bmv_test():
m1 = helper.make_tensor_value_info('1', TensorProto.FLOAT, [3, 6, 7]) m1 = helper.make_tensor_value_info('1', TensorProto.FLOAT, [3, 6, 7])
m2 = helper.make_tensor_value_info('2', TensorProto.FLOAT, [7]) m2 = helper.make_tensor_value_info('2', TensorProto.FLOAT, [7])
...@@ -3330,7 +3485,7 @@ def matmul_bmv_test(): ...@@ -3330,7 +3485,7 @@ def matmul_bmv_test():
return ([node], [m1, m2], [y]) return ([node], [m1, m2], [y])
@onnx_test @onnx_test()
def matmul_mv_test(): def matmul_mv_test():
m1 = helper.make_tensor_value_info('1', TensorProto.FLOAT, [6, 7]) m1 = helper.make_tensor_value_info('1', TensorProto.FLOAT, [6, 7])
m2 = helper.make_tensor_value_info('2', TensorProto.FLOAT, [7]) m2 = helper.make_tensor_value_info('2', TensorProto.FLOAT, [7])
...@@ -3345,7 +3500,7 @@ def matmul_mv_test(): ...@@ -3345,7 +3500,7 @@ def matmul_mv_test():
return ([node], [m1, m2], [y]) return ([node], [m1, m2], [y])
@onnx_test @onnx_test()
def matmul_vbm_test(): def matmul_vbm_test():
m1 = helper.make_tensor_value_info('1', TensorProto.FLOAT, [7]) m1 = helper.make_tensor_value_info('1', TensorProto.FLOAT, [7])
m2 = helper.make_tensor_value_info('2', TensorProto.FLOAT, [5, 7, 8]) m2 = helper.make_tensor_value_info('2', TensorProto.FLOAT, [5, 7, 8])
...@@ -3360,7 +3515,7 @@ def matmul_vbm_test(): ...@@ -3360,7 +3515,7 @@ def matmul_vbm_test():
return ([node], [m1, m2], [y]) return ([node], [m1, m2], [y])
@onnx_test @onnx_test()
def matmul_vm_test(): def matmul_vm_test():
m1 = helper.make_tensor_value_info('1', TensorProto.FLOAT, [7]) m1 = helper.make_tensor_value_info('1', TensorProto.FLOAT, [7])
m2 = helper.make_tensor_value_info('2', TensorProto.FLOAT, [7, 8]) m2 = helper.make_tensor_value_info('2', TensorProto.FLOAT, [7, 8])
...@@ -3375,7 +3530,7 @@ def matmul_vm_test(): ...@@ -3375,7 +3530,7 @@ def matmul_vm_test():
return ([node], [m1, m2], [y]) return ([node], [m1, m2], [y])
@onnx_test @onnx_test()
def matmul_vv_test(): def matmul_vv_test():
m1 = helper.make_tensor_value_info('1', TensorProto.FLOAT, [7]) m1 = helper.make_tensor_value_info('1', TensorProto.FLOAT, [7])
m2 = helper.make_tensor_value_info('2', TensorProto.FLOAT, [7]) m2 = helper.make_tensor_value_info('2', TensorProto.FLOAT, [7])
...@@ -3390,7 +3545,7 @@ def matmul_vv_test(): ...@@ -3390,7 +3545,7 @@ def matmul_vv_test():
return ([node], [m1, m2], [y]) return ([node], [m1, m2], [y])
@onnx_test @onnx_test()
def matmulinteger_test(): def matmulinteger_test():
m1 = helper.make_tensor_value_info('1', TensorProto.INT8, [3, 6, 16]) m1 = helper.make_tensor_value_info('1', TensorProto.INT8, [3, 6, 16])
m2 = helper.make_tensor_value_info('2', TensorProto.INT8, [3, 16, 8]) m2 = helper.make_tensor_value_info('2', TensorProto.INT8, [3, 16, 8])
...@@ -3405,7 +3560,7 @@ def matmulinteger_test(): ...@@ -3405,7 +3560,7 @@ def matmulinteger_test():
return ([node], [m1, m2], [y]) return ([node], [m1, m2], [y])
@onnx_test @onnx_test()
def max_test(): def max_test():
a = helper.make_tensor_value_info('0', TensorProto.FLOAT, [3]) a = helper.make_tensor_value_info('0', TensorProto.FLOAT, [3])
b = helper.make_tensor_value_info('1', TensorProto.FLOAT, [3]) b = helper.make_tensor_value_info('1', TensorProto.FLOAT, [3])
...@@ -3421,7 +3576,7 @@ def max_test(): ...@@ -3421,7 +3576,7 @@ def max_test():
return ([node], [a, b, c], [y]) return ([node], [a, b, c], [y])
@onnx_test @onnx_test()
def maxpool_notset_test(): def maxpool_notset_test():
x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [1, 1, 5, 5]) x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [1, 1, 5, 5])
y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [1, 1, 1, 1]) y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [1, 1, 1, 1])
...@@ -3437,7 +3592,7 @@ def maxpool_notset_test(): ...@@ -3437,7 +3592,7 @@ def maxpool_notset_test():
return ([node], [x], [y]) return ([node], [x], [y])
@onnx_test @onnx_test()
def maxpool_same_upper_test(): def maxpool_same_upper_test():
x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [1, 1, 5, 5]) x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [1, 1, 5, 5])
y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [1, 1, 5, 5]) y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [1, 1, 5, 5])
...@@ -3451,7 +3606,7 @@ def maxpool_same_upper_test(): ...@@ -3451,7 +3606,7 @@ def maxpool_same_upper_test():
return ([node], [x], [y]) return ([node], [x], [y])
@onnx_test @onnx_test()
def mean_broadcast_test(): def mean_broadcast_test():
data_0 = helper.make_tensor_value_info('0', TensorProto.FLOAT, [1, 3, 4]) data_0 = helper.make_tensor_value_info('0', TensorProto.FLOAT, [1, 3, 4])
data_1 = helper.make_tensor_value_info('1', TensorProto.FLOAT, data_1 = helper.make_tensor_value_info('1', TensorProto.FLOAT,
...@@ -3470,7 +3625,7 @@ def mean_broadcast_test(): ...@@ -3470,7 +3625,7 @@ def mean_broadcast_test():
return ([node], [data_0, data_1, data_2, data_3, data_4], [mean]) return ([node], [data_0, data_1, data_2, data_3, data_4], [mean])
@onnx_test @onnx_test()
def mean_fp16_test(): def mean_fp16_test():
data_0 = helper.make_tensor_value_info('0', TensorProto.FLOAT16, [1, 2, 3]) data_0 = helper.make_tensor_value_info('0', TensorProto.FLOAT16, [1, 2, 3])
data_1 = helper.make_tensor_value_info('1', TensorProto.FLOAT16, [1, 2, 3]) data_1 = helper.make_tensor_value_info('1', TensorProto.FLOAT16, [1, 2, 3])
...@@ -3486,7 +3641,7 @@ def mean_fp16_test(): ...@@ -3486,7 +3641,7 @@ def mean_fp16_test():
return ([node], [data_0, data_1, data_2], [mean]) return ([node], [data_0, data_1, data_2], [mean])
@onnx_test @onnx_test()
def mean_invalid_broadcast_test(): def mean_invalid_broadcast_test():
data_0 = helper.make_tensor_value_info('0', TensorProto.FLOAT, [1, 2, 3]) data_0 = helper.make_tensor_value_info('0', TensorProto.FLOAT, [1, 2, 3])
data_1 = helper.make_tensor_value_info('1', TensorProto.FLOAT, [1, 2, 3]) data_1 = helper.make_tensor_value_info('1', TensorProto.FLOAT, [1, 2, 3])
...@@ -3501,7 +3656,7 @@ def mean_invalid_broadcast_test(): ...@@ -3501,7 +3656,7 @@ def mean_invalid_broadcast_test():
return ([node], [data_0, data_1, data_2], [mean]) return ([node], [data_0, data_1, data_2], [mean])
@onnx_test @onnx_test()
def mean_single_input_test(): def mean_single_input_test():
data_0 = helper.make_tensor_value_info('0', TensorProto.FLOAT, [1, 2, 3]) data_0 = helper.make_tensor_value_info('0', TensorProto.FLOAT, [1, 2, 3])
mean = helper.make_tensor_value_info('mean', TensorProto.FLOAT, [1, 2, 3]) mean = helper.make_tensor_value_info('mean', TensorProto.FLOAT, [1, 2, 3])
...@@ -3511,7 +3666,7 @@ def mean_single_input_test(): ...@@ -3511,7 +3666,7 @@ def mean_single_input_test():
return ([node], [data_0], [mean]) return ([node], [data_0], [mean])
@onnx_test @onnx_test()
def mean_test(): def mean_test():
data = [ data = [
helper.make_tensor_value_info(str(i), TensorProto.DOUBLE, [2, 2, 2]) helper.make_tensor_value_info(str(i), TensorProto.DOUBLE, [2, 2, 2])
...@@ -3525,7 +3680,7 @@ def mean_test(): ...@@ -3525,7 +3680,7 @@ def mean_test():
return ([node], data, [mean]) return ([node], data, [mean])
@onnx_test @onnx_test()
def mean_integral_test(): def mean_integral_test():
data = [ data = [
helper.make_tensor_value_info(str(i), TensorProto.INT32, [2, 2, 2]) helper.make_tensor_value_info(str(i), TensorProto.INT32, [2, 2, 2])
...@@ -3539,7 +3694,7 @@ def mean_integral_test(): ...@@ -3539,7 +3694,7 @@ def mean_integral_test():
return ([node], data, [mean]) return ([node], data, [mean])
@onnx_test @onnx_test()
def min_test(): def min_test():
a = helper.make_tensor_value_info('0', TensorProto.FLOAT, [3]) a = helper.make_tensor_value_info('0', TensorProto.FLOAT, [3])
b = helper.make_tensor_value_info('1', TensorProto.FLOAT, [3]) b = helper.make_tensor_value_info('1', TensorProto.FLOAT, [3])
...@@ -3555,7 +3710,7 @@ def min_test(): ...@@ -3555,7 +3710,7 @@ def min_test():
return ([node], [a, b, c], [y]) return ([node], [a, b, c], [y])
@onnx_test @onnx_test()
def mod_test(): def mod_test():
a = helper.make_tensor_value_info('0', TensorProto.INT32, [3, 3, 3]) a = helper.make_tensor_value_info('0', TensorProto.INT32, [3, 3, 3])
b = helper.make_tensor_value_info('1', TensorProto.INT32, [3, 3, 3]) b = helper.make_tensor_value_info('1', TensorProto.INT32, [3, 3, 3])
...@@ -3566,7 +3721,7 @@ def mod_test(): ...@@ -3566,7 +3721,7 @@ def mod_test():
return ([node], [a, b], [y]) return ([node], [a, b], [y])
@onnx_test @onnx_test()
def mod_test_half(): def mod_test_half():
a = helper.make_tensor_value_info('0', TensorProto.FLOAT16, [3, 3, 3]) a = helper.make_tensor_value_info('0', TensorProto.FLOAT16, [3, 3, 3])
b = helper.make_tensor_value_info('1', TensorProto.FLOAT16, [3, 3, 3]) b = helper.make_tensor_value_info('1', TensorProto.FLOAT16, [3, 3, 3])
...@@ -3577,7 +3732,7 @@ def mod_test_half(): ...@@ -3577,7 +3732,7 @@ def mod_test_half():
return ([node], [a, b], [y]) return ([node], [a, b], [y])
@onnx_test @onnx_test()
def mod_test_different_dtypes(): def mod_test_different_dtypes():
a = helper.make_tensor_value_info('0', TensorProto.INT16, [3, 3, 3]) a = helper.make_tensor_value_info('0', TensorProto.INT16, [3, 3, 3])
b = helper.make_tensor_value_info('1', TensorProto.INT32, [3, 3, 3]) b = helper.make_tensor_value_info('1', TensorProto.INT32, [3, 3, 3])
...@@ -3592,7 +3747,7 @@ def mod_test_different_dtypes(): ...@@ -3592,7 +3747,7 @@ def mod_test_different_dtypes():
return ([node], [a, b], [y]) return ([node], [a, b], [y])
@onnx_test @onnx_test()
def mod_test_fmod(): def mod_test_fmod():
a = helper.make_tensor_value_info('0', TensorProto.FLOAT, [3, 3, 3]) a = helper.make_tensor_value_info('0', TensorProto.FLOAT, [3, 3, 3])
b = helper.make_tensor_value_info('1', TensorProto.FLOAT, [3, 3, 3]) b = helper.make_tensor_value_info('1', TensorProto.FLOAT, [3, 3, 3])
...@@ -3608,7 +3763,7 @@ def mod_test_fmod(): ...@@ -3608,7 +3763,7 @@ def mod_test_fmod():
return ([node], [a, b], [y]) return ([node], [a, b], [y])
@onnx_test @onnx_test()
def mod_test_fmod_half(): def mod_test_fmod_half():
a = helper.make_tensor_value_info('0', TensorProto.FLOAT16, [3, 3, 3]) a = helper.make_tensor_value_info('0', TensorProto.FLOAT16, [3, 3, 3])
b = helper.make_tensor_value_info('1', TensorProto.FLOAT16, [3, 3, 3]) b = helper.make_tensor_value_info('1', TensorProto.FLOAT16, [3, 3, 3])
...@@ -3622,7 +3777,7 @@ def mod_test_fmod_half(): ...@@ -3622,7 +3777,7 @@ def mod_test_fmod_half():
return ([node], [a, b], [y]) return ([node], [a, b], [y])
@onnx_test @onnx_test()
def mod_test_fmod_different_dtypes(): def mod_test_fmod_different_dtypes():
a = helper.make_tensor_value_info('0', TensorProto.FLOAT, [3, 3, 3]) a = helper.make_tensor_value_info('0', TensorProto.FLOAT, [3, 3, 3])
b = helper.make_tensor_value_info('1', TensorProto.INT32, [3, 3, 3]) b = helper.make_tensor_value_info('1', TensorProto.INT32, [3, 3, 3])
...@@ -3638,7 +3793,7 @@ def mod_test_fmod_different_dtypes(): ...@@ -3638,7 +3793,7 @@ def mod_test_fmod_different_dtypes():
return ([node], [a, b], [y]) return ([node], [a, b], [y])
@onnx_test @onnx_test()
def multinomial_test(): def multinomial_test():
sample_size = 10 sample_size = 10
seed = 0.0 seed = 0.0
...@@ -3655,7 +3810,7 @@ def multinomial_test(): ...@@ -3655,7 +3810,7 @@ def multinomial_test():
return ([node], [input], [output]) return ([node], [input], [output])
@onnx_test @onnx_test()
def multinomial_generated_seed_test(): def multinomial_generated_seed_test():
sample_size = 10 sample_size = 10
input = helper.make_tensor_value_info("input", TensorProto.FLOAT, [1, 10]) input = helper.make_tensor_value_info("input", TensorProto.FLOAT, [1, 10])
...@@ -3670,7 +3825,7 @@ def multinomial_generated_seed_test(): ...@@ -3670,7 +3825,7 @@ def multinomial_generated_seed_test():
return ([node], [input], [output]) return ([node], [input], [output])
@onnx_test @onnx_test()
def multinomial_dtype_error_test(): def multinomial_dtype_error_test():
sample_size = 10 sample_size = 10
dtype = 0 dtype = 0
...@@ -3687,7 +3842,7 @@ def multinomial_dtype_error_test(): ...@@ -3687,7 +3842,7 @@ def multinomial_dtype_error_test():
return ([node], [input], [output]) return ([node], [input], [output])
@onnx_test @onnx_test()
def multinomial_int64_test(): def multinomial_int64_test():
sample_size = 10 sample_size = 10
dtype = 7 dtype = 7
...@@ -3706,7 +3861,7 @@ def multinomial_int64_test(): ...@@ -3706,7 +3861,7 @@ def multinomial_int64_test():
return ([node], [input], [output]) return ([node], [input], [output])
@onnx_test @onnx_test()
def neg_test(): def neg_test():
x = helper.make_tensor_value_info('0', TensorProto.INT64, [2, 3]) x = helper.make_tensor_value_info('0', TensorProto.INT64, [2, 3])
y = helper.make_tensor_value_info('1', TensorProto.INT64, [2, 3]) y = helper.make_tensor_value_info('1', TensorProto.INT64, [2, 3])
...@@ -3716,7 +3871,7 @@ def neg_test(): ...@@ -3716,7 +3871,7 @@ def neg_test():
return ([node], [x], [y]) return ([node], [x], [y])
@onnx_test @onnx_test()
def neg_dynamic_test(): def neg_dynamic_test():
x = helper.make_tensor_value_info('0', TensorProto.INT64, [None, 3]) x = helper.make_tensor_value_info('0', TensorProto.INT64, [None, 3])
y = helper.make_tensor_value_info('1', TensorProto.INT64, [None, 3]) y = helper.make_tensor_value_info('1', TensorProto.INT64, [None, 3])
...@@ -3726,7 +3881,7 @@ def neg_dynamic_test(): ...@@ -3726,7 +3881,7 @@ def neg_dynamic_test():
return ([node], [x], [y]) return ([node], [x], [y])
@onnx_test @onnx_test()
def nms_test(): def nms_test():
b = helper.make_tensor_value_info('boxes', TensorProto.FLOAT, [1, 6, 4]) b = helper.make_tensor_value_info('boxes', TensorProto.FLOAT, [1, 6, 4])
s = helper.make_tensor_value_info('scores', TensorProto.FLOAT, [1, 1, 6]) s = helper.make_tensor_value_info('scores', TensorProto.FLOAT, [1, 1, 6])
...@@ -3751,7 +3906,7 @@ def nms_test(): ...@@ -3751,7 +3906,7 @@ def nms_test():
return ([node], [b, s, mo, iou, st], [out]) return ([node], [b, s, mo, iou, st], [out])
@onnx_test @onnx_test()
def nms_use_dyn_output_false_test(): def nms_use_dyn_output_false_test():
b = helper.make_tensor_value_info('boxes', TensorProto.FLOAT, [1, 6, 4]) b = helper.make_tensor_value_info('boxes', TensorProto.FLOAT, [1, 6, 4])
s = helper.make_tensor_value_info('scores', TensorProto.FLOAT, [1, 1, 6]) s = helper.make_tensor_value_info('scores', TensorProto.FLOAT, [1, 1, 6])
...@@ -3776,7 +3931,7 @@ def nms_use_dyn_output_false_test(): ...@@ -3776,7 +3931,7 @@ def nms_use_dyn_output_false_test():
return ([node], [b, s, mo, iou, st], [out]) return ([node], [b, s, mo, iou, st], [out])
@onnx_test @onnx_test()
def nms_dynamic_batch_test(): def nms_dynamic_batch_test():
b = helper.make_tensor_value_info('boxes', TensorProto.FLOAT, [None, 6, 4]) b = helper.make_tensor_value_info('boxes', TensorProto.FLOAT, [None, 6, 4])
s = helper.make_tensor_value_info('scores', TensorProto.FLOAT, s = helper.make_tensor_value_info('scores', TensorProto.FLOAT,
...@@ -3803,7 +3958,7 @@ def nms_dynamic_batch_test(): ...@@ -3803,7 +3958,7 @@ def nms_dynamic_batch_test():
return ([node], [b, s, mo, iou, st], [out]) return ([node], [b, s, mo, iou, st], [out])
@onnx_test @onnx_test()
def nms_dynamic_boxes_test(): def nms_dynamic_boxes_test():
b = helper.make_tensor_value_info('boxes', TensorProto.FLOAT, [1, None, 4]) b = helper.make_tensor_value_info('boxes', TensorProto.FLOAT, [1, None, 4])
s = helper.make_tensor_value_info('scores', TensorProto.FLOAT, s = helper.make_tensor_value_info('scores', TensorProto.FLOAT,
...@@ -3828,7 +3983,7 @@ def nms_dynamic_boxes_test(): ...@@ -3828,7 +3983,7 @@ def nms_dynamic_boxes_test():
return ([node], [b, s, mo, iou, st], [out]) return ([node], [b, s, mo, iou, st], [out])
@onnx_test @onnx_test()
def nms_dynamic_classes_test(): def nms_dynamic_classes_test():
b = helper.make_tensor_value_info('boxes', TensorProto.FLOAT, [1, 6, 4]) b = helper.make_tensor_value_info('boxes', TensorProto.FLOAT, [1, 6, 4])
s = helper.make_tensor_value_info('scores', TensorProto.FLOAT, s = helper.make_tensor_value_info('scores', TensorProto.FLOAT,
...@@ -3853,7 +4008,7 @@ def nms_dynamic_classes_test(): ...@@ -3853,7 +4008,7 @@ def nms_dynamic_classes_test():
return ([node], [b, s, mo, iou, st], [out]) return ([node], [b, s, mo, iou, st], [out])
@onnx_test @onnx_test()
def not_test(): def not_test():
x = helper.make_tensor_value_info('0', TensorProto.INT32, [4]) x = helper.make_tensor_value_info('0', TensorProto.INT32, [4])
y = helper.make_tensor_value_info('1', TensorProto.INT32, [4]) y = helper.make_tensor_value_info('1', TensorProto.INT32, [4])
...@@ -3863,7 +4018,7 @@ def not_test(): ...@@ -3863,7 +4018,7 @@ def not_test():
return ([node], [x], [y]) return ([node], [x], [y])
@onnx_test @onnx_test()
def not_bool_test(): def not_bool_test():
x = helper.make_tensor_value_info('0', TensorProto.BOOL, [4]) x = helper.make_tensor_value_info('0', TensorProto.BOOL, [4])
y = helper.make_tensor_value_info('1', TensorProto.BOOL, [4]) y = helper.make_tensor_value_info('1', TensorProto.BOOL, [4])
...@@ -3873,7 +4028,7 @@ def not_bool_test(): ...@@ -3873,7 +4028,7 @@ def not_bool_test():
return ([node], [x], [y]) return ([node], [x], [y])
@onnx_test @onnx_test()
def no_pad_test(): def no_pad_test():
x = helper.make_tensor_value_info('0', TensorProto.FLOAT, [2, 2]) x = helper.make_tensor_value_info('0', TensorProto.FLOAT, [2, 2])
y = helper.make_tensor_value_info('1', TensorProto.FLOAT, [2, 2]) y = helper.make_tensor_value_info('1', TensorProto.FLOAT, [2, 2])
...@@ -3886,7 +4041,7 @@ def no_pad_test(): ...@@ -3886,7 +4041,7 @@ def no_pad_test():
return ([node], [x], [y]) return ([node], [x], [y])
@onnx_test @onnx_test()
def nonzero_dynamic_test(): def nonzero_dynamic_test():
x = helper.make_tensor_value_info('data', TensorProto.BOOL, [2, 2]) x = helper.make_tensor_value_info('data', TensorProto.BOOL, [2, 2])
y = helper.make_tensor_value_info('indices', TensorProto.INT64, [2, 3]) y = helper.make_tensor_value_info('indices', TensorProto.INT64, [2, 3])
...@@ -3898,7 +4053,7 @@ def nonzero_dynamic_test(): ...@@ -3898,7 +4053,7 @@ def nonzero_dynamic_test():
return ([node], [x], [y]) return ([node], [x], [y])
@onnx_test @onnx_test()
def nonzero_test(): def nonzero_test():
data1 = np.array([[1., 0.], [1., 1.]]) data1 = np.array([[1., 0.], [1., 1.]])
data = helper.make_tensor(name='data', data = helper.make_tensor(name='data',
...@@ -3914,7 +4069,7 @@ def nonzero_test(): ...@@ -3914,7 +4069,7 @@ def nonzero_test():
return ([node], [], [y], [data]) return ([node], [], [y], [data])
@onnx_test @onnx_test()
def nonzero_int_test(): def nonzero_int_test():
data1 = np.array([[1, 1, 0], [1, 0, 1]]) data1 = np.array([[1, 1, 0], [1, 0, 1]])
data = helper.make_tensor(name='data', data = helper.make_tensor(name='data',
...@@ -3930,7 +4085,7 @@ def nonzero_int_test(): ...@@ -3930,7 +4085,7 @@ def nonzero_int_test():
return ([node], [], [y], [data]) return ([node], [], [y], [data])
@onnx_test @onnx_test()
def onehot_test(): def onehot_test():
axis_value = 0 axis_value = 0
depth = np.array([3]) depth = np.array([3])
...@@ -3952,7 +4107,7 @@ def onehot_test(): ...@@ -3952,7 +4107,7 @@ def onehot_test():
return ([node], [indices, values], [y], [depth_tensor]) return ([node], [indices, values], [y], [depth_tensor])
@onnx_test @onnx_test()
def pad_test(): def pad_test():
x = helper.make_tensor_value_info('0', TensorProto.FLOAT, [2, 2]) x = helper.make_tensor_value_info('0', TensorProto.FLOAT, [2, 2])
y = helper.make_tensor_value_info('1', TensorProto.FLOAT, [4, 4]) y = helper.make_tensor_value_info('1', TensorProto.FLOAT, [4, 4])
...@@ -3965,7 +4120,7 @@ def pad_test(): ...@@ -3965,7 +4120,7 @@ def pad_test():
return ([node], [x], [y]) return ([node], [x], [y])
@onnx_test @onnx_test()
def pad_3arg_test(): def pad_3arg_test():
values = np.array([1]) values = np.array([1])
val_tensor = helper.make_tensor(name='val', val_tensor = helper.make_tensor(name='val',
...@@ -3997,7 +4152,7 @@ def pad_3arg_test(): ...@@ -3997,7 +4152,7 @@ def pad_3arg_test():
return ([arg_val, arg_pad, node], [x], [y]) return ([arg_val, arg_pad, node], [x], [y])
@onnx_test @onnx_test()
def pad_reflect_test(): def pad_reflect_test():
x = helper.make_tensor_value_info('0', TensorProto.FLOAT, [2, 2]) x = helper.make_tensor_value_info('0', TensorProto.FLOAT, [2, 2])
y = helper.make_tensor_value_info('1', TensorProto.FLOAT, [2, 5]) y = helper.make_tensor_value_info('1', TensorProto.FLOAT, [2, 5])
...@@ -4020,7 +4175,7 @@ def pad_reflect_test(): ...@@ -4020,7 +4175,7 @@ def pad_reflect_test():
return ([arg_pad, node], [x], [y]) return ([arg_pad, node], [x], [y])
@onnx_test @onnx_test()
def pad_reflect_multiaxis_test(): def pad_reflect_multiaxis_test():
x = helper.make_tensor_value_info('0', TensorProto.FLOAT, [2, 3]) x = helper.make_tensor_value_info('0', TensorProto.FLOAT, [2, 3])
y = helper.make_tensor_value_info('1', TensorProto.FLOAT, [4, 5]) y = helper.make_tensor_value_info('1', TensorProto.FLOAT, [4, 5])
...@@ -4043,7 +4198,7 @@ def pad_reflect_multiaxis_test(): ...@@ -4043,7 +4198,7 @@ def pad_reflect_multiaxis_test():
return ([arg_pad, node], [x], [y]) return ([arg_pad, node], [x], [y])
@onnx_test @onnx_test()
def pow_test(): def pow_test():
arg0 = helper.make_tensor_value_info('0', TensorProto.FLOAT, [2, 3, 4, 5]) arg0 = helper.make_tensor_value_info('0', TensorProto.FLOAT, [2, 3, 4, 5])
arg1 = helper.make_tensor_value_info('1', TensorProto.FLOAT, [2, 3, 4, 5]) arg1 = helper.make_tensor_value_info('1', TensorProto.FLOAT, [2, 3, 4, 5])
...@@ -4059,7 +4214,7 @@ def pow_test(): ...@@ -4059,7 +4214,7 @@ def pow_test():
return ([node], [arg0, arg1], [arg_out]) return ([node], [arg0, arg1], [arg_out])
@onnx_test @onnx_test()
def pow_fp32_i64_test(): def pow_fp32_i64_test():
arg0 = helper.make_tensor_value_info('0', TensorProto.FLOAT, [2, 3, 4, 5]) arg0 = helper.make_tensor_value_info('0', TensorProto.FLOAT, [2, 3, 4, 5])
arg1 = helper.make_tensor_value_info('1', TensorProto.INT64, [2, 3, 4, 5]) arg1 = helper.make_tensor_value_info('1', TensorProto.INT64, [2, 3, 4, 5])
...@@ -4075,7 +4230,7 @@ def pow_fp32_i64_test(): ...@@ -4075,7 +4230,7 @@ def pow_fp32_i64_test():
return ([node], [arg0, arg1], [arg_out]) return ([node], [arg0, arg1], [arg_out])
@onnx_test @onnx_test()
def pow_i64_fp32_test(): def pow_i64_fp32_test():
arg0 = helper.make_tensor_value_info('0', TensorProto.INT64, [2, 3, 4, 5]) arg0 = helper.make_tensor_value_info('0', TensorProto.INT64, [2, 3, 4, 5])
arg1 = helper.make_tensor_value_info('1', TensorProto.FLOAT, [2, 3, 4, 5]) arg1 = helper.make_tensor_value_info('1', TensorProto.FLOAT, [2, 3, 4, 5])
...@@ -4091,7 +4246,7 @@ def pow_i64_fp32_test(): ...@@ -4091,7 +4246,7 @@ def pow_i64_fp32_test():
return ([node], [arg0, arg1], [arg_out]) return ([node], [arg0, arg1], [arg_out])
@onnx_test @onnx_test()
def prefix_scan_sum_test(): def prefix_scan_sum_test():
x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [2, 2, 2]) x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [2, 2, 2])
y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [2, 2, 2]) y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [2, 2, 2])
...@@ -4108,7 +4263,7 @@ def prefix_scan_sum_test(): ...@@ -4108,7 +4263,7 @@ def prefix_scan_sum_test():
return ([node], [x], [y], [axis_tensor]) return ([node], [x], [y], [axis_tensor])
@onnx_test @onnx_test()
def prelu_brcst_test(): def prelu_brcst_test():
arg0 = helper.make_tensor_value_info('0', TensorProto.FLOAT, [2, 3, 4, 5]) arg0 = helper.make_tensor_value_info('0', TensorProto.FLOAT, [2, 3, 4, 5])
arg1 = helper.make_tensor_value_info('1', TensorProto.FLOAT, [4, 5]) arg1 = helper.make_tensor_value_info('1', TensorProto.FLOAT, [4, 5])
...@@ -4124,7 +4279,7 @@ def prelu_brcst_test(): ...@@ -4124,7 +4279,7 @@ def prelu_brcst_test():
return ([node], [arg0, arg1], [arg_out]) return ([node], [arg0, arg1], [arg_out])
@onnx_test @onnx_test()
def quantizelinear_test(): def quantizelinear_test():
arg0 = helper.make_tensor_value_info('0', TensorProto.FLOAT, [5]) arg0 = helper.make_tensor_value_info('0', TensorProto.FLOAT, [5])
arg1 = helper.make_tensor_value_info('1', TensorProto.FLOAT, [1]) arg1 = helper.make_tensor_value_info('1', TensorProto.FLOAT, [1])
...@@ -4139,7 +4294,7 @@ def quantizelinear_test(): ...@@ -4139,7 +4294,7 @@ def quantizelinear_test():
return ([node], [arg0, arg1], [arg_out]) return ([node], [arg0, arg1], [arg_out])
@onnx_test @onnx_test()
def quantizelinear_int32_test(): def quantizelinear_int32_test():
arg0 = helper.make_tensor_value_info('0', TensorProto.INT32, [5]) arg0 = helper.make_tensor_value_info('0', TensorProto.INT32, [5])
arg1 = helper.make_tensor_value_info('1', TensorProto.FLOAT, [1]) arg1 = helper.make_tensor_value_info('1', TensorProto.FLOAT, [1])
...@@ -4154,7 +4309,7 @@ def quantizelinear_int32_test(): ...@@ -4154,7 +4309,7 @@ def quantizelinear_int32_test():
return ([node], [arg0, arg1], [arg_out]) return ([node], [arg0, arg1], [arg_out])
@onnx_test @onnx_test()
def quantizelinear_zero_point_test(): def quantizelinear_zero_point_test():
arg0 = helper.make_tensor_value_info('0', TensorProto.FLOAT, [5]) arg0 = helper.make_tensor_value_info('0', TensorProto.FLOAT, [5])
arg1 = helper.make_tensor_value_info('1', TensorProto.FLOAT, [1]) arg1 = helper.make_tensor_value_info('1', TensorProto.FLOAT, [1])
...@@ -4185,17 +4340,17 @@ def make_quantizelinear_axis_graph(axis): ...@@ -4185,17 +4340,17 @@ def make_quantizelinear_axis_graph(axis):
return ([node], [arg0, arg1, arg2], [arg_out]) return ([node], [arg0, arg1, arg2], [arg_out])
@onnx_test @onnx_test()
def quantizelinear_axis_test(): def quantizelinear_axis_test():
return make_quantizelinear_axis_graph(2) return make_quantizelinear_axis_graph(2)
@onnx_test @onnx_test()
def quantizelinear_neg_axis_test(): def quantizelinear_neg_axis_test():
return make_quantizelinear_axis_graph(-2) return make_quantizelinear_axis_graph(-2)
@onnx_test @onnx_test()
def randomnormal_test(): def randomnormal_test():
dtype = 11 dtype = 11
mean = 10.0 mean = 10.0
...@@ -4217,7 +4372,7 @@ def randomnormal_test(): ...@@ -4217,7 +4372,7 @@ def randomnormal_test():
return ([node], [], [output]) return ([node], [], [output])
@onnx_test @onnx_test()
def randomnormal_dtype_error_test(): def randomnormal_dtype_error_test():
dtype = 6 dtype = 6
shape = [2, 3, 4] shape = [2, 3, 4]
...@@ -4233,7 +4388,7 @@ def randomnormal_dtype_error_test(): ...@@ -4233,7 +4388,7 @@ def randomnormal_dtype_error_test():
return ([node], [], [output]) return ([node], [], [output])
@onnx_test @onnx_test()
def randomnormal_generated_seed_test(): def randomnormal_generated_seed_test():
sample_size = 10 sample_size = 10
input = helper.make_tensor_value_info("input", TensorProto.FLOAT, [1, 10]) input = helper.make_tensor_value_info("input", TensorProto.FLOAT, [1, 10])
...@@ -4248,7 +4403,7 @@ def randomnormal_generated_seed_test(): ...@@ -4248,7 +4403,7 @@ def randomnormal_generated_seed_test():
return ([node], [input], [output]) return ([node], [input], [output])
@onnx_test @onnx_test()
def randomnormal_shape_error_test(): def randomnormal_shape_error_test():
dtype = 1 dtype = 1
output = helper.make_tensor_value_info('output', TensorProto.FLOAT, output = helper.make_tensor_value_info('output', TensorProto.FLOAT,
...@@ -4262,7 +4417,7 @@ def randomnormal_shape_error_test(): ...@@ -4262,7 +4417,7 @@ def randomnormal_shape_error_test():
return ([node], [], [output]) return ([node], [], [output])
@onnx_test @onnx_test()
def randomnormallike_test(): def randomnormallike_test():
dtype = 10 dtype = 10
mean = 10.0 mean = 10.0
...@@ -4284,7 +4439,7 @@ def randomnormallike_test(): ...@@ -4284,7 +4439,7 @@ def randomnormallike_test():
return ([node], [input], [output]) return ([node], [input], [output])
@onnx_test @onnx_test()
def randomnormallike_type_error_test(): def randomnormallike_type_error_test():
seed = 0 seed = 0
input = helper.make_tensor_value_info('input', TensorProto.INT32, input = helper.make_tensor_value_info('input', TensorProto.INT32,
...@@ -4300,7 +4455,7 @@ def randomnormallike_type_error_test(): ...@@ -4300,7 +4455,7 @@ def randomnormallike_type_error_test():
return ([node], [input], [output]) return ([node], [input], [output])
@onnx_test @onnx_test()
def randomuniform_test(): def randomuniform_test():
dtype = 11 dtype = 11
high = 1.0 high = 1.0
...@@ -4322,7 +4477,7 @@ def randomuniform_test(): ...@@ -4322,7 +4477,7 @@ def randomuniform_test():
return ([node], [], [output]) return ([node], [], [output])
@onnx_test @onnx_test()
def randomuniform_dtype_error_test(): def randomuniform_dtype_error_test():
dtype = 6 dtype = 6
shape = [2, 3, 4] shape = [2, 3, 4]
...@@ -4338,7 +4493,7 @@ def randomuniform_dtype_error_test(): ...@@ -4338,7 +4493,7 @@ def randomuniform_dtype_error_test():
return ([node], [], [output]) return ([node], [], [output])
@onnx_test @onnx_test()
def randomuniform_generated_seed_test(): def randomuniform_generated_seed_test():
sample_size = 10 sample_size = 10
input = helper.make_tensor_value_info("input", TensorProto.FLOAT, [1, 10]) input = helper.make_tensor_value_info("input", TensorProto.FLOAT, [1, 10])
...@@ -4353,7 +4508,7 @@ def randomuniform_generated_seed_test(): ...@@ -4353,7 +4508,7 @@ def randomuniform_generated_seed_test():
return ([node], [input], [output]) return ([node], [input], [output])
@onnx_test @onnx_test()
def randomuniform_shape_error_test(): def randomuniform_shape_error_test():
dtype = 1 dtype = 1
output = helper.make_tensor_value_info('output', TensorProto.FLOAT, output = helper.make_tensor_value_info('output', TensorProto.FLOAT,
...@@ -4367,7 +4522,7 @@ def randomuniform_shape_error_test(): ...@@ -4367,7 +4522,7 @@ def randomuniform_shape_error_test():
return ([node], [], [output]) return ([node], [], [output])
@onnx_test @onnx_test()
def randomuniformlike_test(): def randomuniformlike_test():
dtype = 10 dtype = 10
high = 10.0 high = 10.0
...@@ -4389,7 +4544,7 @@ def randomuniformlike_test(): ...@@ -4389,7 +4544,7 @@ def randomuniformlike_test():
return ([node], [input], [output]) return ([node], [input], [output])
@onnx_test @onnx_test()
def randomuniformlike_type_error_test(): def randomuniformlike_type_error_test():
seed = 0 seed = 0
input = helper.make_tensor_value_info('input', TensorProto.INT32, input = helper.make_tensor_value_info('input', TensorProto.INT32,
...@@ -4405,7 +4560,7 @@ def randomuniformlike_type_error_test(): ...@@ -4405,7 +4560,7 @@ def randomuniformlike_type_error_test():
return ([node], [input], [output]) return ([node], [input], [output])
@onnx_test @onnx_test()
def range_test(): def range_test():
start_val = np.array([10]) start_val = np.array([10])
...@@ -4448,7 +4603,7 @@ def range_test(): ...@@ -4448,7 +4603,7 @@ def range_test():
return ([start, limit, delta, node], [], [y]) return ([start, limit, delta, node], [], [y])
@onnx_test @onnx_test()
def range_float_test(): def range_float_test():
start_val = np.array([2]) start_val = np.array([2])
...@@ -4491,7 +4646,7 @@ def range_float_test(): ...@@ -4491,7 +4646,7 @@ def range_float_test():
return ([start, limit, delta, node], [], [y]) return ([start, limit, delta, node], [], [y])
@onnx_test @onnx_test()
def recip_test(): def recip_test():
x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [3]) x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [3])
y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [3]) y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [3])
...@@ -4505,7 +4660,7 @@ def recip_test(): ...@@ -4505,7 +4660,7 @@ def recip_test():
return ([node], [x], [y]) return ([node], [x], [y])
@onnx_test @onnx_test()
def reducel1_test(): def reducel1_test():
x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [3, 4, 5, 6]) x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [3, 4, 5, 6])
y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [3, 4, 6]) y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [3, 4, 6])
...@@ -4520,7 +4675,7 @@ def reducel1_test(): ...@@ -4520,7 +4675,7 @@ def reducel1_test():
return ([node], [x], [y]) return ([node], [x], [y])
@onnx_test @onnx_test()
def reducel2_test(): def reducel2_test():
x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [3, 4, 5, 6]) x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [3, 4, 5, 6])
y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [3, 4, 5]) y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [3, 4, 5])
...@@ -4535,7 +4690,7 @@ def reducel2_test(): ...@@ -4535,7 +4690,7 @@ def reducel2_test():
return ([node], [x], [y]) return ([node], [x], [y])
@onnx_test @onnx_test()
def reduce_log_sum_test(): def reduce_log_sum_test():
x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [3, 4, 5, 6]) x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [3, 4, 5, 6])
y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [3, 1, 5, 6]) y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [3, 1, 5, 6])
...@@ -4550,7 +4705,7 @@ def reduce_log_sum_test(): ...@@ -4550,7 +4705,7 @@ def reduce_log_sum_test():
return ([node], [x], [y]) return ([node], [x], [y])
@onnx_test @onnx_test()
def reduce_log_sum_exp_test(): def reduce_log_sum_exp_test():
x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [3, 4, 5, 6]) x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [3, 4, 5, 6])
y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [4, 5, 6]) y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [4, 5, 6])
...@@ -4565,7 +4720,7 @@ def reduce_log_sum_exp_test(): ...@@ -4565,7 +4720,7 @@ def reduce_log_sum_exp_test():
return ([node], [x], [y]) return ([node], [x], [y])
@onnx_test @onnx_test()
def reducemax_test(): def reducemax_test():
x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [3, 4, 5, 6]) x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [3, 4, 5, 6])
y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [3, 4, 6]) y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [3, 4, 6])
...@@ -4580,7 +4735,7 @@ def reducemax_test(): ...@@ -4580,7 +4735,7 @@ def reducemax_test():
return ([node], [x], [y]) return ([node], [x], [y])
@onnx_test @onnx_test()
def reducemean_test(): def reducemean_test():
x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [3, 4, 5, 6]) x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [3, 4, 5, 6])
y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [3, 4]) y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [3, 4])
...@@ -4595,7 +4750,7 @@ def reducemean_test(): ...@@ -4595,7 +4750,7 @@ def reducemean_test():
return ([node], [x], [y]) return ([node], [x], [y])
@onnx_test @onnx_test()
def reducemean_keepdims_test(): def reducemean_keepdims_test():
x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [3, 4, 5, 6]) x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [3, 4, 5, 6])
y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [3, 4, 1, 6]) y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [3, 4, 1, 6])
...@@ -4610,7 +4765,7 @@ def reducemean_keepdims_test(): ...@@ -4610,7 +4765,7 @@ def reducemean_keepdims_test():
return ([node], [x], [y]) return ([node], [x], [y])
@onnx_test @onnx_test()
def reducemin_test(): def reducemin_test():
x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [3, 4, 5, 6]) x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [3, 4, 5, 6])
y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [3, 1, 5, 1]) y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [3, 1, 5, 1])
...@@ -4625,7 +4780,7 @@ def reducemin_test(): ...@@ -4625,7 +4780,7 @@ def reducemin_test():
return ([node], [x], [y]) return ([node], [x], [y])
@onnx_test @onnx_test()
def reduceprod_test(): def reduceprod_test():
x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [3, 4, 5, 6]) x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [3, 4, 5, 6])
y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [3, 4, 1, 6]) y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [3, 4, 1, 6])
...@@ -4640,7 +4795,7 @@ def reduceprod_test(): ...@@ -4640,7 +4795,7 @@ def reduceprod_test():
return ([node], [x], [y]) return ([node], [x], [y])
@onnx_test @onnx_test()
def reducesum_test(): def reducesum_test():
x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [3, 4, 5, 6]) x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [3, 4, 5, 6])
y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [3, 4, 1, 6]) y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [3, 4, 1, 6])
...@@ -4655,7 +4810,7 @@ def reducesum_test(): ...@@ -4655,7 +4810,7 @@ def reducesum_test():
return ([node], [x], [y]) return ([node], [x], [y])
@onnx_test @onnx_test()
def reducesum_empty_axes_test(): def reducesum_empty_axes_test():
x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [3, 4, 5, 6]) x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [3, 4, 5, 6])
y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [3, 4, 1, 6]) y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [3, 4, 1, 6])
...@@ -4674,7 +4829,7 @@ def reducesum_empty_axes_test(): ...@@ -4674,7 +4829,7 @@ def reducesum_empty_axes_test():
return ([node], [x], [y], [axes_tensor]) return ([node], [x], [y], [axes_tensor])
@onnx_test @onnx_test()
def reducesum_noop_test(): def reducesum_noop_test():
x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [3, 4, 5, 6]) x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [3, 4, 5, 6])
y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [3, 4, 1, 6]) y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [3, 4, 1, 6])
...@@ -4693,7 +4848,7 @@ def reducesum_noop_test(): ...@@ -4693,7 +4848,7 @@ def reducesum_noop_test():
return ([node], [x], [y], [axes_tensor]) return ([node], [x], [y], [axes_tensor])
@onnx_test @onnx_test()
def reducesum_keepdims_test(): def reducesum_keepdims_test():
x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [3, 4, 5, 6]) x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [3, 4, 5, 6])
y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [3, 4, 1, 1]) y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [3, 4, 1, 1])
...@@ -4708,7 +4863,7 @@ def reducesum_keepdims_test(): ...@@ -4708,7 +4863,7 @@ def reducesum_keepdims_test():
return ([node], [x], [y]) return ([node], [x], [y])
@onnx_test @onnx_test()
def reducesum_multiaxis_test(): def reducesum_multiaxis_test():
x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [3, 4, 5, 6]) x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [3, 4, 5, 6])
y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [3, 4, 1, 1]) y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [3, 4, 1, 1])
...@@ -4723,7 +4878,7 @@ def reducesum_multiaxis_test(): ...@@ -4723,7 +4878,7 @@ def reducesum_multiaxis_test():
return ([node], [x], [y]) return ([node], [x], [y])
@onnx_test @onnx_test()
def reducesum_square_test(): def reducesum_square_test():
x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [3, 4, 5, 6]) x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [3, 4, 5, 6])
y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [3, 4, 6]) y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [3, 4, 6])
...@@ -4738,7 +4893,7 @@ def reducesum_square_test(): ...@@ -4738,7 +4893,7 @@ def reducesum_square_test():
return ([node], [x], [y]) return ([node], [x], [y])
@onnx_test @onnx_test()
def reshape_test(): def reshape_test():
x = helper.make_tensor_value_info('0', TensorProto.FLOAT, [4, 2, 3]) x = helper.make_tensor_value_info('0', TensorProto.FLOAT, [4, 2, 3])
x_shape = helper.make_tensor_value_info('1', TensorProto.INT64, [2]) x_shape = helper.make_tensor_value_info('1', TensorProto.INT64, [2])
...@@ -4757,7 +4912,7 @@ def reshape_test(): ...@@ -4757,7 +4912,7 @@ def reshape_test():
[helper.make_tensor('1', TensorProto.INT64, [2], [3, 8])]) [helper.make_tensor('1', TensorProto.INT64, [2], [3, 8])])
@onnx_test @onnx_test()
def reshape_non_standard_test(): def reshape_non_standard_test():
x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [2, 3, 4]) x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [2, 3, 4])
y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [4, 3, 2]) y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [4, 3, 2])
...@@ -4777,7 +4932,7 @@ def reshape_non_standard_test(): ...@@ -4777,7 +4932,7 @@ def reshape_non_standard_test():
return ([trans, res], [x], [y]) return ([trans, res], [x], [y])
@onnx_test @onnx_test()
def resize_downsample_f_test(): def resize_downsample_f_test():
scales = np.array([1.0, 1.0, 0.6, 0.6], dtype=np.float32) scales = np.array([1.0, 1.0, 0.6, 0.6], dtype=np.float32)
scale_tensor = helper.make_tensor(name='scales', scale_tensor = helper.make_tensor(name='scales',
...@@ -4799,7 +4954,7 @@ def resize_downsample_f_test(): ...@@ -4799,7 +4954,7 @@ def resize_downsample_f_test():
return ([node], [X], [Y], [scale_tensor]) return ([node], [X], [Y], [scale_tensor])
@onnx_test @onnx_test()
def resize_downsample_c_test(): def resize_downsample_c_test():
scales = np.array([1.0, 1.0, 0.6, 0.6], dtype=np.float32) scales = np.array([1.0, 1.0, 0.6, 0.6], dtype=np.float32)
scale_tensor = helper.make_tensor(name='scales', scale_tensor = helper.make_tensor(name='scales',
...@@ -4820,7 +4975,7 @@ def resize_downsample_c_test(): ...@@ -4820,7 +4975,7 @@ def resize_downsample_c_test():
return ([node], [X], [Y], [scale_tensor]) return ([node], [X], [Y], [scale_tensor])
@onnx_test @onnx_test()
def resize_downsample_linear_test(): def resize_downsample_linear_test():
scales = np.array([1.0, 1.0, 0.6, 0.5], dtype=np.float32) scales = np.array([1.0, 1.0, 0.6, 0.5], dtype=np.float32)
scale_tensor = helper.make_tensor(name='scales', scale_tensor = helper.make_tensor(name='scales',
...@@ -4839,7 +4994,7 @@ def resize_downsample_linear_test(): ...@@ -4839,7 +4994,7 @@ def resize_downsample_linear_test():
return ([node], [X], [Y], [scale_tensor]) return ([node], [X], [Y], [scale_tensor])
@onnx_test @onnx_test()
def resize_nonstd_input_test(): def resize_nonstd_input_test():
scales = np.array([1.0, 1.0, 0.6, 0.6], dtype=np.float32) scales = np.array([1.0, 1.0, 0.6, 0.6], dtype=np.float32)
scale_tensor = helper.make_tensor(name='scales', scale_tensor = helper.make_tensor(name='scales',
...@@ -4865,7 +5020,7 @@ def resize_nonstd_input_test(): ...@@ -4865,7 +5020,7 @@ def resize_nonstd_input_test():
return ([trn, node], [X], [Y], [scale_tensor]) return ([trn, node], [X], [Y], [scale_tensor])
@onnx_test @onnx_test()
def resize_outsize_test(): def resize_outsize_test():
out_lens = np.array([1, 1, 4, 6], dtype=np.int64) out_lens = np.array([1, 1, 4, 6], dtype=np.int64)
out_lens_tensor = helper.make_tensor(name='out_lens', out_lens_tensor = helper.make_tensor(name='out_lens',
...@@ -4888,7 +5043,7 @@ def resize_outsize_test(): ...@@ -4888,7 +5043,7 @@ def resize_outsize_test():
return ([node], [X], [Y], [out_lens_tensor]) return ([node], [X], [Y], [out_lens_tensor])
@onnx_test @onnx_test()
def resize_upsample_linear_ac_test(): def resize_upsample_linear_ac_test():
scales = np.array([1.0, 1.0, 2.0, 2.0], dtype=np.float32) scales = np.array([1.0, 1.0, 2.0, 2.0], dtype=np.float32)
scales_tensor = helper.make_tensor(name='scales', scales_tensor = helper.make_tensor(name='scales',
...@@ -4909,7 +5064,7 @@ def resize_upsample_linear_ac_test(): ...@@ -4909,7 +5064,7 @@ def resize_upsample_linear_ac_test():
return ([node], [X], [Y], [scales_tensor]) return ([node], [X], [Y], [scales_tensor])
@onnx_test @onnx_test()
def resize_upsample_linear_test(): def resize_upsample_linear_test():
scales = np.array([1.0, 1.0, 2.0, 2.0], dtype=np.float32) scales = np.array([1.0, 1.0, 2.0, 2.0], dtype=np.float32)
scales_tensor = helper.make_tensor(name='scales', scales_tensor = helper.make_tensor(name='scales',
...@@ -4928,7 +5083,7 @@ def resize_upsample_linear_test(): ...@@ -4928,7 +5083,7 @@ def resize_upsample_linear_test():
return ([node], [X], [Y], [scales_tensor]) return ([node], [X], [Y], [scales_tensor])
@onnx_test @onnx_test()
def resize_upsample_pf_test(): def resize_upsample_pf_test():
scales = np.array([1.0, 1.0, 2.0, 3.0], dtype=np.float32) scales = np.array([1.0, 1.0, 2.0, 3.0], dtype=np.float32)
scale_tensor = helper.make_tensor(name='scales', scale_tensor = helper.make_tensor(name='scales',
...@@ -4947,7 +5102,7 @@ def resize_upsample_pf_test(): ...@@ -4947,7 +5102,7 @@ def resize_upsample_pf_test():
return ([node], [X], [Y], [scale_tensor]) return ([node], [X], [Y], [scale_tensor])
@onnx_test @onnx_test()
def resize_upsample_pc_test(): def resize_upsample_pc_test():
scales = np.array([1.0, 1.0, 2.0, 1.5], dtype=np.float32) scales = np.array([1.0, 1.0, 2.0, 1.5], dtype=np.float32)
scale_tensor = helper.make_tensor(name='scales', scale_tensor = helper.make_tensor(name='scales',
...@@ -4970,7 +5125,7 @@ def resize_upsample_pc_test(): ...@@ -4970,7 +5125,7 @@ def resize_upsample_pc_test():
return ([node], [X], [Y], [scale_tensor]) return ([node], [X], [Y], [scale_tensor])
@onnx_test @onnx_test()
def reversesequence_4D_test(): def reversesequence_4D_test():
x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [2, 2, 2, 2]) x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [2, 2, 2, 2])
y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [2, 2, 2, 2]) y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [2, 2, 2, 2])
...@@ -4986,7 +5141,7 @@ def reversesequence_4D_test(): ...@@ -4986,7 +5141,7 @@ def reversesequence_4D_test():
return ([node], [x], [y]) return ([node], [x], [y])
@onnx_test @onnx_test()
def reversesequence_batch_test(): def reversesequence_batch_test():
x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [4, 4]) x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [4, 4])
seq_lens = np.array([1, 2, 3, 4]) seq_lens = np.array([1, 2, 3, 4])
...@@ -5014,7 +5169,7 @@ def reversesequence_batch_test(): ...@@ -5014,7 +5169,7 @@ def reversesequence_batch_test():
return ([arg_seq_lens, node], [x], [y]) return ([arg_seq_lens, node], [x], [y])
@onnx_test @onnx_test()
def reversesequence_batch_axis_err_test(): def reversesequence_batch_axis_err_test():
x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [4, 4, 2]) x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [4, 4, 2])
y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [4, 4, 2]) y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [4, 4, 2])
...@@ -5030,7 +5185,7 @@ def reversesequence_batch_axis_err_test(): ...@@ -5030,7 +5185,7 @@ def reversesequence_batch_axis_err_test():
return ([node], [x], [y]) return ([node], [x], [y])
@onnx_test @onnx_test()
def reversesequence_rank_err_test(): def reversesequence_rank_err_test():
x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [4]) x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [4])
y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [4]) y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [4])
...@@ -5044,7 +5199,7 @@ def reversesequence_rank_err_test(): ...@@ -5044,7 +5199,7 @@ def reversesequence_rank_err_test():
return ([node], [x], [y]) return ([node], [x], [y])
@onnx_test @onnx_test()
def reversesequence_sequence_lens_shape_err_test(): def reversesequence_sequence_lens_shape_err_test():
x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [4, 4]) x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [4, 4])
y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [4, 4]) y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [4, 4])
...@@ -5058,7 +5213,7 @@ def reversesequence_sequence_lens_shape_err_test(): ...@@ -5058,7 +5213,7 @@ def reversesequence_sequence_lens_shape_err_test():
return ([node], [x], [y]) return ([node], [x], [y])
@onnx_test @onnx_test()
def reversesequence_same_axis_err_test(): def reversesequence_same_axis_err_test():
x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [4, 4]) x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [4, 4])
y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [4, 4]) y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [4, 4])
...@@ -5074,7 +5229,7 @@ def reversesequence_same_axis_err_test(): ...@@ -5074,7 +5229,7 @@ def reversesequence_same_axis_err_test():
return ([node], [x], [y]) return ([node], [x], [y])
@onnx_test @onnx_test()
def reversesequence_time_axis_err_test(): def reversesequence_time_axis_err_test():
x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [4, 4, 2, 3]) x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [4, 4, 2, 3])
y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [4, 4, 2, 3]) y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [4, 4, 2, 3])
...@@ -5090,7 +5245,7 @@ def reversesequence_time_axis_err_test(): ...@@ -5090,7 +5245,7 @@ def reversesequence_time_axis_err_test():
return ([node], [x], [y]) return ([node], [x], [y])
@onnx_test @onnx_test()
def reversesequence_time_test(): def reversesequence_time_test():
x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [4, 4]) x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [4, 4])
y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [4, 4]) y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [4, 4])
...@@ -5106,7 +5261,7 @@ def reversesequence_time_test(): ...@@ -5106,7 +5261,7 @@ def reversesequence_time_test():
return ([node], [x], [y]) return ([node], [x], [y])
@onnx_test @onnx_test()
def roialign_default_test(): def roialign_default_test():
x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [10, 4, 7, 8]) x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [10, 4, 7, 8])
roi = helper.make_tensor_value_info('rois', TensorProto.FLOAT, [8, 4]) roi = helper.make_tensor_value_info('rois', TensorProto.FLOAT, [8, 4])
...@@ -5120,7 +5275,7 @@ def roialign_default_test(): ...@@ -5120,7 +5275,7 @@ def roialign_default_test():
return ([node], [x, roi, bi], [y]) return ([node], [x, roi, bi], [y])
@onnx_test @onnx_test()
def roialign_test(): def roialign_test():
x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [10, 5, 4, 7]) x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [10, 5, 4, 7])
roi = helper.make_tensor_value_info('rois', TensorProto.FLOAT, [8, 4]) roi = helper.make_tensor_value_info('rois', TensorProto.FLOAT, [8, 4])
...@@ -5141,7 +5296,7 @@ def roialign_test(): ...@@ -5141,7 +5296,7 @@ def roialign_test():
return ([node], [x, roi, bi], [y]) return ([node], [x, roi, bi], [y])
@onnx_test @onnx_test()
def scatter_add_test(): def scatter_add_test():
x = helper.make_tensor_value_info('data', TensorProto.FLOAT, [3, 4, 5, 6]) x = helper.make_tensor_value_info('data', TensorProto.FLOAT, [3, 4, 5, 6])
i = helper.make_tensor_value_info('indices', TensorProto.INT32, i = helper.make_tensor_value_info('indices', TensorProto.INT32,
...@@ -5161,7 +5316,7 @@ def scatter_add_test(): ...@@ -5161,7 +5316,7 @@ def scatter_add_test():
return ([node], [x, i, u], [y]) return ([node], [x, i, u], [y])
@onnx_test @onnx_test()
def scatter_mul_test(): def scatter_mul_test():
x = helper.make_tensor_value_info('data', TensorProto.FLOAT, [3, 4, 5, 6]) x = helper.make_tensor_value_info('data', TensorProto.FLOAT, [3, 4, 5, 6])
i = helper.make_tensor_value_info('indices', TensorProto.INT32, i = helper.make_tensor_value_info('indices', TensorProto.INT32,
...@@ -5181,7 +5336,7 @@ def scatter_mul_test(): ...@@ -5181,7 +5336,7 @@ def scatter_mul_test():
return ([node], [x, i, u], [y]) return ([node], [x, i, u], [y])
@onnx_test @onnx_test()
def scatter_none_test(): def scatter_none_test():
x = helper.make_tensor_value_info('data', TensorProto.FLOAT, [3, 4, 5, 6]) x = helper.make_tensor_value_info('data', TensorProto.FLOAT, [3, 4, 5, 6])
i = helper.make_tensor_value_info('indices', TensorProto.INT32, i = helper.make_tensor_value_info('indices', TensorProto.INT32,
...@@ -5201,7 +5356,7 @@ def scatter_none_test(): ...@@ -5201,7 +5356,7 @@ def scatter_none_test():
return ([node], [x, i, u], [y]) return ([node], [x, i, u], [y])
@onnx_test @onnx_test()
def scatternd_add_test(): def scatternd_add_test():
data = helper.make_tensor_value_info('data', TensorProto.FLOAT, [2, 2, 2]) data = helper.make_tensor_value_info('data', TensorProto.FLOAT, [2, 2, 2])
indices = helper.make_tensor_value_info('indices', TensorProto.INT64, indices = helper.make_tensor_value_info('indices', TensorProto.INT64,
...@@ -5219,7 +5374,7 @@ def scatternd_add_test(): ...@@ -5219,7 +5374,7 @@ def scatternd_add_test():
return ([node], [data, indices, updates], [output]) return ([node], [data, indices, updates], [output])
@onnx_test @onnx_test()
def scatternd_mul_test(): def scatternd_mul_test():
data = helper.make_tensor_value_info('data', TensorProto.FLOAT, [2, 2, 2]) data = helper.make_tensor_value_info('data', TensorProto.FLOAT, [2, 2, 2])
indices = helper.make_tensor_value_info('indices', TensorProto.INT64, indices = helper.make_tensor_value_info('indices', TensorProto.INT64,
...@@ -5237,7 +5392,7 @@ def scatternd_mul_test(): ...@@ -5237,7 +5392,7 @@ def scatternd_mul_test():
return ([node], [data, indices, updates], [output]) return ([node], [data, indices, updates], [output])
@onnx_test @onnx_test()
def scatternd_test(): def scatternd_test():
data = helper.make_tensor_value_info('data', TensorProto.FLOAT, [2, 2, 2]) data = helper.make_tensor_value_info('data', TensorProto.FLOAT, [2, 2, 2])
indices = helper.make_tensor_value_info('indices', TensorProto.INT64, indices = helper.make_tensor_value_info('indices', TensorProto.INT64,
...@@ -5254,7 +5409,7 @@ def scatternd_test(): ...@@ -5254,7 +5409,7 @@ def scatternd_test():
return ([node], [data, indices, updates], [output]) return ([node], [data, indices, updates], [output])
@onnx_test @onnx_test()
def selu_test(): def selu_test():
x = helper.make_tensor_value_info('x', TensorProto.DOUBLE, [2, 3]) x = helper.make_tensor_value_info('x', TensorProto.DOUBLE, [2, 3])
y = helper.make_tensor_value_info('y', TensorProto.DOUBLE, [2, 3]) y = helper.make_tensor_value_info('y', TensorProto.DOUBLE, [2, 3])
...@@ -5268,7 +5423,7 @@ def selu_test(): ...@@ -5268,7 +5423,7 @@ def selu_test():
return ([node], [x], [y]) return ([node], [x], [y])
@onnx_test @onnx_test()
def shape_test(): def shape_test():
x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [3, 4, 5, 6]) x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [3, 4, 5, 6])
y = helper.make_tensor_value_info('y', TensorProto.INT64, [4]) y = helper.make_tensor_value_info('y', TensorProto.INT64, [4])
...@@ -5282,7 +5437,7 @@ def shape_test(): ...@@ -5282,7 +5437,7 @@ def shape_test():
return ([node], [x], [y]) return ([node], [x], [y])
@onnx_test @onnx_test()
def shape_gather_test(): def shape_gather_test():
values = np.array([1]) values = np.array([1])
# value = helper.make_tensor_value_info('value', TensorProto.INT32, [1]) # value = helper.make_tensor_value_info('value', TensorProto.INT32, [1])
...@@ -5317,7 +5472,7 @@ def shape_gather_test(): ...@@ -5317,7 +5472,7 @@ def shape_gather_test():
return ([node_const, node_shape, node_gather], [x], [z]) return ([node_const, node_shape, node_gather], [x], [z])
@onnx_test @onnx_test()
def sign_test(): def sign_test():
x = helper.make_tensor_value_info('x', TensorProto.DOUBLE, [10, 5]) x = helper.make_tensor_value_info('x', TensorProto.DOUBLE, [10, 5])
y = helper.make_tensor_value_info('y', TensorProto.DOUBLE, [10, 5]) y = helper.make_tensor_value_info('y', TensorProto.DOUBLE, [10, 5])
...@@ -5331,7 +5486,7 @@ def sign_test(): ...@@ -5331,7 +5486,7 @@ def sign_test():
return ([node], [x], [y]) return ([node], [x], [y])
@onnx_test @onnx_test()
def sin_test(): def sin_test():
x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [10]) x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [10])
y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [10]) y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [10])
...@@ -5345,7 +5500,7 @@ def sin_test(): ...@@ -5345,7 +5500,7 @@ def sin_test():
return ([node], [x], [y]) return ([node], [x], [y])
@onnx_test @onnx_test()
def sinh_test(): def sinh_test():
x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [10]) x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [10])
y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [10]) y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [10])
...@@ -5359,7 +5514,7 @@ def sinh_test(): ...@@ -5359,7 +5514,7 @@ def sinh_test():
return ([node], [x], [y]) return ([node], [x], [y])
@onnx_test @onnx_test()
def sinh_dynamic_test(): def sinh_dynamic_test():
x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [None]) x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [None])
y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [None]) y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [None])
...@@ -5373,7 +5528,7 @@ def sinh_dynamic_test(): ...@@ -5373,7 +5528,7 @@ def sinh_dynamic_test():
return ([node], [x], [y]) return ([node], [x], [y])
@onnx_test @onnx_test()
def size_float_test(): def size_float_test():
x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [2, 3, 4]) x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [2, 3, 4])
y = helper.make_tensor_value_info('y', TensorProto.INT64, [1]) y = helper.make_tensor_value_info('y', TensorProto.INT64, [1])
...@@ -5385,7 +5540,7 @@ def size_float_test(): ...@@ -5385,7 +5540,7 @@ def size_float_test():
return ([node], [x], [y]) return ([node], [x], [y])
@onnx_test @onnx_test()
def size_half_test(): def size_half_test():
x = helper.make_tensor_value_info('x', TensorProto.FLOAT16, [3, 1]) x = helper.make_tensor_value_info('x', TensorProto.FLOAT16, [3, 1])
y = helper.make_tensor_value_info('y', TensorProto.INT64, [1]) y = helper.make_tensor_value_info('y', TensorProto.INT64, [1])
...@@ -5397,7 +5552,7 @@ def size_half_test(): ...@@ -5397,7 +5552,7 @@ def size_half_test():
return ([node], [x], [y]) return ([node], [x], [y])
@onnx_test @onnx_test()
def size_int_test(): def size_int_test():
x = helper.make_tensor_value_info('x', TensorProto.INT32, [8, 2, 3]) x = helper.make_tensor_value_info('x', TensorProto.INT32, [8, 2, 3])
y = helper.make_tensor_value_info('y', TensorProto.INT64, [1]) y = helper.make_tensor_value_info('y', TensorProto.INT64, [1])
...@@ -5409,7 +5564,7 @@ def size_int_test(): ...@@ -5409,7 +5564,7 @@ def size_int_test():
return ([node], [x], [y]) return ([node], [x], [y])
@onnx_test @onnx_test()
def size_verify_test(): def size_verify_test():
x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [2, 5, 3]) x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [2, 5, 3])
y = helper.make_tensor_value_info('y', TensorProto.INT64, [1]) y = helper.make_tensor_value_info('y', TensorProto.INT64, [1])
...@@ -5421,7 +5576,7 @@ def size_verify_test(): ...@@ -5421,7 +5576,7 @@ def size_verify_test():
return ([node], [x], [y]) return ([node], [x], [y])
@onnx_test @onnx_test()
def slice_test(): def slice_test():
x = helper.make_tensor_value_info('0', TensorProto.FLOAT, [3, 2]) x = helper.make_tensor_value_info('0', TensorProto.FLOAT, [3, 2])
y = helper.make_tensor_value_info('1', TensorProto.FLOAT, [1, 2]) y = helper.make_tensor_value_info('1', TensorProto.FLOAT, [1, 2])
...@@ -5436,7 +5591,7 @@ def slice_test(): ...@@ -5436,7 +5591,7 @@ def slice_test():
return ([node], [x], [y]) return ([node], [x], [y])
@onnx_test @onnx_test()
def slice_3arg_test(): def slice_3arg_test():
x = helper.make_tensor_value_info('0', TensorProto.FLOAT, [5, 5]) x = helper.make_tensor_value_info('0', TensorProto.FLOAT, [5, 5])
y = helper.make_tensor_value_info('1', TensorProto.FLOAT, [2, 5]) y = helper.make_tensor_value_info('1', TensorProto.FLOAT, [2, 5])
...@@ -5468,7 +5623,7 @@ def slice_3arg_test(): ...@@ -5468,7 +5623,7 @@ def slice_3arg_test():
return ([arg_start, arg_end, node], [x], [y]) return ([arg_start, arg_end, node], [x], [y])
@onnx_test @onnx_test()
def slice_5arg_test(): def slice_5arg_test():
step = np.array([1, 1]) step = np.array([1, 1])
step_tensor = helper.make_tensor(name="step", step_tensor = helper.make_tensor(name="step",
...@@ -5521,7 +5676,7 @@ def slice_5arg_test(): ...@@ -5521,7 +5676,7 @@ def slice_5arg_test():
return ([arg_step, arg_axis, arg_end, arg_start, node], [x], [y]) return ([arg_step, arg_axis, arg_end, arg_start, node], [x], [y])
@onnx_test @onnx_test()
def slice_5arg_reverse_test(): def slice_5arg_reverse_test():
step = np.array([-1, 1]) step = np.array([-1, 1])
step_tensor = helper.make_tensor(name="step", step_tensor = helper.make_tensor(name="step",
...@@ -5574,7 +5729,7 @@ def slice_5arg_reverse_test(): ...@@ -5574,7 +5729,7 @@ def slice_5arg_reverse_test():
return ([arg_step, arg_axis, arg_end, arg_start, node], [x], [y]) return ([arg_step, arg_axis, arg_end, arg_start, node], [x], [y])
@onnx_test @onnx_test()
def slice_5arg_step_test(): def slice_5arg_step_test():
step = np.array([-2, 2]) step = np.array([-2, 2])
step_tensor = helper.make_tensor(name="step", step_tensor = helper.make_tensor(name="step",
...@@ -5627,7 +5782,7 @@ def slice_5arg_step_test(): ...@@ -5627,7 +5782,7 @@ def slice_5arg_step_test():
return ([arg_step, arg_axis, arg_end, arg_start, node], [x], [y]) return ([arg_step, arg_axis, arg_end, arg_start, node], [x], [y])
@onnx_test @onnx_test()
def slice_max_end_test(): def slice_max_end_test():
x = helper.make_tensor_value_info('0', TensorProto.FLOAT, [10, 20]) x = helper.make_tensor_value_info('0', TensorProto.FLOAT, [10, 20])
y = helper.make_tensor_value_info('1', TensorProto.FLOAT, [9, 17]) y = helper.make_tensor_value_info('1', TensorProto.FLOAT, [9, 17])
...@@ -5642,7 +5797,7 @@ def slice_max_end_test(): ...@@ -5642,7 +5797,7 @@ def slice_max_end_test():
return ([node], [x], [y]) return ([node], [x], [y])
@onnx_test @onnx_test()
def softmax_test(): def softmax_test():
x = helper.make_tensor_value_info('0', TensorProto.FLOAT, [1, 3]) x = helper.make_tensor_value_info('0', TensorProto.FLOAT, [1, 3])
y = helper.make_tensor_value_info('1', TensorProto.FLOAT, [1, 3]) y = helper.make_tensor_value_info('1', TensorProto.FLOAT, [1, 3])
...@@ -5652,7 +5807,7 @@ def softmax_test(): ...@@ -5652,7 +5807,7 @@ def softmax_test():
return ([node], [x], [y]) return ([node], [x], [y])
@onnx_test @onnx_test()
def softmax_nonstd_input_test(): def softmax_nonstd_input_test():
x = helper.make_tensor_value_info('0', TensorProto.FLOAT, [6, 8]) x = helper.make_tensor_value_info('0', TensorProto.FLOAT, [6, 8])
y = helper.make_tensor_value_info('2', TensorProto.FLOAT, [3, 4]) y = helper.make_tensor_value_info('2', TensorProto.FLOAT, [3, 4])
...@@ -5669,7 +5824,17 @@ def softmax_nonstd_input_test(): ...@@ -5669,7 +5824,17 @@ def softmax_nonstd_input_test():
return ([node0, node1], [x], [y]) return ([node0, node1], [x], [y])
@onnx_test @onnx_test()
def softmax_dyn_test():
x = helper.make_tensor_value_info('0', TensorProto.FLOAT, [None, 3, 4, 4])
y = helper.make_tensor_value_info('1', TensorProto.FLOAT, [None, 3, 4, 4])
node = onnx.helper.make_node('Softmax', inputs=['0'], outputs=['1'])
return ([node], [x], [y])
@onnx_test()
def softsign_test(): def softsign_test():
x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [5]) x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [5])
y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [5]) y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [5])
...@@ -5688,7 +5853,7 @@ def softplus_test(): ...@@ -5688,7 +5853,7 @@ def softplus_test():
return ([node], [x], [y]) return ([node], [x], [y])
@onnx_test @onnx_test()
def softsign_nd_test(): def softsign_nd_test():
x = helper.make_tensor_value_info('x', TensorProto.FLOAT16, [3, 4, 5]) x = helper.make_tensor_value_info('x', TensorProto.FLOAT16, [3, 4, 5])
y = helper.make_tensor_value_info('y', TensorProto.FLOAT16, [3, 4, 5]) y = helper.make_tensor_value_info('y', TensorProto.FLOAT16, [3, 4, 5])
...@@ -5707,7 +5872,7 @@ def softplus_nd_test(): ...@@ -5707,7 +5872,7 @@ def softplus_nd_test():
return ([node], [x], [y]) return ([node], [x], [y])
@onnx_test @onnx_test()
def split_minus_axis_test(): def split_minus_axis_test():
x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [10, 15]) x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [10, 15])
y1 = helper.make_tensor_value_info('y1', TensorProto.FLOAT, [10, 5]) y1 = helper.make_tensor_value_info('y1', TensorProto.FLOAT, [10, 5])
...@@ -5724,7 +5889,7 @@ def split_minus_axis_test(): ...@@ -5724,7 +5889,7 @@ def split_minus_axis_test():
return ([node], [x], [y1, y2, y3]) return ([node], [x], [y1, y2, y3])
@onnx_test @onnx_test()
def split_test(): def split_test():
x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [10, 15]) x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [10, 15])
y1 = helper.make_tensor_value_info('y1', TensorProto.FLOAT, [10, 7]) y1 = helper.make_tensor_value_info('y1', TensorProto.FLOAT, [10, 7])
...@@ -5740,7 +5905,7 @@ def split_test(): ...@@ -5740,7 +5905,7 @@ def split_test():
return ([node], [x], [y1, y2, y3]) return ([node], [x], [y1, y2, y3])
@onnx_test @onnx_test()
def split_test_default(): def split_test_default():
x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [10, 15]) x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [10, 15])
y1 = helper.make_tensor_value_info('y1', TensorProto.FLOAT, [5, 15]) y1 = helper.make_tensor_value_info('y1', TensorProto.FLOAT, [5, 15])
...@@ -5755,7 +5920,7 @@ def split_test_default(): ...@@ -5755,7 +5920,7 @@ def split_test_default():
return ([node], [x], [y1, y2]) return ([node], [x], [y1, y2])
@onnx_test @onnx_test()
def split_test_no_attribute(): def split_test_no_attribute():
x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [300, 15]) x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [300, 15])
y1 = helper.make_tensor_value_info('y1', TensorProto.FLOAT, [75, 15]) y1 = helper.make_tensor_value_info('y1', TensorProto.FLOAT, [75, 15])
...@@ -5782,7 +5947,7 @@ def split_test_no_attribute(): ...@@ -5782,7 +5947,7 @@ def split_test_no_attribute():
return ([const_node, node], [x], [y1, y2, y3, y4]) return ([const_node, node], [x], [y1, y2, y3, y4])
@onnx_test @onnx_test()
def split_test_no_attribute_invalid_split(): def split_test_no_attribute_invalid_split():
x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [300, 15]) x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [300, 15])
y1 = helper.make_tensor_value_info('y1', TensorProto.FLOAT, [75, 15]) y1 = helper.make_tensor_value_info('y1', TensorProto.FLOAT, [75, 15])
...@@ -5809,7 +5974,7 @@ def split_test_no_attribute_invalid_split(): ...@@ -5809,7 +5974,7 @@ def split_test_no_attribute_invalid_split():
return ([const_node, node], [x], [y1, y2, y3, y4]) return ([const_node, node], [x], [y1, y2, y3, y4])
@onnx_test @onnx_test()
def split_test_invalid_split(): def split_test_invalid_split():
x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [10, 15]) x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [10, 15])
y1 = helper.make_tensor_value_info('y1', TensorProto.FLOAT, [10, 7]) y1 = helper.make_tensor_value_info('y1', TensorProto.FLOAT, [10, 7])
...@@ -5825,7 +5990,7 @@ def split_test_invalid_split(): ...@@ -5825,7 +5990,7 @@ def split_test_invalid_split():
return ([node], [x], [y1, y2, y3]) return ([node], [x], [y1, y2, y3])
@onnx_test @onnx_test()
def split_test_no_attribute_invalid_input_split(): def split_test_no_attribute_invalid_input_split():
x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [10, 15]) x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [10, 15])
y1 = helper.make_tensor_value_info('y1', TensorProto.FLOAT, [10, 7]) y1 = helper.make_tensor_value_info('y1', TensorProto.FLOAT, [10, 7])
...@@ -5841,7 +6006,7 @@ def split_test_no_attribute_invalid_input_split(): ...@@ -5841,7 +6006,7 @@ def split_test_no_attribute_invalid_input_split():
return ([node], [x], [y1, y2, y3]) return ([node], [x], [y1, y2, y3])
@onnx_test @onnx_test()
def sqrt_test(): def sqrt_test():
x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [10, 15]) x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [10, 15])
y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [10, 15]) y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [10, 15])
...@@ -5855,7 +6020,7 @@ def sqrt_test(): ...@@ -5855,7 +6020,7 @@ def sqrt_test():
return ([node], [x], [y]) return ([node], [x], [y])
@onnx_test @onnx_test()
def squeeze_axes_input_test(): def squeeze_axes_input_test():
x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [3, 1, 5, 1]) x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [3, 1, 5, 1])
y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [3, 5]) y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [3, 5])
...@@ -5872,7 +6037,7 @@ def squeeze_axes_input_test(): ...@@ -5872,7 +6037,7 @@ def squeeze_axes_input_test():
return ([node], [x], [y], [axes_tensor]) return ([node], [x], [y], [axes_tensor])
@onnx_test @onnx_test()
def squeeze_empty_axes_test(): def squeeze_empty_axes_test():
x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [3, 1, 5, 1]) x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [3, 1, 5, 1])
y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [3, 5]) y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [3, 5])
...@@ -5889,7 +6054,7 @@ def squeeze_empty_axes_test(): ...@@ -5889,7 +6054,7 @@ def squeeze_empty_axes_test():
return ([node], [x], [y], [axes_tensor]) return ([node], [x], [y], [axes_tensor])
@onnx_test @onnx_test()
def squeeze_unsqueeze_test(): def squeeze_unsqueeze_test():
x = helper.make_tensor_value_info('0', TensorProto.FLOAT, x = helper.make_tensor_value_info('0', TensorProto.FLOAT,
[1, 3, 1, 1, 2, 1]) [1, 3, 1, 1, 2, 1])
...@@ -5909,7 +6074,27 @@ def squeeze_unsqueeze_test(): ...@@ -5909,7 +6074,27 @@ def squeeze_unsqueeze_test():
return ([node, node2], [x], [y]) return ([node, node2], [x], [y])
@onnx_test @onnx_test()
def squeeze_unsqueeze_dyn_test():
x = helper.make_tensor_value_info('0', TensorProto.FLOAT,
[1, None, 1, 1, None, 1])
y = helper.make_tensor_value_info('2', TensorProto.FLOAT,
[1, 1, None, 1, None, 1])
node = onnx.helper.make_node('Squeeze',
inputs=['0'],
axes=[0, 2, 3, 5],
outputs=['1'])
node2 = onnx.helper.make_node('Unsqueeze',
inputs=['1'],
axes=[0, 1, 3, 5],
outputs=['2'])
return ([node, node2], [x], [y])
@onnx_test()
def sub_bcast_test(): def sub_bcast_test():
arg0 = helper.make_tensor_value_info('0', TensorProto.FLOAT, [2, 3, 4, 5]) arg0 = helper.make_tensor_value_info('0', TensorProto.FLOAT, [2, 3, 4, 5])
arg1 = helper.make_tensor_value_info('1', TensorProto.FLOAT, [3, 4]) arg1 = helper.make_tensor_value_info('1', TensorProto.FLOAT, [3, 4])
...@@ -5927,7 +6112,7 @@ def sub_bcast_test(): ...@@ -5927,7 +6112,7 @@ def sub_bcast_test():
return ([node], [arg0, arg1], [arg_out]) return ([node], [arg0, arg1], [arg_out])
@onnx_test @onnx_test()
def sub_scalar_test(): def sub_scalar_test():
values = np.array([1]) values = np.array([1])
arg_node = helper.make_tensor_value_info('0', TensorProto.FLOAT, arg_node = helper.make_tensor_value_info('0', TensorProto.FLOAT,
...@@ -5956,7 +6141,7 @@ def sub_scalar_test(): ...@@ -5956,7 +6141,7 @@ def sub_scalar_test():
return ([arg_const, node], [arg_node], [arg_out]) return ([arg_const, node], [arg_node], [arg_out])
@onnx_test @onnx_test()
def sum_int_test(): def sum_int_test():
a = helper.make_tensor_value_info('0', TensorProto.INT16, [3]) a = helper.make_tensor_value_info('0', TensorProto.INT16, [3])
b = helper.make_tensor_value_info('1', TensorProto.UINT16, [3]) b = helper.make_tensor_value_info('1', TensorProto.UINT16, [3])
...@@ -5976,7 +6161,7 @@ def sum_int_test(): ...@@ -5976,7 +6161,7 @@ def sum_int_test():
return ([cnode1, cnode2, node], [a, b, c], [y]) return ([cnode1, cnode2, node], [a, b, c], [y])
@onnx_test @onnx_test()
def sum_test(): def sum_test():
a = helper.make_tensor_value_info('0', TensorProto.FLOAT, [3]) a = helper.make_tensor_value_info('0', TensorProto.FLOAT, [3])
b = helper.make_tensor_value_info('1', TensorProto.FLOAT, [3]) b = helper.make_tensor_value_info('1', TensorProto.FLOAT, [3])
...@@ -5992,7 +6177,7 @@ def sum_test(): ...@@ -5992,7 +6177,7 @@ def sum_test():
return ([node], [a, b, c], [y]) return ([node], [a, b, c], [y])
@onnx_test @onnx_test()
def sum_type_test(): def sum_type_test():
valb = np.array([1, 0]) valb = np.array([1, 0])
t_bool = helper.make_tensor(name="bool", t_bool = helper.make_tensor(name="bool",
...@@ -6086,7 +6271,7 @@ def sum_type_test(): ...@@ -6086,7 +6271,7 @@ def sum_type_test():
]) ])
@onnx_test @onnx_test()
def tan_test(): def tan_test():
x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [10]) x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [10])
y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [10]) y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [10])
...@@ -6100,7 +6285,7 @@ def tan_test(): ...@@ -6100,7 +6285,7 @@ def tan_test():
return ([node], [x], [y]) return ([node], [x], [y])
@onnx_test @onnx_test()
def tanh_test(): def tanh_test():
x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [1]) x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [1])
y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [1]) y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [1])
...@@ -6114,7 +6299,7 @@ def tanh_test(): ...@@ -6114,7 +6299,7 @@ def tanh_test():
return ([node], [x], [y]) return ([node], [x], [y])
@onnx_test @onnx_test()
def thresholdedrelu_default_test(): def thresholdedrelu_default_test():
x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [2, 2, 3]) x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [2, 2, 3])
y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [2, 2, 3]) y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [2, 2, 3])
...@@ -6126,7 +6311,7 @@ def thresholdedrelu_default_test(): ...@@ -6126,7 +6311,7 @@ def thresholdedrelu_default_test():
return ([node], [x], [y]) return ([node], [x], [y])
@onnx_test @onnx_test()
def thresholdedrelu_test(): def thresholdedrelu_test():
x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [2, 2, 3]) x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [2, 2, 3])
y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [2, 2, 3]) y = helper.make_tensor_value_info('y', TensorProto.FLOAT, [2, 2, 3])
...@@ -6140,7 +6325,7 @@ def thresholdedrelu_test(): ...@@ -6140,7 +6325,7 @@ def thresholdedrelu_test():
return ([node], [x], [y]) return ([node], [x], [y])
@onnx_test @onnx_test()
def thresholdedrelu_int_test(): def thresholdedrelu_int_test():
x = helper.make_tensor_value_info('x', TensorProto.INT32, [2, 2, 3]) x = helper.make_tensor_value_info('x', TensorProto.INT32, [2, 2, 3])
y = helper.make_tensor_value_info('y', TensorProto.INT32, [2, 2, 3]) y = helper.make_tensor_value_info('y', TensorProto.INT32, [2, 2, 3])
...@@ -6154,7 +6339,7 @@ def thresholdedrelu_int_test(): ...@@ -6154,7 +6339,7 @@ def thresholdedrelu_int_test():
return ([node], [x], [y]) return ([node], [x], [y])
@onnx_test @onnx_test()
def tile_test(): def tile_test():
x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [2, 2]) x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [2, 2])
y = helper.make_tensor_value_info('y', TensorProto.INT64, [2]) y = helper.make_tensor_value_info('y', TensorProto.INT64, [2])
...@@ -6166,7 +6351,7 @@ def tile_test(): ...@@ -6166,7 +6351,7 @@ def tile_test():
[helper.make_tensor('y', TensorProto.INT64, [2], [1, 2])]) [helper.make_tensor('y', TensorProto.INT64, [2], [1, 2])])
@onnx_test @onnx_test()
def tile_test_3x2(): def tile_test_3x2():
x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [2, 2]) x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [2, 2])
y = helper.make_tensor_value_info('y', TensorProto.INT64, [2]) y = helper.make_tensor_value_info('y', TensorProto.INT64, [2])
...@@ -6178,7 +6363,7 @@ def tile_test_3x2(): ...@@ -6178,7 +6363,7 @@ def tile_test_3x2():
[helper.make_tensor('y', TensorProto.INT64, [2], [3, 2])]) [helper.make_tensor('y', TensorProto.INT64, [2], [3, 2])])
@onnx_test @onnx_test()
def topk_attrk_test(): def topk_attrk_test():
x = helper.make_tensor_value_info('data', TensorProto.FLOAT, [2, 5, 3, 2]) x = helper.make_tensor_value_info('data', TensorProto.FLOAT, [2, 5, 3, 2])
val = helper.make_tensor_value_info('val', TensorProto.FLOAT, [2, 2, 3, 2]) val = helper.make_tensor_value_info('val', TensorProto.FLOAT, [2, 2, 3, 2])
...@@ -6192,7 +6377,7 @@ def topk_attrk_test(): ...@@ -6192,7 +6377,7 @@ def topk_attrk_test():
return ([node], [x], [val, ind]) return ([node], [x], [val, ind])
@onnx_test @onnx_test()
def topk_neg_axis_test(): def topk_neg_axis_test():
k = np.array([3]) k = np.array([3])
x = helper.make_tensor_value_info('data', TensorProto.FLOAT, [3, 4, 5, 6]) x = helper.make_tensor_value_info('data', TensorProto.FLOAT, [3, 4, 5, 6])
...@@ -6213,7 +6398,7 @@ def topk_neg_axis_test(): ...@@ -6213,7 +6398,7 @@ def topk_neg_axis_test():
return ([node], [x], [val, ind], [k_tensor]) return ([node], [x], [val, ind], [k_tensor])
@onnx_test @onnx_test()
def topk_test(): def topk_test():
k = np.array([4]) k = np.array([4])
x = helper.make_tensor_value_info('data', TensorProto.FLOAT, [2, 5, 3, 2]) x = helper.make_tensor_value_info('data', TensorProto.FLOAT, [2, 5, 3, 2])
...@@ -6247,7 +6432,7 @@ def transpose_default_perm_test(): ...@@ -6247,7 +6432,7 @@ def transpose_default_perm_test():
return ([node], [x], [y]) return ([node], [x], [y])
@onnx_test @onnx_test()
def transpose_invalid_perm_test(): def transpose_invalid_perm_test():
x = helper.make_tensor_value_info('0', TensorProto.FLOAT, [1, 2, 4, 3]) x = helper.make_tensor_value_info('0', TensorProto.FLOAT, [1, 2, 4, 3])
y = helper.make_tensor_value_info('1', TensorProto.FLOAT, [1, 3, 2, 2]) y = helper.make_tensor_value_info('1', TensorProto.FLOAT, [1, 3, 2, 2])
...@@ -6262,7 +6447,7 @@ def transpose_invalid_perm_test(): ...@@ -6262,7 +6447,7 @@ def transpose_invalid_perm_test():
return ([node], [x], [y]) return ([node], [x], [y])
@onnx_test @onnx_test()
def transpose_test(): def transpose_test():
x = helper.make_tensor_value_info('0', TensorProto.FLOAT, [1, 2, 2, 3]) x = helper.make_tensor_value_info('0', TensorProto.FLOAT, [1, 2, 2, 3])
y = helper.make_tensor_value_info('1', TensorProto.FLOAT, [1, 3, 2, 2]) y = helper.make_tensor_value_info('1', TensorProto.FLOAT, [1, 3, 2, 2])
...@@ -6277,6 +6462,21 @@ def transpose_test(): ...@@ -6277,6 +6462,21 @@ def transpose_test():
return ([node], [x], [y]) return ([node], [x], [y])
@onnx_test()
def transpose_dyn_test():
x = helper.make_tensor_value_info('0', TensorProto.FLOAT, [None, 2, 2, 3])
y = helper.make_tensor_value_info('1', TensorProto.FLOAT, [None, 3, 2, 2])
node = onnx.helper.make_node(
'Transpose',
perm=[0, 3, 1, 2],
inputs=['0'],
outputs=['1'],
)
return ([node], [x], [y])
@onnx_test @onnx_test
def transpose_gather_test(): def transpose_gather_test():
x = helper.make_tensor_value_info('data', TensorProto.FLOAT, [3, 5, 4, 6]) x = helper.make_tensor_value_info('data', TensorProto.FLOAT, [3, 5, 4, 6])
...@@ -6307,7 +6507,7 @@ def transpose_gather_test(): ...@@ -6307,7 +6507,7 @@ def transpose_gather_test():
return ([td, ti, node], [x, i], [y]) return ([td, ti, node], [x, i], [y])
@onnx_test @onnx_test()
def undefined_test(): def undefined_test():
x = helper.make_tensor_value_info('0', TensorProto.FLOAT, [2, 3, 4, 5]) x = helper.make_tensor_value_info('0', TensorProto.FLOAT, [2, 3, 4, 5])
y = helper.make_tensor_value_info('1', TensorProto.FLOAT, [2, 3, 4, 5]) y = helper.make_tensor_value_info('1', TensorProto.FLOAT, [2, 3, 4, 5])
...@@ -6317,7 +6517,7 @@ def undefined_test(): ...@@ -6317,7 +6517,7 @@ def undefined_test():
return ([node], [x], [y]) return ([node], [x], [y])
@onnx_test @onnx_test()
def unknown_test(): def unknown_test():
x = helper.make_tensor_value_info('0', TensorProto.FLOAT, [2, 3, 4, 5]) x = helper.make_tensor_value_info('0', TensorProto.FLOAT, [2, 3, 4, 5])
y = helper.make_tensor_value_info('1', TensorProto.FLOAT, [3, 4]) y = helper.make_tensor_value_info('1', TensorProto.FLOAT, [3, 4])
...@@ -6333,7 +6533,7 @@ def unknown_test(): ...@@ -6333,7 +6533,7 @@ def unknown_test():
return ([node, node2], [x, y], [a]) return ([node, node2], [x, y], [a])
@onnx_test @onnx_test()
def unknown_aten_test(): def unknown_aten_test():
x = helper.make_tensor_value_info('0', TensorProto.FLOAT, [2, 3, 4, 5]) x = helper.make_tensor_value_info('0', TensorProto.FLOAT, [2, 3, 4, 5])
y = helper.make_tensor_value_info('1', TensorProto.FLOAT, [3, 4]) y = helper.make_tensor_value_info('1', TensorProto.FLOAT, [3, 4])
...@@ -6350,7 +6550,7 @@ def unknown_aten_test(): ...@@ -6350,7 +6550,7 @@ def unknown_aten_test():
return ([node], [x, y], [a]) return ([node], [x, y], [a])
@onnx_test @onnx_test()
def upsample_linear_test(): def upsample_linear_test():
scales = np.array([1.0, 1.0, 2.0, 2.0], dtype=np.float32) scales = np.array([1.0, 1.0, 2.0, 2.0], dtype=np.float32)
scales_tensor = helper.make_tensor(name='scales', scales_tensor = helper.make_tensor(name='scales',
...@@ -6369,7 +6569,7 @@ def upsample_linear_test(): ...@@ -6369,7 +6569,7 @@ def upsample_linear_test():
return ([node], [X], [Y], [scales_tensor]) return ([node], [X], [Y], [scales_tensor])
@onnx_test @onnx_test()
def upsample_test(): def upsample_test():
scales = np.array([1.0, 1.0, 2.0, 3.0], dtype=np.float32) scales = np.array([1.0, 1.0, 2.0, 3.0], dtype=np.float32)
scale_tensor = helper.make_tensor(name='scales', scale_tensor = helper.make_tensor(name='scales',
...@@ -6390,7 +6590,7 @@ def upsample_test(): ...@@ -6390,7 +6590,7 @@ def upsample_test():
return ([node], [X], [Y], [scale_tensor]) return ([node], [X], [Y], [scale_tensor])
@onnx_test @onnx_test()
def variable_batch_test(): def variable_batch_test():
x = helper.make_tensor_value_info('0', TensorProto.FLOAT, x = helper.make_tensor_value_info('0', TensorProto.FLOAT,
[None, 3, 16, 16]) [None, 3, 16, 16])
...@@ -6402,7 +6602,7 @@ def variable_batch_test(): ...@@ -6402,7 +6602,7 @@ def variable_batch_test():
return ([node], [x], [y]) return ([node], [x], [y])
@onnx_test @onnx_test()
def variable_batch_leq_zero_test(): def variable_batch_leq_zero_test():
x = helper.make_tensor_value_info('0', TensorProto.FLOAT, [0, 3, 16, 16]) x = helper.make_tensor_value_info('0', TensorProto.FLOAT, [0, 3, 16, 16])
y = helper.make_tensor_value_info('1', TensorProto.FLOAT, [-1, 3, 16, 16]) y = helper.make_tensor_value_info('1', TensorProto.FLOAT, [-1, 3, 16, 16])
...@@ -6413,7 +6613,7 @@ def variable_batch_leq_zero_test(): ...@@ -6413,7 +6613,7 @@ def variable_batch_leq_zero_test():
return ([node], [x, y], [z]) return ([node], [x, y], [z])
@onnx_test @onnx_test()
def where_test(): def where_test():
c = helper.make_tensor_value_info('c', TensorProto.BOOL, [2]) c = helper.make_tensor_value_info('c', TensorProto.BOOL, [2])
x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [2, 2, 2]) x = helper.make_tensor_value_info('x', TensorProto.FLOAT, [2, 2, 2])
......
test/onnx/onnx_test.cpp
View file @ 913ae362
...@@ -181,6 +181,24 @@ TEST_CASE(argmax_test) ...@@ -181,6 +181,24 @@ TEST_CASE(argmax_test)
EXPECT(p == prog); EXPECT(p == prog);
} }
TEST_CASE(argmax_dyn_test)
{
migraphx::program p;
auto* mm = p.get_main_module();
auto l0 = mm->add_parameter(
"x",
migraphx::shape{migraphx::shape::float_type, {{1, 4, 0}, {4, 4, 0}, {5, 5, 0}, {6, 6, 0}}});
auto ins = mm->add_instruction(migraphx::make_op("argmax", {{"axis", 2}}), l0);
auto ret = mm->add_instruction(migraphx::make_op("squeeze", {{"axes", {2}}}), ins);
mm->add_return({ret});
migraphx::onnx_options options;
options.default_dyn_dim_value = {1, 4, 0};
auto prog = parse_onnx("argmax_dyn_test.onnx", options);
EXPECT(p == prog);
}
TEST_CASE(argmin_test) TEST_CASE(argmin_test)
{ {
migraphx::program p; migraphx::program p;
...@@ -273,6 +291,51 @@ TEST_CASE(averagepool_3d_test) ...@@ -273,6 +291,51 @@ TEST_CASE(averagepool_3d_test)
EXPECT(p == prog); EXPECT(p == prog);
} }
TEST_CASE(averagepool_dyn_test)
{
migraphx::program p;
auto* mm = p.get_main_module();
auto l0 = mm->add_parameter(
"0",
{migraphx::shape::float_type, {{1, 4, 0}, {3, 3, 0}, {5, 5, 0}, {5, 5, 0}, {5, 5, 0}}});
auto ret = mm->add_instruction(migraphx::make_op("pooling",
{{"mode", migraphx::op::pooling_mode::average},
{"padding", {0, 0, 0, 0, 0, 0}},
{"stride", {1, 1, 1}},
{"lengths", {3, 3, 3}}}),
l0);
mm->add_return({ret});
migraphx::onnx_options options;
options.default_dyn_dim_value = {1, 4, 0};
auto prog = migraphx::parse_onnx("averagepool_dyn_test.onnx", options);
EXPECT(p == prog);
}
TEST_CASE(averagepool_dyn_autopad_error_test)
{
migraphx::onnx_options options;
options.default_dyn_dim_value = {1, 4, 0};
EXPECT(test::throws(
[&] { migraphx::parse_onnx("averagepool_dyn_autopad_error_test.onnx", options); }));
}
TEST_CASE(averagepool_dyn_asym_padding_error_test)
{
migraphx::onnx_options options;
options.default_dyn_dim_value = {1, 4, 0};
EXPECT(test::throws(
[&] { migraphx::parse_onnx("averagepool_dyn_asym_padding_error_test.onnx", options); }));
}
TEST_CASE(averagepool_dyn_cip_error_test)
{
migraphx::onnx_options options;
options.default_dyn_dim_value = {1, 4, 0};
EXPECT(test::throws(
[&] { migraphx::parse_onnx("averagepool_dyn_cip_error_test.onnx", options); }));
}
TEST_CASE(averagepool_notset_test) TEST_CASE(averagepool_notset_test)
{ {
migraphx::program p; migraphx::program p;
...@@ -1755,6 +1818,16 @@ migraphx::program create_external_data_prog() ...@@ -1755,6 +1818,16 @@ migraphx::program create_external_data_prog()
return p; return p;
} }
TEST_CASE(external_constant_test)
{
migraphx::program p;
auto* mm = p.get_main_module();
mm->add_literal(migraphx::literal{{migraphx::shape::int64_type, {3}}, {0, 1, 2}});
auto prog = optimize_onnx("external_constant_test.onnx");
EXPECT(p == prog);
}
TEST_CASE(external_data_test) TEST_CASE(external_data_test)
{ {
migraphx::program p = create_external_data_prog(); migraphx::program p = create_external_data_prog();
...@@ -1914,6 +1987,23 @@ TEST_CASE(flatten_nonstd_test) ...@@ -1914,6 +1987,23 @@ TEST_CASE(flatten_nonstd_test)
EXPECT(p == prog); EXPECT(p == prog);
} }
TEST_CASE(flatten_dyn_test)
{
migraphx::program p;
auto* mm = p.get_main_module();
auto l0 = mm->add_parameter(
"0",
migraphx::shape{migraphx::shape::float_type, {{1, 4, 0}, {3, 3, 0}, {4, 4, 0}, {5, 5, 0}}});
auto c0 = mm->add_instruction(migraphx::make_op("contiguous"), l0);
auto ret = mm->add_instruction(migraphx::make_op("flatten", {{"axis", 2}}), c0);
mm->add_return({ret});
migraphx::onnx_options options;
options.default_dyn_dim_value = {1, 4, 0};
auto prog = parse_onnx("flatten_dyn_test.onnx", options);
EXPECT(p == prog);
}
TEST_CASE(floor_test) TEST_CASE(floor_test)
{ {
migraphx::program p; migraphx::program p;
...@@ -2144,6 +2234,28 @@ TEST_CASE(globalavgpool_test) ...@@ -2144,6 +2234,28 @@ TEST_CASE(globalavgpool_test)
EXPECT(p == prog); EXPECT(p == prog);
} }
TEST_CASE(globalavgpool_dyn_test)
{
migraphx::program p;
auto* mm = p.get_main_module();
auto input =
mm->add_parameter("0",
migraphx::shape{migraphx::shape::float_type,
{{1, 4, 0}, {3, 3, 0}, {16, 16, 0}, {16, 16, 0}}});
auto ret = mm->add_instruction(migraphx::make_op("pooling",
{{"mode", migraphx::op::pooling_mode::average},
{"lengths", {16, 16}},
{"padding", {0, 0, 0, 0}}}),
input);
mm->add_return({ret});
migraphx::onnx_options options;
options.default_dyn_dim_value = {1, 4, 0};
auto prog = parse_onnx("globalavgpool_dyn_test.onnx", options);
EXPECT(p == prog);
}
TEST_CASE(globallppool_test) TEST_CASE(globallppool_test)
{ {
migraphx::program p; migraphx::program p;
...@@ -2161,6 +2273,29 @@ TEST_CASE(globallppool_test) ...@@ -2161,6 +2273,29 @@ TEST_CASE(globallppool_test)
EXPECT(p == prog); EXPECT(p == prog);
} }
TEST_CASE(globallppool_dyn_test)
{
migraphx::program p;
auto* mm = p.get_main_module();
auto input =
mm->add_parameter("0",
migraphx::shape{migraphx::shape::float_type,
{{1, 1, 0}, {3, 3, 0}, {16, 32, 0}, {16, 32, 0}}});
auto ret = mm->add_instruction(migraphx::make_op("pooling",
{{"mode", migraphx::op::pooling_mode::lpnorm},
{"dyn_global", true},
{"padding", {0, 0, 0, 0}},
{"lengths", {}}}),
input);
mm->add_return({ret});
migraphx::onnx_options options;
options.default_dyn_dim_value = {16, 32, 0};
auto prog = migraphx::parse_onnx("globallppool_dyn_test.onnx", options);
EXPECT(p == prog);
}
TEST_CASE(globalmaxpool_test) TEST_CASE(globalmaxpool_test)
{ {
migraphx::program p; migraphx::program p;
...@@ -2178,6 +2313,28 @@ TEST_CASE(globalmaxpool_test) ...@@ -2178,6 +2313,28 @@ TEST_CASE(globalmaxpool_test)
EXPECT(p == prog); EXPECT(p == prog);
} }
TEST_CASE(globalmaxpool_dyn_test)
{
migraphx::program p;
auto* mm = p.get_main_module();
auto input =
mm->add_parameter("0",
migraphx::shape{migraphx::shape::float_type,
{{1, 4, 0}, {3, 3, 0}, {32, 32, 0}, {32, 32, 0}}});
auto ret = mm->add_instruction(migraphx::make_op("pooling",
{{"mode", migraphx::op::pooling_mode::max},
{"lengths", {32, 32}},
{"padding", {0, 0, 0, 0}}}),
input);
mm->add_return({ret});
migraphx::onnx_options options;
options.default_dyn_dim_value = {1, 4, 0};
auto prog = parse_onnx("globalmaxpool_dyn_test.onnx", options);
EXPECT(p == prog);
}
TEST_CASE(greater_test) TEST_CASE(greater_test)
{ {
migraphx::program p; migraphx::program p;
...@@ -5492,6 +5649,23 @@ TEST_CASE(softmax_nonstd_input_test) ...@@ -5492,6 +5649,23 @@ TEST_CASE(softmax_nonstd_input_test)
EXPECT(p == prog); EXPECT(p == prog);
} }
TEST_CASE(softmax_dyn_test)
{
migraphx::program p;
auto* mm = p.get_main_module();
auto l0 = mm->add_parameter(
"0",
migraphx::shape{migraphx::shape::float_type, {{1, 4, 0}, {3, 3, 0}, {4, 4, 0}, {4, 4, 0}}});
auto ret = mm->add_instruction(migraphx::make_op("softmax", {{"axis", -1}}), l0);
mm->add_return({ret});
migraphx::onnx_options options;
options.default_dyn_dim_value = {1, 4, 0};
auto prog = migraphx::parse_onnx("softmax_dyn_test.onnx", options);
EXPECT(p == prog);
}
TEST_CASE(softplus_test) TEST_CASE(softplus_test)
{ {
migraphx::program p; migraphx::program p;
...@@ -5690,6 +5864,29 @@ TEST_CASE(squeeze_unsqueeze_test) ...@@ -5690,6 +5864,29 @@ TEST_CASE(squeeze_unsqueeze_test)
EXPECT(p == prog); EXPECT(p == prog);
} }
TEST_CASE(squeeze_unsqueeze_dyn_test)
{
migraphx::program p;
auto* mm = p.get_main_module();
std::vector<int64_t> squeeze_axes{0, 2, 3, 5};
std::vector<int64_t> unsqueeze_axes{0, 1, 3, 5};
auto l0 = mm->add_parameter(
"0",
migraphx::shape{migraphx::shape::float_type,
{{1, 1, 0}, {1, 4, 0}, {1, 1, 0}, {1, 1, 0}, {1, 4, 0}, {1, 1, 0}}});
auto c0 = mm->add_instruction(migraphx::make_op("contiguous"), l0);
auto l1 = mm->add_instruction(migraphx::make_op("squeeze", {{"axes", squeeze_axes}}), c0);
auto c1 = mm->add_instruction(migraphx::make_op("contiguous"), l1);
auto ret = mm->add_instruction(migraphx::make_op("unsqueeze", {{"axes", unsqueeze_axes}}), c1);
mm->add_return({ret});
migraphx::onnx_options options;
options.default_dyn_dim_value = {1, 4, 0};
auto prog = parse_onnx("squeeze_unsqueeze_dyn_test.onnx", options);
EXPECT(p == prog);
}
TEST_CASE(squeeze_axes_input_test) TEST_CASE(squeeze_axes_input_test)
{ {
migraphx::program p; migraphx::program p;
...@@ -5973,6 +6170,24 @@ TEST_CASE(transpose_test) ...@@ -5973,6 +6170,24 @@ TEST_CASE(transpose_test)
EXPECT(p == prog); EXPECT(p == prog);
} }
TEST_CASE(transpose_dyn_test)
{
migraphx::program p;
auto* mm = p.get_main_module();
auto input = mm->add_parameter(
"0",
migraphx::shape{migraphx::shape::float_type, {{1, 4, 0}, {2, 2, 0}, {2, 2, 0}, {3, 3, 0}}});
std::vector<int64_t> perm{0, 3, 1, 2};
auto t0 = mm->add_instruction(migraphx::make_op("transpose", {{"permutation", perm}}), input);
mm->add_return({t0});
migraphx::onnx_options options;
options.default_dyn_dim_value = {1, 4, 0};
auto prog = migraphx::parse_onnx("transpose_dyn_test.onnx", options);
EXPECT(p == prog);
}
TEST_CASE(topk_attrk_test) TEST_CASE(topk_attrk_test)
{ {
migraphx::program p; migraphx::program p;
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment