Home
last modified time | relevance | path

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

/frameworks/ml/nn/common/operations/
DHashtableLookup.cpp40 output_ = GetOutput(operation, operands, kOutputTensor); in HashtableLookup()
41 hits_ = GetOutput(operation, operands, kHitsTensor); in HashtableLookup()
DLSHProjectionTest.cpp95 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()
DRNN.cpp36 hidden_state_out_ = GetOutput(operation, operands, kHiddenStateOutTensor); in RNN()
37 output_ = GetOutput(operation, operands, kOutputTensor); in RNN()
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()
DHashtableLookupTest.cpp148 const std::vector<float>& GetOutput() const { return Output_; } in GetOutput() function in android::nn::wrapper::HashtableLookupOpModel
173 EXPECT_THAT(m.GetOutput(), ElementsAreArray(ArrayFloatNear({ in TEST()
DSVDF.cpp56 state_out_ = GetOutput(operation, operands, kStateOutTensor); in SVDF()
57 output_ = GetOutput(operation, operands, kOutputTensor); in SVDF()
DEmbeddingLookup.cpp31 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
381 EXPECT_THAT(lstm.GetOutput(), ElementsAreArray(ArrayFloatNear(expected))); in TEST()
487 EXPECT_THAT(lstm.GetOutput(), ElementsAreArray(ArrayFloatNear(expected))); in TEST()
1108 EXPECT_THAT(lstm.GetOutput(), ElementsAreArray(ArrayFloatNear(expected))); in TEST()
DLSTM.cpp75 output_state_out_ = GetOutput(operation, operands, kOutputStateOutTensor); in LSTMCell()
76 cell_state_out_ = GetOutput(operation, operands, kCellStateOutTensor); in LSTMCell()
77 output_ = GetOutput(operation, operands, kOutputTensor); in LSTMCell()
79 scratch_buffer_ = GetOutput(operation, operands, kScratchBufferTensor); in LSTMCell()
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()
DSVDFTest.cpp251 const std::vector<float>& GetOutput() const { return Output_; } in GetOutput() function in android::nn::wrapper::SVDFOpModel
317 EXPECT_THAT(svdf.GetOutput(), ElementsAreArray(ArrayFloatNear(expected))); in TEST()
DLSHProjection.cpp35 output_ = GetOutput(operation, operands, kOutputTensor); in LSHProjection()
/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.h149 inline RunTimeOperandInfo *GetOutput(const Operation &operation, in GetOutput() function