| /external/tensorflow/tensorflow/python/ops/ |
| D | rnn_grad_test.py | 39 hidden_size = 8
41 [input_size + hidden_size, 4 * hidden_size])
42 b = deterministic_random_uniform([4 * hidden_size])
44 cs_prev = h_prev = deterministic_random_uniform([batch_size, hidden_size])
69 hidden_size = np.random.randint(1, 32)
71 [input_size + hidden_size, 4 * hidden_size])
72 b = deterministic_random_uniform([4 * hidden_size])
74 cs_prev = h_prev = deterministic_random_uniform([batch_size, hidden_size])
76 cs_grad = deterministic_random_uniform([batch_size, hidden_size])
77 h_grad = deterministic_random_uniform([batch_size, hidden_size])
|
| /external/tensorflow/tensorflow/python/framework/experimental/ |
| D | unified_api_test.py | 402 num_iters, hidden_layers, hidden_size, batch_size): argument
427 [hidden_size if i else image_size, hidden_size]), cast))
429 random_ops.random_uniform([hidden_size, num_classes]), cast)
451 hidden_size=100,
467 hidden_size=100,
481 hidden_size=100,
497 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 | 306 SAFTM_CHECK_EQ(model_->hidden_size(), model_->hidden_bias_size()); in EmbeddingNetwork()
307 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/compiler/xla/stream_executor/cuda/ |
| D | cuda_dnn.cc | 1235 int hidden_size, int input_size, int cell_size, in CudnnRnnDescriptor() argument
1246 hidden_size_(hidden_size), in CudnnRnnDescriptor()
1264 const CudnnHandle& cudnn, int num_layers, int hidden_size, int input_size, in Create() argument
1278 auto proj_size = hidden_size; in Create()
1279 hidden_size = std::max(hidden_size, cell_size); in Create()
1320 /*hiddenSize=*/hidden_size, /*projSize=*/proj_size, in Create()
1327 /*hiddenSize=*/hidden_size, /*numLayers=*/num_layers, in Create()
1333 if (proj_size < hidden_size) { in Create()
1373 num_layers, hidden_size, input_size, cell_size, in Create()
1381 int hidden_size() const { return hidden_size_; } in hidden_size() function in stream_executor::gpu::CudnnRnnDescriptor
[all …]
|
| D | cuda_dnn.h | 58 int num_layers, int hidden_size, int input_size, int cell_size,
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 1817 int hidden_size, int input_size, in MIOpenRnnDescriptor() argument
1825 hidden_size_(hidden_size), in MIOpenRnnDescriptor()
1835 rnn_desc_ /*rnnDesc*/, hidden_size /*hiddenSize*/, in MIOpenRnnDescriptor()
1860 int hidden_size() const { return hidden_size_; } in hidden_size() function in stream_executor::gpu::MIOpenRnnDescriptor
2020 int hidden_size = 0; member
2044 model_dims->hidden_size = rnn_desc.hidden_size(); in ExtractAndCheckRnnForward()
2053 input_h_desc.data_size() == model_dims->hidden_size)) { in ExtractAndCheckRnnForward()
2066 model_dims->hidden_size * model_dims->dir_count)) { in ExtractAndCheckRnnForward()
2570 int num_layers, int hidden_size, int input_size, int cell_size, in createRnnDescriptor() argument
2583 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/compiler/xla/stream_executor/ |
| D | stream_executor_pimpl.cc | 376 int num_layers, int hidden_size, int input_size, int cell_size, in createRnnDescriptor() argument
388 num_layers, hidden_size, input_size, cell_size, batch_size, input_mode, in createRnnDescriptor()
|
| D | stream_executor_pimpl.h | 420 int num_layers, int hidden_size, int input_size, int cell_size,
|
| D | dnn.h | 2038 createRnnDescriptor(int num_layers, int hidden_size, int input_size, in createRnnDescriptor() argument
|