| /external/tensorflow/tensorflow/python/ops/ |
| D | rnn_grad_test.py | 43 hidden_size = 8
45 [input_size + hidden_size, 4 * hidden_size])
46 b = deterministic_random_uniform([4 * hidden_size])
48 cs_prev = h_prev = deterministic_random_uniform([batch_size, hidden_size])
73 hidden_size = np.random.randint(1, 32)
75 [input_size + hidden_size, 4 * hidden_size])
76 b = deterministic_random_uniform([4 * hidden_size])
78 cs_prev = h_prev = deterministic_random_uniform([batch_size, hidden_size])
80 cs_grad = deterministic_random_uniform([batch_size, hidden_size])
81 h_grad = deterministic_random_uniform([batch_size, hidden_size])
|
| /external/tensorflow/tensorflow/python/framework/experimental/ |
| D | unified_api_test.py | 405 num_iters, hidden_layers, hidden_size, batch_size): argument
430 [hidden_size if i else image_size, hidden_size]), cast))
432 random_ops.random_uniform([hidden_size, num_classes]), cast)
454 hidden_size=100,
470 hidden_size=100,
484 hidden_size=100,
500 hidden_size=100,
|
| /external/libtextclassifier/native/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/native/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/python/keras/layers/ |
| D | multi_head_attention_test.py | 202 batch_size, hidden_size = 3, 8
204 query_shape = [batch_size] + q_dims + [hidden_size]
205 value_shape = [batch_size] + v_dims + [hidden_size]
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 1060 int hidden_size, int input_size, int cell_size, in CudnnRnnDescriptor() argument
1071 hidden_size_(hidden_size), in CudnnRnnDescriptor()
1089 const CudnnHandle& cudnn, int num_layers, int hidden_size, int input_size, in Create() argument
1103 auto proj_size = hidden_size; in Create()
1104 hidden_size = std::max(hidden_size, cell_size); in Create()
1145 /*hiddenSize=*/hidden_size, /*projSize=*/proj_size, in Create()
1152 /*hiddenSize=*/hidden_size, /*numLayers=*/num_layers, in Create()
1158 if (proj_size < hidden_size) { in Create()
1195 num_layers, hidden_size, input_size, cell_size, in Create()
1203 int hidden_size() const { return hidden_size_; } in hidden_size() function in stream_executor::gpu::CudnnRnnDescriptor
[all …]
|
| D | cuda_dnn.h | 51 int num_layers, int hidden_size, int input_size, int cell_size,
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 1763 int hidden_size, int input_size, in MIOpenRnnDescriptor() argument
1771 hidden_size_(hidden_size), in MIOpenRnnDescriptor()
1781 rnn_desc_ /*rnnDesc*/, hidden_size /*hiddenSize*/, in MIOpenRnnDescriptor()
1806 int hidden_size() const { return hidden_size_; } in hidden_size() function in stream_executor::gpu::MIOpenRnnDescriptor
1966 int hidden_size = 0; member
1990 model_dims->hidden_size = rnn_desc.hidden_size(); in ExtractAndCheckRnnForward()
1999 input_h_desc.data_size() == model_dims->hidden_size)) { in ExtractAndCheckRnnForward()
2012 model_dims->hidden_size * model_dims->dir_count)) { in ExtractAndCheckRnnForward()
2516 int num_layers, int hidden_size, int input_size, int cell_size, in createRnnDescriptor() argument
2529 bool use_projection = cell_size != 0 && hidden_size < cell_size; in createRnnDescriptor()
[all …]
|
| D | rocm_dnn.h | 85 int num_layers, int hidden_size, int input_size, int cell_size,
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | stream_executor_pimpl.cc | 366 int num_layers, int hidden_size, int input_size, int cell_size, in createRnnDescriptor() argument
378 num_layers, hidden_size, input_size, cell_size, batch_size, input_mode, in createRnnDescriptor()
|
| D | stream_executor_pimpl.h | 416 int num_layers, int hidden_size, int input_size, int cell_size,
|
| D | dnn.h | 2110 createRnnDescriptor(int num_layers, int hidden_size, int input_size, in createRnnDescriptor() argument
|