| /packages/modules/NeuralNetworks/common/operations/ |
| D | BidirectionalSequenceLSTM.cpp | 82 input_ = GetInput(operation, operands, kInputTensor); in BidirectionalSequenceLSTM()
176 if (input_->type == OperandType::TENSOR_FLOAT32) { in BidirectionalSequenceLSTM()
253 if (input_->type == OperandType::TENSOR_FLOAT32) { in Prepare()
267 NN_CHECK(NumDimensions(input_) == 3); in Prepare()
268 const uint32_t max_time = SizeOfDimension(input_, params_.time_major ? 0 : 1); in Prepare()
269 const uint32_t n_batch = SizeOfDimension(input_, params_.time_major ? 1 : 0); in Prepare()
270 const uint32_t n_fw_input = SizeOfDimension(input_, 2); in Prepare()
288 input_, fw_input_to_input_weights_, fw_input_to_forget_weights_, in Prepare()
336 NN_CHECK_EQ(aux_input_->shape().dimensions[0], input_->shape().dimensions[0]); in Prepare()
337 NN_CHECK_EQ(aux_input_->shape().dimensions[1], input_->shape().dimensions[1]); in Prepare()
[all …]
|
| D | Multinomial.cpp | 58 input_ = GetInput(operation, operands, kInputTensor); in Multinomial()
88 switch (input_->type) { in Eval()
90 std::vector<float> inputDataFloat32(getNumberOfElements(input_->shape())); in Eval()
91 convertFloat16ToFloat32(GetBuffer<_Float16>(input_), &inputDataFloat32); in Eval()
96 EvalFloat32(GetBuffer<float>(input_)); in Eval()
100 LOG(ERROR) << "Unsupported data type: " << static_cast<int>(input_->type); in Eval()
108 const int batch_size = SizeOfDimension(input_, 0); in EvalFloat32()
109 const int class_size = SizeOfDimension(input_, 1); in EvalFloat32()
|
| D | SVDF.cpp | 33 input_ = GetInput(operation, operands, kInputTensor); in SVDF()
112 switch (input_->type) { in Eval()
114 std::vector<float> inputDataFloat32(getNumberOfElements(input_->shape())); in Eval()
115 convertFloat16ToFloat32(reinterpret_cast<_Float16*>(input_->buffer), &inputDataFloat32); in Eval()
145 EvalFloat32(reinterpret_cast<float*>(input_->buffer), in Eval()
155 LOG(ERROR) << "Unsupported data type: " << static_cast<int>(input_->type); in Eval()
168 const int batch_size = SizeOfDimension(input_, 0); in EvalFloat32()
169 const int input_size = SizeOfDimension(input_, 1); in EvalFloat32()
|
| D | RNN.cpp | 32 input_ = GetInput(operation, operands, kInputTensor); in RNN()
82 switch (input_->type) { in Eval()
84 RNNStep<_Float16>(reinterpret_cast<_Float16*>(input_->buffer), input_->shape(), in Eval()
96 RNNStep<float>(reinterpret_cast<float*>(input_->buffer), input_->shape(), in Eval()
108 LOG(ERROR) << "Unsupported data type: " << static_cast<int>(input_->type); in Eval()
|
| D | MultinomialTest.cpp | 69 input_.push_back(srng.RandDouble()); in Invoke()
71 ASSERT_EQ(execution.setInput(Multinomial::kInputTensor, input_.data(), in Invoke()
72 sizeof(float) * input_.size()), in Invoke()
91 const std::vector<float>& GetInput() const { return input_; } in GetInput()
101 std::vector<float> input_; member in android::nn::wrapper::MultinomialOpModel
|
| D | LSTM.cpp | 55 input_ = GetInput(operation, operands, kInputTensor); in LSTMCell()
92 if (input_->type == OperandType::TENSOR_FLOAT32) { in LSTMCell()
133 const RunTimeOperandInfo* input_, const RunTimeOperandInfo* input_to_input_weights, in CheckInputTensorDimensions() argument
337 if (input_->type == OperandType::TENSOR_FLOAT32) { in Prepare()
347 NN_CHECK(NumDimensions(input_) > 1); in Prepare()
348 const uint32_t n_batch = SizeOfDimension(input_, 0); in Prepare()
349 const uint32_t n_input = SizeOfDimension(input_, 1); in Prepare()
361 input_, input_to_input_weights_, input_to_forget_weights_, input_to_cell_weights_, in Prepare()
373 const Shape& inputShape = input_->shape(); in Prepare()
991 switch (input_->type) { in Eval()
[all …]
|
| D | QuantizedLSTM.cpp | 221 input_ = GetInput(operation, operands, kInputTensor); in QuantizedLSTMCell()
379 SizeOfDimension(input_, 1) + SizeOfDimension(prevOutput_, 1)}; in eval()
389 concatTempShape.dimensions = {SizeOfDimension(input_, 0), getSizeOfDimension(weightsShape, 1)}; in eval()
392 activationTempShape.dimensions = {SizeOfDimension(input_, 0), in eval()
412 GetBuffer<const uint8_t>(input_), convertShapeToDims(input_->shape()), in eval()
|
| D | LSHProjection.cpp | 34 input_ = GetInput(operation, operands, kInputTensor); in LSHProjection()
167 DenseLshProjection<T>(hash_, input_, weight_, out_buf); in Eval()
171 SparseLshProjection<T>(type_, hash_, input_, weight_, out_buf); in Eval()
|
| D | QuantizedLSTMTest.cpp | 77 initializeInputData(inputOperandTypeParams[QuantizedLSTMCell::kInputTensor], &input_); in QuantizedLSTMOpModel()
97 ASSERT_EQ(setInputTensor(&execution, QuantizedLSTMCell::kInputTensor, input_), in invoke()
157 void setInput(const std::vector<uint8_t>& input) { input_ = input; } in setInput()
203 std::vector<uint8_t> input_; member in android::nn::wrapper::QuantizedLSTMOpModel
|
| D | LSTM.h | 185 const RunTimeOperandInfo* input_, const RunTimeOperandInfo* input_to_input_weights,
207 const RunTimeOperandInfo* input_; variable
|
| D | Multinomial.h | 49 RunTimeOperandInfo* input_; variable
|
| D | SVDF.h | 65 const RunTimeOperandInfo* input_; variable
|
| D | LSHProjection.h | 58 const RunTimeOperandInfo* input_; variable
|
| D | QuantizedLSTM.h | 65 const RunTimeOperandInfo* input_;
|
| D | RNN.h | 67 const RunTimeOperandInfo* input_; variable
|
| D | BidirectionalSequenceLSTM.h | 160 const RunTimeOperandInfo* input_; variable
|
| /packages/modules/NeuralNetworks/runtime/test/specs/V1_2/ |
| D | quantized_lstm.mod.py | 26 input_ = Input("input", InputType) variable
61 input_,
79 input_: [166, 179, 50, 150],
113 input_ = Input("input", InputType) variable
158 model = model.Operation("QUANTIZED_16BIT_LSTM", input_, input_to_input_weights,
168 input_: [166, 179],
|
| /packages/modules/adb/daemon/ |
| D | shell_service.cpp | 210 std::unique_ptr<ShellProtocol> input_, output_; member in __anon543b8fee0111::Subprocess
477 input_ = std::make_unique<ShellProtocol>(protocol_sfd_); in ConnectProtocolEndpoints()
479 if (!input_ || !output_) { in ConnectProtocolEndpoints()
674 if (!input_->Read()) { in PassInput()
683 switch (input_->id()) { in PassInput()
686 if (sscanf(input_->data(), "%dx%d,%dx%d", in PassInput()
697 input_bytes_left_ = input_->data_length(); in PassInput()
721 int index = input_->data_length() - input_bytes_left_; in PassInput()
722 int bytes = adb_write(stdinout_sfd_, input_->data() + index, input_bytes_left_); in PassInput()
|