Commit 311ae1d1 authored by Paul's avatar Paul
Browse files

Formatting

parent 87aed719
...@@ -322,12 +322,12 @@ struct miopen_softmax ...@@ -322,12 +322,12 @@ struct miopen_softmax
auto x_desc = make_tensor(args[0].get_shape()); auto x_desc = make_tensor(args[0].get_shape());
auto y_desc = make_tensor(output_shape); auto y_desc = make_tensor(output_shape);
miopenSoftmaxForward(ctx.handle.get(), miopenSoftmaxForward(ctx.handle.get(),
&alpha, &alpha,
x_desc.get(), x_desc.get(),
args[0].implicit(), args[0].implicit(),
&beta, &beta,
y_desc.get(), y_desc.get(),
args[1].implicit()); args[1].implicit());
return args[1]; return args[1];
} }
...@@ -439,8 +439,7 @@ struct miopen_apply ...@@ -439,8 +439,7 @@ struct miopen_apply
instruction_ref apply_softmax(instruction_ref ins) instruction_ref apply_softmax(instruction_ref ins)
{ {
auto output = insert_allocation(ins, ins->get_shape()); auto output = insert_allocation(ins, ins->get_shape());
return prog->replace_instruction( return prog->replace_instruction(ins, miopen_softmax{}, ins->inputs().at(0), output);
ins, miopen_softmax{}, ins->inputs().at(0), output);
} }
instruction_ref apply_add(instruction_ref ins) instruction_ref apply_add(instruction_ref ins)
......
...@@ -240,7 +240,7 @@ struct test_softmax ...@@ -240,7 +240,7 @@ struct test_softmax
migraph::program create_program() const migraph::program create_program() const
{ {
migraph::program p; migraph::program p;
auto x = p.add_parameter("x", migraph::shape{migraph::shape::float_type, {5, 3, 4, 2}}); auto x = p.add_parameter("x", migraph::shape{migraph::shape::float_type, {5, 3, 4, 2}});
p.add_instruction(migraph::softmax{}, x); p.add_instruction(migraph::softmax{}, x);
return p; return 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