33 std::vector<ITensorHandle*> outputs)
39 m_Input0 = MakeDecoder<InType>(inputInfo0);
40 m_Input1 = MakeDecoder<InType>(inputInfo1);
42 m_Output = MakeEncoder<OutType>(outputInfo);
69 m_Input0->Reset(inputs[0]->
Map());
70 m_Input1->Reset(inputs[1]->
Map());
71 m_Output->Reset(outputs[0]->
Map());
84 EqualFunction(inShape0, inShape1, outShape, *m_Input0, *m_Input1, *m_Output);
89 GreaterFunction(inShape0, inShape1, outShape, *m_Input0, *m_Input1, *m_Output);
94 GreaterOrEqualFunction(inShape0, inShape1, outShape, *m_Input0, *m_Input1, *m_Output);
99 LessFunction(inShape0, inShape1, outShape, *m_Input0, *m_Input1, *m_Output);
104 LessOrEqualFunction(inShape0, inShape1, outShape, *m_Input0, *m_Input1, *m_Output);
109 NotEqualFunction(inShape0, inShape1, outShape, *m_Input0, *m_Input1, *m_Output);
const TensorShape & GetShape() const
CPU Execution: Reference C++ kernels.
Copyright (c) 2021 ARM Limited and Contributors.
LayerDescriptor m_Parameters
#define ARMNN_SCOPED_PROFILING_EVENT(backendId, name)
void Execute() const override
std::vector< ITensorHandle * > m_Inputs
ComparisonQueueDescriptor m_Data
ComparisonOperation m_Operation
Specifies the comparison operation to execute.
RefComparisonWorkload(const ComparisonQueueDescriptor &descriptor, const WorkloadInfo &info)
std::vector< ITensorHandle * > m_Outputs
void PostAllocationConfigure() override
constexpr char const * GetComparisonOperationAsCString(ComparisonOperation operation)
std::vector< ITensorHandle * > m_Outputs
Contains information about TensorInfos of a layer.
std::vector< ITensorHandle * > m_Inputs
void ExecuteAsync(WorkingMemDescriptor &workingMemDescriptor) override
const TensorInfo & GetTensorInfo(const ITensorHandle *tensorHandle)
float32 helpers