Skip to content

Commit

Permalink
[circlechef] Removed beta(bias) from RmsNorm (#14195)
Browse files Browse the repository at this point in the history
This commit removes beta(bias) from RmsNorm

ONE-DCO-1.0-Signed-off-by: Seockho Kim [email protected]
  • Loading branch information
seockho-kim authored Oct 10, 2024
1 parent ba37de2 commit 2dc21f4
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions compiler/circlechef/circle/src/Op/RmsNorm.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -24,12 +24,12 @@ namespace circlechef
void CircleOpRmsNorm::filler(const circle::Operator *op, CircleImport *import,
circlechef::ModelRecipe *model_recipe) const
{
// index 1 and 2 maybe constant
// index 1 maybe constant
const std::vector<int32_t> &inputs = as_index_vector(op->inputs());
assert(inputs.size() == 3);
// TODO: Enable assert after recipe updated.
// assert(inputs.size() == 2);

import->set_tensor_filler(inputs[1]); // set gaussian filler
import->set_tensor_filler(inputs[2]);
}

circlechef::Operation *CircleOpRmsNorm::build(const circle::Operator *op, CircleImport *import,
Expand Down

0 comments on commit 2dc21f4

Please sign in to comment.