Searched refs:kOutputRoiTensor (Results 1 – 3 of 3) sorted by relevance
| /packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | GenerateProposals.cpp | 533 Shape roiOutShape = context->getOutputShape(kOutputRoiTensor); in boxWithNmsLimitWriteOutput()
535 NN_RET_CHECK(context->setOutputShape(kOutputRoiTensor, roiOutShape)); in boxWithNmsLimitWriteOutput()
550 T_Roi* roiOutPtr = context->getOutputBuffer<T_Roi>(kOutputRoiTensor); in boxWithNmsLimitWriteOutput()
671 Shape outputRoiShape = context->getOutputShape(kOutputRoiTensor); in prepare()
708 NN_RET_CHECK(context->setOutputShape(kOutputRoiTensor, outputRoiShape)); in prepare()
741 context->getOutputBuffer<_Float16>(kOutputRoiTensor), in execute()
742 context->getOutputShape(kOutputRoiTensor), in execute()
763 context->getOutputBuffer<float>(kOutputRoiTensor), in execute()
764 context->getOutputShape(kOutputRoiTensor), in execute()
785 context->getOutputBuffer<uint16_t>(kOutputRoiTensor), in execute()
[all …]
|
| /packages/modules/NeuralNetworks/common/types/operations/include/ |
| D | GenerateProposals.h | 56 constexpr uint32_t kOutputRoiTensor = 1; variable
81 constexpr uint32_t kOutputRoiTensor = 1; variable
108 constexpr uint32_t kOutputRoiTensor = 1; variable
|
| /packages/modules/NeuralNetworks/runtime/test/fuzzing/operation_signatures/ |
| D | BoundingBox.cpp | 39 static const OperandSignature kOutputRoiTensor = {.type = RandomOperandType::OUTPUT, variable
237 .outputs = {OUTPUT_DEFAULT, kOutputRoiTensor}, \
|