Searched defs:hidden_size (Results 1 – 6 of 6) sorted by relevance
| /external/libtextclassifier/native/lang_id/common/flatbuffers/ |
| D | embedding-network-params-from-flatbuffer.h | 87 int hidden_size() const override { in hidden_size() function
|
| /external/tensorflow/tensorflow/python/framework/experimental/ |
| D | unified_api_test.py | 402 num_iters, hidden_layers, hidden_size, batch_size): argument
|
| /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()
|
| D | dnn.h | 2038 createRnnDescriptor(int num_layers, int hidden_size, int input_size, in createRnnDescriptor()
|
| /external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/ |
| D | cuda_dnn.cc | 1235 int hidden_size, int input_size, int cell_size, in CudnnRnnDescriptor()
1264 const CudnnHandle& cudnn, int num_layers, int hidden_size, int input_size, in Create()
1381 int hidden_size() const { return hidden_size_; } in hidden_size() function in stream_executor::gpu::CudnnRnnDescriptor
1765 int hidden_size = 0; member
2414 int num_layers, int hidden_size, int input_size, int cell_size, in createRnnDescriptor()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 1817 int hidden_size, int input_size, 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
2570 int num_layers, int hidden_size, int input_size, int cell_size, in createRnnDescriptor()
|