Searched refs:expectedOutputTensor (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/android-nn-driver/test/1.1/ |
| H A D | Transpose.cpp | 27 const TestTensor & expectedOutputTensor, armnn::Compute computeDevice) in TransposeTestImpl() argument
39 AddOutputOperand<HalPolicy>(model, expectedOutputTensor.GetDimensions()); in TransposeTestImpl()
62 outloc.length = expectedOutputTensor.GetNumElements() * sizeof(float); in TransposeTestImpl()
65 output.dimensions = expectedOutputTensor.GetDimensions(); in TransposeTestImpl()
78 …android::sp<IMemory> outMemory = AddPoolAndGetData<float>(expectedOutputTensor.GetNumElements(), r… in TransposeTestImpl()
86 const float * expectedOutput = expectedOutputTensor.GetData(); in TransposeTestImpl()
87 for (unsigned int i = 0; i < expectedOutputTensor.GetNumElements(); ++i) in TransposeTestImpl()
|
| /aosp_15_r20/external/android-nn-driver/test/ |
| H A D | Concat.cpp | 25 const TestTensor & expectedOutputTensor, in ConcatTestImpl() argument
42 AddOutputOperand<HalPolicy>(model, expectedOutputTensor.GetDimensions()); in ConcatTestImpl()
95 outloc.length = expectedOutputTensor.GetNumElements() * sizeof(float); in ConcatTestImpl()
98 output.dimensions = expectedOutputTensor.GetDimensions(); in ConcatTestImpl()
116 …android::sp<IMemory> outMemory = AddPoolAndGetData<float>(expectedOutputTensor.GetNumElements(), r… in ConcatTestImpl()
127 const float * expectedOutput = expectedOutputTensor.GetData(); in ConcatTestImpl()
128 for (unsigned int i=0; i<expectedOutputTensor.GetNumElements();++i) in ConcatTestImpl()
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_unary_ufuncs.py | 980 expectedOutputTensor = torch.tensor(expectedOutput, dtype=dtype, device=device)
984 torch.nn.functional.hardswish(inputTensor), expectedOutputTensor
990 self.assertEqual(inputTensorCpy, expectedOutputTensor)
|