• Home
  • Raw
  • Download

Lines Matching refs:inputTensorInfo

23         const armnn::TensorInfo inputTensorInfo,  in ReduceTestCommon()  argument
32 auto inputTensor = ConvertToDataType<ArmnnType>(inputData, inputTensorInfo); in ReduceTestCommon()
36 …ue_ptr<armnn::ITensorHandle> inputHandle = tensorHandleFactory.CreateTensorHandle(inputTensorInfo); in ReduceTestCommon()
46 resolvedAxis = inputTensorInfo.GetNumDimensions() + static_cast<uint32_t>(vAxis[i]); in ReduceTestCommon()
60 AddInputToWorkload(descriptor, info, inputTensorInfo, inputHandle.get()); in ReduceTestCommon()
93 armnn::TensorInfo inputTensorInfo(inputShape, ArmnnType); in ReduceSumSimpleTest() local
97 inputTensorInfo.SetQuantizationScale(1.0f); in ReduceSumSimpleTest()
98 inputTensorInfo.SetQuantizationOffset(0); in ReduceSumSimpleTest()
109 inputTensorInfo, in ReduceSumSimpleTest()
126 armnn::TensorInfo inputTensorInfo(inputShape, ArmnnType); in ReduceSumSingleAxisTest1() local
130 inputTensorInfo.SetQuantizationScale(1.0f); in ReduceSumSingleAxisTest1()
131 inputTensorInfo.SetQuantizationOffset(0); in ReduceSumSingleAxisTest1()
150 inputTensorInfo, in ReduceSumSingleAxisTest1()
167 armnn::TensorInfo inputTensorInfo(inputShape, ArmnnType); in ReduceSumSingleAxisTest2() local
171 inputTensorInfo.SetQuantizationScale(1.0f); in ReduceSumSingleAxisTest2()
172 inputTensorInfo.SetQuantizationOffset(0); in ReduceSumSingleAxisTest2()
209 inputTensorInfo, in ReduceSumSingleAxisTest2()
226 armnn::TensorInfo inputTensorInfo(inputShape, ArmnnType); in ReduceSumSingleAxisTest3() local
230 inputTensorInfo.SetQuantizationScale(1.0f); in ReduceSumSingleAxisTest3()
231 inputTensorInfo.SetQuantizationOffset(0); in ReduceSumSingleAxisTest3()
273 inputTensorInfo, in ReduceSumSingleAxisTest3()
291 armnn::TensorInfo inputTensorInfo(inputShape, ArmnnType); in ReduceSumMultipleAxisTest() local
295 inputTensorInfo.SetQuantizationScale(1.0f); in ReduceSumMultipleAxisTest()
296 inputTensorInfo.SetQuantizationOffset(0); in ReduceSumMultipleAxisTest()
314 inputTensorInfo, in ReduceSumMultipleAxisTest()