Lines Matching refs:tensorHandleFactory
27 const armnn::ITensorHandleFactory& tensorHandleFactory, in GatherTestImpl() argument
42 …std::unique_ptr<armnn::ITensorHandle> paramsHandle = tensorHandleFactory.CreateTensorHandle(params… in GatherTestImpl()
43 …std::unique_ptr<armnn::ITensorHandle> indicesHandle = tensorHandleFactory.CreateTensorHandle(indic… in GatherTestImpl()
44 …std::unique_ptr<armnn::ITensorHandle> outputHandle = tensorHandleFactory.CreateTensorHandle(output… in GatherTestImpl()
74 const armnn::ITensorHandleFactory& tensorHandleFactory) in Gather1dParamsTestImpl()
94 tensorHandleFactory, in Gather1dParamsTestImpl()
106 const armnn::ITensorHandleFactory& tensorHandleFactory) in GatherMultiDimParamsTestImpl()
127 tensorHandleFactory, in GatherMultiDimParamsTestImpl()
139 const armnn::ITensorHandleFactory& tensorHandleFactory) in GatherMultiDimParamsMultiDimIndicesTestImpl()
187 tensorHandleFactory, in GatherMultiDimParamsMultiDimIndicesTestImpl()
203 const armnn::ITensorHandleFactory& tensorHandleFactory) in Gather1dParamsTestImpl()
218 tensorHandleFactory, in Gather1dParamsTestImpl()
230 const armnn::ITensorHandleFactory& tensorHandleFactory) in GatherMultiDimParamsTestImpl()
246 tensorHandleFactory, in GatherMultiDimParamsTestImpl()
258 const armnn::ITensorHandleFactory& tensorHandleFactory) in GatherMultiDimParamsMultiDimIndicesTestImpl()
300 tensorHandleFactory, in GatherMultiDimParamsMultiDimIndicesTestImpl()
315 const armnn::ITensorHandleFactory& tensorHandleFactory) in Gather1dParamsFloat32Test() argument
318 workloadFactory, memoryManager, tensorHandleFactory); in Gather1dParamsFloat32Test()
324 const armnn::ITensorHandleFactory& tensorHandleFactory) in Gather1dParamsFloat16Test() argument
327 workloadFactory, memoryManager, tensorHandleFactory); in Gather1dParamsFloat16Test()
333 const armnn::ITensorHandleFactory& tensorHandleFactory) in Gather1dParamsUint8Test() argument
336 workloadFactory, memoryManager, tensorHandleFactory); in Gather1dParamsUint8Test()
342 const armnn::ITensorHandleFactory& tensorHandleFactory) in Gather1dParamsInt16Test() argument
345 workloadFactory, memoryManager, tensorHandleFactory); in Gather1dParamsInt16Test()
351 const armnn::ITensorHandleFactory& tensorHandleFactory) in Gather1dParamsInt32Test() argument
354 workloadFactory, memoryManager, tensorHandleFactory); in Gather1dParamsInt32Test()
360 const armnn::ITensorHandleFactory& tensorHandleFactory) in GatherMultiDimParamsFloat32Test() argument
363 workloadFactory, memoryManager, tensorHandleFactory); in GatherMultiDimParamsFloat32Test()
369 const armnn::ITensorHandleFactory& tensorHandleFactory) in GatherMultiDimParamsFloat16Test() argument
372 workloadFactory, memoryManager, tensorHandleFactory); in GatherMultiDimParamsFloat16Test()
378 const armnn::ITensorHandleFactory& tensorHandleFactory) in GatherMultiDimParamsUint8Test() argument
381 workloadFactory, memoryManager, tensorHandleFactory); in GatherMultiDimParamsUint8Test()
387 const armnn::ITensorHandleFactory& tensorHandleFactory) in GatherMultiDimParamsInt16Test() argument
390 workloadFactory, memoryManager, tensorHandleFactory); in GatherMultiDimParamsInt16Test()
396 const armnn::ITensorHandleFactory& tensorHandleFactory) in GatherMultiDimParamsInt32Test() argument
399 workloadFactory, memoryManager, tensorHandleFactory); in GatherMultiDimParamsInt32Test()
405 const armnn::ITensorHandleFactory& tensorHandleFactory) in GatherMultiDimParamsMultiDimIndicesFloat32Test() argument
408 workloadFactory, memoryManager, tensorHandleFactory); in GatherMultiDimParamsMultiDimIndicesFloat32Test()
414 const armnn::ITensorHandleFactory& tensorHandleFactory) in GatherMultiDimParamsMultiDimIndicesFloat16Test() argument
417 workloadFactory, memoryManager, tensorHandleFactory); in GatherMultiDimParamsMultiDimIndicesFloat16Test()
423 const armnn::ITensorHandleFactory& tensorHandleFactory) in GatherMultiDimParamsMultiDimIndicesUint8Test() argument
426 workloadFactory, memoryManager, tensorHandleFactory); in GatherMultiDimParamsMultiDimIndicesUint8Test()
432 const armnn::ITensorHandleFactory& tensorHandleFactory) in GatherMultiDimParamsMultiDimIndicesInt16Test() argument
435 workloadFactory, memoryManager, tensorHandleFactory); in GatherMultiDimParamsMultiDimIndicesInt16Test()
441 const armnn::ITensorHandleFactory& tensorHandleFactory) in GatherMultiDimParamsMultiDimIndicesInt32Test() argument
444 workloadFactory, memoryManager, tensorHandleFactory); in GatherMultiDimParamsMultiDimIndicesInt32Test()