Searched refs:elementwiseUnaryQueueDescriptor (Results 1 – 5 of 5) sorted by relevance
| /external/armnn/src/backends/backendsCommon/test/ |
| D | DefaultAsyncExecuteTest.cpp | 117 ElementwiseUnaryQueueDescriptor elementwiseUnaryQueueDescriptor; in CreateWorkload() local
118 elementwiseUnaryQueueDescriptor.m_Inputs = std::vector<ITensorHandle*>{inputTensor}; in CreateWorkload()
119 elementwiseUnaryQueueDescriptor.m_Outputs = std::vector<ITensorHandle*>{outputTensor}; in CreateWorkload()
121 return std::make_unique<Workload>(elementwiseUnaryQueueDescriptor, workloadInfo); in CreateWorkload()
|
| /external/armnn/src/backends/neon/ |
| D | NeonWorkloadFactory.cpp | 338 auto elementwiseUnaryQueueDescriptor in CreateWorkload() local
341 switch(elementwiseUnaryQueueDescriptor->m_Parameters.m_Operation) in CreateWorkload()
346 absQueueDescriptor.m_Inputs = elementwiseUnaryQueueDescriptor->m_Inputs; in CreateWorkload()
347 absQueueDescriptor.m_Outputs = elementwiseUnaryQueueDescriptor->m_Outputs; in CreateWorkload()
352 … return std::make_unique<NeonExpWorkload>(*elementwiseUnaryQueueDescriptor, info); in CreateWorkload()
354 … return std::make_unique<NeonLogicalNotWorkload>(*elementwiseUnaryQueueDescriptor, info); in CreateWorkload()
356 … return std::make_unique<NeonLogWorkload>(*elementwiseUnaryQueueDescriptor, info); in CreateWorkload()
358 … return std::make_unique<NeonNegWorkload>(*elementwiseUnaryQueueDescriptor, info); in CreateWorkload()
362 rsqrtQueueDescriptor.m_Inputs = elementwiseUnaryQueueDescriptor->m_Inputs; in CreateWorkload()
363 rsqrtQueueDescriptor.m_Outputs = elementwiseUnaryQueueDescriptor->m_Outputs; in CreateWorkload()
[all …]
|
| /external/armnn/src/backends/cl/ |
| D | ClWorkloadFactory.cpp | 479 auto elementwiseUnaryQueueDescriptor in CreateWorkload() local
482 switch(elementwiseUnaryQueueDescriptor->m_Parameters.m_Operation) in CreateWorkload()
487 absQueueDescriptor.m_Inputs = elementwiseUnaryQueueDescriptor->m_Inputs; in CreateWorkload()
488 absQueueDescriptor.m_Outputs = elementwiseUnaryQueueDescriptor->m_Outputs; in CreateWorkload()
493 …return std::make_unique<ClExpWorkload>(*elementwiseUnaryQueueDescriptor, info, m_CLCompileContext); in CreateWorkload()
495 …return std::make_unique<ClLogWorkload>(*elementwiseUnaryQueueDescriptor, info, m_CLCompileContext); in CreateWorkload()
497 return std::make_unique<ClLogicalNotWorkload>(*elementwiseUnaryQueueDescriptor, in CreateWorkload()
501 …return std::make_unique<ClNegWorkload>(*elementwiseUnaryQueueDescriptor, info, m_CLCompileContext); in CreateWorkload()
505 rsqrtQueueDescriptor.m_Inputs = elementwiseUnaryQueueDescriptor->m_Inputs; in CreateWorkload()
506 rsqrtQueueDescriptor.m_Outputs = elementwiseUnaryQueueDescriptor->m_Outputs; in CreateWorkload()
[all …]
|
| /external/armnn/src/backends/reference/ |
| D | RefWorkloadFactory.cpp | 313 auto elementwiseUnaryQueueDescriptor in CreateWorkload() local
315 … if ((*elementwiseUnaryQueueDescriptor).m_Parameters.m_Operation == UnaryOperation::LogicalNot) in CreateWorkload()
317 … return std::make_unique<RefLogicalUnaryWorkload>(*elementwiseUnaryQueueDescriptor, info); in CreateWorkload()
319 … return std::make_unique<RefElementwiseUnaryWorkload>(*elementwiseUnaryQueueDescriptor, info); in CreateWorkload()
|
| /external/armnn/src/backends/backendsCommon/ |
| D | WorkloadFactory.cpp | 1710 auto elementwiseUnaryQueueDescriptor in CreateWorkload() local
1712 return CreateElementwiseUnary(*elementwiseUnaryQueueDescriptor, info); in CreateWorkload()
|