Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/delegates/xnnpack/
Dreshape_tester.h79 int32_t input_size_ = 1; variable
Dfully_connected_tester.h108 int32_t input_size_ = 1; variable
Dreduce_tester.h109 int32_t input_size_; variable
/external/tensorflow/tensorflow/lite/kernels/
Dlog_softmax_test.cc58 int input_size_; member in tflite::__anoneb2c93520111::LogSoftmaxOpModel
Dsoftmax_test.cc59 int input_size_; member in tflite::__anon8d5b88b10111::SoftmaxOpModel
Dsvdf_test.cc194 int input_size_; member in tflite::__anon7126f9820111::BaseSVDFOpModel
531 int input_size_; member in tflite::__anon7126f9820111::IntegerSVDFOpModel
Dquant_basic_lstm_test.cc156 int input_size_; member in tflite::__anond96182980111::QuantizedLSTMOpModel
Dbatch_matmul_test.cc400 int input_size_; member in tflite::__anon83a19b600111::HybridBatchMatMulOpModel
765 int input_size_; member in tflite::__anon83a19b600111::QuantizedBatchMatMulOpModel
Dfully_connected_test.cc211 int input_size_; member in tflite::__anonae46fc4d0111::BaseFullyConnectedOpModel
355 int input_size_; member in tflite::__anonae46fc4d0111::HybridFullyConnectedOpModel
1234 int input_size_; member in tflite::__anonae46fc4d0111::SparseFullyConnectedOpModel
Dbasic_rnn_test.cc229 int input_size_; member in tflite::__anon3123f5fe0111::RNNOpModel
Dunidirectional_sequence_rnn_test.cc245 int input_size_; member in tflite::__anone81b2e4e0111::UnidirectionalRNNOpModel
Dlstm_eval_test.cc151 std::vector<int32_t> input_size_ = {n_batch_, n_input_}; member in tflite::__anon39d466b60111::BaseLstmParam
Dbidirectional_sequence_rnn_test.cc843 int input_size_; member in tflite::__anon7adf078d0111::BidirectionalRNNOpModel
/external/tensorflow/tensorflow/core/kernels/rnn/
Dgru_ops.h64 const int input_size_; member
Dlstm_ops.h156 const int input_size_; member
/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/tests/
Dmatmul_test.cc104 int input_size_; member in tflite::FullyConnectedOpModel
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate_signed_quantization_test.cc125 int input_size_; member in tflite::__anonc4d0941d0111::HybridFullyConnectedOpModel
Dnnapi_delegate_test.cc2791 int input_size_; member in tflite::__anonde5f765e0111::RNNOpModel
2984 int input_size_; member in tflite::__anonde5f765e0111::BaseSVDFOpModel
/external/tensorflow/tensorflow/core/kernels/
Dcudnn_rnn_ops.cc208 const int input_size_; member in tensorflow::__anond88b5f210111::CudnnRnnParameters
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.cc1233 int input_size_; member in stream_executor::gpu::CudnnRnnDescriptor
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.cc1832 int input_size_; member in stream_executor::gpu::MIOpenRnnDescriptor