Home
last modified time | relevance | path

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

/frameworks/ml/nn/common/operations/
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 …]
DRNNTest.cpp138 inputs.push_back(model_.addOperand(&InputTy)); in BasicRNNOpModel()
140 inputs.push_back(model_.addOperand(&WeightTy)); in BasicRNNOpModel()
142 inputs.push_back(model_.addOperand(&RecurrentWeightTy)); in BasicRNNOpModel()
144 inputs.push_back(model_.addOperand(&BiasTy)); in BasicRNNOpModel()
146 inputs.push_back(model_.addOperand(&HiddenStateTy)); in BasicRNNOpModel()
148 inputs.push_back(model_.addOperand(&ActionParamTy)); in BasicRNNOpModel()
152 outputs.push_back(model_.addOperand(&HiddenStateTy)); in BasicRNNOpModel()
154 outputs.push_back(model_.addOperand(&OutputTy)); in BasicRNNOpModel()
161 model_.addOperation(ANEURALNETWORKS_RNN, inputs, outputs); in BasicRNNOpModel()
162 model_.identifyInputsAndOutputs(inputs, outputs); in BasicRNNOpModel()
[all …]
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
DHashtableLookupTest.cpp70 inputs.push_back(model_.addOperand(&LookupTy)); in HashtableLookupOpModel()
73 inputs.push_back(model_.addOperand(&KeyTy)); in HashtableLookupOpModel()
76 inputs.push_back(model_.addOperand(&ValueTy)); in HashtableLookupOpModel()
85 outputs.push_back(model_.addOperand(&OutputOpndTy)); in HashtableLookupOpModel()
88 outputs.push_back(model_.addOperand(&HitsOpndTy)); in HashtableLookupOpModel()
102 model_.addOperation(ANEURALNETWORKS_HASHTABLE_LOOKUP, inputs, outputs); in HashtableLookupOpModel()
103 model_.identifyInputsAndOutputs(inputs, outputs); in HashtableLookupOpModel()
105 model_.finish(); in HashtableLookupOpModel()
109 ASSERT_TRUE(model_.isValid()); in Invoke()
111 Compilation compilation(&model_); in Invoke()
[all …]
DEmbeddingLookupTest.cpp66 inputs.push_back(model_.addOperand(&LookupTy)); in EmbeddingLookupOpModel()
69 inputs.push_back(model_.addOperand(&ValueTy)); in EmbeddingLookupOpModel()
74 outputs.push_back(model_.addOperand(&OutputOpndTy)); in EmbeddingLookupOpModel()
87 model_.addOperation(ANEURALNETWORKS_EMBEDDING_LOOKUP, inputs, outputs); in EmbeddingLookupOpModel()
88 model_.identifyInputsAndOutputs(inputs, outputs); in EmbeddingLookupOpModel()
90 model_.finish(); in EmbeddingLookupOpModel()
94 ASSERT_TRUE(model_.isValid()); in Invoke()
96 Compilation compilation(&model_); in Invoke()
141 Model model_; member in android::nn::wrapper::EmbeddingLookupOpModel
DSVDFTest.cpp186 inputs.push_back(model_.addOperand(&X##OpndTy)); in SVDFOpModel()
194 inputs.push_back(model_.addOperand(&RankParamTy)); in SVDFOpModel()
196 inputs.push_back(model_.addOperand(&ActivationParamTy)); in SVDFOpModel()
207 outputs.push_back(model_.addOperand(&X##OpndTy)); in SVDFOpModel()
230 model_.addOperation(ANEURALNETWORKS_SVDF, inputs, outputs); in SVDFOpModel()
231 model_.identifyInputsAndOutputs(inputs, outputs); in SVDFOpModel()
233 model_.finish(); in SVDFOpModel()
237 ASSERT_TRUE(model_.isValid()); in Invoke()
239 Compilation compilation(&model_); in Invoke()
298 Model model_; member in android::nn::wrapper::SVDFOpModel
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 …]
DLSTMTest.cpp102 inputs.push_back(model_.addOperand(&X##OpndTy)); in LSTMOpModel()
110 inputs.push_back(model_.addOperand(&ActivationOpndTy)); in LSTMOpModel()
112 inputs.push_back(model_.addOperand(&CellClipOpndTy)); in LSTMOpModel()
114 inputs.push_back(model_.addOperand(&ProjClipOpndTy)); in LSTMOpModel()
129 outputs.push_back(model_.addOperand(&X##OpndTy)); in LSTMOpModel()
135 model_.addOperation(ANEURALNETWORKS_LSTM, inputs, outputs); in LSTMOpModel()
136 model_.identifyInputsAndOutputs(inputs, outputs); in LSTMOpModel()
158 model_.finish(); in LSTMOpModel()
190 ASSERT_TRUE(model_.isValid()); in Invoke()
195 Compilation compilation(&model_); in Invoke()
[all …]
DQuantizedLSTMTest.cpp56 inputs.push_back(model_.addOperand(&curType)); in QuantizedLSTMOpModel()
68 outputs.push_back(model_.addOperand(&cellStateOutOperandType)); in QuantizedLSTMOpModel()
71 outputs.push_back(model_.addOperand(&outputOperandType)); in QuantizedLSTMOpModel()
73 model_.addOperation(ANEURALNETWORKS_QUANTIZED_16BIT_LSTM, inputs, outputs); in QuantizedLSTMOpModel()
74 model_.identifyInputsAndOutputs(inputs, outputs); in QuantizedLSTMOpModel()
85 model_.finish(); in QuantizedLSTMOpModel()
89 ASSERT_TRUE(model_.isValid()); in invoke()
91 Compilation compilation(&model_); in invoke()
200 Model model_; member in android::nn::wrapper::QuantizedLSTMOpModel