Lines Matching refs:context
71 bool executeLessTyped(IOperationExecutionContext* context) { in executeLessTyped() argument
73 std::less<ComparisonType>(), context->getInputBuffer<DataType>(kInputTensor1), in executeLessTyped()
74 context->getInputShape(kInputTensor1), context->getInputBuffer<DataType>(kInputTensor2), in executeLessTyped()
75 context->getInputShape(kInputTensor2), context->getOutputBuffer<bool8>(kOutputTensor), in executeLessTyped()
76 context->getOutputShape(kOutputTensor)); in executeLessTyped()
80 bool executeLessEqualTyped(IOperationExecutionContext* context) { in executeLessEqualTyped() argument
82 std::less_equal<ComparisonType>(), context->getInputBuffer<DataType>(kInputTensor1), in executeLessEqualTyped()
83 context->getInputShape(kInputTensor1), context->getInputBuffer<DataType>(kInputTensor2), in executeLessEqualTyped()
84 context->getInputShape(kInputTensor2), context->getOutputBuffer<bool8>(kOutputTensor), in executeLessEqualTyped()
85 context->getOutputShape(kOutputTensor)); in executeLessEqualTyped()
89 bool executeEqualTyped(IOperationExecutionContext* context) { in executeEqualTyped() argument
91 std::equal_to<ComparisonType>(), context->getInputBuffer<DataType>(kInputTensor1), in executeEqualTyped()
92 context->getInputShape(kInputTensor1), context->getInputBuffer<DataType>(kInputTensor2), in executeEqualTyped()
93 context->getInputShape(kInputTensor2), context->getOutputBuffer<bool8>(kOutputTensor), in executeEqualTyped()
94 context->getOutputShape(kOutputTensor)); in executeEqualTyped()
98 bool executeNotEqualTyped(IOperationExecutionContext* context) { in executeNotEqualTyped() argument
100 std::not_equal_to<ComparisonType>(), context->getInputBuffer<DataType>(kInputTensor1), in executeNotEqualTyped()
101 context->getInputShape(kInputTensor1), context->getInputBuffer<DataType>(kInputTensor2), in executeNotEqualTyped()
102 context->getInputShape(kInputTensor2), context->getOutputBuffer<bool8>(kOutputTensor), in executeNotEqualTyped()
103 context->getOutputShape(kOutputTensor)); in executeNotEqualTyped()
107 bool executeGreaterEqualTyped(IOperationExecutionContext* context) { in executeGreaterEqualTyped() argument
109 std::greater_equal<ComparisonType>(), context->getInputBuffer<DataType>(kInputTensor1), in executeGreaterEqualTyped()
110 context->getInputShape(kInputTensor1), context->getInputBuffer<DataType>(kInputTensor2), in executeGreaterEqualTyped()
111 context->getInputShape(kInputTensor2), context->getOutputBuffer<bool8>(kOutputTensor), in executeGreaterEqualTyped()
112 context->getOutputShape(kOutputTensor)); in executeGreaterEqualTyped()
116 bool executeGreaterTyped(IOperationExecutionContext* context) { in executeGreaterTyped() argument
118 std::greater<ComparisonType>(), context->getInputBuffer<DataType>(kInputTensor1), in executeGreaterTyped()
119 context->getInputShape(kInputTensor1), context->getInputBuffer<DataType>(kInputTensor2), in executeGreaterTyped()
120 context->getInputShape(kInputTensor2), context->getOutputBuffer<bool8>(kOutputTensor), in executeGreaterTyped()
121 context->getOutputShape(kOutputTensor)); in executeGreaterTyped()
126 Result<Version> validate(const IOperationValidationContext* context) { in validate() argument
127 NN_RET_CHECK_EQ(context->getNumInputs(), kNumInputs); in validate()
128 NN_RET_CHECK_EQ(context->getNumOutputs(), kNumOutputs); in validate()
129 OperandType inputType = context->getInputType(kInputTensor1); in validate()
136 NN_RET_CHECK(validateInputTypes(context, {inputType, inputType})); in validate()
137 NN_RET_CHECK(validateOutputTypes(context, {OperandType::TENSOR_BOOL8})); in validate()
145 bool prepare(IOperationExecutionContext* context) { in prepare() argument
146 Shape input1 = context->getInputShape(kInputTensor1); in prepare()
147 Shape input2 = context->getInputShape(kInputTensor2); in prepare()
148 Shape output = context->getOutputShape(kOutputTensor); in prepare()
150 return context->setOutputShape(kOutputTensor, output); in prepare()
153 bool executeLess(IOperationExecutionContext* context) { in executeLess() argument
154 switch (context->getInputType(kInputTensor1)) { in executeLess()
156 return executeLessTyped<_Float16, _Float16>(context); in executeLess()
158 return executeLessTyped<float, float>(context); in executeLess()
160 return executeLessTyped<int32_t, int32_t>(context); in executeLess()
162 return executeLessTyped<uint8_t, float>(context); in executeLess()
164 return executeLessTyped<int8_t, float>(context); in executeLess()
166 return executeLessTyped<bool8, bool8>(context); in executeLess()
172 bool executeLessEqual(IOperationExecutionContext* context) { in executeLessEqual() argument
173 switch (context->getInputType(kInputTensor1)) { in executeLessEqual()
175 return executeLessEqualTyped<_Float16, _Float16>(context); in executeLessEqual()
177 return executeLessEqualTyped<float, float>(context); in executeLessEqual()
179 return executeLessEqualTyped<int32_t, int32_t>(context); in executeLessEqual()
181 return executeLessEqualTyped<uint8_t, float>(context); in executeLessEqual()
183 return executeLessEqualTyped<int8_t, float>(context); in executeLessEqual()
185 return executeLessEqualTyped<bool8, bool8>(context); in executeLessEqual()
191 bool executeEqual(IOperationExecutionContext* context) { in executeEqual() argument
192 switch (context->getInputType(kInputTensor1)) { in executeEqual()
194 return executeEqualTyped<_Float16, _Float16>(context); in executeEqual()
196 return executeEqualTyped<float, float>(context); in executeEqual()
198 return executeEqualTyped<int32_t, int32_t>(context); in executeEqual()
200 return executeEqualTyped<uint8_t, float>(context); in executeEqual()
202 return executeEqualTyped<int8_t, float>(context); in executeEqual()
204 return executeEqualTyped<bool8, bool8>(context); in executeEqual()
210 bool executeNotEqual(IOperationExecutionContext* context) { in executeNotEqual() argument
211 switch (context->getInputType(kInputTensor1)) { in executeNotEqual()
213 return executeNotEqualTyped<_Float16, _Float16>(context); in executeNotEqual()
215 return executeNotEqualTyped<float, float>(context); in executeNotEqual()
217 return executeNotEqualTyped<int32_t, int32_t>(context); in executeNotEqual()
219 return executeNotEqualTyped<uint8_t, float>(context); in executeNotEqual()
221 return executeNotEqualTyped<int8_t, float>(context); in executeNotEqual()
223 return executeNotEqualTyped<bool8, bool8>(context); in executeNotEqual()
229 bool executeGreaterEqual(IOperationExecutionContext* context) { in executeGreaterEqual() argument
230 switch (context->getInputType(kInputTensor1)) { in executeGreaterEqual()
232 return executeGreaterEqualTyped<_Float16, _Float16>(context); in executeGreaterEqual()
234 return executeGreaterEqualTyped<float, float>(context); in executeGreaterEqual()
236 return executeGreaterEqualTyped<int32_t, int32_t>(context); in executeGreaterEqual()
238 return executeGreaterEqualTyped<uint8_t, float>(context); in executeGreaterEqual()
240 return executeGreaterEqualTyped<int8_t, float>(context); in executeGreaterEqual()
242 return executeGreaterEqualTyped<bool8, bool8>(context); in executeGreaterEqual()
248 bool executeGreater(IOperationExecutionContext* context) { in executeGreater() argument
249 switch (context->getInputType(kInputTensor1)) { in executeGreater()
251 return executeGreaterTyped<_Float16, _Float16>(context); in executeGreater()
253 return executeGreaterTyped<float, float>(context); in executeGreater()
255 return executeGreaterTyped<int32_t, int32_t>(context); in executeGreater()
257 return executeGreaterTyped<uint8_t, float>(context); in executeGreater()
259 return executeGreaterTyped<int8_t, float>(context); in executeGreater()
261 return executeGreaterTyped<bool8, bool8>(context); in executeGreater()