1 /*
2 * Copyright (C) 2018 The Android Open Source Project
3 *
4 * Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at
7 *
8 * http://www.apache.org/licenses/LICENSE-2.0
9 *
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
15 */
16
17 #define LOG_TAG "Operations"
18
19 #include "OperationResolver.h"
20 #include "OperationsUtils.h"
21
22 namespace android {
23 namespace nn {
24 namespace logical_not {
25
26 constexpr uint32_t kNumInputs = 1;
27 constexpr uint32_t kInputTensor = 0;
28
29 constexpr uint32_t kNumOutputs = 1;
30 constexpr uint32_t kOutputTensor = 0;
31
32 namespace {
33
compute(const bool8 * input,const Shape & shape,bool8 * output)34 bool compute(const bool8* input, const Shape& shape, bool8* output) {
35 const auto size = getNumberOfElements(shape);
36 for (uint32_t i = 0; i < size; ++i) {
37 output[i] = input[i] == 0;
38 }
39 return true;
40 }
41
42 } // namespace
43
validate(const IOperationValidationContext * context)44 Result<Version> validate(const IOperationValidationContext* context) {
45 NN_RET_CHECK_EQ(context->getNumInputs(), kNumInputs);
46 NN_RET_CHECK_EQ(context->getNumOutputs(), kNumOutputs);
47 OperandType inputType = context->getInputType(kInputTensor);
48 NN_RET_CHECK(inputType == OperandType::TENSOR_BOOL8)
49 << "Unsupported tensor type for LOGICAL_NOT";
50 NN_RET_CHECK(validateInputTypes(context, {inputType}));
51 NN_RET_CHECK(validateOutputTypes(context, {inputType}));
52 return Version::ANDROID_Q;
53 }
54
prepare(IOperationExecutionContext * context)55 bool prepare(IOperationExecutionContext* context) {
56 Shape input = context->getInputShape(kInputTensor);
57 Shape output = context->getOutputShape(kOutputTensor);
58 NN_RET_CHECK(SetShape(input, &output));
59 return context->setOutputShape(kOutputTensor, output);
60 }
61
execute(IOperationExecutionContext * context)62 bool execute(IOperationExecutionContext* context) {
63 return compute(context->getInputBuffer<bool8>(kInputTensor),
64 context->getInputShape(kInputTensor),
65 context->getOutputBuffer<bool8>(kOutputTensor));
66 }
67
68 } // namespace logical_not
69
70 NN_REGISTER_OPERATION(LOGICAL_NOT, "LOGICAL_NOT", logical_not::validate, logical_not::prepare,
71 logical_not::execute);
72
73 } // namespace nn
74 } // namespace android
75