"examples/vision/python_super_resolution/README.md" did not exist on "78eaf2b80d39277a59ff600573949740439259d3"
Unverified Commit f8b56a66 authored by kahmed10's avatar kahmed10 Committed by GitHub
Browse files

Extra tf ops (#679)

* initial testing

* add new ops

* remove comment

* remove extra changes

* remove extra changes

* add tests

* formatting

* add tests
parent 2466dd6f
......@@ -5,4 +5,4 @@
shape:
(
stopgradient StopGradient0*
T0"
\ No newline at end of file
T0"¸
\ No newline at end of file
:
0 Placeholder*
shape:*
dtype0
0 Placeholder*
dtype0*
shape:
:
1 Placeholder*
shape:*
dtype0
1 Placeholder*
dtype0*
shape:

sub1Sub01*
T0"
\ No newline at end of file
T0"
\ No newline at end of file
......@@ -5,4 +5,4 @@
shape:

tanhTanh0*
T0"
\ No newline at end of file
T0"
\ No newline at end of file
......@@ -41,6 +41,17 @@ TEST_CASE(add_test)
EXPECT(p == prog);
}
TEST_CASE(addv2_test)
{
migraphx::program p;
auto l0 = p.add_parameter("0", migraphx::shape{migraphx::shape::float_type, {1, 2, 2, 3}});
auto l1 = p.add_parameter("1", migraphx::shape{migraphx::shape::float_type, {1, 2, 2, 3}});
p.add_instruction(migraphx::op::add{}, l0, l1);
auto prog = optimize_tf("addv2_test.pb", false);
EXPECT(p == prog);
}
TEST_CASE(add_bcast_test)
{
......@@ -146,6 +157,29 @@ TEST_CASE(batchnorm_test)
EXPECT(p == prog);
}
TEST_CASE(batchnormv3_test)
{
float epsilon = 1.0e-5f;
float momentum = 0.9f;
migraphx::program p;
migraphx::op::batch_norm_inference op{
epsilon, momentum, migraphx::op::batch_norm_inference::spatial};
migraphx::shape s0{migraphx::shape::float_type, {32}};
auto l0 = p.add_parameter("0", migraphx::shape{migraphx::shape::float_type, {1, 32, 16, 16}});
std::vector<float> const_vals(32);
std::fill(const_vals.begin(), const_vals.end(), 1.0f);
auto l2 = p.add_parameter("2", s0);
auto l3 = p.add_parameter("3", s0);
auto l4 = p.add_parameter("4", s0);
auto l1 = p.add_literal(migraphx::literal{s0, const_vals});
p.add_instruction(op, l0, l1, l2, l3, l4);
auto prog = optimize_tf("batchnormv3_test.pb", true);
EXPECT(p == prog);
}
TEST_CASE(biasadd_test)
{
migraphx::program p;
......@@ -399,6 +433,14 @@ TEST_CASE(onehot_test)
EXPECT(p == prog);
}
TEST_CASE(noop_test)
{
migraphx::program p;
auto prog = optimize_tf("noop_test.pb", false);
EXPECT(p == prog);
}
TEST_CASE(pack_test)
{
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