Home
last modified time | relevance | path

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

12

/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A Dsvdf_test.cc194 int input_size_; member in tflite::__anona04cb8d10111::BaseSVDFOpModel
531 int input_size_; member in tflite::__anona04cb8d10111::IntegerSVDFOpModel
H A Dbatch_matmul_test.cc440 int input_size_; member in tflite::__anond27d39af0111::HybridBatchMatMulOpModel
972 int input_size_; member in tflite::__anond27d39af0111::QuantizedBatchMatMulOpModel
H A Dlog_softmax_test.cc58 int input_size_; member in tflite::__anon04d623c10111::LogSoftmaxOpModel
H A Dfully_connected_test.cc256 int input_size_; member in tflite::__anonaa112c3c0111::BaseFullyConnectedOpModel
463 int input_size_; member in tflite::__anonaa112c3c0111::HybridFullyConnectedOpModel
1608 int input_size_; member in tflite::__anonaa112c3c0111::SparseFullyConnectedOpModel
H A Dsoftmax_test.cc59 int input_size_; member in tflite::__anon0f4839a00111::SoftmaxOpModel
H A Dquant_basic_lstm_test.cc156 int input_size_; member in tflite::__anon4e71b0670111::QuantizedLSTMOpModel
H A Dbasic_rnn_test.cc229 int input_size_; member in tflite::__anone1009ead0111::RNNOpModel
H A Dunidirectional_sequence_rnn_test.cc245 int input_size_; member in tflite::__anon6f5dbcbd0111::UnidirectionalRNNOpModel
H A Dlstm_eval_test.cc151 std::vector<int32_t> input_size_ = {n_batch_, n_input_}; member in tflite::__anone9b10f650111::BaseLstmParam
H A Dbidirectional_sequence_rnn_test.cc843 int input_size_; member in tflite::__anona5c2041c0111::BidirectionalRNNOpModel
/aosp_15_r20/external/pytorch/aten/src/ATen/native/mkldnn/
H A DOpContext.h36 std::vector<int64_t> input_size_; variable
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/xnnpack/
H A Dreshape_tester.h79 int32_t input_size_ = 1; variable
H A Dreduce_tester.h109 int32_t input_size_; variable
H A Dfully_connected_tester.h143 int32_t input_size_ = 1; variable
H A Dquantized_reduce_tester.h143 int32_t input_size_; variable
H A Dquantized_fully_connected_tester.h174 int32_t input_size_ = 1; variable
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/rnn/
H A Dgru_ops.h64 const int input_size_; member
H A Dlstm_ops.h156 const int input_size_; member
/aosp_15_r20/packages/modules/NeuralNetworks/common/cpu_operations/
DRNNTest.cpp227 const uint32_t input_size_; member in android::nn::wrapper::BasicRNNOpModel
DSVDFTest.cpp304 const uint32_t input_size_; member in android::nn::wrapper::SVDFOpModel
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/tests/
H A Dmatmul_test.cc103 int input_size_; member in tflite::FullyConnectedOpModel
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/nnapi/
H A Dnnapi_delegate_signed_quantization_test.cc129 int input_size_; member in tflite::__anonf5e0ceac0111::HybridFullyConnectedOpModel
H A Dnnapi_delegate_test.cc2868 int input_size_; member in tflite::__anon1a74836d0111::RNNOpModel
3061 int input_size_; member in tflite::__anon1a74836d0111::BaseSVDFOpModel
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dcudnn_rnn_ops.cc211 const int input_size_; member in tensorflow::__anon980e2df00111::CudnnRnnParameters
/aosp_15_r20/external/tensorflow/tensorflow/stream_executor/rocm/
H A Drocm_dnn.cc1886 int input_size_; member in stream_executor::gpu::MIOpenRnnDescriptor

12