Searched defs:hidden_size (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/python/grappler/ |
| D | graph_placer_test.py | 41 hidden_size=256, argument
|
| /external/tensorflow/tensorflow/contrib/autograph/examples/benchmarks/ |
| D | cartpole_benchmark.py | 74 def __init__(self, hidden_size): argument
275 def __init__(self, hidden_size): argument
|
| /external/libtextclassifier/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/cuda/ |
| D | cuda_dnn.cc | 1005 int hidden_size, int input_size, int batch_size, in CudnnRnnDescriptor()
1033 const CudnnHandle& cudnn, int num_layers, int hidden_size, int input_size, in Create()
1109 int hidden_size() const { return hidden_size_; } in hidden_size() function in stream_executor::gpu::CudnnRnnDescriptor
1405 int hidden_size = 0; member
1817 int num_layers, int hidden_size, int input_size, int batch_size, in createRnnDescriptor()
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | stream_executor_pimpl.cc | 382 int num_layers, int hidden_size, int input_size, int batch_size, in createRnnDescriptor()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 1682 int hidden_size, int input_size, in MIOpenRnnDescriptor()
1725 int hidden_size() const { return hidden_size_; } in hidden_size() function in stream_executor::gpu::MIOpenRnnDescriptor
1885 int hidden_size = 0; member
2276 int num_layers, int hidden_size, int input_size, int batch_size, in createRnnDescriptor()
|