Home
last modified time | relevance | path

Searched defs:tensorSize (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/armnn/delegate/test/
H A DTestUtils.cpp11 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 Dandroid_neuralnetworks_cts_ModelAssetTest.cpp159 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 DShape.cu131 IndexType tensorSize[MaxDims]; member
/aosp_15_r20/external/pytorch/torch/csrc/jit/serialization/
H A Dexport.cpp1318 size_t tensorSize = static_cast<size_t>(c10::multiply_integers( in EncodeTensor() local
/aosp_15_r20/external/armnn/src/armnnDeserializer/
H A DDeserializer.cpp161 const unsigned int tensorSize, in CheckConstTensorSize()