Home
last modified time | relevance | path

Searched defs:input_size_ (Results 1 – 13 of 13) sorted by relevance

/external/tensorflow/tensorflow/contrib/rnn/kernels/
Dgru_ops.h64 const int input_size_; member
Dlstm_ops.h144 const int input_size_; member
/external/tensorflow/tensorflow/lite/kernels/
Dlog_softmax_test.cc58 int input_size_; member in tflite::__anon9d13c1740111::LogSoftmaxOpModel
Dsoftmax_test.cc58 int input_size_; member in tflite::__anona7ecf5d30111::SoftmaxOpModel
Dbasic_rnn_test.cc230 int input_size_; member in tflite::__anon35c533a00111::RNNOpModel
Dfully_connected_test.cc195 int input_size_; member in tflite::__anon3e7c6b6f0111::BaseFullyConnectedOpModel
327 int input_size_; member in tflite::__anon3e7c6b6f0111::HybridFullyConnectedOpModel
Dsvdf_test.cc193 int input_size_; member in tflite::__anond1a0a9e40111::BaseSVDFOpModel
Dunidirectional_sequence_rnn_test.cc244 int input_size_; member in tflite::__anon2f8e60700111::UnidirectionalRNNOpModel
Dbidirectional_sequence_rnn_test.cc780 int input_size_; member in tflite::__anon65c37d6f0111::BidirectionalRNNOpModel
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate_test.cc650 int input_size_; member in tflite::__anon6d004b400111::SoftmaxOpModel
1957 int input_size_; member in tflite::__anon6d004b400111::RNNOpModel
2151 int input_size_; member in tflite::__anon6d004b400111::BaseSVDFOpModel
/external/tensorflow/tensorflow/core/kernels/
Dcudnn_rnn_ops.cc208 const int input_size_; member in tensorflow::__anon454a70830111::CudnnRnnParameters
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.cc1138 int input_size_; member in stream_executor::gpu::CudnnRnnDescriptor
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.cc1751 int input_size_; member in stream_executor::gpu::MIOpenRnnDescriptor