Skip to content

Commit

Permalink
remove unused parameter in GRU
Browse files Browse the repository at this point in the history
  • Loading branch information
chunseoklee committed Oct 25, 2024
1 parent 53e40da commit 223b60d
Show file tree
Hide file tree
Showing 2 changed files with 4 additions and 10 deletions.
9 changes: 3 additions & 6 deletions compiler/luci-interpreter/pal/linux/PALGRU.h
Original file line number Diff line number Diff line change
Expand Up @@ -62,8 +62,7 @@ void calculateGRU(const float *input_data, const float *weight_input_data,
const tflite::RuntimeShape &input_shape, const tflite::RuntimeShape &output_shape,
const tflite::RuntimeShape &weight_input_shape,
const tflite::RuntimeShape &weight_hidden_shape, float *output_input_data,
float *output_hidden_data, const tflite::RuntimeShape &output_shape_fc,
float *intermediate_buffer)
float *output_hidden_data, const tflite::RuntimeShape &output_shape_fc)
{
tflite::FullyConnectedParams op_params{};
// As FC nodes doesn't have any activations inside GRU, let' use just numeric limits
Expand Down Expand Up @@ -150,8 +149,7 @@ void GRU(const float *input_data, const float *weight_input_data, const float *w
const float *hidden_state_data, float *output_data, float *output_input_data,
float *output_hidden_data, const tflite::RuntimeShape &input_shape,
const tflite::RuntimeShape &output_shape, const tflite::RuntimeShape &weight_input_shape,
const tflite::RuntimeShape &weight_hidden_shape, const size_t intermediate_buffer_size,
float *intermediate_buffer)
const tflite::RuntimeShape &weight_hidden_shape)
{
const int32_t time = input_shape.Dims(0);

Expand All @@ -165,8 +163,7 @@ void GRU(const float *input_data, const float *weight_input_data, const float *w
{
calculateGRU(input_data, weight_input_data, weight_hidden_data, bias_input_data,
bias_hidden_data, output_data, input_shape, output_shape, weight_input_shape,
weight_hidden_shape, output_input_data, output_hidden_data, output_shape_fc,
intermediate_buffer);
weight_hidden_shape, output_input_data, output_hidden_data, output_shape_fc);
input_data += input_shape.Dims(2);
}
}
Expand Down
5 changes: 1 addition & 4 deletions compiler/luci-interpreter/src/kernels/GRU.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -40,8 +40,6 @@ void GRU::configure()
auto hidden_input_shape = getTensorShape(hidden_input());
LUCI_INTERPRETER_CHECK(hidden_hidden_shape.Dims(0) == hidden_input_shape.Dims(0));

const int32_t div_factor = 3;

auto output_shape = getTensorShape(output());
auto state_shape = getTensorShape(state());

Expand Down Expand Up @@ -77,8 +75,7 @@ void GRU::evalFloat() const
getTensorData<float>(hidden_hidden_bias()), getTensorData<float>(state()),
getTensorData<float>(output()), reinterpret_cast<float *>(output_input_data),
reinterpret_cast<float *>(output_hidden_data), getTensorShape(input()),
getTensorShape(output()), getTensorShape(hidden_input()), getTensorShape(hidden_hidden()), 0,
nullptr);
getTensorShape(output()), getTensorShape(hidden_input()), getTensorShape(hidden_hidden()));

delete output_hidden_data;
delete output_input_data;
Expand Down

0 comments on commit 223b60d

Please sign in to comment.