Function TEST_CASE

test/op/builder/batchnorm_test.cpp:27–40

#include <op_builder_test_utils.hpp>

TEST_CASE(batchnorm_rank_0_op_builder_test)
{
    migraphx::module mm;

    mm.add_parameter("x", {migraphx::shape::half_type, {}});
    mm.add_parameter("scale", {migraphx::shape::float_type, {3}});
    mm.add_parameter("bias", {migraphx::shape::float_type, {3}});
    mm.add_parameter("mean", {migraphx::shape::float_type, {3}});
    mm.add_parameter("variance", {migraphx::shape::float_type, {3}});

    EXPECT(test::throws<migraphx::exception>(
        [&] { make_op_module("batchnorm", {}, mm.get_parameters()); },
        "rank 0 input tensor, unhandled data format"));
}

TEST_CASE(batchnorm_rank_1_op_builder_test)
{