| /external/tensorflow/tensorflow/python/grappler/ |
| D | hierarchical_controller.py | 118 hidden_size=512,
195 self.embedding_size + self.hparams.hidden_size / 2,
196 2 * self.hparams.hidden_size
199 self.embedding_size + self.hparams.hidden_size / 2,
200 2 * self.hparams.hidden_size
203 "device_embeddings", [self.num_devices, self.hparams.hidden_size])
206 [2 * self.hparams.hidden_size, 4 * self.hparams.hidden_size])
208 "device_softmax", [2 * self.hparams.hidden_size, self.num_devices])
210 [1, self.hparams.hidden_size])
224 "attn_w_1", [self.hparams.hidden_size, self.hparams.hidden_size])
[all …]
|
| D | graph_placer_test.py | 41 hidden_size=256, argument
50 a = input_size if layer_id == 0 else hidden_size
51 b = hidden_size if layer_id < num_layers - 1 else num_classes
|
| /external/tensorflow/tensorflow/contrib/mpi_collectives/ |
| D | mpi_allreduce_test.py | 60 hidden_size = batch_size
64 inputs = tf.placeholder(tf.float32, shape=(batch_size, hidden_size),
72 shape=(hidden_size, out_size),
91 my_reducer = tf.Variable(initial_value=np.ones((hidden_size, out_size)),
112 input_feed = np.ones((batch_size, hidden_size), dtype=np.float32)
129 my_correct = np.zeros((batch_size, hidden_size), dtype=np.float32)
132 our_correct = np.zeros((batch_size, hidden_size), dtype=np.float32)
|
| /external/tensorflow/tensorflow/contrib/autograph/examples/benchmarks/ |
| D | cartpole_benchmark.py | 74 def __init__(self, hidden_size): argument
77 hidden_size, activation=tf.nn.elu)
275 def __init__(self, hidden_size): argument
278 hidden_size, activation=tf.nn.elu)
424 network = GraphPolicyNetwork(hidden_size=5)
441 network = EagerPolicyNetwork(hidden_size=5)
468 network = GraphPolicyNetwork(hidden_size=model_size)
482 network = EagerPolicyNetwork(hidden_size=model_size)
|
| /external/libtextclassifier/lang_id/common/ |
| D | embedding-network-params.h | 116 CheckIndex(i, hidden_size(), "hidden layer"); in GetHiddenLayerMatrix()
224 virtual int hidden_size() const = 0;
|
| D | embedding-network.cc | 303 SAFTM_CHECK_EQ(model_->hidden_size(), model_->hidden_bias_size()); in EmbeddingNetwork()
304 for (int i = 0; i < model_->hidden_size(); ++i) { in EmbeddingNetwork()
|
| /external/libtextclassifier/lang_id/common/flatbuffers/ |
| D | embedding-network-params-from-flatbuffer.h | 87 int hidden_size() const override { in hidden_size() function
112 int hidden_bias_size() const override { return hidden_size(); } in hidden_bias_size()
|
| D | embedding-network-params-from-flatbuffer.cc | 219 for (int i = 0; i < hidden_size(); ++i) { in ValidityChecking()
|
| /external/tensorflow/tensorflow/contrib/eager/python/examples/nmt_with_attention/ |
| D | nmt_with_attention.ipynb | 355 …ut of shape *(batch_size, max_length, hidden_size)* and the encoder hidden state of shape *(batch_…
457 " # enc_output shape == (batch_size, max_length, hidden_size)\n",
471 " # context_vector shape after sum == (batch_size, hidden_size)\n",
478 " # x shape after concatenation == (batch_size, 1, embedding_dim + hidden_size)\n",
484 " # output shape == (batch_size * 1, hidden_size)\n",
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 1005 int hidden_size, int input_size, int batch_size, in CudnnRnnDescriptor() argument
1016 hidden_size_(hidden_size), in CudnnRnnDescriptor()
1033 const CudnnHandle& cudnn, int num_layers, int hidden_size, int input_size, in Create() argument
1048 cudnn.handle(), /*rnnDesc=*/rnn_desc.get(), /*hiddenSize=*/hidden_size, in Create()
1101 num_layers, hidden_size, input_size, batch_size, in Create()
1109 int hidden_size() const { return hidden_size_; } in hidden_size() function in stream_executor::gpu::CudnnRnnDescriptor
1405 int hidden_size = 0; member
1430 model_dims.hidden_size = rnn_desc.hidden_size(); in ExtractAndCheckRnnForward()
1439 input_h_desc.data_size() == model_dims.hidden_size)) { in ExtractAndCheckRnnForward()
1450 model_dims.hidden_size * model_dims.dir_count)) { in ExtractAndCheckRnnForward()
[all …]
|
| D | cuda_dnn.h | 51 int num_layers, int hidden_size, int input_size, int batch_size,
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 1682 int hidden_size, int input_size, in MIOpenRnnDescriptor() argument
1690 hidden_size_(hidden_size), in MIOpenRnnDescriptor()
1700 rnn_desc_ /*rnnDesc*/, hidden_size /*hiddenSize*/, in MIOpenRnnDescriptor()
1725 int hidden_size() const { return hidden_size_; } in hidden_size() function in stream_executor::gpu::MIOpenRnnDescriptor
1885 int hidden_size = 0; member
1909 model_dims->hidden_size = rnn_desc.hidden_size(); in ExtractAndCheckRnnForward()
1918 input_h_desc.data_size() == model_dims->hidden_size)) { in ExtractAndCheckRnnForward()
1931 model_dims->hidden_size * model_dims->dir_count)) { in ExtractAndCheckRnnForward()
2276 int num_layers, int hidden_size, int input_size, int batch_size, in createRnnDescriptor() argument
2285 miopen.handle(), num_layers, hidden_size, input_size, in createRnnDescriptor()
|
| D | rocm_dnn.h | 49 int num_layers, int hidden_size, int input_size, int batch_size,
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | stream_executor_pimpl.cc | 382 int num_layers, int hidden_size, int input_size, int batch_size, in createRnnDescriptor() argument
393 num_layers, hidden_size, input_size, batch_size, input_mode, in createRnnDescriptor()
|
| D | stream_executor_pimpl.h | 408 int num_layers, int hidden_size, int input_size, int batch_size,
|
| D | dnn.h | 2042 createRnnDescriptor(int num_layers, int hidden_size, int input_size,
|