Function TEST_CASE

test/op/builder/celu_test.cpp:27–56

#include <op_builder_test_utils.hpp>

TEST_CASE(celu_happy_path_op_builder_test)
{
    migraphx::module mm;

    const float alpha       = 0.8;
    const migraphx::shape s = {migraphx::shape::float_type, {3}};

    auto x = mm.add_parameter("x", s);

    const auto& input_lens = s.lens();
    const auto& input_type = s.type();
    auto zero_lit =
        mm.add_instruction(migraphx::make_op("multibroadcast", {{"out_lens", input_lens}}),
                           mm.add_literal(migraphx::literal{migraphx::shape{input_type}, {0.}}));
    auto one_lit =
        mm.add_instruction(migraphx::make_op("multibroadcast", {{"out_lens", input_lens}}),
                           mm.add_literal(migraphx::literal{migraphx::shape{input_type}, {1.}}));
    auto alpha_lit =
        mm.add_instruction(migraphx::make_op("multibroadcast", {{"out_lens", input_lens}}),
                           mm.add_literal(migraphx::literal{migraphx::shape{input_type}, {alpha}}));
    auto linear_part = mm.add_instruction(migraphx::make_op("max"), zero_lit, x);
    auto divi        = mm.add_instruction(migraphx::make_op("div"), x, alpha_lit);
    auto expo        = mm.add_instruction(migraphx::make_op("exp"), divi);
    auto sub         = mm.add_instruction(migraphx::make_op("sub"), expo, one_lit);
    auto mul         = mm.add_instruction(migraphx::make_op("mul"), alpha_lit, sub);
    auto exp_part    = mm.add_instruction(migraphx::make_op("min"), zero_lit, mul);
    mm.add_instruction(migraphx::make_op("add"), linear_part, exp_part);

    EXPECT(mm == make_op_module("celu", {{"alpha", alpha}}, mm.get_parameters()));
}

TEST_CASE(celu_zero_alpha_op_builder_test)
{