| /packages/modules/NeuralNetworks/common/operations/ |
| D | LSHProjectionTest.cpp | 96 const std::vector<int>& GetOutput() const { return Output_; } in GetOutput() function in android::nn::wrapper::LSHProjectionOpModel
148 EXPECT_THAT(m.GetOutput(), ElementsAre(1, 1, 1, 0, 1, 1, 1, 0)); in TEST()
159 EXPECT_THAT(m.GetOutput(), ElementsAre(1, 2, 2, 0)); in TEST()
170 EXPECT_THAT(m.GetOutput(), ElementsAre(1, 6, 10, 12)); in TEST()
|
| D | HashtableLookup.cpp | 41 output_ = GetOutput(operation, operands, kOutputTensor); in HashtableLookup()
42 hits_ = GetOutput(operation, operands, kHitsTensor); in HashtableLookup()
|
| D | SVDFTest.cpp | 293 const std::vector<float>& GetOutput() const { return Output_; } in GetOutput() function in android::nn::wrapper::SVDFOpModel
357 EXPECT_THAT(svdf.GetOutput(), ElementsAreArray(ArrayFloatNear(expected))); in TEST()
416 EXPECT_THAT(svdf.GetOutput(), ElementsAreArray(ArrayFloatNear(expected))); in TEST()
|
| D | LSTMTest.cpp | 186 const std::vector<float>& GetOutput() const { return Output_; } in GetOutput() function in android::nn::wrapper::LSTMOpModel
372 EXPECT_THAT(lstm.GetOutput(), ElementsAreArray(ArrayFloatNear(expected))); in TEST()
469 EXPECT_THAT(lstm.GetOutput(), ElementsAreArray(ArrayFloatNear(expected))); in TEST()
1004 EXPECT_THAT(lstm.GetOutput(), ElementsAreArray(ArrayFloatNear(expected))); in TEST()
|
| D | BidirectionalSequenceLSTM.cpp | 191 fw_output_ = GetOutput(operation, operands, kFwOutputTensor); in BidirectionalSequenceLSTM()
193 bw_output_ = GetOutput(operation, operands, kBwOutputTensor); in BidirectionalSequenceLSTM()
200 GetOutput(operation, operands, kFwOutputActivationStateTensor - delta); in BidirectionalSequenceLSTM()
201 fw_output_cell_state_ = GetOutput(operation, operands, kFwOutputCellStateTensor - delta); in BidirectionalSequenceLSTM()
203 GetOutput(operation, operands, kBwOutputActivationStateTensor - delta); in BidirectionalSequenceLSTM()
204 bw_output_cell_state_ = GetOutput(operation, operands, kBwOutputCellStateTensor - delta); in BidirectionalSequenceLSTM()
|
| D | MultinomialTest.cpp | 92 const std::vector<uint32_t>& GetOutput() const { return output_; } in GetOutput() function in android::nn::wrapper::MultinomialOpModel
114 std::vector<uint32_t> output = multinomial.GetOutput(); in TEST()
|
| D | EmbeddingLookupTest.cpp | 139 const std::vector<float>& GetOutput() const { return Output_; } in GetOutput() function in android::nn::wrapper::EmbeddingLookupOpModel
164 EXPECT_THAT(m.GetOutput(), ElementsAreArray(ArrayFloatNear({ in TEST()
|
| D | HashtableLookupTest.cpp | 152 const std::vector<float>& GetOutput() const { return Output_; } in GetOutput() function in android::nn::wrapper::HashtableLookupOpModel
177 EXPECT_THAT(m.GetOutput(), ElementsAreArray(ArrayFloatNear({ in TEST()
|
| D | RNNTest.cpp | 186 const std::vector<float>& GetOutput() const { return Output_; } in GetOutput() function in android::nn::wrapper::BasicRNNOpModel
300 EXPECT_THAT(rnn.GetOutput(), ElementsAreArray(ArrayFloatNear(expected))); in TEST()
|
| D | SVDF.cpp | 45 state_out_ = GetOutput(operation, operands, kStateOutTensor); in SVDF()
46 output_ = GetOutput(operation, operands, kOutputTensor); in SVDF()
|
| D | RNN.cpp | 41 hidden_state_out_ = GetOutput(operation, operands, kHiddenStateOutTensor); in RNN()
42 output_ = GetOutput(operation, operands, kOutputTensor); in RNN()
|
| D | LayerNormLSTMTest.cpp | 197 const std::vector<float>& GetOutput() const { return Output_; } in GetOutput() function in android::nn::wrapper::LayerNormLSTMOpModel
421 EXPECT_THAT(lstm.GetOutput(), ElementsAreArray(ArrayFloatNear(expected))); in TEST()
|
| D | EmbeddingLookup.cpp | 32 output_ = GetOutput(operation, operands, kOutputTensor); in EmbeddingLookup()
|
| D | LSTM.cpp | 124 output_state_out_ = GetOutput(operation, operands, kOutputStateOutTensor); in LSTMCell()
125 cell_state_out_ = GetOutput(operation, operands, kCellStateOutTensor); in LSTMCell()
126 output_ = GetOutput(operation, operands, kOutputTensor); in LSTMCell()
128 scratch_buffer_ = GetOutput(operation, operands, kScratchBufferTensor); in LSTMCell()
|
| D | QuantizedLSTM.cpp | 241 cellStateOut_ = GetOutput(operation, operands, kCellStateOutTensor); in QuantizedLSTMCell()
242 output_ = GetOutput(operation, operands, kOutputTensor); in QuantizedLSTMCell()
|
| D | Multinomial.cpp | 62 output_ = GetOutput(operation, operands, kOutputTensor); in Multinomial()
|
| D | LSHProjection.cpp | 41 output_ = GetOutput(operation, operands, kOutputTensor); in LSHProjection()
|
| /packages/modules/NeuralNetworks/common/include/ |
| D | CpuExecutor.h | 288 inline RunTimeOperandInfo* GetOutput(const Operation& operation, RunTimeOperandInfo* operands, in GetOutput() function
|