Home
last modified time | relevance | path

Searched refs:add_variable_wrapped (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/lite/experimental/examples/lstm/
Drnn_cell.py109 def add_variable_wrapped(name, shape, initializer, index): function
114 self._input_weights = add_variable_wrapped(
116 self._recurrent_weights = add_variable_wrapped(
118 self._bias = add_variable_wrapped(
310 def add_variable_wrapped(name, shape, initializer, index, partitioner): function
324 self.input_to_input_w = add_variable_wrapped(
327 self.input_to_forget_w = add_variable_wrapped(
330 self.input_to_cell_w = add_variable_wrapped(
333 self.input_to_output_w = add_variable_wrapped(
336 self.cell_to_input_w = add_variable_wrapped(
[all …]