Home
last modified time | relevance | path

Searched refs:model_ (Results 1 – 11 of 11) sorted by relevance

/aosp12/packages/modules/NeuralNetworks/common/operations/
H A DRNNTest.cpp139 inputs.push_back(model_.addOperand(&InputTy)); in BasicRNNOpModel()
141 inputs.push_back(model_.addOperand(&WeightTy)); in BasicRNNOpModel()
145 inputs.push_back(model_.addOperand(&BiasTy)); in BasicRNNOpModel()
147 inputs.push_back(model_.addOperand(&HiddenStateTy)); in BasicRNNOpModel()
149 inputs.push_back(model_.addOperand(&ActionParamTy)); in BasicRNNOpModel()
155 outputs.push_back(model_.addOperand(&OutputTy)); in BasicRNNOpModel()
163 model_.identifyInputsAndOutputs(inputs, outputs); in BasicRNNOpModel()
165 model_.finish(); in BasicRNNOpModel()
193 ASSERT_TRUE(model_.isValid()); in Invoke()
197 Compilation compilation(&model_); in Invoke()
[all …]
H A DLSHProjectionTest.cpp51 inputs.push_back(model_.addOperand(&HashTy)); in LSHProjectionOpModel()
53 inputs.push_back(model_.addOperand(&InputTy)); in LSHProjectionOpModel()
55 inputs.push_back(model_.addOperand(&WeightTy)); in LSHProjectionOpModel()
58 inputs.push_back(model_.addOperand(&TypeParamTy)); in LSHProjectionOpModel()
81 outputs.push_back(model_.addOperand(&OutputTy)); in LSHProjectionOpModel()
83 model_.addOperation(ANEURALNETWORKS_LSH_PROJECTION, inputs, outputs); in LSHProjectionOpModel()
84 model_.identifyInputsAndOutputs(inputs, outputs); in LSHProjectionOpModel()
86 model_.finish(); in LSHProjectionOpModel()
99 ASSERT_TRUE(model_.isValid()); in Invoke()
101 Compilation compilation(&model_); in Invoke()
[all …]
H A DMultinomialTest.cpp43 inputs.push_back(model_.addOperand(&logitsType)); in MultinomialOpModel()
45 inputs.push_back(model_.addOperand(&samplesType)); in MultinomialOpModel()
47 inputs.push_back(model_.addOperand(&seedsType)); in MultinomialOpModel()
51 outputs.push_back(model_.addOperand(&outputType)); in MultinomialOpModel()
53 model_.addOperation(ANEURALNETWORKS_RANDOM_MULTINOMIAL, inputs, outputs); in MultinomialOpModel()
54 model_.identifyInputsAndOutputs(inputs, outputs); in MultinomialOpModel()
55 model_.finish(); in MultinomialOpModel()
59 ASSERT_TRUE(model_.isValid()); in Invoke()
61 Compilation compilation(&model_); in Invoke()
95 Model model_; member in android::nn::wrapper::MultinomialOpModel
H A DHashtableLookupTest.cpp71 inputs.push_back(model_.addOperand(&LookupTy)); in HashtableLookupOpModel()
74 inputs.push_back(model_.addOperand(&KeyTy)); in HashtableLookupOpModel()
77 inputs.push_back(model_.addOperand(&ValueTy)); in HashtableLookupOpModel()
86 outputs.push_back(model_.addOperand(&OutputOpndTy)); in HashtableLookupOpModel()
89 outputs.push_back(model_.addOperand(&HitsOpndTy)); in HashtableLookupOpModel()
103 model_.addOperation(ANEURALNETWORKS_HASHTABLE_LOOKUP, inputs, outputs); in HashtableLookupOpModel()
104 model_.identifyInputsAndOutputs(inputs, outputs); in HashtableLookupOpModel()
106 model_.finish(); in HashtableLookupOpModel()
110 ASSERT_TRUE(model_.isValid()); in Invoke()
112 Compilation compilation(&model_); in Invoke()
[all …]
H A DEmbeddingLookupTest.cpp67 inputs.push_back(model_.addOperand(&LookupTy)); in EmbeddingLookupOpModel()
70 inputs.push_back(model_.addOperand(&ValueTy)); in EmbeddingLookupOpModel()
75 outputs.push_back(model_.addOperand(&OutputOpndTy)); in EmbeddingLookupOpModel()
88 model_.addOperation(ANEURALNETWORKS_EMBEDDING_LOOKUP, inputs, outputs); in EmbeddingLookupOpModel()
89 model_.identifyInputsAndOutputs(inputs, outputs); in EmbeddingLookupOpModel()
91 model_.finish(); in EmbeddingLookupOpModel()
95 ASSERT_TRUE(model_.isValid()); in Invoke()
97 Compilation compilation(&model_); in Invoke()
142 Model model_; member in android::nn::wrapper::EmbeddingLookupOpModel
H A DSVDFTest.cpp188 inputs.push_back(model_.addOperand(&X##OpndTy)); in SVDFOpModel()
196 inputs.push_back(model_.addOperand(&RankParamTy)); in SVDFOpModel()
198 inputs.push_back(model_.addOperand(&ActivationParamTy)); in SVDFOpModel()
209 outputs.push_back(model_.addOperand(&X##OpndTy)); in SVDFOpModel()
232 model_.addOperation(ANEURALNETWORKS_SVDF, inputs, outputs); in SVDFOpModel()
233 model_.identifyInputsAndOutputs(inputs, outputs); in SVDFOpModel()
235 model_.finish(); in SVDFOpModel()
239 ASSERT_TRUE(model_.isValid()); in Invoke()
241 Compilation compilation(&model_); in Invoke()
300 Model model_; member in android::nn::wrapper::SVDFOpModel
H A DLayerNormLSTMTest.cpp108 inputs.push_back(model_.addOperand(&X##OpndTy)); in LayerNormLSTMOpModel()
114 inputs.push_back(model_.addOperand(&ActivationOpndTy)); in LayerNormLSTMOpModel()
116 inputs.push_back(model_.addOperand(&CellClipOpndTy)); in LayerNormLSTMOpModel()
118 inputs.push_back(model_.addOperand(&ProjClipOpndTy)); in LayerNormLSTMOpModel()
138 outputs.push_back(model_.addOperand(&X##OpndTy)); in LayerNormLSTMOpModel()
144 model_.addOperation(ANEURALNETWORKS_LSTM, inputs, outputs); in LayerNormLSTMOpModel()
145 model_.identifyInputsAndOutputs(inputs, outputs); in LayerNormLSTMOpModel()
167 model_.finish(); in LayerNormLSTMOpModel()
200 ASSERT_TRUE(model_.isValid()); in Invoke()
205 Compilation compilation(&model_); in Invoke()
[all …]
H A DLSTMTest.cpp101 inputs.push_back(model_.addOperand(&X##OpndTy)); in LSTMOpModel()
109 inputs.push_back(model_.addOperand(&ActivationOpndTy)); in LSTMOpModel()
111 inputs.push_back(model_.addOperand(&CellClipOpndTy)); in LSTMOpModel()
113 inputs.push_back(model_.addOperand(&ProjClipOpndTy)); in LSTMOpModel()
128 outputs.push_back(model_.addOperand(&X##OpndTy)); in LSTMOpModel()
134 model_.addOperation(ANEURALNETWORKS_LSTM, inputs, outputs); in LSTMOpModel()
135 model_.identifyInputsAndOutputs(inputs, outputs); in LSTMOpModel()
157 model_.finish(); in LSTMOpModel()
189 ASSERT_TRUE(model_.isValid()); in Invoke()
194 Compilation compilation(&model_); in Invoke()
[all …]
H A DQuantizedLSTMTest.cpp57 inputs.push_back(model_.addOperand(&curType)); in QuantizedLSTMOpModel()
69 outputs.push_back(model_.addOperand(&cellStateOutOperandType)); in QuantizedLSTMOpModel()
72 outputs.push_back(model_.addOperand(&outputOperandType)); in QuantizedLSTMOpModel()
74 model_.addOperation(ANEURALNETWORKS_QUANTIZED_16BIT_LSTM, inputs, outputs); in QuantizedLSTMOpModel()
75 model_.identifyInputsAndOutputs(inputs, outputs); in QuantizedLSTMOpModel()
86 model_.finish(); in QuantizedLSTMOpModel()
90 ASSERT_TRUE(model_.isValid()); in invoke()
92 Compilation compilation(&model_); in invoke()
201 Model model_; member in android::nn::wrapper::QuantizedLSTMOpModel
/aosp12/system/bt/vendor_libs/test_vendor_lib/model/setup/
H A Dtest_command_handler.cc141 size_t dev_index = model_.Add(new_dev); in Add()
167 model_.AddRemote(args[0], port, phy_type); in AddRemote()
176 model_.Del(dev_index); in Del()
183 model_.AddPhy(Phy::Type::LOW_ENERGY); in AddPhy()
185 model_.AddPhy(Phy::Type::BR_EDR); in AddPhy()
195 model_.DelPhy(phy_index); in DelPhy()
208 model_.AddDeviceToPhy(dev_index, phy_index); in AddDeviceToPhy()
235 send_response_(model_.List()); in List()
274 model_.StartTimer(); in StartTimer()
283 model_.StopTimer(); in StopTimer()
[all …]
H A Dtest_command_handler.h91 TestModel& model_;