33 std::vector<ITensorHandle*> outputs)
39 m_Input0 = MakeDecoder<InType>(inputInfo0);
40 m_Input1 = MakeDecoder<InType>(inputInfo1);
42 m_Output = MakeEncoder<OutType>(outputInfo);
70 m_Input0->Reset(inputs[0]->
Map());
71 m_Input1->Reset(inputs[1]->
Map());
72 m_Output->Reset(outputs[0]->
Map());
85 EqualFunction(inShape0, inShape1, outShape, *m_Input0, *m_Input1, *m_Output);
90 GreaterFunction(inShape0, inShape1, outShape, *m_Input0, *m_Input1, *m_Output);
95 GreaterOrEqualFunction(inShape0, inShape1, outShape, *m_Input0, *m_Input1, *m_Output);
100 LessFunction(inShape0, inShape1, outShape, *m_Input0, *m_Input1, *m_Output);
105 LessOrEqualFunction(inShape0, inShape1, outShape, *m_Input0, *m_Input1, *m_Output);
110 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
void ExecuteAsync(ExecutionData &executionData) 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
const TensorInfo & GetTensorInfo(const ITensorHandle *tensorHandle)
float32 helpers