Searched defs:hidden_size (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/python/grappler/ |
| D | graph_placer_test.py | 41 hidden_size=256, argument
|
| /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/stream_executor/ |
| D | stream_executor_pimpl.cc | 355 int num_layers, int hidden_size, int input_size, int cell_size, in createRnnDescriptor()
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 1043 int hidden_size, int input_size, int cell_size, in CudnnRnnDescriptor()
1072 const CudnnHandle& cudnn, int num_layers, int hidden_size, int input_size, in Create()
1174 int hidden_size() const { return hidden_size_; } in hidden_size() function in stream_executor::gpu::CudnnRnnDescriptor
1565 int hidden_size = 0; member
2068 int num_layers, int hidden_size, int input_size, int cell_size, in createRnnDescriptor()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 1726 int hidden_size, int input_size, in MIOpenRnnDescriptor()
1769 int hidden_size() const { return hidden_size_; } in hidden_size() function in stream_executor::gpu::MIOpenRnnDescriptor
1929 int hidden_size = 0; member
2320 int num_layers, int hidden_size, int input_size, int cell_size, in createRnnDescriptor()
|