Searched refs:timestep (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | gru_v2_test.py | 85 timestep = 4
92 input_shape=(timestep, input_shape),
99 shape=[timestep, input_shape], dtype=dtypes.float32)
145 timestep = 4
151 input_shape=(timestep, input_shape),
156 shape=[timestep, input_shape], dtype=dtypes.float32)
194 timestep = 3
198 x = np.random.random((batch, timestep, input_dim))
202 shape=[timestep, input_dim], dtype=dtypes.float32)
224 timestep = 4
[all …]
|
| D | lstm_v2_test.py | 313 timestep = 4
319 input_shape=(timestep, input_shape),
324 shape=[timestep, input_shape], dtype=dtypes.float32)
413 timestep = 4
416 x_train = np.random.random((batch, timestep, input_shape))
420 shape=[timestep, input_shape], dtype=dtypes.float32)
449 timestep = 4
456 input_shape=(timestep, input_shape),
463 shape=[timestep, input_shape], dtype=dtypes.float32)
484 timestep = 3
[all …]
|
| D | recurrent_v2_test.py | 46 timestep = 12
48 x = np.random.randint(0, vocab_size, size=(batch_size, timestep))
49 y = np.random.randint(0, vocab_size, size=(batch_size, timestep))
56 batch_input_shape=[batch_size, timestep]),
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | rnn_test.py | 368 timestep = 4
373 input_shape=(timestep, input_shape),
379 dtypes.float32, shape=(None, timestep, input_shape))
385 self.assertEqual(outputs.shape.as_list(), [None, timestep, output_shape])
402 timestep = 4
407 input_shape=(timestep, input_shape),
413 dtypes.float32, shape=(None, timestep, input_shape))
419 self.assertEqual(outputs.shape.as_list(), [None, timestep, output_shape])
436 timestep = 4
441 input_shape=(timestep, input_shape),
[all …]
|
| /external/tensorflow/tensorflow/contrib/seq2seq/python/kernel_tests/ |
| D | attention_wrapper_v2_test.py | 44 self.timestep = 5
48 self.memory = np.random.randn(self.batch, self.timestep,
51 self.state = np.random.randn(self.batch, self.timestep).astype(np.float32)
63 self.assertEqual(attention_score[0].shape, (self.batch, self.timestep))
64 self.assertEqual(attention_score[1].shape, (self.batch, self.timestep))
94 self.assertEqual(score_val[0].shape, (self.batch, self.timestep))
95 self.assertEqual(score_val[1].shape, (self.batch, self.timestep))
134 inputs = keras.layers.Input(shape=[self.timestep])
144 state = keras.layers.Input(shape=[self.timestep])
148 x = np.random.randint(vocab, size=(self.batch, self.timestep))
[all …]
|
| /external/tensorflow/tensorflow/contrib/opt/python/training/ |
| D | shampoo.py | 38 def GetParam(var, timestep): argument
40 return var(timestep)
|
| /external/tensorflow/tensorflow/contrib/timeseries/python/timeseries/state_space_models/ |
| D | state_space_model.py | 815 for timestep, observation_model in enumerate(observation_models):
825 observations[:, timestep] = numpy.random.normal(loc=observation_mean,
828 observations[:, timestep] = observation_mean
|
| /external/tensorflow/tensorflow/contrib/rnn/python/kernel_tests/ |
| D | rnn_cell_test.py | 1482 timestep = 4
1487 input_shape=(timestep, input_shape),
1493 dtypes.float32, shape=(None, timestep, input_shape))
1499 self.assertEqual(outputs.shape.as_list(), [None, timestep, output_shape])
1549 timestep = 4
1554 input_shape=(timestep, input_shape),
1560 dtypes.float32, shape=(None, timestep, input_shape))
1566 self.assertEqual(outputs.shape.as_list(), [None, timestep, output_shape])
|
| /external/tensorflow/tensorflow/core/protobuf/tpu/ |
| D | optimization_parameters.proto | 81 // Here, t is the current timestep.
|