Commit c6c7bf33 authored by Shucai Xiao's avatar Shucai Xiao
Browse files

clang format

parent 0da78d29
...@@ -1264,8 +1264,8 @@ struct lstm ...@@ -1264,8 +1264,8 @@ struct lstm
std::size_t hidden_size = 1; std::size_t hidden_size = 1;
std::vector<operation> actv_funcs{sigmoid{}, tanh{}, tanh{}}; std::vector<operation> actv_funcs{sigmoid{}, tanh{}, tanh{}};
rnn_direction direction = rnn_direction::forward; rnn_direction direction = rnn_direction::forward;
float clip = 0.0f; float clip = 0.0f;
int input_forget = 0; int input_forget = 0;
std::string name() const { return "lstm"; } std::string name() const { return "lstm"; }
shape compute_shape(std::vector<shape> inputs) const shape compute_shape(std::vector<shape> inputs) const
......
...@@ -683,8 +683,8 @@ void rewrite_rnn::apply_lstm(program& prog, instruction_ref ins) const ...@@ -683,8 +683,8 @@ void rewrite_rnn::apply_lstm(program& prog, instruction_ref ins) const
migraphx::shape pph_shape{type, {1, 3 * hidden_size}}; migraphx::shape pph_shape{type, {1, 3 * hidden_size}};
std::vector<float> pph_data(pph_shape.elements(), 0.0); std::vector<float> pph_data(pph_shape.elements(), 0.0);
auto actv_funcs = lstm_actv_funcs(ins); auto actv_funcs = lstm_actv_funcs(ins);
auto lstm_op = any_cast<op::lstm>(ins->get_operator()); auto lstm_op = any_cast<op::lstm>(ins->get_operator());
op::rnn_direction dirct = lstm_op.direction; op::rnn_direction dirct = lstm_op.direction;
instruction_ref last_output{}; instruction_ref last_output{};
......
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