Home
last modified time | relevance | path

Searched refs:zero_state (Results 1 – 25 of 47) sorted by relevance

12

/external/tensorflow/tensorflow/contrib/seq2seq/python/kernel_tests/
Ddecoder_v2_test.py63 initial_state = cell.zero_state(
130 zero_state = cell.zero_state(dtype=dtypes.float32, batch_size=batch_size)
136 inputs, initial_state=zero_state, sequence_length=sequence_length)
142 initial_state=zero_state)
Ddecoder_test.py59 initial_state=cell.zero_state(
133 zero_state = cell.zero_state(dtype=dtypes.float32, batch_size=batch_size)
136 cell=cell, helper=helper, initial_state=zero_state)
153 initial_state=zero_state,
Dbasic_decoder_test.py65 initial_state=cell.zero_state(
145 initial_state=cell.zero_state(
221 initial_state=cell.zero_state(
294 initial_state=cell.zero_state(
396 initial_state=cell.zero_state(
535 initial_state=cell.zero_state(
616 initial_state=cell.zero_state(
Dbasic_decoder_v2_test.py65 initial_state = cell.zero_state(dtype=dtypes.float32,
144 initial_state = cell.zero_state(
221 initial_state = cell.zero_state(
295 initial_state = cell.zero_state(
396 initial_state = cell.zero_state(
537 initial_state = cell.zero_state(
616 initial_state = cell.zero_state(
Dattention_wrapper_test.py96 static_state = cell.zero_state(batch_size, dtypes.float32)
98 state = cell.zero_state(array_ops.shape(memory)[0], dtypes.float32)
208 initial_state=cell.zero_state(
249 cell.zero_state(batch_size, dtypes.float32))
308 initial_state=cell.zero_state(
437 initial_state=cell.zero_state(
1033 initial_state = attention_cell.zero_state(
Dbeam_search_decoder_test.py499 initial_state = cell.zero_state(batch_size, dtypes.float32)
522 cell_state = cell.zero_state(
613 initial_state = cell.zero_state(batch_size, dtypes.float32)
636 cell_state = cell.zero_state(
Dattention_wrapper_v2_test.py362 cell.zero_state(batch_size, dtypes.float32))
416 initial_state=cell.zero_state(dtype=dtype, batch_size=self.batch),
443 initial_state=cell.zero_state(dtype=dtype, batch_size=self.batch),
/external/tensorflow/tensorflow/contrib/rnn/python/ops/
Dcore_rnn_cell.py240 def zero_state(self, batch_size, dtype): member in EmbeddingWrapper
242 return self._cell.zero_state(batch_size, dtype)
316 def zero_state(self, batch_size, dtype): member in InputProjectionWrapper
318 return self._cell.zero_state(batch_size, dtype)
372 def zero_state(self, batch_size, dtype): member in OutputProjectionWrapper
374 return self._cell.zero_state(batch_size, dtype)
Dlstm_ops.py199 zero_state = None
201 zero_state = array_ops.constant(
204 cs_prev = zero_state
206 h_prev = zero_state
/external/tensorflow/tensorflow/contrib/rnn/python/kernel_tests/
Drnn_cell_test.py818 zero_state = ((zeros, zeros), zeros, attn_state_zeros)
820 zero_state = array_ops.zeros(
828 output, state = cell(inputs, zero_state)
873 zero_state = ((zeros, zeros), zeros, attn_state_zeros)
875 zero_state = constant_op.constant(
888 output, state = cell(inputs, zero_state)
959 zero_state = ((zeros1, zeros2), zeros3, attn_state_zeros)
961 zero_state = array_ops.concat([
962 zero_state[0][0], zero_state[0][1], zero_state[1], zero_state[2]
966 output, state = cell(inputs, zero_state)
[all …]
/external/tensorflow/tensorflow/python/ops/
Drnn_cell_impl.py310 return self.zero_state(batch_size, dtype)
312 def zero_state(self, batch_size, dtype): member in RNNCell
1347 def zero_state(self, batch_size, dtype): member in DropoutWrapperBase
1349 return self.cell.zero_state(batch_size, dtype)
1476 def zero_state(self, batch_size, dtype): member in ResidualWrapperBase
1478 return self.cell.zero_state(batch_size, dtype)
1552 def zero_state(self, batch_size, dtype): member in DeviceWrapperBase
1555 return self.cell.zero_state(batch_size, dtype)
1649 def zero_state(self, batch_size, dtype): member in MultiRNNCell
1652 return tuple(cell.zero_state(batch_size, dtype) for cell in self._cells)
[all …]
/external/tensorflow/tensorflow/contrib/legacy_seq2seq/python/ops/
Dseq2seq.py532 zero_state = cell.zero_state(batch_size, dtype)
533 if nest.is_sequence(zero_state):
535 structure=zero_state, flat_sequence=state_list)
/external/tensorflow/tensorflow/contrib/recurrent/python/ops/
Dfunctional_rnn.py139 self._zero_state = rnn_cell.zero_state
292 initial_state = cell.zero_state(batch_size, dtype)
/external/tensorflow/tensorflow/python/kernel_tests/
Drnn_cell_test.py733 initial_state=cell.zero_state(batch_size, dtypes.float64))
831 initial_state=cell.zero_state(batch_size, dtypes.float64))
1064 test_zero = cell.zero_state(1, dtypes.float32)
2069 next_state = cell.zero_state(batch_size, dtypes.float32)
2178 next_state = cell.zero_state(batch_size, dtypes.float32)
2220 next_state = cell.zero_state(batch_size, dtypes.float32)
2263 next_state = cell.zero_state(batch_size, dtypes.float32)
2331 next_state = cell.zero_state(batch_size, dtypes.float32)
2694 zero_state = cell.zero_state(1, dtypes.float32)
2695 self.assertTrue(isinstance(zero_state, tuple))
[all …]
/external/tensorflow/tensorflow/lite/experimental/examples/lstm/
Drnn.py239 state = cell.zero_state(batch_size, dtype)
/external/tensorflow/tensorflow/python/ops/parallel_for/
Dgradients_test.py79 init_state = cell.zero_state(batch_size, dtypes.float32)
96 initial_state = cell.zero_state(batch_size, dtypes.float32)
192 init_state = cell.zero_state(batch_size, dtypes.float32)
/external/tensorflow/tensorflow/contrib/model_pruning/python/
Dstrip_pruning_vars_test.py121 initial_state=cell.zero_state(batch_size, dtypes.float32))
/external/tensorflow/tensorflow/contrib/timeseries/examples/
Dlstm.py119 in self._lstm_cell.zero_state(batch_size=1, dtype=self.dtype)])
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.nn.-r-n-n-cell-dropout-wrapper.pbtxt203 name: "zero_state"
Dtensorflow.nn.-r-n-n-cell-device-wrapper.pbtxt199 name: "zero_state"
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.nn.rnn_cell.-basic-r-n-n-cell.pbtxt207 name: "zero_state"
Dtensorflow.nn.rnn_cell.-basic-l-s-t-m-cell.pbtxt207 name: "zero_state"
Dtensorflow.nn.rnn_cell.-residual-wrapper.pbtxt208 name: "zero_state"
Dtensorflow.nn.rnn_cell.-multi-r-n-n-cell.pbtxt206 name: "zero_state"
Dtensorflow.nn.rnn_cell.-device-wrapper.pbtxt208 name: "zero_state"

12