/frameworks/ml/nn/common/operations/ |
D | MultinomialTest.cpp | 83 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
|
D | HashtableLookup.cpp | 45 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()
|
D | RNN.cpp | 46 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()
|
D | EmbeddingLookup.cpp | 36 output_ = GetOutput(operation, operands, kOutputTensor); in EmbeddingLookup() 51 memcpy(output_->buffer + i * row_bytes, value_->buffer + idx * row_bytes, row_bytes); in Eval()
|
D | SVDF.cpp | 48 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()
|
D | QuantizedLSTMTest.cpp | 83 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
|
D | Multinomial.cpp | 60 output_ = GetOutput(operation, operands, kOutputTensor); in Multinomial() 140 auto* output_ptr_batch = GetBuffer<int32_t>(output_) + b * sample_count_; in EvalFloat32()
|
D | EmbeddingLookup.h | 44 RunTimeOperandInfo* output_; variable
|
D | HashtableLookup.h | 47 RunTimeOperandInfo* output_; variable
|
D | Multinomial.h | 55 RunTimeOperandInfo* output_; variable
|
D | SVDF.h | 72 RunTimeOperandInfo* output_; variable
|
D | LSHProjection.cpp | 42 output_ = GetOutput(operation, operands, kOutputTensor); in LSHProjection() 164 int32_t* out_buf = reinterpret_cast<int32_t*>(output_->buffer); in Eval()
|
D | LSHProjection.h | 61 RunTimeOperandInfo* output_; variable
|
D | QuantizedLSTM.h | 86 RunTimeOperandInfo* output_; variable
|
D | RNN.h | 74 RunTimeOperandInfo* output_; variable
|
D | QuantizedLSTM.cpp | 245 output_ = GetOutput(operation, operands, kOutputTensor); in QuantizedLSTMCell() 422 GetBuffer<uint8_t>(output_), convertShapeToDims(output_->shape()), concatTemp.data(), in eval()
|
D | LSTM.h | 241 RunTimeOperandInfo* output_; variable
|
D | LSTM.cpp | 126 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/ |
D | dumpsys_test.cpp | 75 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
|