| /packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | Multinomial.cpp | 62 input_ = GetInput(operation, operands, kInputTensor); in Multinomial()
92 switch (input_->type) { in Eval()
94 std::vector<float> inputDataFloat32(getNumberOfElements(input_->shape())); in Eval()
95 convertFloat16ToFloat32(GetBuffer<_Float16>(input_), &inputDataFloat32); in Eval()
100 EvalFloat32(GetBuffer<float>(input_)); in Eval()
104 LOG(ERROR) << "Unsupported data type: " << static_cast<int>(input_->type); in Eval()
112 const uint32_t batch_size = SizeOfDimension(input_, 0); in EvalFloat32()
113 const uint32_t class_size = SizeOfDimension(input_, 1); in EvalFloat32()
|
| 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 | SVDF.cpp | 35 input_ = GetInput(operation, operands, kInputTensor); in SVDF()
114 switch (input_->type) { in Eval()
116 std::vector<float> inputDataFloat32(getNumberOfElements(input_->shape())); in Eval()
117 convertFloat16ToFloat32(reinterpret_cast<_Float16*>(input_->buffer), &inputDataFloat32); in Eval()
147 EvalFloat32(reinterpret_cast<float*>(input_->buffer), in Eval()
157 LOG(ERROR) << "Unsupported data type: " << static_cast<int>(input_->type); in Eval()
170 const int batch_size = SizeOfDimension(input_, 0); in EvalFloat32()
171 const int input_size = SizeOfDimension(input_, 1); in EvalFloat32()
|
| D | MultinomialTest.cpp | 74 input_.push_back(srng.RandDouble()); in Invoke()
76 ASSERT_EQ(execution.setInput(Multinomial::kInputTensor, input_.data(), in Invoke()
77 sizeof(float) * input_.size()), in Invoke()
96 const std::vector<float>& GetInput() const { return input_; } in GetInput()
106 std::vector<float> input_; member in android::nn::wrapper::MultinomialOpModel
|
| 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 | LSTM.cpp | 55 input_ = GetInput(operation, operands, kInputTensor); in LSTMCell()
92 if (input_->type == OperandType::TENSOR_FLOAT32) { in LSTMCell()
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()
993 switch (input_->type) { in Eval()
995 LSTMEvalFloat32(params_, GetBuffer<const float>(input_), input_->shape(), in Eval()
[all …]
|
| D | QuantizedLSTM.cpp | 225 input_ = GetInput(operation, operands, kInputTensor); in QuantizedLSTMCell()
383 SizeOfDimension(input_, 1) + SizeOfDimension(prevOutput_, 1)}; in eval()
393 concatTempShape.dimensions = {SizeOfDimension(input_, 0), getSizeOfDimension(weightsShape, 1)}; in eval()
396 activationTempShape.dimensions = {SizeOfDimension(input_, 0), in eval()
416 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
|
| /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 __anon0e077fa70111::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()
|
| /packages/modules/NeuralNetworks/common/types/operations/include/ |
| D | LSTM.h | 184 const RunTimeOperandInfo* input_, const RunTimeOperandInfo* input_to_input_weights,
206 const RunTimeOperandInfo* input_; variable
|
| D | Multinomial.h | 50 RunTimeOperandInfo* input_; variable
|
| D | SVDF.h | 65 const RunTimeOperandInfo* input_; variable
|
| D | LSHProjection.h | 59 const RunTimeOperandInfo* input_; variable
|
| D | QuantizedLSTM.h | 65 const RunTimeOperandInfo* input_;
|
| D | RNN.h | 73 const RunTimeOperandInfo* input_; variable
|
| D | BidirectionalSequenceLSTM.h | 160 const RunTimeOperandInfo* input_; variable
|