Searched refs:sizeOfElement (Results 1 – 2 of 2) sorted by relevance
| /frameworks/ml/nn/common/ |
| D | Utils.cpp | 365 uint32_t sizeOfElement = tableLookup(kSizeOfDataType, kSizeOfDataTypeOEM, n); in nonExtensionOperandSizeOfData() local
367 ? sizeOfElement in nonExtensionOperandSizeOfData()
368 : sizeOfTensorData(sizeOfElement, dimensions); in nonExtensionOperandSizeOfData()
372 static std::pair<bool, uint32_t> sizeOfTensorDataHelper(uint32_t sizeOfElement, in sizeOfTensorDataHelper() argument
377 uint64_t size = static_cast<uint64_t>(sizeOfElement); in sizeOfTensorDataHelper()
386 uint32_t sizeOfTensorData(uint32_t sizeOfElement, const std::vector<uint32_t>& dimensions) { in sizeOfTensorData() argument
387 const auto [overflow, size] = sizeOfTensorDataHelper(sizeOfElement, dimensions); in sizeOfTensorData()
396 uint32_t sizeOfElement = tableLookup(kSizeOfDataType, kSizeOfDataTypeOEM, n); in nonExtensionOperandSizeOfDataOverflowsUInt32() local
399 : sizeOfTensorDataOverflowsUInt32(sizeOfElement, dimensions); in nonExtensionOperandSizeOfDataOverflowsUInt32()
402 bool sizeOfTensorDataOverflowsUInt32(uint32_t sizeOfElement, in sizeOfTensorDataOverflowsUInt32() argument
[all …]
|
| /frameworks/ml/nn/common/include/ |
| D | Utils.h | 278 uint32_t sizeOfTensorData(uint32_t sizeOfElement, const std::vector<uint32_t>& dimensions);
|