Searched refs:nonExtensionOperandSizeOfData (Results 1 – 13 of 13) sorted by relevance
| /frameworks/ml/nn/common/operations/ |
| D | ExpandDims.cpp | 43 nonExtensionOperandSizeOfData(inputShape.type, inputShape.dimensions)); in eval()
|
| D | EmbeddingLookup.cpp | 42 const int total_bytes = nonExtensionOperandSizeOfData(value_->type, value_->dimensions); in Eval()
|
| D | HashtableLookup.cpp | 53 nonExtensionOperandSizeOfData(value_->type, value_->dimensions) / num_rows; in Eval()
|
| D | LSHProjection.cpp | 99 int input_item_bytes = nonExtensionOperandSizeOfData(input->type, input->dimensions) / in runningSignBit()
|
| D | Reshape.cpp | 37 size_t count = nonExtensionOperandSizeOfData(inputShape.type, inputShape.dimensions); in copyData()
|
| /frameworks/ml/nn/common/include/ |
| D | Utils.h | 256 uint32_t nonExtensionOperandSizeOfData(hal::OperandType type,
267 inline uint32_t nonExtensionOperandSizeOfData(const hal::Operand& operand) {
268 return nonExtensionOperandSizeOfData(operand.type, operand.dimensions);
|
| D | CpuExecutor.h | 91 return length >= nonExtensionOperandSizeOfData(type, dimensions); in isSufficient()
|
| /frameworks/ml/nn/runtime/test/android_fuzzing/ |
| D | Converter.cpp | 86 const size_t bufferSize = makeEmpty ? 0 : nonExtensionOperandSizeOfData(halType, dimensions); in convert()
|
| /frameworks/ml/nn/runtime/ |
| D | TypeManager.cpp | 282 return nonExtensionOperandSizeOfData(type, dimensions); in getSizeOfData()
|
| /frameworks/ml/nn/common/ |
| D | CpuExecutor.cpp | 209 uint32_t length = nonExtensionOperandSizeOfData(info->type, info->dimensions); in setInfoAndAllocateIfNeeded()
220 uint32_t length = nonExtensionOperandSizeOfData(info->type, info->dimensions); in setInfoAndAllocateIfNeeded()
1826 bodyOutputHasUnknownShape[i] = nonExtensionOperandSizeOfData(operand) == 0; in executeWhileOperation()
|
| D | ValidateHal.cpp | 417 uint32_t expectedLength = nonExtensionOperandSizeOfData(operand); in validateOperands()
|
| D | Utils.cpp | 362 uint32_t nonExtensionOperandSizeOfData(OperandType type, const std::vector<uint32_t>& dimensions) { in nonExtensionOperandSizeOfData() function
769 NN_RET_CHECK_NE(nonExtensionOperandSizeOfData(operand.type, operand.dimensions), 0u); in validateControlFlowOperandUnknownSize()
|
| /frameworks/ml/nn/driver/sample/ |
| D | SampleDriver.cpp | 272 uint32_t size = nonExtensionOperandSizeOfData(operand.type, operand.dimensions); in allocate()
|