| /frameworks/ml/nn/runtime/ |
| H A D | ModelBuilder.h | 45 uint32_t outputCount, const uint32_t* outputs);
46 int identifyInputsAndOutputs(uint32_t inputCount, const uint32_t* inputs, uint32_t outputCount,
67 uint32_t outputCount() const { return static_cast<uint32_t>(mOutputIndexes.size()); } function in class:android::nn::ModelBuilder
|
| H A D | ModelBuilder.cpp | 201 const uint32_t* inputs, uint32_t outputCount,
212 outputCount, outputs, mOperands);
226 .outputs = hidl_vec<uint32_t>(outputs, outputs + outputCount),
236 uint32_t outputCount, const uint32_t* outputs) {
246 n = validateOperandList(outputCount, outputs, operandCount(),
281 !setArguments(&mOutputIndexes, outputCount, outputs, OperandLifeTime::MODEL_OUTPUT)) {
200 addOperation(ANeuralNetworksOperationType type, uint32_t inputCount, const uint32_t* inputs, uint32_t outputCount, const uint32_t* outputs) argument
235 identifyInputsAndOutputs(uint32_t inputCount, const uint32_t* inputs, uint32_t outputCount, const uint32_t* outputs) argument
|
| H A D | NeuralNetworks.cpp | 337 const uint32_t* inputs, uint32_t outputCount,
344 return m->addOperation(type, inputCount, inputs, outputCount, outputs);
348 const uint32_t* inputs, uint32_t outputCount,
355 return m->identifyInputsAndOutputs(inputCount, inputs, outputCount, outputs);
335 ANeuralNetworksModel_addOperation(ANeuralNetworksModel* model, ANeuralNetworksOperationType type, uint32_t inputCount, const uint32_t* inputs, uint32_t outputCount, const uint32_t* outputs) argument
347 ANeuralNetworksModel_identifyInputsAndOutputs(ANeuralNetworksModel* model, uint32_t inputCount, const uint32_t* inputs, uint32_t outputCount, const uint32_t* outputs) argument
|
| H A D | ExecutionPlan.cpp | 241 const uint32_t outputCount = static_cast<uint32_t>(operation.outputs.size()); local
243 std::vector<uint32_t> outputs(outputCount);
266 outputCount, outputs.data());
377 fromModel->outputCount(),
410 for (unsigned i = 0, e = fromModel->outputCount(); i < e; ++i) {
|
| /frameworks/ml/nn/common/ |
| H A D | Utils.cpp | 366 uint32_t outputCount, const uint32_t* outputIndexes,
373 n = validateOperandList(outputCount, outputIndexes, static_cast<uint32_t>(operands.size()),
379 auto logInvalidInOutNumber = [opType, inputCount, outputCount](int expIn, int expOut) {
382 << outputCount << ", expected " << expOut << ") for operation "
391 if (inputCount != 3 || outputCount != 1) {
416 outputCount, outputIndexes,
420 if (inputCount != 3 || outputCount != 1) {
445 outputCount, outputIndexes,
449 if (inputCount != 1 || outputCount != 1) {
467 outputCount, outputIndexe
364 validateOperation(ANeuralNetworksOperationType opType, uint32_t inputCount, const uint32_t* inputIndexes, uint32_t outputCount, const uint32_t* outputIndexes, const std::vector<Operand>& operands) argument
[all...] |
| /frameworks/ml/nn/runtime/test/ |
| H A D | TestPartitioningRandom.cpp | 168 uint32_t outputCount() const { function in class:android::__anon1348::TestModel
169 return builder()->outputCount();
417 const uint32_t outputCount = stats.mBuilder->outputCount(); local
422 << ", outputCount = " << outputCount
423 << " (" << (double(outputCount) / operandCount) << ")";
848 for (uint32_t outputIdx = 0, outputCount = outputs.size(); outputIdx < outputCount;
1038 std::vector<InputOutputDescriptor> ioDescriptors(model.inputCount() + model.outputCount());
[all...] |
| /frameworks/base/core/jni/android/opengl/ |
| H A D | util.cpp | 471 int outputCount; local
493 outputCount = 0;
496 if (outputCount < resultsCapacity) {
499 outputCount++;
503 return outputCount;
|