Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/delegates/xnnpack/
Dreshape_tester.h79 int32_t input_size_ = 1; variable
Dreduce_tester.h109 int32_t input_size_; variable
Dfully_connected_tester.h143 int32_t input_size_ = 1; variable
Dquantized_reduce_tester.h143 int32_t input_size_; variable
Dquantized_fully_connected_tester.h174 int32_t input_size_ = 1; variable
/external/tensorflow/tensorflow/lite/kernels/
Dlog_softmax_test.cc58 int input_size_; member in tflite::__anon394565300111::LogSoftmaxOpModel
Dsoftmax_test.cc59 int input_size_; member in tflite::__anon72ca1b8f0111::SoftmaxOpModel
Dsvdf_test.cc194 int input_size_; member in tflite::__anon10ad49200111::BaseSVDFOpModel
531 int input_size_; member in tflite::__anon10ad49200111::IntegerSVDFOpModel
Dquant_basic_lstm_test.cc156 int input_size_; member in tflite::__anon427e2f360111::QuantizedLSTMOpModel
Dbasic_rnn_test.cc229 int input_size_; member in tflite::__anon2c82b85c0111::RNNOpModel
Dbatch_matmul_test.cc440 int input_size_; member in tflite::__anon94d4a8fe0111::HybridBatchMatMulOpModel
972 int input_size_; member in tflite::__anon94d4a8fe0111::QuantizedBatchMatMulOpModel
Dunidirectional_sequence_rnn_test.cc245 int input_size_; member in tflite::__anona0a7fc2c0111::UnidirectionalRNNOpModel
Dfully_connected_test.cc256 int input_size_; member in tflite::__anon1e118d2b0111::BaseFullyConnectedOpModel
463 int input_size_; member in tflite::__anon1e118d2b0111::HybridFullyConnectedOpModel
1608 int input_size_; member in tflite::__anon1e118d2b0111::SparseFullyConnectedOpModel
Dlstm_eval_test.cc151 std::vector<int32_t> input_size_ = {n_batch_, n_input_}; member in tflite::__anon353329140111::BaseLstmParam
Dbidirectional_sequence_rnn_test.cc843 int input_size_; member in tflite::__anon8ffa91ab0111::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.cc103 int input_size_; member in tflite::FullyConnectedOpModel
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate_signed_quantization_test.cc129 int input_size_; member in tflite::__anon59489a3b0111::HybridFullyConnectedOpModel
Dnnapi_delegate_test.cc2868 int input_size_; member in tflite::__anon4fbea17c0111::RNNOpModel
3061 int input_size_; member in tflite::__anon4fbea17c0111::BaseSVDFOpModel
/external/tensorflow/tensorflow/core/kernels/
Dcudnn_rnn_ops.cc211 const int input_size_; member in tensorflow::__anon6bcc4dbf0111::CudnnRnnParameters
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.cc1886 int input_size_; member in stream_executor::gpu::MIOpenRnnDescriptor
/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
Dcuda_dnn.cc1411 int input_size_; member in stream_executor::gpu::CudnnRnnDescriptor