Commit 5cce443f authored by wsttiger's avatar wsttiger
Browse files

Formatting

parent c527d0b5
......@@ -12,8 +12,9 @@ shape miopen_convolution::compute_shape(const std::vector<shape>& inputs) const
check_shapes{inputs, *this}.has(4).standard();
return op.compute_shape({inputs.at(0), inputs.at(1)});
}
argument
miopen_convolution::compute(context& ctx, const shape& output_shape, const std::vector<argument>& args) const
argument miopen_convolution::compute(context& ctx,
const shape& output_shape,
const std::vector<argument>& args) const
{
auto x_desc = make_tensor(args[0].get_shape());
auto w_desc = make_tensor(args[1].get_shape());
......@@ -36,7 +37,9 @@ miopen_convolution::compute(context& ctx, const shape& output_shape, const std::
return args[3];
}
shape miopen_convolution::compile(context& ctx, const shape& output_shape, std::vector<instruction_ref> inputs)
shape miopen_convolution::compile(context& ctx,
const shape& output_shape,
std::vector<instruction_ref> inputs)
{
shape workspace_shape{};
auto x_desc = make_tensor(inputs[0]->get_shape());
......@@ -76,4 +79,3 @@ shape miopen_convolution::compile(context& ctx, const shape& output_shape, std::
} // namespace gpu
} // namespace migraph
......@@ -12,8 +12,9 @@ shape miopen_gemm::compute_shape(const std::vector<shape>& inputs) const
check_shapes{inputs, *this}.has(3);
return op.compute_shape({inputs.at(0), inputs.at(1)});
}
argument
miopen_gemm::compute(context& ctx, const shape& output_shape, const std::vector<argument>& args) const
argument miopen_gemm::compute(context& ctx,
const shape& output_shape,
const std::vector<argument>& args) const
{
float alpha = 1.0f;
float beta = 0.0f;
......@@ -45,4 +46,3 @@ miopen_gemm::compute(context& ctx, const shape& output_shape, const std::vector<
} // namespace gpu
} // namespace migraph
......@@ -20,7 +20,6 @@
namespace migraph {
namespace gpu {
struct miopen_pooling
{
pooling op;
......
......@@ -12,8 +12,9 @@ shape miopen_pooling::compute_shape(const std::vector<shape>& inputs) const
check_shapes{inputs, *this}.has(2).standard();
return op.compute_shape({inputs.at(0)});
}
argument
miopen_pooling::compute(context& ctx, const shape& output_shape, const std::vector<argument>& args) const
argument miopen_pooling::compute(context& ctx,
const shape& output_shape,
const std::vector<argument>& args) const
{
auto x_desc = make_tensor(args[0].get_shape());
auto y_desc = make_tensor(output_shape);
......@@ -38,4 +39,3 @@ miopen_pooling::compute(context& ctx, const shape& output_shape, const std::vect
} // namespace gpu
} // namespace migraph
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