Home
last modified time | relevance | path

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

/frameworks/ml/nn/common/operations/
DLSHProjectionTest.cpp95 const std::vector<int>& GetOutput() const { return Output_; } in GetOutput() function in android::nn::wrapper::LSHProjectionOpModel
147 EXPECT_THAT(m.GetOutput(), ElementsAre(1, 1, 1, 0, 1, 1, 1, 0)); in TEST()
158 EXPECT_THAT(m.GetOutput(), ElementsAre(1, 2, 2, 0)); in TEST()
169 EXPECT_THAT(m.GetOutput(), ElementsAre(1, 6, 10, 12)); in TEST()
DHashtableLookup.cpp41 output_ = GetOutput(operation, operands, kOutputTensor); in HashtableLookup()
42 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.cpp315 const std::vector<float>& GetOutput() const { return Output_; } in GetOutput() function in android::nn::wrapper::SVDFOpModel
381 EXPECT_THAT(svdf.GetOutput(), ElementsAreArray(ArrayFloatNear(expected))); in TEST()
443 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
164 EXPECT_THAT(m.GetOutput(), in TEST()
DEmbeddingLookup.cpp32 output_ = GetOutput(operation, operands, kOutputTensor); in EmbeddingLookup()
DLSTMTest.cpp183 const std::vector<float> &GetOutput() const { return Output_; } in GetOutput() function in android::nn::wrapper::LSTMOpModel
379 EXPECT_THAT(lstm.GetOutput(), ElementsAreArray(ArrayFloatNear(expected))); in TEST()
485 EXPECT_THAT(lstm.GetOutput(), ElementsAreArray(ArrayFloatNear(expected))); in TEST()
1106 EXPECT_THAT(lstm.GetOutput(), ElementsAreArray(ArrayFloatNear(expected))); in TEST()
DHashtableLookupTest.cpp150 const std::vector<float>& GetOutput() const { return Output_; } in GetOutput() function in android::nn::wrapper::HashtableLookupOpModel
175 EXPECT_THAT(m.GetOutput(), ElementsAreArray(ArrayFloatNear({ in TEST()
DRNNTest.cpp209 const std::vector<float>& GetOutput() const { return Output_; } in GetOutput() function in android::nn::wrapper::BasicRNNOpModel
331 EXPECT_THAT(rnn.GetOutput(), ElementsAreArray(ArrayFloatNear(expected))); in TEST()
DSVDF.cpp41 state_out_ = GetOutput(operation, operands, kStateOutTensor); in SVDF()
42 output_ = GetOutput(operation, operands, kOutputTensor); in SVDF()
DRNN.cpp40 hidden_state_out_ = GetOutput(operation, operands, kHiddenStateOutTensor); in RNN()
41 output_ = GetOutput(operation, operands, kOutputTensor); in RNN()
DLayerNormLSTMTest.cpp198 const std::vector<float>& GetOutput() const { return Output_; } in GetOutput() function in android::nn::wrapper::LayerNormLSTMOpModel
422 EXPECT_THAT(lstm.GetOutput(), ElementsAreArray(ArrayFloatNear(expected))); in TEST()
DMultinomial.cpp53 output_ = GetOutput(operation, operands, kOutputTensor); in Multinomial()
DQuantizedLSTM.cpp238 cellStateOut_ = GetOutput(operation, operands, kCellStateOutTensor); in QuantizedLSTMCell()
239 output_ = GetOutput(operation, operands, kOutputTensor); in QuantizedLSTMCell()
DLSHProjection.cpp37 output_ = GetOutput(operation, operands, kOutputTensor); in LSHProjection()
DLSTM.cpp115 output_state_out_ = GetOutput(operation, operands, kOutputStateOutTensor); in LSTMCell()
116 cell_state_out_ = GetOutput(operation, operands, kCellStateOutTensor); in LSTMCell()
117 output_ = GetOutput(operation, operands, kOutputTensor); in LSTMCell()
119 scratch_buffer_ = GetOutput(operation, operands, kScratchBufferTensor); in LSTMCell()
DBidirectionalSequenceLSTM.cpp154 fw_output_ = GetOutput(operation, operands, kFwOutputTensor); in BidirectionalSequenceLSTM()
156 bw_output_ = GetOutput(operation, operands, kBwOutputTensor); in BidirectionalSequenceLSTM()
/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.h267 inline RunTimeOperandInfo *GetOutput(const Operation &operation, in GetOutput() function