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::__anonad02274b0111::LogSoftmaxOpModel
Dsoftmax_test.cc59 int input_size_; member in tflite::__anonbc7c082a0111::SoftmaxOpModel
Dsvdf_test.cc194 int input_size_; member in tflite::__anon2a1e039b0111::BaseSVDFOpModel
531 int input_size_; member in tflite::__anon2a1e039b0111::IntegerSVDFOpModel
Dquant_basic_lstm_test.cc156 int input_size_; member in tflite::__anon337f7a310111::QuantizedLSTMOpModel
Dbatch_matmul_test.cc400 int input_size_; member in tflite::__anon8029ae790111::HybridBatchMatMulOpModel
765 int input_size_; member in tflite::__anon8029ae790111::QuantizedBatchMatMulOpModel
Dfully_connected_test.cc211 int input_size_; member in tflite::__anonc847e4c60111::BaseFullyConnectedOpModel
355 int input_size_; member in tflite::__anonc847e4c60111::HybridFullyConnectedOpModel
1234 int input_size_; member in tflite::__anonc847e4c60111::SparseFullyConnectedOpModel
Dbasic_rnn_test.cc229 int input_size_; member in tflite::__anonaa6237b70111::RNNOpModel
Dunidirectional_sequence_rnn_test.cc245 int input_size_; member in tflite::__anona6a394470111::UnidirectionalRNNOpModel
Dlstm_eval_test.cc151 std::vector<int32_t> input_size_ = {n_batch_, n_input_}; member in tflite::__anonb312a86f0111::BaseLstmParam
Dbidirectional_sequence_rnn_test.cc843 int input_size_; member in tflite::__anon048612660111::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::__anonb9fc30f60111::HybridFullyConnectedOpModel
Dnnapi_delegate_test.cc2791 int input_size_; member in tflite::__anon0c68fcb70111::RNNOpModel
2984 int input_size_; member in tflite::__anon0c68fcb70111::BaseSVDFOpModel
/external/tensorflow/tensorflow/core/kernels/
Dcudnn_rnn_ops.cc208 const int input_size_; member in tensorflow::__anonebbbcdba0111::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