From 406cb3a31202bf63284c32ba65d679bcb5135eb1 Mon Sep 17 00:00:00 2001 From: Balyshev Artem <43214667+BalyshevArtem@users.noreply.github.com> Date: Fri, 22 Mar 2024 07:48:56 +0300 Subject: [PATCH] [res] Add a CircleRecipe for GRU (#12785) This pr adds a CircleRecipe for GRU op. ONE-DCO-1.0-Signed-off-by: Artem Balyshev --- res/CircleRecipes/GRU_000/test.recipe | 77 ++++++++++++++++++++++++++ res/CircleRecipes/GRU_000/test.reverse | 0 2 files changed, 77 insertions(+) create mode 100644 res/CircleRecipes/GRU_000/test.recipe create mode 100644 res/CircleRecipes/GRU_000/test.reverse diff --git a/res/CircleRecipes/GRU_000/test.recipe b/res/CircleRecipes/GRU_000/test.recipe new file mode 100644 index 00000000000..3ef639bfbea --- /dev/null +++ b/res/CircleRecipes/GRU_000/test.recipe @@ -0,0 +1,77 @@ +operand { + name: "ifm" + type: FLOAT32 + shape { dim: 3 dim: 1 dim: 2 } +} +operand { + name: "hidden_hidden" + type: FLOAT32 + shape { dim: 12 dim: 4 } + filler { + tag: "gaussian" + arg: "0.0" + arg: "1.0" + } +} +operand { + name: "hidden_hidden_bias" + type: FLOAT32 + shape { dim: 12 } + filler { + tag: "gaussian" + arg: "0.0" + arg: "1.0" + } +} +operand { + name: "hidden_input" + type: FLOAT32 + shape { dim: 12 dim: 2 } + filler { + tag: "gaussian" + arg: "0.0" + arg: "1.0" + } +} +operand { + name: "hidden_input_bias" + type: FLOAT32 + shape { dim: 12 } + filler { + tag: "gaussian" + arg: "0.0" + arg: "1.0" + } +} +operand { + name: "state" + type: FLOAT32 + shape { dim: 1 dim: 4 } + filler { + tag: "gaussian" + arg: "0.0" + arg: "1.0" + } +} +operand { + name: "ofm" + type: FLOAT32 + shape { dim: 1 dim: 1 dim: 4 } +} +operation { + type: "GRU" + gru_options { + activation: NONE + return_sequences: false + time_major: false + } + input: "ifm" + input: "hidden_hidden" + input: "hidden_hidden_bias" + input: "hidden_input" + input: "hidden_input_bias" + input: "state" + output: "ofm" +} +input: "ifm" +output: "ofm" diff --git a/res/CircleRecipes/GRU_000/test.reverse b/res/CircleRecipes/GRU_000/test.reverse new file mode 100644 index 00000000000..e69de29bb2d