Home
last modified time | relevance | path

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

/frameworks/ml/nn/common/operations/
DMultinomialTest.cpp83 output_.insert(output_.end(), batch_size_ * sample_size_, 0); in Invoke()
84 ASSERT_EQ(execution.setOutput(Multinomial::kOutputTensor, output_.data(), in Invoke()
85 sizeof(uint32_t) * output_.size()), in Invoke()
92 const std::vector<uint32_t>& GetOutput() const { return output_; } in GetOutput()
102 std::vector<uint32_t> output_; member in android::nn::wrapper::MultinomialOpModel
DHashtableLookup.cpp45 output_ = GetOutput(operation, operands, kOutputTensor); in HashtableLookup()
65 memset(output_->buffer + i * row_bytes, 0, row_bytes); in Eval()
68 memcpy(output_->buffer + i * row_bytes, value_->buffer + idx * row_bytes, row_bytes); in Eval()
DRNN.cpp46 output_ = GetOutput(operation, operands, kOutputTensor); in RNN()
94 reinterpret_cast<_Float16*>(output_->buffer)); in Eval()
95 memcpy(hidden_state_out_->buffer, output_->buffer, in Eval()
96 sizeof(_Float16) * getNumberOfElements(output_->shape())); in Eval()
106 reinterpret_cast<float*>(output_->buffer)); in Eval()
107 memcpy(hidden_state_out_->buffer, output_->buffer, in Eval()
108 sizeof(float) * getNumberOfElements(output_->shape())); in Eval()
DEmbeddingLookup.cpp36 output_ = GetOutput(operation, operands, kOutputTensor); in EmbeddingLookup()
51 memcpy(output_->buffer + i * row_bytes, value_->buffer + idx * row_bytes, row_bytes); in Eval()
DSVDF.cpp48 output_ = GetOutput(operation, operands, kOutputTensor); in SVDF()
133 std::vector<float> outputDataFloat32(getNumberOfElements(output_->shape())); in Eval()
141 reinterpret_cast<_Float16*>(output_->buffer)); in Eval()
152 reinterpret_cast<float*>(output_->buffer), in Eval()
DQuantizedLSTMTest.cpp83 output_.resize(numBatches * outputSize, 0); in QuantizedLSTMOpModel()
142 ASSERT_EQ(setOutputTensor(&execution, QuantizedLSTMCell::kOutputTensor, &output_), in invoke()
148 prevOutput_ = output_; in invoke()
194 std::vector<uint8_t> getOutput() { return output_; } in getOutput()
219 std::vector<uint8_t> output_; member in android::nn::wrapper::QuantizedLSTMOpModel
DMultinomial.cpp60 output_ = GetOutput(operation, operands, kOutputTensor); in Multinomial()
140 auto* output_ptr_batch = GetBuffer<int32_t>(output_) + b * sample_count_; in EvalFloat32()
DEmbeddingLookup.h44 RunTimeOperandInfo* output_; variable
DHashtableLookup.h47 RunTimeOperandInfo* output_; variable
DMultinomial.h55 RunTimeOperandInfo* output_; variable
DSVDF.h72 RunTimeOperandInfo* output_; variable
DLSHProjection.cpp42 output_ = GetOutput(operation, operands, kOutputTensor); in LSHProjection()
164 int32_t* out_buf = reinterpret_cast<int32_t*>(output_->buffer); in Eval()
DLSHProjection.h61 RunTimeOperandInfo* output_; variable
DQuantizedLSTM.h86 RunTimeOperandInfo* output_; variable
DRNN.h74 RunTimeOperandInfo* output_; variable
DQuantizedLSTM.cpp245 output_ = GetOutput(operation, operands, kOutputTensor); in QuantizedLSTMCell()
422 GetBuffer<uint8_t>(output_), convertShapeToDims(output_->shape()), concatTemp.data(), in eval()
DLSTM.h241 RunTimeOperandInfo* output_; variable
DLSTM.cpp126 output_ = GetOutput(operation, operands, kOutputTensor); in LSTMCell()
1029 GetBuffer<float>(output_), GetBuffer<float>(scratch_buffer_)); in Eval()
1064 GetBuffer<_Float16>(cell_state_out_), GetBuffer<_Float16>(output_), in Eval()
/frameworks/native/cmds/dumpsys/tests/
Ddumpsys_test.cpp75 explicit WriteOnFdAction(const std::string& output) : output_(output) { in WriteOnFdAction()
79 android::base::WriteStringToFd(output_, fd); in Perform()
83 std::string output_; member in WriteOnFdAction