16 template<
typename Workload>
17 void CheckInputOutput(std::unique_ptr<Workload> workload,
const TensorInfo& inputInfo,
const TensorInfo& outputInfo)
19 auto queueDescriptor = workload->GetData();
20 auto inputHandle = PolymorphicDowncast<RefTensorHandle*>(queueDescriptor.m_Inputs[0]);
21 auto outputHandle = PolymorphicDowncast<RefTensorHandle*>(queueDescriptor.m_Outputs[0]);
22 BOOST_TEST((inputHandle->GetTensorInfo() == inputInfo));
23 BOOST_TEST((outputHandle->GetTensorInfo() == outputInfo));
26 template <
typename Workload>
27 void CheckInputsOutput(std::unique_ptr<Workload> workload,
32 auto queueDescriptor = workload->GetData();
33 auto inputHandle0 = PolymorphicDowncast<RefTensorHandle*>(queueDescriptor.m_Inputs[0]);
34 auto inputHandle1 = PolymorphicDowncast<RefTensorHandle*>(queueDescriptor.m_Inputs[1]);
35 auto outputHandle = PolymorphicDowncast<RefTensorHandle*>(queueDescriptor.m_Outputs[0]);
36 BOOST_TEST((inputHandle0->GetTensorInfo() == inputInfo0));
37 BOOST_TEST((inputHandle1->GetTensorInfo() == inputInfo1));
38 BOOST_TEST((outputHandle->GetTensorInfo() == outputInfo));
43 std::shared_ptr<RefMemoryManager> memoryManager = std::make_shared<RefMemoryManager>();
52 template <
typename ActivationWorkloadType, armnn::DataType DataType>
53 static void RefCreateActivationWorkloadTest()
57 auto workload = CreateActivationWorkloadTest<ActivationWorkloadType, DataType>(factory, graph);
60 CheckInputOutput(std::move(workload),
67 RefCreateActivationWorkloadTest<RefActivationWorkload, armnn::DataType::Float32>();
72 RefCreateActivationWorkloadTest<RefActivationWorkload, armnn::DataType::QAsymmU8>();
75 template <
typename WorkloadType,
76 typename DescriptorType,
79 static void RefCreateElementwiseWorkloadTest()
83 auto workload = CreateElementwiseWorkloadTest<WorkloadType, DescriptorType, LayerType, DataType>(
86 CheckInputsOutput(std::move(workload),
204 template <
typename BatchNormalizationWorkloadType, armnn::DataType DataType>
205 static void RefCreateBatchNormalizationWorkloadTest(
DataLayout dataLayout)
209 auto workload = CreateBatchNormalizationWorkloadTest<BatchNormalizationWorkloadType, DataType>(factory,
218 case DataLayout::NHWC:
219 inputShape = { 2, 4, 4, 3 };
220 outputShape = { 2, 4, 4, 3 };
222 case DataLayout::NCHW:
224 inputShape = { 2, 3, 4, 4 };
225 outputShape = { 2, 3, 4, 4 };
235 RefCreateBatchNormalizationWorkloadTest<RefBatchNormalizationWorkload,armnn::DataType::Float32>
241 RefCreateBatchNormalizationWorkloadTest<RefBatchNormalizationWorkload, armnn::DataType::Float32>
247 RefCreateBatchNormalizationWorkloadTest<RefBatchNormalizationWorkload,armnn::DataType::Float16>
253 RefCreateBatchNormalizationWorkloadTest<RefBatchNormalizationWorkload, armnn::DataType::Float16>
259 RefCreateBatchNormalizationWorkloadTest<RefBatchNormalizationWorkload, armnn::DataType::QAsymmU8>
265 RefCreateBatchNormalizationWorkloadTest<RefBatchNormalizationWorkload, armnn::DataType::QAsymmU8>
271 RefCreateBatchNormalizationWorkloadTest<RefBatchNormalizationWorkload, armnn::DataType::QSymmS16>
277 RefCreateBatchNormalizationWorkloadTest<RefBatchNormalizationWorkload, armnn::DataType::QSymmS16>
285 auto workload = CreateConvertFp16ToFp32WorkloadTest<RefConvertFp16ToFp32Workload>(factory, graph);
289 std::move(workload),
TensorInfo({1, 3, 2, 3}, DataType::Float16),
TensorInfo({1, 3, 2, 3}, DataType::Float32));
296 auto workload = CreateConvertFp32ToFp16WorkloadTest<RefConvertFp32ToFp16Workload>(factory, graph);
300 std::move(workload),
TensorInfo({1, 3, 2, 3}, DataType::Float32),
TensorInfo({1, 3, 2, 3}, DataType::Float16));
303 static void RefCreateConvolution2dWorkloadTest(
DataLayout dataLayout = DataLayout::NCHW)
307 auto workload = CreateConvolution2dWorkloadTest<RefConvolution2dWorkload, DataType::Float32>
308 (factory, graph, dataLayout);
310 TensorShape inputShape = (dataLayout == DataLayout::NCHW) ? std::initializer_list<unsigned int>({2, 3, 8, 16})
311 : std::initializer_list<unsigned int>({2, 8, 16, 3});
312 TensorShape outputShape = (dataLayout == DataLayout::NCHW) ? std::initializer_list<unsigned int>({2, 2, 2, 10})
313 : std::initializer_list<unsigned int>({2, 2, 10, 2});
316 CheckInputOutput(std::move(workload),
323 RefCreateConvolution2dWorkloadTest(DataLayout::NCHW);
328 RefCreateConvolution2dWorkloadTest(DataLayout::NHWC);
331 static void RefCreateDepthwiseConvolutionWorkloadTest(
DataLayout dataLayout)
335 auto workload = CreateDepthwiseConvolution2dWorkloadTest<RefDepthwiseConvolution2dWorkload, DataType::Float32>
336 (factory, graph, dataLayout);
338 TensorShape inputShape = (dataLayout == DataLayout::NCHW) ? std::initializer_list<unsigned int>({ 2, 2, 5, 5 })
339 : std::initializer_list<unsigned int>({ 2, 5, 5, 2 });
340 TensorShape outputShape = (dataLayout == DataLayout::NCHW) ? std::initializer_list<unsigned int>({ 2, 2, 5, 5 })
341 : std::initializer_list<unsigned int>({ 2, 5, 5, 2 });
344 CheckInputOutput(std::move(workload),
351 RefCreateDepthwiseConvolutionWorkloadTest(DataLayout::NHWC);
354 template <
typename FullyConnectedWorkloadType, armnn::DataType DataType>
355 static void RefCreateFullyConnectedWorkloadTest()
359 auto workload = CreateFullyConnectedWorkloadTest<FullyConnectedWorkloadType, DataType>(factory, graph);
364 CheckInputOutput(std::move(workload),
371 RefCreateFullyConnectedWorkloadTest<RefFullyConnectedWorkload, armnn::DataType::Float32>();
376 RefCreateFullyConnectedWorkloadTest<RefFullyConnectedWorkload, armnn::DataType::QAsymmU8>();
381 RefCreateFullyConnectedWorkloadTest<RefFullyConnectedWorkload, armnn::DataType::QSymmS16>();
384 template <
typename NormalizationWorkloadType, armnn::DataType DataType>
385 static void RefCreateNormalizationWorkloadTest(
DataLayout dataLayout)
389 auto workload = CreateNormalizationWorkloadTest<NormalizationWorkloadType, DataType>(factory, graph, dataLayout);
396 case DataLayout::NHWC:
397 inputShape = { 3, 1, 5, 5 };
398 outputShape = { 3, 1, 5, 5 };
400 case DataLayout::NCHW:
402 inputShape = { 3, 5, 5, 1 };
403 outputShape = { 3, 5, 5, 1 };
413 RefCreateNormalizationWorkloadTest<RefNormalizationWorkload, armnn::DataType::Float32>(DataLayout::NCHW);
418 RefCreateNormalizationWorkloadTest<RefNormalizationWorkload, armnn::DataType::Float32>(DataLayout::NHWC);
423 RefCreateNormalizationWorkloadTest<RefNormalizationWorkload, armnn::DataType::QAsymmU8>(DataLayout::NCHW);
428 RefCreateNormalizationWorkloadTest<RefNormalizationWorkload, armnn::DataType::QAsymmU8>(DataLayout::NHWC);
433 RefCreateNormalizationWorkloadTest<RefNormalizationWorkload, armnn::DataType::QSymmS16>(DataLayout::NCHW);
438 RefCreateNormalizationWorkloadTest<RefNormalizationWorkload, armnn::DataType::QSymmS16>(DataLayout::NHWC);
441 template <
typename Pooling2dWorkloadType, armnn::DataType DataType>
442 static void RefCreatePooling2dWorkloadTest(
DataLayout dataLayout)
446 auto workload = CreatePooling2dWorkloadTest<Pooling2dWorkloadType, DataType>(factory, graph, dataLayout);
453 case DataLayout::NHWC:
454 inputShape = { 3, 5, 5, 2 };
455 outputShape = { 3, 2, 4, 2 };
457 case DataLayout::NCHW:
459 inputShape = { 3, 2, 5, 5 };
460 outputShape = { 3, 2, 2, 4 };
464 CheckInputOutput(std::move(workload),
471 RefCreatePooling2dWorkloadTest<RefPooling2dWorkload, armnn::DataType::Float32>(DataLayout::NCHW);
476 RefCreatePooling2dWorkloadTest<RefPooling2dWorkload, armnn::DataType::Float32>(DataLayout::NHWC);
481 RefCreatePooling2dWorkloadTest<RefPooling2dWorkload, armnn::DataType::QAsymmU8>(DataLayout::NCHW);
486 RefCreatePooling2dWorkloadTest<RefPooling2dWorkload, armnn::DataType::QAsymmU8>(DataLayout::NHWC);
491 RefCreatePooling2dWorkloadTest<RefPooling2dWorkload, armnn::DataType::QSymmS16>(DataLayout::NCHW);
496 RefCreatePooling2dWorkloadTest<RefPooling2dWorkload, armnn::DataType::QSymmS16>(DataLayout::NHWC);
499 template <
typename SoftmaxWorkloadType, armnn::DataType DataType>
500 static void RefCreateSoftmaxWorkloadTest()
504 auto workload = CreateSoftmaxWorkloadTest<SoftmaxWorkloadType, DataType>(factory, graph);
512 tensorInfo.SetQuantizationScale(1.f / 256);
516 tensorInfo.SetQuantizationOffset(-128);
517 tensorInfo.SetQuantizationScale(1.f / 256);
527 RefCreateSoftmaxWorkloadTest<RefSoftmaxWorkload, armnn::DataType::Float32>();
532 RefCreateSoftmaxWorkloadTest<RefSoftmaxWorkload, armnn::DataType::Float16>();
537 RefCreateSoftmaxWorkloadTest<RefSoftmaxWorkload, armnn::DataType::QAsymmU8>();
542 RefCreateSoftmaxWorkloadTest<RefSoftmaxWorkload, armnn::DataType::QSymmS16>();
545 template <
typename SplitterWorkloadType, armnn::DataType DataType>
546 static void RefCreateSplitterWorkloadTest()
550 auto workload = CreateSplitterWorkloadTest<SplitterWorkloadType, DataType>(factory, graph);
554 auto inputHandle = PolymorphicDowncast<RefTensorHandle*>(queueDescriptor.m_Inputs[0]);
557 auto outputHandle0 = PolymorphicDowncast<RefTensorHandle*>(queueDescriptor.m_Outputs[0]);
560 auto outputHandle1 = PolymorphicDowncast<RefTensorHandle*>(queueDescriptor.m_Outputs[1]);
563 auto outputHandle2 = PolymorphicDowncast<RefTensorHandle*>(queueDescriptor.m_Outputs[2]);
569 RefCreateSplitterWorkloadTest<RefSplitterWorkload, armnn::DataType::Float32>();
574 RefCreateSplitterWorkloadTest<RefSplitterWorkload, armnn::DataType::Float16>();
579 RefCreateSplitterWorkloadTest<RefSplitterWorkload, armnn::DataType::QAsymmU8>();
582 template <
typename SplitterWorkloadType,
typename ConcatWorkloadType, armnn::DataType DataType>
583 static void RefCreateSplitterConcatWorkloadTest()
593 auto workloads = CreateSplitterConcatWorkloadTest<SplitterWorkloadType, ConcatWorkloadType, DataType>
596 auto wlSplitter = std::move(workloads.first);
597 auto wlConcat = std::move(workloads.second);
610 bool validDataPointers = (sOut0 == mIn1) && (sOut1 == mIn0);
612 BOOST_TEST(validDataPointers);
617 RefCreateSplitterConcatWorkloadTest<RefSplitterWorkload, RefConcatWorkload, DataType::Float32>();
622 RefCreateSplitterConcatWorkloadTest<RefSplitterWorkload, RefConcatWorkload, DataType::Float16>();
627 RefCreateSplitterConcatWorkloadTest<RefSplitterWorkload, RefConcatWorkload, DataType::QAsymmU8>();
630 template <
typename SplitterWorkloadType,
typename ActivationWorkloadType, armnn::DataType DataType>
631 static void RefCreateSingleOutputMultipleInputsTest()
638 std::unique_ptr<SplitterWorkloadType> wlSplitter;
639 std::unique_ptr<ActivationWorkloadType> wlActiv0_0;
640 std::unique_ptr<ActivationWorkloadType> wlActiv0_1;
641 std::unique_ptr<ActivationWorkloadType> wlActiv1_0;
642 std::unique_ptr<ActivationWorkloadType> wlActiv1_1;
644 CreateSplitterMultipleInputsOneOutputWorkloadTest<SplitterWorkloadType,
645 ActivationWorkloadType,
DataType>(factory, graph, wlSplitter, wlActiv0_0, wlActiv0_1, wlActiv1_0, wlActiv1_1);
657 BOOST_TEST(activ0_0Im);
658 BOOST_TEST(activ0_1Im);
659 BOOST_TEST(activ1_0Im);
660 BOOST_TEST(activ1_1Im);
662 bool validDataPointers = (sOut0 == activ0_0Im) && (sOut0 == activ0_1Im) &&
663 (sOut1 == activ1_0Im) && (sOut1 == activ1_1Im);
665 BOOST_TEST(validDataPointers);
680 template <
typename ResizeBilinearWorkloadType, armnn::DataType DataType>
681 static void RefCreateResizeBilinearTest(
DataLayout dataLayout)
685 auto workload = CreateResizeBilinearWorkloadTest<ResizeBilinearWorkloadType, DataType>(factory, graph, dataLayout);
692 case DataLayout::NHWC:
693 inputShape = { 2, 4, 4, 3 };
694 outputShape = { 2, 2, 2, 3 };
696 case DataLayout::NCHW:
698 inputShape = { 2, 3, 4, 4 };
699 outputShape = { 2, 3, 2, 2 };
703 CheckInputOutput(std::move(workload),
710 RefCreateResizeBilinearTest<RefResizeWorkload, armnn::DataType::Float32>(DataLayout::NCHW);
715 RefCreateResizeBilinearTest<RefResizeWorkload, armnn::DataType::Float16>(DataLayout::NCHW);
720 RefCreateResizeBilinearTest<RefResizeWorkload, armnn::DataType::QAsymmU8>(DataLayout::NCHW);
725 RefCreateResizeBilinearTest<RefResizeWorkload, armnn::DataType::QSymmS16>(DataLayout::NCHW);
730 RefCreateResizeBilinearTest<RefResizeWorkload, armnn::DataType::Float32>(DataLayout::NHWC);
733 template <
typename BatchToSpaceNdWorkloadType, armnn::DataType DataType>
734 static void RefCreateBatchToSpaceNdTest()
739 auto workload = CreateBatchToSpaceNdWorkloadTest<BatchToSpaceNdWorkloadType, DataType>(factory, graph);
741 CheckInputOutput(std::move(workload),
748 RefCreateBatchToSpaceNdTest<RefBatchToSpaceNdWorkload, armnn::DataType::Float32>();
753 RefCreateBatchToSpaceNdTest<RefBatchToSpaceNdWorkload, armnn::DataType::Float16>();
758 RefCreateBatchToSpaceNdTest<RefBatchToSpaceNdWorkload, armnn::DataType::QAsymmU8>();
763 RefCreateBatchToSpaceNdTest<RefBatchToSpaceNdWorkload, armnn::DataType::QSymmS16>();
766 template <
typename L2NormalizationWorkloadType, armnn::DataType DataType>
767 static void RefCreateL2NormalizationTest(
DataLayout dataLayout)
772 CreateL2NormalizationWorkloadTest<L2NormalizationWorkloadType, DataType>(factory, graph, dataLayout);
779 case DataLayout::NHWC:
780 inputShape = { 5, 50, 67, 20 };
781 outputShape = { 5, 50, 67, 20 };
783 case DataLayout::NCHW:
785 inputShape = { 5, 20, 50, 67 };
786 outputShape = { 5, 20, 50, 67 };
796 RefCreateL2NormalizationTest<RefL2NormalizationWorkload, armnn::DataType::Float32>(DataLayout::NCHW);
801 RefCreateL2NormalizationTest<RefL2NormalizationWorkload, armnn::DataType::Float32>(DataLayout::NHWC);
806 RefCreateL2NormalizationTest<RefL2NormalizationWorkload, armnn::DataType::QSymmS16>(DataLayout::NCHW);
811 RefCreateL2NormalizationTest<RefL2NormalizationWorkload, armnn::DataType::QSymmS16>(DataLayout::NHWC);
816 RefCreateL2NormalizationTest<RefL2NormalizationWorkload, armnn::DataType::QAsymmU8>(DataLayout::NCHW);
821 RefCreateL2NormalizationTest<RefL2NormalizationWorkload, armnn::DataType::QAsymmU8>(DataLayout::NHWC);
824 template <
typename ReshapeWorkloadType, armnn::DataType DataType>
825 static void RefCreateReshapeWorkloadTest()
829 auto workload = CreateReshapeWorkloadTest<ReshapeWorkloadType, DataType>(factory, graph);
840 RefCreateReshapeWorkloadTest<RefReshapeWorkload, armnn::DataType::Float32>();
845 RefCreateReshapeWorkloadTest<RefReshapeWorkload, armnn::DataType::QAsymmU8>();
850 RefCreateReshapeWorkloadTest<RefReshapeWorkload, armnn::DataType::QSymmS16>();
853 template <
typename ConcatWorkloadType, armnn::DataType DataType>
855 unsigned int concatAxis)
859 auto workload = CreateConcatWorkloadTest<ConcatWorkloadType, DataType>(factory, graph, outputShape, concatAxis);
861 CheckInputsOutput(std::move(workload),
869 RefCreateConcatWorkloadTest<RefConcatWorkload, armnn::DataType::Float32>({ 4, 3, 2, 5 }, 0);
874 RefCreateConcatWorkloadTest<RefConcatWorkload, armnn::DataType::Float16>({ 4, 3, 2, 5 }, 0);
879 RefCreateConcatWorkloadTest<RefConcatWorkload, armnn::DataType::QAsymmU8>({ 4, 3, 2, 5 }, 0);
884 RefCreateConcatWorkloadTest<RefConcatWorkload, armnn::DataType::QSymmS16>({ 4, 3, 2, 5 }, 0);
889 RefCreateConcatWorkloadTest<RefConcatWorkload, armnn::DataType::Float32>({ 2, 6, 2, 5 }, 1);
894 RefCreateConcatWorkloadTest<RefConcatWorkload, armnn::DataType::QAsymmU8>({ 2, 6, 2, 5 }, 1);
899 RefCreateConcatWorkloadTest<RefConcatWorkload, armnn::DataType::Float32>({ 2, 3, 4, 5 }, 2);
904 RefCreateConcatWorkloadTest<RefConcatWorkload, armnn::DataType::QAsymmU8>({ 2, 3, 4, 5 }, 2);
909 RefCreateConcatWorkloadTest<RefConcatWorkload, armnn::DataType::Float32>({ 2, 3, 2, 10 }, 3);
914 RefCreateConcatWorkloadTest<RefConcatWorkload, armnn::DataType::QAsymmU8>({ 2, 3, 2, 10 }, 3);
917 template <
typename ConstantWorkloadType, armnn::DataType DataType>
922 auto workload = CreateConstantWorkloadTest<ConstantWorkloadType, DataType>(factory, graph, outputShape);
925 auto queueDescriptor = workload->GetData();
926 auto outputHandle = PolymorphicDowncast<RefTensorHandle*>(queueDescriptor.m_Outputs[0]);
932 RefCreateConstantWorkloadTest<RefConstantWorkload, armnn::DataType::QAsymmU8>({ 2, 3, 2, 10 });
937 RefCreateConstantWorkloadTest<RefConstantWorkload, armnn::DataType::QSymmS16>({ 2, 3, 2, 10 });
942 RefCreateConstantWorkloadTest<RefConstantWorkload, armnn::DataType::Float32>({ 2, 3, 2, 10 });
947 RefCreateConstantWorkloadTest<RefConstantWorkload, armnn::DataType::Signed32>({ 2, 3, 2, 10 });
957 auto workload = CreatePreluWorkloadTest<RefPreluWorkload>(factory,
965 auto queueDescriptor = workload->GetData();
966 auto outputHandle = PolymorphicDowncast<RefTensorHandle*>(queueDescriptor.m_Outputs[0]);
967 BOOST_TEST((outputHandle->GetTensorInfo() ==
TensorInfo(outputShape, dataType)));
972 RefCreatePreluWorkloadTest({ 1, 4, 1, 2 }, { 5, 4, 3, 1 }, { 5, 4, 3, 2 },
armnn::DataType::Float32);
977 RefCreatePreluWorkloadTest({ 1, 4, 1, 2 }, { 5, 4, 3, 1 }, { 5, 4, 3, 2 },
armnn::DataType::Float16);
982 RefCreatePreluWorkloadTest({ 1, 4, 1, 2 }, { 5, 4, 3, 1 }, { 5, 4, 3, 2 },
armnn::DataType::QAsymmU8);
987 RefCreatePreluWorkloadTest({ 1, 4, 1, 2 }, { 5, 4, 3, 1 }, { 5, 4, 3, 2 },
armnn::DataType::QSymmS16);
992 BOOST_CHECK_THROW(RefCreatePreluWorkloadTest({ 1, 4, 7, 2 }, { 5, 4, 3, 1 }, { 5, 4, 3, 2 },
999 BOOST_CHECK_THROW(RefCreatePreluWorkloadTest({ 1, 4, 7, 2 }, { 5, 4, 3, 1 }, { 5, 4, 3, 2 },
1006 BOOST_CHECK_THROW(RefCreatePreluWorkloadTest({ 1, 4, 7, 2 }, { 5, 4, 3, 1 }, { 5, 4, 3, 2 },
1013 BOOST_CHECK_THROW(RefCreatePreluWorkloadTest({ 1, 4, 7, 2 }, { 5, 4, 3, 1 }, { 5, 4, 3, 2 },
1018 template <
typename SpaceToDepthWorkloadType, armnn::DataType DataType>
1019 static void RefCreateSpaceToDepthWorkloadTest()
1024 auto workload = CreateSpaceToDepthWorkloadTest<SpaceToDepthWorkloadType, DataType>(factory, graph);
1026 CheckInputOutput(std::move(workload),
1033 RefCreateSpaceToDepthWorkloadTest<RefSpaceToDepthWorkload, armnn::DataType::Float32>();
1038 RefCreateSpaceToDepthWorkloadTest<RefSpaceToDepthWorkload, armnn::DataType::Float16>();
1043 RefCreateSpaceToDepthWorkloadTest<RefSpaceToDepthWorkload, armnn::DataType::QAsymmU8>();
1048 RefCreateSpaceToDepthWorkloadTest<RefSpaceToDepthWorkload, armnn::DataType::QSymmS16>();
1051 template <armnn::DataType DataType>
1055 unsigned int numInputs)
1059 auto workload = CreateStackWorkloadTest<RefStackWorkload, DataType>(factory,
1068 for (
unsigned int i = 0; i < numInputs; ++i)
1070 auto inputHandle = PolymorphicDowncast<RefTensorHandle*>(queueDescriptor.m_Inputs[i]);
1073 auto outputHandle = PolymorphicDowncast<RefTensorHandle*>(queueDescriptor.m_Outputs[0]);
1079 RefCreateStackWorkloadTest<armnn::DataType::Float32>({ 3, 4, 5 }, { 3, 4, 2, 5 }, 2, 2);
1084 RefCreateStackWorkloadTest<armnn::DataType::QAsymmU8>({ 3, 4, 5 }, { 3, 4, 2, 5 }, 2, 2);
1089 RefCreateStackWorkloadTest<armnn::DataType::QSymmS16>({ 3, 4, 5 }, { 3, 4, 2, 5 }, 2, 2);
1092 template <
typename QLstmWorkloadType>
1093 static void RefCreateQLstmWorkloadTest()
1098 auto workload = CreateQLstmWorkloadTest<QLstmWorkloadType>(factory, graph);
1107 auto inputHandle = boost::polymorphic_downcast<RefTensorHandle*>(queueDescriptor.m_Inputs[0]);
1108 auto cellStateOutHandle = boost::polymorphic_downcast<RefTensorHandle*>(queueDescriptor.m_Outputs[1]);
1109 auto outputHandle = boost::polymorphic_downcast<RefTensorHandle*>(queueDescriptor.m_Outputs[2]);
1111 BOOST_TEST((inputHandle->GetTensorInfo() == inputInfo));
1112 BOOST_TEST((cellStateOutHandle->GetTensorInfo() == cellStateInfo));
1113 BOOST_TEST((outputHandle->GetTensorInfo() == outputInfo));
1118 RefCreateQLstmWorkloadTest<RefQLstmWorkload>();
BOOST_AUTO_TEST_SUITE(TensorflowLiteParser)
RefElementwiseWorkload< std::divides< float >, DivisionQueueDescriptor, StringMapping::RefDivisionWorkload_Execute > RefDivisionWorkload
RefElementwiseWorkload< std::minus< float >, SubtractionQueueDescriptor, StringMapping::RefSubtractionWorkload_Execute > RefSubtractionWorkload
RefElementwiseWorkload< std::plus< float >, AdditionQueueDescriptor, StringMapping::RefAdditionWorkload_Execute > RefAdditionWorkload
RefElementwiseWorkload< std::multiplies< float >, MultiplicationQueueDescriptor, StringMapping::RefMultiplicationWorkload_Execute > RefMultiplicationWorkload
This layer represents an addition operation.
BOOST_AUTO_TEST_SUITE_END()
This layer represents a subtraction operation.
This layer represents a division operation.
BOOST_AUTO_TEST_CASE(CreateActivationFloat32Workload)
void SetQuantizationOffset(int32_t offset)
This layer represents a multiplication operation.