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.cpp99 const std::vector<int> &GetOutput() const { return Output_; } in GetOutput() function in android::nn::wrapper::LSHProjectionOpModel
152 EXPECT_THAT(m.GetOutput(), ElementsAre(1, 1, 1, 0, 1, 1, 1, 0)); in TEST()
163 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()
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()
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()
DEmbeddingLookup.cpp31 output_ = GetOutput(operation, operands, kOutputTensor); in EmbeddingLookup()
DSVDF.cpp51 state_out_ = GetOutput(operation, operands, kStateOutTensor); in SVDF()
52 output_ = GetOutput(operation, operands, kOutputTensor); in SVDF()
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()
DLSTM.cpp74 output_state_out_ = GetOutput(operation, operands, kOutputStateOutTensor); in LSTMCell()
75 cell_state_out_ = GetOutput(operation, operands, kCellStateOutTensor); in LSTMCell()
76 output_ = GetOutput(operation, operands, kOutputTensor); in LSTMCell()
78 scratch_buffer_ = GetOutput(operation, operands, kScratchBufferTensor); in LSTMCell()
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.h217 inline RunTimeOperandInfo *GetOutput(const Operation &operation, in GetOutput() function