Searched defs:tensorSize (Results 1 – 5 of 5) sorted by relevance
/aosp_15_r20/external/armnn/delegate/test/ |
H A D | TestUtils.cpp | 11 void CompareData(bool tensor1[], bool tensor2[], size_t tensorSize) in CompareData() 20 void CompareData(std::vector<bool>& tensor1, std::vector<bool>& tensor2, size_t tensorSize) in CompareData() 29 void CompareData(float tensor1[], float tensor2[], size_t tensorSize) in CompareData() 37 void CompareData(float tensor1[], float tensor2[], size_t tensorSize, float percentTolerance) in CompareData() 46 void CompareData(uint8_t tensor1[], uint8_t tensor2[], size_t tensorSize) in CompareData() 55 void CompareData(int16_t tensor1[], int16_t tensor2[], size_t tensorSize) in CompareData() 64 void CompareData(int32_t tensor1[], int32_t tensor2[], size_t tensorSize) in CompareData() 73 void CompareData(int8_t tensor1[], int8_t tensor2[], size_t tensorSize) in CompareData() 82 void CompareData(Half tensor1[], Half tensor2[], size_t tensorSize) in CompareData() 90 void CompareData(TfLiteFloat16 tensor1[], TfLiteFloat16 tensor2[], size_t tensorSize) in CompareData() [all …]
|
/aosp_15_r20/cts/tests/tests/neuralnetworks/java_test/jni/ |
H A D | android_neuralnetworks_cts_ModelAssetTest.cpp | 159 const uint32_t tensorSize = 1 * 256 * 256 * 4 * kDataTypeToElementSizeMap.at(dataType); in createAndCompileAddModel() local
|
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
H A D | Shape.cu | 131 IndexType tensorSize[MaxDims]; member
|
/aosp_15_r20/external/pytorch/torch/csrc/jit/serialization/ |
H A D | export.cpp | 1318 size_t tensorSize = static_cast<size_t>(c10::multiply_integers( in EncodeTensor() local
|
/aosp_15_r20/external/armnn/src/armnnDeserializer/ |
H A D | Deserializer.cpp | 161 const unsigned int tensorSize, in CheckConstTensorSize()
|