| /frameworks/ml/nn/common/operations/ |
| D | LSTM.cpp | 40 std::vector<RunTimeOperandInfo>& operands) { in LSTMCell() argument
41 input_ = GetInput(operation, operands, kInputTensor); in LSTMCell()
43 input_to_input_weights_ = GetInput(operation, operands, kInputToInputWeightsTensor); // optional in LSTMCell()
44 input_to_forget_weights_ = GetInput(operation, operands, kInputToForgetWeightsTensor); in LSTMCell()
45 input_to_cell_weights_ = GetInput(operation, operands, kInputToCellWeightsTensor); in LSTMCell()
46 input_to_output_weights_ = GetInput(operation, operands, kInputToOutputWeightsTensor); in LSTMCell()
49 GetInput(operation, operands, kRecurrentToInputWeightsTensor); // optional in LSTMCell()
50 recurrent_to_forget_weights_ = GetInput(operation, operands, kRecurrentToForgetWeightsTensor); in LSTMCell()
51 recurrent_to_cell_weights_ = GetInput(operation, operands, kRecurrentToCellWeightsTensor); in LSTMCell()
52 recurrent_to_output_weights_ = GetInput(operation, operands, kRecurrentToOutputWeightsTensor); in LSTMCell()
[all …]
|
| D | RNN.cpp | 26 std::vector<RunTimeOperandInfo>& operands) { in RNN() argument
27 input_ = GetInput(operation, operands, kInputTensor); in RNN()
28 weights_ = GetInput(operation, operands, kWeightsTensor); in RNN()
29 recurrent_weights_ = GetInput(operation, operands, kRecurrentWeightsTensor); in RNN()
30 hidden_state_in_ = GetInput(operation, operands, kHiddenStateInTensor); in RNN()
31 bias_ = GetInput(operation, operands, kBiasTensor); in RNN()
34 getScalarData<int32_t>(operands[operation.inputs[kActivationParam]])); in RNN()
36 hidden_state_out_ = GetOutput(operation, operands, kHiddenStateOutTensor); in RNN()
37 output_ = GetOutput(operation, operands, kOutputTensor); in RNN()
41 std::vector<RunTimeOperandInfo> &operands, in Prepare() argument
[all …]
|
| D | SVDF.cpp | 40 std::vector<RunTimeOperandInfo>& operands) { in SVDF() argument
41 input_ = GetInput(operation, operands, kInputTensor); in SVDF()
42 weights_feature_ = GetInput(operation, operands, kWeightsFeatureTensor); in SVDF()
43 weights_time_ = GetInput(operation, operands, kWeightsTimeTensor); in SVDF()
44 bias_ = GetInput(operation, operands, kBiasTensor); in SVDF()
45 state_in_ = GetInput(operation, operands, kStateInTensor); in SVDF()
47 params_.rank_ = getScalarData<int>(*GetInput(operation, operands, kRankParam)); in SVDF()
49 *GetInput(operation, operands, kActivationParam))); in SVDF()
51 state_out_ = GetOutput(operation, operands, kStateOutTensor); in SVDF()
52 output_ = GetOutput(operation, operands, kOutputTensor); in SVDF()
[all …]
|
| D | LSHProjection.cpp | 27 std::vector<RunTimeOperandInfo>& operands) { in LSHProjection() argument
28 input_ = GetInput(operation, operands, kInputTensor); in LSHProjection()
29 weight_ = GetInput(operation, operands, kWeightTensor); in LSHProjection()
30 hash_ = GetInput(operation, operands, kHashTensor); in LSHProjection()
33 getScalarData<int32_t>(*GetInput(operation, operands, kTypeParam))); in LSHProjection()
35 output_ = GetOutput(operation, operands, kOutputTensor); in LSHProjection()
39 std::vector<RunTimeOperandInfo>& operands, in Prepare() argument
41 const int num_inputs = NumInputsWithValues(operation, operands); in Prepare()
45 const RunTimeOperandInfo *hash = GetInput(operation, operands, kHashTensor); in Prepare()
50 const RunTimeOperandInfo* input = GetInput(operation, operands, kInputTensor); in Prepare()
[all …]
|
| D | HashtableLookup.cpp | 35 std::vector<RunTimeOperandInfo>& operands) { in HashtableLookup() argument
36 lookup_ = GetInput(operation, operands, kLookupTensor); in HashtableLookup()
37 key_ = GetInput(operation, operands, kKeyTensor); in HashtableLookup()
38 value_ = GetInput(operation, operands, kValueTensor); in HashtableLookup()
40 output_ = GetOutput(operation, operands, kOutputTensor); in HashtableLookup()
41 hits_ = GetOutput(operation, operands, kHitsTensor); in HashtableLookup()
|
| D | EmbeddingLookup.cpp | 27 std::vector<RunTimeOperandInfo>& operands) { in EmbeddingLookup() argument
28 value_ = GetInput(operation, operands, kValueTensor); in EmbeddingLookup()
29 lookup_ = GetInput(operation, operands, kLookupTensor); in EmbeddingLookup()
31 output_ = GetOutput(operation, operands, kOutputTensor); in EmbeddingLookup()
|
| /frameworks/ml/nn/common/ |
| D | Utils.cpp | 266 LOG(INFO) << "operands" << toString(model.operands); in logModelToInfo()
276 LOG(INFO) << "operands" << toString(model.operands); in logModelToInfo()
335 int validateOperationOperandTypes(const std::vector<Operand>& operands, in validateOperationOperandTypes() argument
345 if (operands[inOperandIndexes[i]].type != inExpectedTypes[i]) { in validateOperationOperandTypes()
347 << toString(operands[inOperandIndexes[i]].type) in validateOperationOperandTypes()
353 if (operands[outOperandIndexes[i]].type != outExpectedInTypes[i]) { in validateOperationOperandTypes()
355 << toString(operands[outOperandIndexes[i]].type) in validateOperationOperandTypes()
367 const std::vector<Operand>& operands) { in validateOperation() argument
368 int n = validateOperandList(inputCount, inputIndexes, static_cast<uint32_t>(operands.size()), in validateOperation()
373 n = validateOperandList(outputCount, outputIndexes, static_cast<uint32_t>(operands.size()), in validateOperation()
[all …]
|
| D | ValidateHal.cpp | 57 static bool validateOperands(const hidl_vec<Operand>& operands, in validateOperands() argument
62 for (auto& operand : operands) { in validateOperands()
310 const hidl_vec<Operand>& operands) { in validateOperations() argument
311 const size_t operandCount = operands.size(); in validateOperations()
326 op.outputs.size() > 0 ? op.outputs.data() : nullptr, operands); in validateOperations()
332 const Operand& operand = operands[i]; in validateOperations()
350 const Operand& operand = operands[i]; in validateOperations()
381 const hidl_vec<Operand>& operands, OperandLifeTime lifetime) { in validateModelInputOutputs() argument
382 const size_t operandCount = operands.size(); in validateModelInputOutputs()
388 const Operand& operand = operands[i]; in validateModelInputOutputs()
[all …]
|
| /frameworks/ml/nn/common/include/ |
| D | CpuExecutor.h | 191 std::vector<RunTimeOperandInfo> &operands) { in NumInputsWithValues() argument
194 [&operands](uint32_t i) { in NumInputsWithValues()
195 return !IsNullInput(&operands[i]); in NumInputsWithValues()
212 std::vector<RunTimeOperandInfo> &operands, in GetInput() argument
214 return &operands[operation.inputs[index]]; in GetInput()
218 std::vector<RunTimeOperandInfo> &operands, in GetOutput() argument
220 return &operands[operation.outputs[index]]; in GetOutput()
|
| /frameworks/rs/rsov/compiler/spirit/ |
| D | generate.py | 59 operands = inst.get('operands')
62 for type, var, quantifier, comment in generate_member_list(operands):
83 operands = inst.get('operands')
87 for type, var, quantifier, comment in generate_member_list(operands):
105 for type, var, quantifier, comment in generate_member_list(operands):
189 def generate_member_list(operands): argument
191 if operands is None:
194 for operand in operands:
377 operands = inst.get('operands')
378 members = generate_member_list(operands)
|
| /frameworks/ml/nn/runtime/test/generated/vts_models/ |
| D | floor.model.cpp | 4 const std::vector<Operand> operands = { in createTestModel() local
39 .operands = operands, in createTestModel()
|
| D | l2_normalization.model.cpp | 4 const std::vector<Operand> operands = { in createTestModel() local
39 .operands = operands, in createTestModel()
|
| D | l2_normalization_2.model.cpp | 4 const std::vector<Operand> operands = { in createTestModel() local
39 .operands = operands, in createTestModel()
|
| D | l2_normalization_2_relaxed.model.cpp | 4 const std::vector<Operand> operands = { in createTestModel() local
39 .operands = operands, in createTestModel()
|
| D | l2_normalization_large.model.cpp | 4 const std::vector<Operand> operands = { in createTestModel() local
39 .operands = operands, in createTestModel()
|
| D | dequantize_relaxed.model.cpp | 4 const std::vector<Operand> operands = { in createTestModel() local
39 .operands = operands, in createTestModel()
|
| D | l2_normalization_relaxed.model.cpp | 4 const std::vector<Operand> operands = { in createTestModel() local
39 .operands = operands, in createTestModel()
|
| D | logistic_float_1.model.cpp | 4 const std::vector<Operand> operands = { in createTestModel() local
39 .operands = operands, in createTestModel()
|
| D | floor_relaxed.model.cpp | 4 const std::vector<Operand> operands = { in createTestModel() local
39 .operands = operands, in createTestModel()
|
| D | l2_normalization_large_relaxed.model.cpp | 4 const std::vector<Operand> operands = { in createTestModel() local
39 .operands = operands, in createTestModel()
|
| D | logistic_float_2.model.cpp | 4 const std::vector<Operand> operands = { in createTestModel() local
39 .operands = operands, in createTestModel()
|
| D | logistic_float_2_relaxed.model.cpp | 4 const std::vector<Operand> operands = { in createTestModel() local
39 .operands = operands, in createTestModel()
|
| D | logistic_quant8_1.model.cpp | 4 const std::vector<Operand> operands = { in createTestModel() local
39 .operands = operands, in createTestModel()
|
| D | logistic_quant8_2.model.cpp | 4 const std::vector<Operand> operands = { in createTestModel() local
39 .operands = operands, in createTestModel()
|
| D | relu1_float_1.model.cpp | 4 const std::vector<Operand> operands = { in createTestModel() local
39 .operands = operands, in createTestModel()
|