Home
last modified time | relevance | path

Searched refs:GetOutput (Results 1 – 19 of 19) sorted by relevance

/frameworks/ml/nn/common/operations/
DLSHProjectionTest.cpp96 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()
DHashtableLookup.cpp45 output_ = GetOutput(operation, operands, kOutputTensor); in HashtableLookup()
46 hits_ = GetOutput(operation, operands, kHitsTensor); in HashtableLookup()
DMultinomialTest.cpp92 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()
DSVDFTest.cpp291 const std::vector<float>& GetOutput() const { return Output_; } in GetOutput() function in android::nn::wrapper::SVDFOpModel
355 EXPECT_THAT(svdf.GetOutput(), ElementsAreArray(ArrayFloatNear(expected))); in TEST()
414 EXPECT_THAT(svdf.GetOutput(), ElementsAreArray(ArrayFloatNear(expected))); in TEST()
DEmbeddingLookupTest.cpp138 const std::vector<float>& GetOutput() const { return Output_; } in GetOutput() function in android::nn::wrapper::EmbeddingLookupOpModel
163 EXPECT_THAT(m.GetOutput(), ElementsAreArray(ArrayFloatNear({ in TEST()
DLSTMTest.cpp187 const std::vector<float>& GetOutput() const { return Output_; } in GetOutput() function in android::nn::wrapper::LSTMOpModel
373 EXPECT_THAT(lstm.GetOutput(), ElementsAreArray(ArrayFloatNear(expected))); in TEST()
470 EXPECT_THAT(lstm.GetOutput(), ElementsAreArray(ArrayFloatNear(expected))); in TEST()
1005 EXPECT_THAT(lstm.GetOutput(), ElementsAreArray(ArrayFloatNear(expected))); in TEST()
DEmbeddingLookup.cpp36 output_ = GetOutput(operation, operands, kOutputTensor); in EmbeddingLookup()
DHashtableLookupTest.cpp151 const std::vector<float>& GetOutput() const { return Output_; } in GetOutput() function in android::nn::wrapper::HashtableLookupOpModel
176 EXPECT_THAT(m.GetOutput(), ElementsAreArray(ArrayFloatNear({ in TEST()
DBidirectionalSequenceLSTM.cpp192 fw_output_ = GetOutput(operation, operands, kFwOutputTensor); in BidirectionalSequenceLSTM()
194 bw_output_ = GetOutput(operation, operands, kBwOutputTensor); in BidirectionalSequenceLSTM()
201 GetOutput(operation, operands, kFwOutputActivationStateTensor - delta); in BidirectionalSequenceLSTM()
202 fw_output_cell_state_ = GetOutput(operation, operands, kFwOutputCellStateTensor - delta); in BidirectionalSequenceLSTM()
204 GetOutput(operation, operands, kBwOutputActivationStateTensor - delta); in BidirectionalSequenceLSTM()
205 bw_output_cell_state_ = GetOutput(operation, operands, kBwOutputCellStateTensor - delta); in BidirectionalSequenceLSTM()
DRNNTest.cpp185 const std::vector<float>& GetOutput() const { return Output_; } in GetOutput() function in android::nn::wrapper::BasicRNNOpModel
299 EXPECT_THAT(rnn.GetOutput(), ElementsAreArray(ArrayFloatNear(expected))); in TEST()
DSVDF.cpp47 state_out_ = GetOutput(operation, operands, kStateOutTensor); in SVDF()
48 output_ = GetOutput(operation, operands, kOutputTensor); in SVDF()
DRNN.cpp45 hidden_state_out_ = GetOutput(operation, operands, kHiddenStateOutTensor); in RNN()
46 output_ = GetOutput(operation, operands, kOutputTensor); in RNN()
DLayerNormLSTMTest.cpp197 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()
DMultinomial.cpp60 output_ = GetOutput(operation, operands, kOutputTensor); in Multinomial()
DQuantizedLSTM.cpp244 cellStateOut_ = GetOutput(operation, operands, kCellStateOutTensor); in QuantizedLSTMCell()
245 output_ = GetOutput(operation, operands, kOutputTensor); in QuantizedLSTMCell()
DLSHProjection.cpp42 output_ = GetOutput(operation, operands, kOutputTensor); in LSHProjection()
DLSTM.cpp124 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()
/frameworks/rs/tests/lldb/tests/harness/
Dtest_base.py181 error = error if error else res.GetOutput()
185 output = res.GetOutput() or ''
/frameworks/ml/nn/common/include/
DCpuExecutor.h289 inline RunTimeOperandInfo* GetOutput(const hal::Operation& operation, RunTimeOperandInfo* operands, in GetOutput() function