• Home
  • Raw
  • Download

Lines Matching refs:inputTensorInfo

24         const armnn::TensorInfo inputTensorInfo,  in ArgMinMaxTestCommon()  argument
30 std::vector<T> inputTensor = ConvertToDataType<ArmnnType>(inputData, inputTensorInfo); in ArgMinMaxTestCommon()
33 …e_ptr<armnn::ITensorHandle> inputHandle = tensorHandleFactory.CreateTensorHandle(inputTensorInfo); in ArgMinMaxTestCommon()
41 AddInputToWorkload(descriptor, info, inputTensorInfo, inputHandle.get()); in ArgMinMaxTestCommon()
74 armnn::TensorInfo inputTensorInfo(inputShape, ArmnnType); in ArgMaxSimpleTest() local
78 inputTensorInfo.SetQuantizationScale(1.0f); in ArgMaxSimpleTest()
79 inputTensorInfo.SetQuantizationOffset(0); in ArgMaxSimpleTest()
89 inputTensorInfo, outputTensorInfo, in ArgMaxSimpleTest()
102 armnn::TensorInfo inputTensorInfo(inputShape, ArmnnType); in ArgMinSimpleTest() local
106 inputTensorInfo.SetQuantizationScale(1.0f); in ArgMinSimpleTest()
107 inputTensorInfo.SetQuantizationOffset(0); in ArgMinSimpleTest()
117 inputTensorInfo, outputTensorInfo, in ArgMinSimpleTest()
130 armnn::TensorInfo inputTensorInfo(inputShape, ArmnnType); in ArgMinChannelTest() local
134 inputTensorInfo.SetQuantizationScale(1.0f); in ArgMinChannelTest()
135 inputTensorInfo.SetQuantizationOffset(0); in ArgMinChannelTest()
153 inputTensorInfo, outputTensorInfo, in ArgMinChannelTest()
166 armnn::TensorInfo inputTensorInfo(inputShape, ArmnnType); in ArgMaxChannelTest() local
170 inputTensorInfo.SetQuantizationScale(1.0f); in ArgMaxChannelTest()
171 inputTensorInfo.SetQuantizationOffset(0); in ArgMaxChannelTest()
189 inputTensorInfo, outputTensorInfo, in ArgMaxChannelTest()
202 armnn::TensorInfo inputTensorInfo(inputShape, ArmnnType); in ArgMaxHeightTest() local
207 inputTensorInfo.SetQuantizationScale(1.0f); in ArgMaxHeightTest()
208 inputTensorInfo.SetQuantizationOffset(0); in ArgMaxHeightTest()
225 inputTensorInfo, outputTensorInfo, in ArgMaxHeightTest()
238 armnn::TensorInfo inputTensorInfo(inputShape, ArmnnType); in ArgMinWidthTest() local
243 inputTensorInfo.SetQuantizationScale(1.0f); in ArgMinWidthTest()
244 inputTensorInfo.SetQuantizationOffset(0); in ArgMinWidthTest()
261 inputTensorInfo, outputTensorInfo, in ArgMinWidthTest()