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 | 405 num_iters, hidden_layers, hidden_size, batch_size): argument
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 1060 int hidden_size, int input_size, int cell_size, in CudnnRnnDescriptor()
1089 const CudnnHandle& cudnn, int num_layers, int hidden_size, int input_size, in Create()
1203 int hidden_size() const { return hidden_size_; } in hidden_size() function in stream_executor::gpu::CudnnRnnDescriptor
1590 int hidden_size = 0; member
2071 int num_layers, int hidden_size, int input_size, int cell_size, in createRnnDescriptor()
|
| /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()
|
| D | dnn.h | 2110 createRnnDescriptor(int num_layers, int hidden_size, int input_size, in createRnnDescriptor()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 1763 int hidden_size, int input_size, 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
2516 int num_layers, int hidden_size, int input_size, int cell_size, in createRnnDescriptor()
|