Support DIV/SUB/MINIMUM/MAXIMUM/SQUARED_DIFFERENCE in XNNPACK delegate
PiperOrigin-RevId: 315226922 Change-Id: Ibd94b650b67c3808edca34d370d121e8671bf21f
This commit is contained in:
parent
b997946576
commit
c7f8f9ecfb
@ -254,6 +254,21 @@ cc_test(
|
|||||||
],
|
],
|
||||||
)
|
)
|
||||||
|
|
||||||
|
cc_test(
|
||||||
|
name = "div_test",
|
||||||
|
srcs = ["div_test.cc"],
|
||||||
|
linkopts = select({
|
||||||
|
"//tensorflow:emscripten": EMSCRIPTEN_LINKOPTS,
|
||||||
|
"//conditions:default": [],
|
||||||
|
}),
|
||||||
|
deps = [
|
||||||
|
":binary_elementwise_tester",
|
||||||
|
":test_main",
|
||||||
|
":xnnpack_delegate_test_mode",
|
||||||
|
"@com_google_googletest//:gtest",
|
||||||
|
],
|
||||||
|
)
|
||||||
|
|
||||||
cc_test(
|
cc_test(
|
||||||
name = "fully_connected_test",
|
name = "fully_connected_test",
|
||||||
srcs = ["fully_connected_test.cc"],
|
srcs = ["fully_connected_test.cc"],
|
||||||
@ -314,6 +329,36 @@ cc_test(
|
|||||||
],
|
],
|
||||||
)
|
)
|
||||||
|
|
||||||
|
cc_test(
|
||||||
|
name = "maximum_test",
|
||||||
|
srcs = ["maximum_test.cc"],
|
||||||
|
linkopts = select({
|
||||||
|
"//tensorflow:emscripten": EMSCRIPTEN_LINKOPTS,
|
||||||
|
"//conditions:default": [],
|
||||||
|
}),
|
||||||
|
deps = [
|
||||||
|
":binary_elementwise_tester",
|
||||||
|
":test_main",
|
||||||
|
":xnnpack_delegate_test_mode",
|
||||||
|
"@com_google_googletest//:gtest",
|
||||||
|
],
|
||||||
|
)
|
||||||
|
|
||||||
|
cc_test(
|
||||||
|
name = "minimum_test",
|
||||||
|
srcs = ["minimum_test.cc"],
|
||||||
|
linkopts = select({
|
||||||
|
"//tensorflow:emscripten": EMSCRIPTEN_LINKOPTS,
|
||||||
|
"//conditions:default": [],
|
||||||
|
}),
|
||||||
|
deps = [
|
||||||
|
":binary_elementwise_tester",
|
||||||
|
":test_main",
|
||||||
|
":xnnpack_delegate_test_mode",
|
||||||
|
"@com_google_googletest//:gtest",
|
||||||
|
],
|
||||||
|
)
|
||||||
|
|
||||||
cc_test(
|
cc_test(
|
||||||
name = "mul_test",
|
name = "mul_test",
|
||||||
srcs = ["mul_test.cc"],
|
srcs = ["mul_test.cc"],
|
||||||
@ -404,4 +449,34 @@ cc_test(
|
|||||||
],
|
],
|
||||||
)
|
)
|
||||||
|
|
||||||
|
cc_test(
|
||||||
|
name = "squared_difference_test",
|
||||||
|
srcs = ["squared_difference_test.cc"],
|
||||||
|
linkopts = select({
|
||||||
|
"//tensorflow:emscripten": EMSCRIPTEN_LINKOPTS,
|
||||||
|
"//conditions:default": [],
|
||||||
|
}),
|
||||||
|
deps = [
|
||||||
|
":binary_elementwise_tester",
|
||||||
|
":test_main",
|
||||||
|
":xnnpack_delegate_test_mode",
|
||||||
|
"@com_google_googletest//:gtest",
|
||||||
|
],
|
||||||
|
)
|
||||||
|
|
||||||
|
cc_test(
|
||||||
|
name = "sub_test",
|
||||||
|
srcs = ["sub_test.cc"],
|
||||||
|
linkopts = select({
|
||||||
|
"//tensorflow:emscripten": EMSCRIPTEN_LINKOPTS,
|
||||||
|
"//conditions:default": [],
|
||||||
|
}),
|
||||||
|
deps = [
|
||||||
|
":binary_elementwise_tester",
|
||||||
|
":test_main",
|
||||||
|
":xnnpack_delegate_test_mode",
|
||||||
|
"@com_google_googletest//:gtest",
|
||||||
|
],
|
||||||
|
)
|
||||||
|
|
||||||
tflite_portable_test_suite_combined(combine_conditions = {"deps": [":test_main"]})
|
tflite_portable_test_suite_combined(combine_conditions = {"deps": [":test_main"]})
|
||||||
|
@ -116,6 +116,12 @@ Below is the list of current operators and limitations:
|
|||||||
* Fused `NONE`, `RELU`, `RELU_N1_TO_1`, and `RELU6` activations are supported,
|
* Fused `NONE`, `RELU`, `RELU_N1_TO_1`, and `RELU6` activations are supported,
|
||||||
but fused `TANH` and `SIGN_BIT` activations are not.
|
but fused `TANH` and `SIGN_BIT` activations are not.
|
||||||
|
|
||||||
|
### `DIV`
|
||||||
|
|
||||||
|
* Inputs and outputs must be in 32-bit floating-point format.
|
||||||
|
* Fused `NONE`, `RELU`, `RELU_N1_TO_1`, and `RELU6` activations are supported,
|
||||||
|
but fused `TANH` and `SIGN_BIT` activations are not.
|
||||||
|
|
||||||
### `FULLY_CONNECTED`
|
### `FULLY_CONNECTED`
|
||||||
|
|
||||||
* Inputs and outputs must be in 32-bit floating-point format.
|
* Inputs and outputs must be in 32-bit floating-point format.
|
||||||
@ -139,6 +145,14 @@ Below is the list of current operators and limitations:
|
|||||||
* Fused `NONE`, `RELU`, `RELU_N1_TO_1`, and `RELU6` activations are supported,
|
* Fused `NONE`, `RELU`, `RELU_N1_TO_1`, and `RELU6` activations are supported,
|
||||||
but fused `TANH` and `SIGN_BIT` activations are not.
|
but fused `TANH` and `SIGN_BIT` activations are not.
|
||||||
|
|
||||||
|
### `MAXIMUM`
|
||||||
|
|
||||||
|
* Inputs and outputs must be in 32-bit floating-point format.
|
||||||
|
|
||||||
|
### `MINIMUM`
|
||||||
|
|
||||||
|
* Inputs and outputs must be in 32-bit floating-point format.
|
||||||
|
|
||||||
### `MUL`
|
### `MUL`
|
||||||
|
|
||||||
* Inputs and outputs must be in 32-bit floating-point format.
|
* Inputs and outputs must be in 32-bit floating-point format.
|
||||||
@ -176,6 +190,16 @@ Below is the list of current operators and limitations:
|
|||||||
* Inputs and outputs must be in 32-bit floating-point format.
|
* Inputs and outputs must be in 32-bit floating-point format.
|
||||||
* Only `beta = 1.0` is supported.
|
* Only `beta = 1.0` is supported.
|
||||||
|
|
||||||
|
### `SQUARED_DIFFERENCE`
|
||||||
|
|
||||||
|
* Inputs and outputs must be in 32-bit floating-point format.
|
||||||
|
|
||||||
|
### `SUB`
|
||||||
|
|
||||||
|
* Inputs and outputs must be in 32-bit floating-point format.
|
||||||
|
* Fused `NONE`, `RELU`, `RELU_N1_TO_1`, and `RELU6` activations are supported,
|
||||||
|
but fused `TANH` and `SIGN_BIT` activations are not.
|
||||||
|
|
||||||
### Other limitations
|
### Other limitations
|
||||||
|
|
||||||
* Dynamically allocated (with `kTfLiteDynamic` allocation type) inputs and
|
* Dynamically allocated (with `kTfLiteDynamic` allocation type) inputs and
|
||||||
|
840
tensorflow/lite/delegates/xnnpack/div_test.cc
Normal file
840
tensorflow/lite/delegates/xnnpack/div_test.cc
Normal file
@ -0,0 +1,840 @@
|
|||||||
|
/* Copyright 2020 The TensorFlow Authors. All Rights Reserved.
|
||||||
|
|
||||||
|
Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
|
you may not use this file except in compliance with the License.
|
||||||
|
You may obtain a copy of the License at
|
||||||
|
|
||||||
|
http://www.apache.org/licenses/LICENSE-2.0
|
||||||
|
|
||||||
|
Unless required by applicable law or agreed to in writing, software
|
||||||
|
distributed under the License is distributed on an "AS IS" BASIS,
|
||||||
|
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||||
|
See the License for the specific language governing permissions and
|
||||||
|
limitations under the License.
|
||||||
|
==============================================================================*/
|
||||||
|
|
||||||
|
#include <cstdint>
|
||||||
|
#include <functional>
|
||||||
|
#include <memory>
|
||||||
|
#include <random>
|
||||||
|
|
||||||
|
#include <gtest/gtest.h>
|
||||||
|
#include "tensorflow/lite/delegates/xnnpack/binary_elementwise_tester.h"
|
||||||
|
#include "tensorflow/lite/delegates/xnnpack/xnnpack_delegate.h"
|
||||||
|
|
||||||
|
namespace tflite {
|
||||||
|
namespace xnnpack {
|
||||||
|
|
||||||
|
TEST(Div, 4DBy4D) {
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(nullptr),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto height = shape_rng();
|
||||||
|
const auto width = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, height, width, channels})
|
||||||
|
.Input2Shape({batch, height, width, channels})
|
||||||
|
.Test(BuiltinOperator_DIV, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(Div, 4DBy4DBroadcastChannels) {
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(nullptr),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto height = shape_rng();
|
||||||
|
const auto width = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({1, 1, 1, channels})
|
||||||
|
.Input2Shape({batch, height, width, channels})
|
||||||
|
.Test(BuiltinOperator_DIV, xnnpack_delegate.get());
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, height, width, channels})
|
||||||
|
.Input2Shape({1, 1, 1, channels})
|
||||||
|
.Test(BuiltinOperator_DIV, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(Div, 4DBy4DBroadcastWidth) {
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(nullptr),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto height = shape_rng();
|
||||||
|
const auto width = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({1, 1, width, 1})
|
||||||
|
.Input2Shape({batch, height, width, channels})
|
||||||
|
.Test(BuiltinOperator_DIV, xnnpack_delegate.get());
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, height, width, channels})
|
||||||
|
.Input2Shape({1, 1, width, 1})
|
||||||
|
.Test(BuiltinOperator_DIV, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(Div, 4DBy4DBroadcastHeight) {
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(nullptr),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto height = shape_rng();
|
||||||
|
const auto width = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({1, height, 1, 1})
|
||||||
|
.Input2Shape({batch, height, width, channels})
|
||||||
|
.Test(BuiltinOperator_DIV, xnnpack_delegate.get());
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, height, width, channels})
|
||||||
|
.Input2Shape({1, height, 1, 1})
|
||||||
|
.Test(BuiltinOperator_DIV, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(Div, 4DBy4DBroadcastBatch) {
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(nullptr),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto height = shape_rng();
|
||||||
|
const auto width = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, 1, 1, 1})
|
||||||
|
.Input2Shape({batch, height, width, channels})
|
||||||
|
.Test(BuiltinOperator_DIV, xnnpack_delegate.get());
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, height, width, channels})
|
||||||
|
.Input2Shape({batch, 1, 1, 1})
|
||||||
|
.Test(BuiltinOperator_DIV, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(Div, 4DBy4DBroadcastHeightWidthChannels) {
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(nullptr),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto height = shape_rng();
|
||||||
|
const auto width = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({1, height, width, channels})
|
||||||
|
.Input2Shape({batch, height, width, channels})
|
||||||
|
.Test(BuiltinOperator_DIV, xnnpack_delegate.get());
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, height, width, channels})
|
||||||
|
.Input2Shape({1, height, width, channels})
|
||||||
|
.Test(BuiltinOperator_DIV, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(Div, 4DBy3D) {
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(nullptr),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto height = shape_rng();
|
||||||
|
const auto width = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({height, width, channels})
|
||||||
|
.Input2Shape({batch, height, width, channels})
|
||||||
|
.Test(BuiltinOperator_DIV, xnnpack_delegate.get());
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, height, width, channels})
|
||||||
|
.Input2Shape({height, width, channels})
|
||||||
|
.Test(BuiltinOperator_DIV, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(Div, 4DBy2D) {
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(nullptr),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto height = shape_rng();
|
||||||
|
const auto width = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({width, channels})
|
||||||
|
.Input2Shape({batch, height, width, channels})
|
||||||
|
.Test(BuiltinOperator_DIV, xnnpack_delegate.get());
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, height, width, channels})
|
||||||
|
.Input2Shape({width, channels})
|
||||||
|
.Test(BuiltinOperator_DIV, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(Div, 4DBy1D) {
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(nullptr),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto height = shape_rng();
|
||||||
|
const auto width = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({channels})
|
||||||
|
.Input2Shape({batch, height, width, channels})
|
||||||
|
.Test(BuiltinOperator_DIV, xnnpack_delegate.get());
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, height, width, channels})
|
||||||
|
.Input2Shape({channels})
|
||||||
|
.Test(BuiltinOperator_DIV, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(Div, 4DBy0D) {
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(nullptr),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto height = shape_rng();
|
||||||
|
const auto width = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({})
|
||||||
|
.Input2Shape({batch, height, width, channels})
|
||||||
|
.Test(BuiltinOperator_DIV, xnnpack_delegate.get());
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, height, width, channels})
|
||||||
|
.Input2Shape({})
|
||||||
|
.Test(BuiltinOperator_DIV, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(Div, 2DBy2D) {
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(nullptr),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, channels})
|
||||||
|
.Input2Shape({batch, channels})
|
||||||
|
.Test(BuiltinOperator_DIV, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(Div, 2DBy1D) {
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(nullptr),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({channels})
|
||||||
|
.Input2Shape({batch, channels})
|
||||||
|
.Test(BuiltinOperator_DIV, xnnpack_delegate.get());
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, channels})
|
||||||
|
.Input2Shape({channels})
|
||||||
|
.Test(BuiltinOperator_DIV, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(Div, 2DBy0D) {
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(nullptr),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({})
|
||||||
|
.Input2Shape({batch, channels})
|
||||||
|
.Test(BuiltinOperator_DIV, xnnpack_delegate.get());
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, channels})
|
||||||
|
.Input2Shape({})
|
||||||
|
.Test(BuiltinOperator_DIV, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(Div, 4DByStatic4D) {
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(nullptr),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto height = shape_rng();
|
||||||
|
const auto width = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, height, width, channels})
|
||||||
|
.Input2Shape({batch, height, width, channels})
|
||||||
|
.Input1Static(true)
|
||||||
|
.Test(BuiltinOperator_DIV, xnnpack_delegate.get());
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, height, width, channels})
|
||||||
|
.Input2Shape({batch, height, width, channels})
|
||||||
|
.Input2Static(true)
|
||||||
|
.Test(BuiltinOperator_DIV, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(Div, 4DByStatic4DBroadcastChannels) {
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(nullptr),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto height = shape_rng();
|
||||||
|
const auto width = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({1, 1, 1, channels})
|
||||||
|
.Input2Shape({batch, height, width, channels})
|
||||||
|
.Input1Static(true)
|
||||||
|
.Test(BuiltinOperator_DIV, xnnpack_delegate.get());
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, height, width, channels})
|
||||||
|
.Input2Shape({1, 1, 1, channels})
|
||||||
|
.Input2Static(true)
|
||||||
|
.Test(BuiltinOperator_DIV, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(Div, 4DByStatic4DBroadcastWidth) {
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(nullptr),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto height = shape_rng();
|
||||||
|
const auto width = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({1, 1, width, 1})
|
||||||
|
.Input2Shape({batch, height, width, channels})
|
||||||
|
.Input1Static(true)
|
||||||
|
.Test(BuiltinOperator_DIV, xnnpack_delegate.get());
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, height, width, channels})
|
||||||
|
.Input2Shape({1, 1, width, 1})
|
||||||
|
.Input2Static(true)
|
||||||
|
.Test(BuiltinOperator_DIV, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(Div, 4DByStatic4DBroadcastHeight) {
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(nullptr),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto height = shape_rng();
|
||||||
|
const auto width = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({1, height, 1, 1})
|
||||||
|
.Input2Shape({batch, height, width, channels})
|
||||||
|
.Input1Static(true)
|
||||||
|
.Test(BuiltinOperator_DIV, xnnpack_delegate.get());
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, height, width, channels})
|
||||||
|
.Input2Shape({1, height, 1, 1})
|
||||||
|
.Input2Static(true)
|
||||||
|
.Test(BuiltinOperator_DIV, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(Div, 4DByStatic4DBroadcastBatch) {
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(nullptr),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto height = shape_rng();
|
||||||
|
const auto width = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, 1, 1, 1})
|
||||||
|
.Input2Shape({batch, height, width, channels})
|
||||||
|
.Input1Static(true)
|
||||||
|
.Test(BuiltinOperator_DIV, xnnpack_delegate.get());
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, height, width, channels})
|
||||||
|
.Input2Shape({batch, 1, 1, 1})
|
||||||
|
.Input2Static(true)
|
||||||
|
.Test(BuiltinOperator_DIV, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(Div, 4DByStatic4DBroadcastHeightWidthChannels) {
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(nullptr),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto height = shape_rng();
|
||||||
|
const auto width = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({1, height, width, channels})
|
||||||
|
.Input2Shape({batch, height, width, channels})
|
||||||
|
.Input1Static(true)
|
||||||
|
.Test(BuiltinOperator_DIV, xnnpack_delegate.get());
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, height, width, channels})
|
||||||
|
.Input2Shape({1, height, width, channels})
|
||||||
|
.Input2Static(true)
|
||||||
|
.Test(BuiltinOperator_DIV, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(Div, 4DByStatic3D) {
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(nullptr),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto height = shape_rng();
|
||||||
|
const auto width = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({height, width, channels})
|
||||||
|
.Input2Shape({batch, height, width, channels})
|
||||||
|
.Input1Static(true)
|
||||||
|
.Test(BuiltinOperator_DIV, xnnpack_delegate.get());
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, height, width, channels})
|
||||||
|
.Input2Shape({height, width, channels})
|
||||||
|
.Input2Static(true)
|
||||||
|
.Test(BuiltinOperator_DIV, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(Div, 4DByStatic2D) {
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(nullptr),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto height = shape_rng();
|
||||||
|
const auto width = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({width, channels})
|
||||||
|
.Input2Shape({batch, height, width, channels})
|
||||||
|
.Input1Static(true)
|
||||||
|
.Test(BuiltinOperator_DIV, xnnpack_delegate.get());
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, height, width, channels})
|
||||||
|
.Input2Shape({width, channels})
|
||||||
|
.Input2Static(true)
|
||||||
|
.Test(BuiltinOperator_DIV, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(Div, 4DByStatic1D) {
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(nullptr),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto height = shape_rng();
|
||||||
|
const auto width = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({channels})
|
||||||
|
.Input2Shape({batch, height, width, channels})
|
||||||
|
.Input1Static(true)
|
||||||
|
.Test(BuiltinOperator_DIV, xnnpack_delegate.get());
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, height, width, channels})
|
||||||
|
.Input2Shape({channels})
|
||||||
|
.Input2Static(true)
|
||||||
|
.Test(BuiltinOperator_DIV, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(Div, 4DByStatic0D) {
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(nullptr),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto height = shape_rng();
|
||||||
|
const auto width = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({})
|
||||||
|
.Input2Shape({batch, height, width, channels})
|
||||||
|
.Input1Static(true)
|
||||||
|
.Test(BuiltinOperator_DIV, xnnpack_delegate.get());
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, height, width, channels})
|
||||||
|
.Input2Shape({})
|
||||||
|
.Input2Static(true)
|
||||||
|
.Test(BuiltinOperator_DIV, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(Div, 2DByStatic2D) {
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(nullptr),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, channels})
|
||||||
|
.Input2Shape({batch, channels})
|
||||||
|
.Input1Static(true)
|
||||||
|
.Test(BuiltinOperator_DIV, xnnpack_delegate.get());
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, channels})
|
||||||
|
.Input2Shape({batch, channels})
|
||||||
|
.Input2Static(true)
|
||||||
|
.Test(BuiltinOperator_DIV, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(Div, 2DByStatic1D) {
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(nullptr),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({channels})
|
||||||
|
.Input2Shape({batch, channels})
|
||||||
|
.Input1Static(true)
|
||||||
|
.Test(BuiltinOperator_DIV, xnnpack_delegate.get());
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, channels})
|
||||||
|
.Input2Shape({channels})
|
||||||
|
.Input2Static(true)
|
||||||
|
.Test(BuiltinOperator_DIV, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(Div, 2DByStatic0D) {
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(nullptr),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({})
|
||||||
|
.Input2Shape({batch, channels})
|
||||||
|
.Input1Static(true)
|
||||||
|
.Test(BuiltinOperator_DIV, xnnpack_delegate.get());
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, channels})
|
||||||
|
.Input2Shape({})
|
||||||
|
.Input2Static(true)
|
||||||
|
.Test(BuiltinOperator_DIV, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(Div, FP16Weights) {
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(nullptr),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto height = shape_rng();
|
||||||
|
const auto width = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, height, width, channels})
|
||||||
|
.Input2Shape({batch, height, width, channels})
|
||||||
|
.Input1Static(true)
|
||||||
|
.FP16Weights()
|
||||||
|
.Test(BuiltinOperator_DIV, xnnpack_delegate.get());
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, height, width, channels})
|
||||||
|
.Input2Shape({batch, height, width, channels})
|
||||||
|
.Input2Static(true)
|
||||||
|
.FP16Weights()
|
||||||
|
.Test(BuiltinOperator_DIV, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(Div, ReluActivation) {
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(nullptr),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto height = shape_rng();
|
||||||
|
const auto width = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, height, width, channels})
|
||||||
|
.Input2Shape({batch, height, width, channels})
|
||||||
|
.ReluActivation()
|
||||||
|
.Test(BuiltinOperator_DIV, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(Div, Relu6Activation) {
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(nullptr),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto height = shape_rng();
|
||||||
|
const auto width = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, height, width, channels})
|
||||||
|
.Input2Shape({batch, height, width, channels})
|
||||||
|
.Relu6Activation()
|
||||||
|
.Test(BuiltinOperator_DIV, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(Div, ReluMinus1To1Activation) {
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(nullptr),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto height = shape_rng();
|
||||||
|
const auto width = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, height, width, channels})
|
||||||
|
.Input2Shape({batch, height, width, channels})
|
||||||
|
.ReluMinus1To1Activation()
|
||||||
|
.Test(BuiltinOperator_DIV, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(Div, DISABLED_TanhActivation) {
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(nullptr),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto height = shape_rng();
|
||||||
|
const auto width = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, height, width, channels})
|
||||||
|
.Input2Shape({batch, height, width, channels})
|
||||||
|
.TanhActivation()
|
||||||
|
.Test(BuiltinOperator_DIV, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(Div, DISABLED_SignBitActivation) {
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(nullptr),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto height = shape_rng();
|
||||||
|
const auto width = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, height, width, channels})
|
||||||
|
.Input2Shape({batch, height, width, channels})
|
||||||
|
.SignBitActivation()
|
||||||
|
.Test(BuiltinOperator_DIV, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(Div, MultiThreading) {
|
||||||
|
TfLiteXNNPackDelegateOptions delegate_options =
|
||||||
|
TfLiteXNNPackDelegateOptionsDefault();
|
||||||
|
delegate_options.num_threads = 2;
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(&delegate_options),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto height = shape_rng();
|
||||||
|
const auto width = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, height, width, channels})
|
||||||
|
.Input2Shape({batch, height, width, channels})
|
||||||
|
.Test(BuiltinOperator_DIV, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
} // namespace xnnpack
|
||||||
|
} // namespace tflite
|
735
tensorflow/lite/delegates/xnnpack/maximum_test.cc
Normal file
735
tensorflow/lite/delegates/xnnpack/maximum_test.cc
Normal file
@ -0,0 +1,735 @@
|
|||||||
|
/* Copyright 2020 The TensorFlow Authors. All Rights Reserved.
|
||||||
|
|
||||||
|
Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
|
you may not use this file except in compliance with the License.
|
||||||
|
You may obtain a copy of the License at
|
||||||
|
|
||||||
|
http://www.apache.org/licenses/LICENSE-2.0
|
||||||
|
|
||||||
|
Unless required by applicable law or agreed to in writing, software
|
||||||
|
distributed under the License is distributed on an "AS IS" BASIS,
|
||||||
|
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||||
|
See the License for the specific language governing permissions and
|
||||||
|
limitations under the License.
|
||||||
|
==============================================================================*/
|
||||||
|
|
||||||
|
#include <cstdint>
|
||||||
|
#include <functional>
|
||||||
|
#include <memory>
|
||||||
|
#include <random>
|
||||||
|
|
||||||
|
#include <gtest/gtest.h>
|
||||||
|
#include "tensorflow/lite/delegates/xnnpack/binary_elementwise_tester.h"
|
||||||
|
#include "tensorflow/lite/delegates/xnnpack/xnnpack_delegate.h"
|
||||||
|
|
||||||
|
namespace tflite {
|
||||||
|
namespace xnnpack {
|
||||||
|
|
||||||
|
TEST(Maximum, 4DBy4D) {
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(nullptr),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto height = shape_rng();
|
||||||
|
const auto width = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, height, width, channels})
|
||||||
|
.Input2Shape({batch, height, width, channels})
|
||||||
|
.Test(BuiltinOperator_MAXIMUM, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(Maximum, 4DBy4DBroadcastChannels) {
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(nullptr),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto height = shape_rng();
|
||||||
|
const auto width = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({1, 1, 1, channels})
|
||||||
|
.Input2Shape({batch, height, width, channels})
|
||||||
|
.Test(BuiltinOperator_MAXIMUM, xnnpack_delegate.get());
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, height, width, channels})
|
||||||
|
.Input2Shape({1, 1, 1, channels})
|
||||||
|
.Test(BuiltinOperator_MAXIMUM, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(Maximum, 4DBy4DBroadcastWidth) {
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(nullptr),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto height = shape_rng();
|
||||||
|
const auto width = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({1, 1, width, 1})
|
||||||
|
.Input2Shape({batch, height, width, channels})
|
||||||
|
.Test(BuiltinOperator_MAXIMUM, xnnpack_delegate.get());
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, height, width, channels})
|
||||||
|
.Input2Shape({1, 1, width, 1})
|
||||||
|
.Test(BuiltinOperator_MAXIMUM, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(Maximum, 4DBy4DBroadcastHeight) {
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(nullptr),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto height = shape_rng();
|
||||||
|
const auto width = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({1, height, 1, 1})
|
||||||
|
.Input2Shape({batch, height, width, channels})
|
||||||
|
.Test(BuiltinOperator_MAXIMUM, xnnpack_delegate.get());
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, height, width, channels})
|
||||||
|
.Input2Shape({1, height, 1, 1})
|
||||||
|
.Test(BuiltinOperator_MAXIMUM, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(Maximum, 4DBy4DBroadcastBatch) {
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(nullptr),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto height = shape_rng();
|
||||||
|
const auto width = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, 1, 1, 1})
|
||||||
|
.Input2Shape({batch, height, width, channels})
|
||||||
|
.Test(BuiltinOperator_MAXIMUM, xnnpack_delegate.get());
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, height, width, channels})
|
||||||
|
.Input2Shape({batch, 1, 1, 1})
|
||||||
|
.Test(BuiltinOperator_MAXIMUM, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(Maximum, 4DBy4DBroadcastHeightWidthChannels) {
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(nullptr),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto height = shape_rng();
|
||||||
|
const auto width = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({1, height, width, channels})
|
||||||
|
.Input2Shape({batch, height, width, channels})
|
||||||
|
.Test(BuiltinOperator_MAXIMUM, xnnpack_delegate.get());
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, height, width, channels})
|
||||||
|
.Input2Shape({1, height, width, channels})
|
||||||
|
.Test(BuiltinOperator_MAXIMUM, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(Maximum, 4DBy3D) {
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(nullptr),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto height = shape_rng();
|
||||||
|
const auto width = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({height, width, channels})
|
||||||
|
.Input2Shape({batch, height, width, channels})
|
||||||
|
.Test(BuiltinOperator_MAXIMUM, xnnpack_delegate.get());
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, height, width, channels})
|
||||||
|
.Input2Shape({height, width, channels})
|
||||||
|
.Test(BuiltinOperator_MAXIMUM, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(Maximum, 4DBy2D) {
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(nullptr),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto height = shape_rng();
|
||||||
|
const auto width = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({width, channels})
|
||||||
|
.Input2Shape({batch, height, width, channels})
|
||||||
|
.Test(BuiltinOperator_MAXIMUM, xnnpack_delegate.get());
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, height, width, channels})
|
||||||
|
.Input2Shape({width, channels})
|
||||||
|
.Test(BuiltinOperator_MAXIMUM, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(Maximum, 4DBy1D) {
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(nullptr),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto height = shape_rng();
|
||||||
|
const auto width = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({channels})
|
||||||
|
.Input2Shape({batch, height, width, channels})
|
||||||
|
.Test(BuiltinOperator_MAXIMUM, xnnpack_delegate.get());
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, height, width, channels})
|
||||||
|
.Input2Shape({channels})
|
||||||
|
.Test(BuiltinOperator_MAXIMUM, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(Maximum, 4DBy0D) {
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(nullptr),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto height = shape_rng();
|
||||||
|
const auto width = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({})
|
||||||
|
.Input2Shape({batch, height, width, channels})
|
||||||
|
.Test(BuiltinOperator_MAXIMUM, xnnpack_delegate.get());
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, height, width, channels})
|
||||||
|
.Input2Shape({})
|
||||||
|
.Test(BuiltinOperator_MAXIMUM, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(Maximum, 2DBy2D) {
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(nullptr),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, channels})
|
||||||
|
.Input2Shape({batch, channels})
|
||||||
|
.Test(BuiltinOperator_MAXIMUM, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(Maximum, 2DBy1D) {
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(nullptr),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({channels})
|
||||||
|
.Input2Shape({batch, channels})
|
||||||
|
.Test(BuiltinOperator_MAXIMUM, xnnpack_delegate.get());
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, channels})
|
||||||
|
.Input2Shape({channels})
|
||||||
|
.Test(BuiltinOperator_MAXIMUM, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(Maximum, 2DBy0D) {
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(nullptr),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({})
|
||||||
|
.Input2Shape({batch, channels})
|
||||||
|
.Test(BuiltinOperator_MAXIMUM, xnnpack_delegate.get());
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, channels})
|
||||||
|
.Input2Shape({})
|
||||||
|
.Test(BuiltinOperator_MAXIMUM, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(Maximum, 4DByStatic4D) {
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(nullptr),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto height = shape_rng();
|
||||||
|
const auto width = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, height, width, channels})
|
||||||
|
.Input2Shape({batch, height, width, channels})
|
||||||
|
.Input1Static(true)
|
||||||
|
.Test(BuiltinOperator_MAXIMUM, xnnpack_delegate.get());
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, height, width, channels})
|
||||||
|
.Input2Shape({batch, height, width, channels})
|
||||||
|
.Input2Static(true)
|
||||||
|
.Test(BuiltinOperator_MAXIMUM, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(Maximum, 4DByStatic4DBroadcastChannels) {
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(nullptr),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto height = shape_rng();
|
||||||
|
const auto width = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({1, 1, 1, channels})
|
||||||
|
.Input2Shape({batch, height, width, channels})
|
||||||
|
.Input1Static(true)
|
||||||
|
.Test(BuiltinOperator_MAXIMUM, xnnpack_delegate.get());
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, height, width, channels})
|
||||||
|
.Input2Shape({1, 1, 1, channels})
|
||||||
|
.Input2Static(true)
|
||||||
|
.Test(BuiltinOperator_MAXIMUM, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(Maximum, 4DByStatic4DBroadcastWidth) {
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(nullptr),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto height = shape_rng();
|
||||||
|
const auto width = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({1, 1, width, 1})
|
||||||
|
.Input2Shape({batch, height, width, channels})
|
||||||
|
.Input1Static(true)
|
||||||
|
.Test(BuiltinOperator_MAXIMUM, xnnpack_delegate.get());
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, height, width, channels})
|
||||||
|
.Input2Shape({1, 1, width, 1})
|
||||||
|
.Input2Static(true)
|
||||||
|
.Test(BuiltinOperator_MAXIMUM, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(Maximum, 4DByStatic4DBroadcastHeight) {
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(nullptr),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto height = shape_rng();
|
||||||
|
const auto width = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({1, height, 1, 1})
|
||||||
|
.Input2Shape({batch, height, width, channels})
|
||||||
|
.Input1Static(true)
|
||||||
|
.Test(BuiltinOperator_MAXIMUM, xnnpack_delegate.get());
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, height, width, channels})
|
||||||
|
.Input2Shape({1, height, 1, 1})
|
||||||
|
.Input2Static(true)
|
||||||
|
.Test(BuiltinOperator_MAXIMUM, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(Maximum, 4DByStatic4DBroadcastBatch) {
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(nullptr),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto height = shape_rng();
|
||||||
|
const auto width = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, 1, 1, 1})
|
||||||
|
.Input2Shape({batch, height, width, channels})
|
||||||
|
.Input1Static(true)
|
||||||
|
.Test(BuiltinOperator_MAXIMUM, xnnpack_delegate.get());
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, height, width, channels})
|
||||||
|
.Input2Shape({batch, 1, 1, 1})
|
||||||
|
.Input2Static(true)
|
||||||
|
.Test(BuiltinOperator_MAXIMUM, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(Maximum, 4DByStatic4DBroadcastHeightWidthChannels) {
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(nullptr),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto height = shape_rng();
|
||||||
|
const auto width = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({1, height, width, channels})
|
||||||
|
.Input2Shape({batch, height, width, channels})
|
||||||
|
.Input1Static(true)
|
||||||
|
.Test(BuiltinOperator_MAXIMUM, xnnpack_delegate.get());
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, height, width, channels})
|
||||||
|
.Input2Shape({1, height, width, channels})
|
||||||
|
.Input2Static(true)
|
||||||
|
.Test(BuiltinOperator_MAXIMUM, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(Maximum, 4DByStatic3D) {
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(nullptr),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto height = shape_rng();
|
||||||
|
const auto width = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({height, width, channels})
|
||||||
|
.Input2Shape({batch, height, width, channels})
|
||||||
|
.Input1Static(true)
|
||||||
|
.Test(BuiltinOperator_MAXIMUM, xnnpack_delegate.get());
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, height, width, channels})
|
||||||
|
.Input2Shape({height, width, channels})
|
||||||
|
.Input2Static(true)
|
||||||
|
.Test(BuiltinOperator_MAXIMUM, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(Maximum, 4DByStatic2D) {
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(nullptr),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto height = shape_rng();
|
||||||
|
const auto width = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({width, channels})
|
||||||
|
.Input2Shape({batch, height, width, channels})
|
||||||
|
.Input1Static(true)
|
||||||
|
.Test(BuiltinOperator_MAXIMUM, xnnpack_delegate.get());
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, height, width, channels})
|
||||||
|
.Input2Shape({width, channels})
|
||||||
|
.Input2Static(true)
|
||||||
|
.Test(BuiltinOperator_MAXIMUM, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(Maximum, 4DByStatic1D) {
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(nullptr),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto height = shape_rng();
|
||||||
|
const auto width = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({channels})
|
||||||
|
.Input2Shape({batch, height, width, channels})
|
||||||
|
.Input1Static(true)
|
||||||
|
.Test(BuiltinOperator_MAXIMUM, xnnpack_delegate.get());
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, height, width, channels})
|
||||||
|
.Input2Shape({channels})
|
||||||
|
.Input2Static(true)
|
||||||
|
.Test(BuiltinOperator_MAXIMUM, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(Maximum, 4DByStatic0D) {
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(nullptr),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto height = shape_rng();
|
||||||
|
const auto width = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({})
|
||||||
|
.Input2Shape({batch, height, width, channels})
|
||||||
|
.Input1Static(true)
|
||||||
|
.Test(BuiltinOperator_MAXIMUM, xnnpack_delegate.get());
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, height, width, channels})
|
||||||
|
.Input2Shape({})
|
||||||
|
.Input2Static(true)
|
||||||
|
.Test(BuiltinOperator_MAXIMUM, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(Maximum, 2DByStatic2D) {
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(nullptr),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, channels})
|
||||||
|
.Input2Shape({batch, channels})
|
||||||
|
.Input1Static(true)
|
||||||
|
.Test(BuiltinOperator_MAXIMUM, xnnpack_delegate.get());
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, channels})
|
||||||
|
.Input2Shape({batch, channels})
|
||||||
|
.Input2Static(true)
|
||||||
|
.Test(BuiltinOperator_MAXIMUM, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(Maximum, 2DByStatic1D) {
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(nullptr),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({channels})
|
||||||
|
.Input2Shape({batch, channels})
|
||||||
|
.Input1Static(true)
|
||||||
|
.Test(BuiltinOperator_MAXIMUM, xnnpack_delegate.get());
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, channels})
|
||||||
|
.Input2Shape({channels})
|
||||||
|
.Input2Static(true)
|
||||||
|
.Test(BuiltinOperator_MAXIMUM, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(Maximum, 2DByStatic0D) {
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(nullptr),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({})
|
||||||
|
.Input2Shape({batch, channels})
|
||||||
|
.Input1Static(true)
|
||||||
|
.Test(BuiltinOperator_MAXIMUM, xnnpack_delegate.get());
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, channels})
|
||||||
|
.Input2Shape({})
|
||||||
|
.Input2Static(true)
|
||||||
|
.Test(BuiltinOperator_MAXIMUM, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(Maximum, FP16Weights) {
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(nullptr),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto height = shape_rng();
|
||||||
|
const auto width = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, height, width, channels})
|
||||||
|
.Input2Shape({batch, height, width, channels})
|
||||||
|
.Input1Static(true)
|
||||||
|
.FP16Weights()
|
||||||
|
.Test(BuiltinOperator_MAXIMUM, xnnpack_delegate.get());
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, height, width, channels})
|
||||||
|
.Input2Shape({batch, height, width, channels})
|
||||||
|
.Input2Static(true)
|
||||||
|
.FP16Weights()
|
||||||
|
.Test(BuiltinOperator_MAXIMUM, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(Maximum, MultiThreading) {
|
||||||
|
TfLiteXNNPackDelegateOptions delegate_options =
|
||||||
|
TfLiteXNNPackDelegateOptionsDefault();
|
||||||
|
delegate_options.num_threads = 2;
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(&delegate_options),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto height = shape_rng();
|
||||||
|
const auto width = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, height, width, channels})
|
||||||
|
.Input2Shape({batch, height, width, channels})
|
||||||
|
.Test(BuiltinOperator_MAXIMUM, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
} // namespace xnnpack
|
||||||
|
} // namespace tflite
|
735
tensorflow/lite/delegates/xnnpack/minimum_test.cc
Normal file
735
tensorflow/lite/delegates/xnnpack/minimum_test.cc
Normal file
@ -0,0 +1,735 @@
|
|||||||
|
/* Copyright 2020 The TensorFlow Authors. All Rights Reserved.
|
||||||
|
|
||||||
|
Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
|
you may not use this file except in compliance with the License.
|
||||||
|
You may obtain a copy of the License at
|
||||||
|
|
||||||
|
http://www.apache.org/licenses/LICENSE-2.0
|
||||||
|
|
||||||
|
Unless required by applicable law or agreed to in writing, software
|
||||||
|
distributed under the License is distributed on an "AS IS" BASIS,
|
||||||
|
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||||
|
See the License for the specific language governing permissions and
|
||||||
|
limitations under the License.
|
||||||
|
==============================================================================*/
|
||||||
|
|
||||||
|
#include <cstdint>
|
||||||
|
#include <functional>
|
||||||
|
#include <memory>
|
||||||
|
#include <random>
|
||||||
|
|
||||||
|
#include <gtest/gtest.h>
|
||||||
|
#include "tensorflow/lite/delegates/xnnpack/binary_elementwise_tester.h"
|
||||||
|
#include "tensorflow/lite/delegates/xnnpack/xnnpack_delegate.h"
|
||||||
|
|
||||||
|
namespace tflite {
|
||||||
|
namespace xnnpack {
|
||||||
|
|
||||||
|
TEST(Minimum, 4DBy4D) {
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(nullptr),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto height = shape_rng();
|
||||||
|
const auto width = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, height, width, channels})
|
||||||
|
.Input2Shape({batch, height, width, channels})
|
||||||
|
.Test(BuiltinOperator_MINIMUM, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(Minimum, 4DBy4DBroadcastChannels) {
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(nullptr),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto height = shape_rng();
|
||||||
|
const auto width = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({1, 1, 1, channels})
|
||||||
|
.Input2Shape({batch, height, width, channels})
|
||||||
|
.Test(BuiltinOperator_MINIMUM, xnnpack_delegate.get());
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, height, width, channels})
|
||||||
|
.Input2Shape({1, 1, 1, channels})
|
||||||
|
.Test(BuiltinOperator_MINIMUM, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(Minimum, 4DBy4DBroadcastWidth) {
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(nullptr),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto height = shape_rng();
|
||||||
|
const auto width = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({1, 1, width, 1})
|
||||||
|
.Input2Shape({batch, height, width, channels})
|
||||||
|
.Test(BuiltinOperator_MINIMUM, xnnpack_delegate.get());
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, height, width, channels})
|
||||||
|
.Input2Shape({1, 1, width, 1})
|
||||||
|
.Test(BuiltinOperator_MINIMUM, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(Minimum, 4DBy4DBroadcastHeight) {
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(nullptr),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto height = shape_rng();
|
||||||
|
const auto width = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({1, height, 1, 1})
|
||||||
|
.Input2Shape({batch, height, width, channels})
|
||||||
|
.Test(BuiltinOperator_MINIMUM, xnnpack_delegate.get());
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, height, width, channels})
|
||||||
|
.Input2Shape({1, height, 1, 1})
|
||||||
|
.Test(BuiltinOperator_MINIMUM, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(Minimum, 4DBy4DBroadcastBatch) {
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(nullptr),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto height = shape_rng();
|
||||||
|
const auto width = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, 1, 1, 1})
|
||||||
|
.Input2Shape({batch, height, width, channels})
|
||||||
|
.Test(BuiltinOperator_MINIMUM, xnnpack_delegate.get());
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, height, width, channels})
|
||||||
|
.Input2Shape({batch, 1, 1, 1})
|
||||||
|
.Test(BuiltinOperator_MINIMUM, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(Minimum, 4DBy4DBroadcastHeightWidthChannels) {
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(nullptr),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto height = shape_rng();
|
||||||
|
const auto width = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({1, height, width, channels})
|
||||||
|
.Input2Shape({batch, height, width, channels})
|
||||||
|
.Test(BuiltinOperator_MINIMUM, xnnpack_delegate.get());
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, height, width, channels})
|
||||||
|
.Input2Shape({1, height, width, channels})
|
||||||
|
.Test(BuiltinOperator_MINIMUM, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(Minimum, 4DBy3D) {
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(nullptr),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto height = shape_rng();
|
||||||
|
const auto width = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({height, width, channels})
|
||||||
|
.Input2Shape({batch, height, width, channels})
|
||||||
|
.Test(BuiltinOperator_MINIMUM, xnnpack_delegate.get());
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, height, width, channels})
|
||||||
|
.Input2Shape({height, width, channels})
|
||||||
|
.Test(BuiltinOperator_MINIMUM, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(Minimum, 4DBy2D) {
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(nullptr),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto height = shape_rng();
|
||||||
|
const auto width = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({width, channels})
|
||||||
|
.Input2Shape({batch, height, width, channels})
|
||||||
|
.Test(BuiltinOperator_MINIMUM, xnnpack_delegate.get());
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, height, width, channels})
|
||||||
|
.Input2Shape({width, channels})
|
||||||
|
.Test(BuiltinOperator_MINIMUM, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(Minimum, 4DBy1D) {
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(nullptr),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto height = shape_rng();
|
||||||
|
const auto width = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({channels})
|
||||||
|
.Input2Shape({batch, height, width, channels})
|
||||||
|
.Test(BuiltinOperator_MINIMUM, xnnpack_delegate.get());
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, height, width, channels})
|
||||||
|
.Input2Shape({channels})
|
||||||
|
.Test(BuiltinOperator_MINIMUM, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(Minimum, 4DBy0D) {
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(nullptr),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto height = shape_rng();
|
||||||
|
const auto width = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({})
|
||||||
|
.Input2Shape({batch, height, width, channels})
|
||||||
|
.Test(BuiltinOperator_MINIMUM, xnnpack_delegate.get());
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, height, width, channels})
|
||||||
|
.Input2Shape({})
|
||||||
|
.Test(BuiltinOperator_MINIMUM, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(Minimum, 2DBy2D) {
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(nullptr),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, channels})
|
||||||
|
.Input2Shape({batch, channels})
|
||||||
|
.Test(BuiltinOperator_MINIMUM, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(Minimum, 2DBy1D) {
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(nullptr),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({channels})
|
||||||
|
.Input2Shape({batch, channels})
|
||||||
|
.Test(BuiltinOperator_MINIMUM, xnnpack_delegate.get());
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, channels})
|
||||||
|
.Input2Shape({channels})
|
||||||
|
.Test(BuiltinOperator_MINIMUM, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(Minimum, 2DBy0D) {
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(nullptr),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({})
|
||||||
|
.Input2Shape({batch, channels})
|
||||||
|
.Test(BuiltinOperator_MINIMUM, xnnpack_delegate.get());
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, channels})
|
||||||
|
.Input2Shape({})
|
||||||
|
.Test(BuiltinOperator_MINIMUM, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(Minimum, 4DByStatic4D) {
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(nullptr),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto height = shape_rng();
|
||||||
|
const auto width = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, height, width, channels})
|
||||||
|
.Input2Shape({batch, height, width, channels})
|
||||||
|
.Input1Static(true)
|
||||||
|
.Test(BuiltinOperator_MINIMUM, xnnpack_delegate.get());
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, height, width, channels})
|
||||||
|
.Input2Shape({batch, height, width, channels})
|
||||||
|
.Input2Static(true)
|
||||||
|
.Test(BuiltinOperator_MINIMUM, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(Minimum, 4DByStatic4DBroadcastChannels) {
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(nullptr),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto height = shape_rng();
|
||||||
|
const auto width = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({1, 1, 1, channels})
|
||||||
|
.Input2Shape({batch, height, width, channels})
|
||||||
|
.Input1Static(true)
|
||||||
|
.Test(BuiltinOperator_MINIMUM, xnnpack_delegate.get());
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, height, width, channels})
|
||||||
|
.Input2Shape({1, 1, 1, channels})
|
||||||
|
.Input2Static(true)
|
||||||
|
.Test(BuiltinOperator_MINIMUM, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(Minimum, 4DByStatic4DBroadcastWidth) {
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(nullptr),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto height = shape_rng();
|
||||||
|
const auto width = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({1, 1, width, 1})
|
||||||
|
.Input2Shape({batch, height, width, channels})
|
||||||
|
.Input1Static(true)
|
||||||
|
.Test(BuiltinOperator_MINIMUM, xnnpack_delegate.get());
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, height, width, channels})
|
||||||
|
.Input2Shape({1, 1, width, 1})
|
||||||
|
.Input2Static(true)
|
||||||
|
.Test(BuiltinOperator_MINIMUM, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(Minimum, 4DByStatic4DBroadcastHeight) {
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(nullptr),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto height = shape_rng();
|
||||||
|
const auto width = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({1, height, 1, 1})
|
||||||
|
.Input2Shape({batch, height, width, channels})
|
||||||
|
.Input1Static(true)
|
||||||
|
.Test(BuiltinOperator_MINIMUM, xnnpack_delegate.get());
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, height, width, channels})
|
||||||
|
.Input2Shape({1, height, 1, 1})
|
||||||
|
.Input2Static(true)
|
||||||
|
.Test(BuiltinOperator_MINIMUM, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(Minimum, 4DByStatic4DBroadcastBatch) {
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(nullptr),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto height = shape_rng();
|
||||||
|
const auto width = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, 1, 1, 1})
|
||||||
|
.Input2Shape({batch, height, width, channels})
|
||||||
|
.Input1Static(true)
|
||||||
|
.Test(BuiltinOperator_MINIMUM, xnnpack_delegate.get());
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, height, width, channels})
|
||||||
|
.Input2Shape({batch, 1, 1, 1})
|
||||||
|
.Input2Static(true)
|
||||||
|
.Test(BuiltinOperator_MINIMUM, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(Minimum, 4DByStatic4DBroadcastHeightWidthChannels) {
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(nullptr),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto height = shape_rng();
|
||||||
|
const auto width = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({1, height, width, channels})
|
||||||
|
.Input2Shape({batch, height, width, channels})
|
||||||
|
.Input1Static(true)
|
||||||
|
.Test(BuiltinOperator_MINIMUM, xnnpack_delegate.get());
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, height, width, channels})
|
||||||
|
.Input2Shape({1, height, width, channels})
|
||||||
|
.Input2Static(true)
|
||||||
|
.Test(BuiltinOperator_MINIMUM, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(Minimum, 4DByStatic3D) {
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(nullptr),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto height = shape_rng();
|
||||||
|
const auto width = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({height, width, channels})
|
||||||
|
.Input2Shape({batch, height, width, channels})
|
||||||
|
.Input1Static(true)
|
||||||
|
.Test(BuiltinOperator_MINIMUM, xnnpack_delegate.get());
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, height, width, channels})
|
||||||
|
.Input2Shape({height, width, channels})
|
||||||
|
.Input2Static(true)
|
||||||
|
.Test(BuiltinOperator_MINIMUM, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(Minimum, 4DByStatic2D) {
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(nullptr),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto height = shape_rng();
|
||||||
|
const auto width = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({width, channels})
|
||||||
|
.Input2Shape({batch, height, width, channels})
|
||||||
|
.Input1Static(true)
|
||||||
|
.Test(BuiltinOperator_MINIMUM, xnnpack_delegate.get());
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, height, width, channels})
|
||||||
|
.Input2Shape({width, channels})
|
||||||
|
.Input2Static(true)
|
||||||
|
.Test(BuiltinOperator_MINIMUM, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(Minimum, 4DByStatic1D) {
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(nullptr),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto height = shape_rng();
|
||||||
|
const auto width = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({channels})
|
||||||
|
.Input2Shape({batch, height, width, channels})
|
||||||
|
.Input1Static(true)
|
||||||
|
.Test(BuiltinOperator_MINIMUM, xnnpack_delegate.get());
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, height, width, channels})
|
||||||
|
.Input2Shape({channels})
|
||||||
|
.Input2Static(true)
|
||||||
|
.Test(BuiltinOperator_MINIMUM, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(Minimum, 4DByStatic0D) {
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(nullptr),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto height = shape_rng();
|
||||||
|
const auto width = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({})
|
||||||
|
.Input2Shape({batch, height, width, channels})
|
||||||
|
.Input1Static(true)
|
||||||
|
.Test(BuiltinOperator_MINIMUM, xnnpack_delegate.get());
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, height, width, channels})
|
||||||
|
.Input2Shape({})
|
||||||
|
.Input2Static(true)
|
||||||
|
.Test(BuiltinOperator_MINIMUM, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(Minimum, 2DByStatic2D) {
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(nullptr),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, channels})
|
||||||
|
.Input2Shape({batch, channels})
|
||||||
|
.Input1Static(true)
|
||||||
|
.Test(BuiltinOperator_MINIMUM, xnnpack_delegate.get());
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, channels})
|
||||||
|
.Input2Shape({batch, channels})
|
||||||
|
.Input2Static(true)
|
||||||
|
.Test(BuiltinOperator_MINIMUM, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(Minimum, 2DByStatic1D) {
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(nullptr),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({channels})
|
||||||
|
.Input2Shape({batch, channels})
|
||||||
|
.Input1Static(true)
|
||||||
|
.Test(BuiltinOperator_MINIMUM, xnnpack_delegate.get());
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, channels})
|
||||||
|
.Input2Shape({channels})
|
||||||
|
.Input2Static(true)
|
||||||
|
.Test(BuiltinOperator_MINIMUM, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(Minimum, 2DByStatic0D) {
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(nullptr),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({})
|
||||||
|
.Input2Shape({batch, channels})
|
||||||
|
.Input1Static(true)
|
||||||
|
.Test(BuiltinOperator_MINIMUM, xnnpack_delegate.get());
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, channels})
|
||||||
|
.Input2Shape({})
|
||||||
|
.Input2Static(true)
|
||||||
|
.Test(BuiltinOperator_MINIMUM, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(Minimum, FP16Weights) {
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(nullptr),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto height = shape_rng();
|
||||||
|
const auto width = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, height, width, channels})
|
||||||
|
.Input2Shape({batch, height, width, channels})
|
||||||
|
.Input1Static(true)
|
||||||
|
.FP16Weights()
|
||||||
|
.Test(BuiltinOperator_MINIMUM, xnnpack_delegate.get());
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, height, width, channels})
|
||||||
|
.Input2Shape({batch, height, width, channels})
|
||||||
|
.Input2Static(true)
|
||||||
|
.FP16Weights()
|
||||||
|
.Test(BuiltinOperator_MINIMUM, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(Minimum, MultiThreading) {
|
||||||
|
TfLiteXNNPackDelegateOptions delegate_options =
|
||||||
|
TfLiteXNNPackDelegateOptionsDefault();
|
||||||
|
delegate_options.num_threads = 2;
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(&delegate_options),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto height = shape_rng();
|
||||||
|
const auto width = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, height, width, channels})
|
||||||
|
.Input2Shape({batch, height, width, channels})
|
||||||
|
.Test(BuiltinOperator_MINIMUM, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
} // namespace xnnpack
|
||||||
|
} // namespace tflite
|
735
tensorflow/lite/delegates/xnnpack/squared_difference_test.cc
Normal file
735
tensorflow/lite/delegates/xnnpack/squared_difference_test.cc
Normal file
@ -0,0 +1,735 @@
|
|||||||
|
/* Copyright 2020 The TensorFlow Authors. All Rights Reserved.
|
||||||
|
|
||||||
|
Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
|
you may not use this file except in compliance with the License.
|
||||||
|
You may obtain a copy of the License at
|
||||||
|
|
||||||
|
http://www.apache.org/licenses/LICENSE-2.0
|
||||||
|
|
||||||
|
Unless required by applicable law or agreed to in writing, software
|
||||||
|
distributed under the License is distributed on an "AS IS" BASIS,
|
||||||
|
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||||
|
See the License for the specific language governing permissions and
|
||||||
|
limitations under the License.
|
||||||
|
==============================================================================*/
|
||||||
|
|
||||||
|
#include <cstdint>
|
||||||
|
#include <functional>
|
||||||
|
#include <memory>
|
||||||
|
#include <random>
|
||||||
|
|
||||||
|
#include <gtest/gtest.h>
|
||||||
|
#include "tensorflow/lite/delegates/xnnpack/binary_elementwise_tester.h"
|
||||||
|
#include "tensorflow/lite/delegates/xnnpack/xnnpack_delegate.h"
|
||||||
|
|
||||||
|
namespace tflite {
|
||||||
|
namespace xnnpack {
|
||||||
|
|
||||||
|
TEST(SquaredDifference, 4DBy4D) {
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(nullptr),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto height = shape_rng();
|
||||||
|
const auto width = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, height, width, channels})
|
||||||
|
.Input2Shape({batch, height, width, channels})
|
||||||
|
.Test(BuiltinOperator_SQUARED_DIFFERENCE, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(SquaredDifference, 4DBy4DBroadcastChannels) {
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(nullptr),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto height = shape_rng();
|
||||||
|
const auto width = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({1, 1, 1, channels})
|
||||||
|
.Input2Shape({batch, height, width, channels})
|
||||||
|
.Test(BuiltinOperator_SQUARED_DIFFERENCE, xnnpack_delegate.get());
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, height, width, channels})
|
||||||
|
.Input2Shape({1, 1, 1, channels})
|
||||||
|
.Test(BuiltinOperator_SQUARED_DIFFERENCE, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(SquaredDifference, 4DBy4DBroadcastWidth) {
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(nullptr),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto height = shape_rng();
|
||||||
|
const auto width = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({1, 1, width, 1})
|
||||||
|
.Input2Shape({batch, height, width, channels})
|
||||||
|
.Test(BuiltinOperator_SQUARED_DIFFERENCE, xnnpack_delegate.get());
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, height, width, channels})
|
||||||
|
.Input2Shape({1, 1, width, 1})
|
||||||
|
.Test(BuiltinOperator_SQUARED_DIFFERENCE, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(SquaredDifference, 4DBy4DBroadcastHeight) {
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(nullptr),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto height = shape_rng();
|
||||||
|
const auto width = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({1, height, 1, 1})
|
||||||
|
.Input2Shape({batch, height, width, channels})
|
||||||
|
.Test(BuiltinOperator_SQUARED_DIFFERENCE, xnnpack_delegate.get());
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, height, width, channels})
|
||||||
|
.Input2Shape({1, height, 1, 1})
|
||||||
|
.Test(BuiltinOperator_SQUARED_DIFFERENCE, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(SquaredDifference, 4DBy4DBroadcastBatch) {
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(nullptr),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto height = shape_rng();
|
||||||
|
const auto width = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, 1, 1, 1})
|
||||||
|
.Input2Shape({batch, height, width, channels})
|
||||||
|
.Test(BuiltinOperator_SQUARED_DIFFERENCE, xnnpack_delegate.get());
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, height, width, channels})
|
||||||
|
.Input2Shape({batch, 1, 1, 1})
|
||||||
|
.Test(BuiltinOperator_SQUARED_DIFFERENCE, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(SquaredDifference, 4DBy4DBroadcastHeightWidthChannels) {
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(nullptr),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto height = shape_rng();
|
||||||
|
const auto width = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({1, height, width, channels})
|
||||||
|
.Input2Shape({batch, height, width, channels})
|
||||||
|
.Test(BuiltinOperator_SQUARED_DIFFERENCE, xnnpack_delegate.get());
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, height, width, channels})
|
||||||
|
.Input2Shape({1, height, width, channels})
|
||||||
|
.Test(BuiltinOperator_SQUARED_DIFFERENCE, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(SquaredDifference, 4DBy3D) {
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(nullptr),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto height = shape_rng();
|
||||||
|
const auto width = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({height, width, channels})
|
||||||
|
.Input2Shape({batch, height, width, channels})
|
||||||
|
.Test(BuiltinOperator_SQUARED_DIFFERENCE, xnnpack_delegate.get());
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, height, width, channels})
|
||||||
|
.Input2Shape({height, width, channels})
|
||||||
|
.Test(BuiltinOperator_SQUARED_DIFFERENCE, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(SquaredDifference, 4DBy2D) {
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(nullptr),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto height = shape_rng();
|
||||||
|
const auto width = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({width, channels})
|
||||||
|
.Input2Shape({batch, height, width, channels})
|
||||||
|
.Test(BuiltinOperator_SQUARED_DIFFERENCE, xnnpack_delegate.get());
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, height, width, channels})
|
||||||
|
.Input2Shape({width, channels})
|
||||||
|
.Test(BuiltinOperator_SQUARED_DIFFERENCE, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(SquaredDifference, 4DBy1D) {
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(nullptr),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto height = shape_rng();
|
||||||
|
const auto width = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({channels})
|
||||||
|
.Input2Shape({batch, height, width, channels})
|
||||||
|
.Test(BuiltinOperator_SQUARED_DIFFERENCE, xnnpack_delegate.get());
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, height, width, channels})
|
||||||
|
.Input2Shape({channels})
|
||||||
|
.Test(BuiltinOperator_SQUARED_DIFFERENCE, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(SquaredDifference, 4DBy0D) {
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(nullptr),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto height = shape_rng();
|
||||||
|
const auto width = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({})
|
||||||
|
.Input2Shape({batch, height, width, channels})
|
||||||
|
.Test(BuiltinOperator_SQUARED_DIFFERENCE, xnnpack_delegate.get());
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, height, width, channels})
|
||||||
|
.Input2Shape({})
|
||||||
|
.Test(BuiltinOperator_SQUARED_DIFFERENCE, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(SquaredDifference, 2DBy2D) {
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(nullptr),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, channels})
|
||||||
|
.Input2Shape({batch, channels})
|
||||||
|
.Test(BuiltinOperator_SQUARED_DIFFERENCE, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(SquaredDifference, 2DBy1D) {
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(nullptr),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({channels})
|
||||||
|
.Input2Shape({batch, channels})
|
||||||
|
.Test(BuiltinOperator_SQUARED_DIFFERENCE, xnnpack_delegate.get());
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, channels})
|
||||||
|
.Input2Shape({channels})
|
||||||
|
.Test(BuiltinOperator_SQUARED_DIFFERENCE, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(SquaredDifference, 2DBy0D) {
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(nullptr),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({})
|
||||||
|
.Input2Shape({batch, channels})
|
||||||
|
.Test(BuiltinOperator_SQUARED_DIFFERENCE, xnnpack_delegate.get());
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, channels})
|
||||||
|
.Input2Shape({})
|
||||||
|
.Test(BuiltinOperator_SQUARED_DIFFERENCE, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(SquaredDifference, 4DByStatic4D) {
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(nullptr),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto height = shape_rng();
|
||||||
|
const auto width = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, height, width, channels})
|
||||||
|
.Input2Shape({batch, height, width, channels})
|
||||||
|
.Input1Static(true)
|
||||||
|
.Test(BuiltinOperator_SQUARED_DIFFERENCE, xnnpack_delegate.get());
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, height, width, channels})
|
||||||
|
.Input2Shape({batch, height, width, channels})
|
||||||
|
.Input2Static(true)
|
||||||
|
.Test(BuiltinOperator_SQUARED_DIFFERENCE, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(SquaredDifference, 4DByStatic4DBroadcastChannels) {
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(nullptr),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto height = shape_rng();
|
||||||
|
const auto width = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({1, 1, 1, channels})
|
||||||
|
.Input2Shape({batch, height, width, channels})
|
||||||
|
.Input1Static(true)
|
||||||
|
.Test(BuiltinOperator_SQUARED_DIFFERENCE, xnnpack_delegate.get());
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, height, width, channels})
|
||||||
|
.Input2Shape({1, 1, 1, channels})
|
||||||
|
.Input2Static(true)
|
||||||
|
.Test(BuiltinOperator_SQUARED_DIFFERENCE, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(SquaredDifference, 4DByStatic4DBroadcastWidth) {
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(nullptr),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto height = shape_rng();
|
||||||
|
const auto width = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({1, 1, width, 1})
|
||||||
|
.Input2Shape({batch, height, width, channels})
|
||||||
|
.Input1Static(true)
|
||||||
|
.Test(BuiltinOperator_SQUARED_DIFFERENCE, xnnpack_delegate.get());
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, height, width, channels})
|
||||||
|
.Input2Shape({1, 1, width, 1})
|
||||||
|
.Input2Static(true)
|
||||||
|
.Test(BuiltinOperator_SQUARED_DIFFERENCE, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(SquaredDifference, 4DByStatic4DBroadcastHeight) {
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(nullptr),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto height = shape_rng();
|
||||||
|
const auto width = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({1, height, 1, 1})
|
||||||
|
.Input2Shape({batch, height, width, channels})
|
||||||
|
.Input1Static(true)
|
||||||
|
.Test(BuiltinOperator_SQUARED_DIFFERENCE, xnnpack_delegate.get());
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, height, width, channels})
|
||||||
|
.Input2Shape({1, height, 1, 1})
|
||||||
|
.Input2Static(true)
|
||||||
|
.Test(BuiltinOperator_SQUARED_DIFFERENCE, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(SquaredDifference, 4DByStatic4DBroadcastBatch) {
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(nullptr),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto height = shape_rng();
|
||||||
|
const auto width = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, 1, 1, 1})
|
||||||
|
.Input2Shape({batch, height, width, channels})
|
||||||
|
.Input1Static(true)
|
||||||
|
.Test(BuiltinOperator_SQUARED_DIFFERENCE, xnnpack_delegate.get());
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, height, width, channels})
|
||||||
|
.Input2Shape({batch, 1, 1, 1})
|
||||||
|
.Input2Static(true)
|
||||||
|
.Test(BuiltinOperator_SQUARED_DIFFERENCE, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(SquaredDifference, 4DByStatic4DBroadcastHeightWidthChannels) {
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(nullptr),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto height = shape_rng();
|
||||||
|
const auto width = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({1, height, width, channels})
|
||||||
|
.Input2Shape({batch, height, width, channels})
|
||||||
|
.Input1Static(true)
|
||||||
|
.Test(BuiltinOperator_SQUARED_DIFFERENCE, xnnpack_delegate.get());
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, height, width, channels})
|
||||||
|
.Input2Shape({1, height, width, channels})
|
||||||
|
.Input2Static(true)
|
||||||
|
.Test(BuiltinOperator_SQUARED_DIFFERENCE, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(SquaredDifference, 4DByStatic3D) {
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(nullptr),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto height = shape_rng();
|
||||||
|
const auto width = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({height, width, channels})
|
||||||
|
.Input2Shape({batch, height, width, channels})
|
||||||
|
.Input1Static(true)
|
||||||
|
.Test(BuiltinOperator_SQUARED_DIFFERENCE, xnnpack_delegate.get());
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, height, width, channels})
|
||||||
|
.Input2Shape({height, width, channels})
|
||||||
|
.Input2Static(true)
|
||||||
|
.Test(BuiltinOperator_SQUARED_DIFFERENCE, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(SquaredDifference, 4DByStatic2D) {
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(nullptr),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto height = shape_rng();
|
||||||
|
const auto width = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({width, channels})
|
||||||
|
.Input2Shape({batch, height, width, channels})
|
||||||
|
.Input1Static(true)
|
||||||
|
.Test(BuiltinOperator_SQUARED_DIFFERENCE, xnnpack_delegate.get());
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, height, width, channels})
|
||||||
|
.Input2Shape({width, channels})
|
||||||
|
.Input2Static(true)
|
||||||
|
.Test(BuiltinOperator_SQUARED_DIFFERENCE, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(SquaredDifference, 4DByStatic1D) {
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(nullptr),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto height = shape_rng();
|
||||||
|
const auto width = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({channels})
|
||||||
|
.Input2Shape({batch, height, width, channels})
|
||||||
|
.Input1Static(true)
|
||||||
|
.Test(BuiltinOperator_SQUARED_DIFFERENCE, xnnpack_delegate.get());
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, height, width, channels})
|
||||||
|
.Input2Shape({channels})
|
||||||
|
.Input2Static(true)
|
||||||
|
.Test(BuiltinOperator_SQUARED_DIFFERENCE, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(SquaredDifference, 4DByStatic0D) {
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(nullptr),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto height = shape_rng();
|
||||||
|
const auto width = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({})
|
||||||
|
.Input2Shape({batch, height, width, channels})
|
||||||
|
.Input1Static(true)
|
||||||
|
.Test(BuiltinOperator_SQUARED_DIFFERENCE, xnnpack_delegate.get());
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, height, width, channels})
|
||||||
|
.Input2Shape({})
|
||||||
|
.Input2Static(true)
|
||||||
|
.Test(BuiltinOperator_SQUARED_DIFFERENCE, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(SquaredDifference, 2DByStatic2D) {
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(nullptr),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, channels})
|
||||||
|
.Input2Shape({batch, channels})
|
||||||
|
.Input1Static(true)
|
||||||
|
.Test(BuiltinOperator_SQUARED_DIFFERENCE, xnnpack_delegate.get());
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, channels})
|
||||||
|
.Input2Shape({batch, channels})
|
||||||
|
.Input2Static(true)
|
||||||
|
.Test(BuiltinOperator_SQUARED_DIFFERENCE, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(SquaredDifference, 2DByStatic1D) {
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(nullptr),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({channels})
|
||||||
|
.Input2Shape({batch, channels})
|
||||||
|
.Input1Static(true)
|
||||||
|
.Test(BuiltinOperator_SQUARED_DIFFERENCE, xnnpack_delegate.get());
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, channels})
|
||||||
|
.Input2Shape({channels})
|
||||||
|
.Input2Static(true)
|
||||||
|
.Test(BuiltinOperator_SQUARED_DIFFERENCE, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(SquaredDifference, 2DByStatic0D) {
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(nullptr),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({})
|
||||||
|
.Input2Shape({batch, channels})
|
||||||
|
.Input1Static(true)
|
||||||
|
.Test(BuiltinOperator_SQUARED_DIFFERENCE, xnnpack_delegate.get());
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, channels})
|
||||||
|
.Input2Shape({})
|
||||||
|
.Input2Static(true)
|
||||||
|
.Test(BuiltinOperator_SQUARED_DIFFERENCE, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(SquaredDifference, FP16Weights) {
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(nullptr),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto height = shape_rng();
|
||||||
|
const auto width = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, height, width, channels})
|
||||||
|
.Input2Shape({batch, height, width, channels})
|
||||||
|
.Input1Static(true)
|
||||||
|
.FP16Weights()
|
||||||
|
.Test(BuiltinOperator_SQUARED_DIFFERENCE, xnnpack_delegate.get());
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, height, width, channels})
|
||||||
|
.Input2Shape({batch, height, width, channels})
|
||||||
|
.Input2Static(true)
|
||||||
|
.FP16Weights()
|
||||||
|
.Test(BuiltinOperator_SQUARED_DIFFERENCE, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(SquaredDifference, MultiThreading) {
|
||||||
|
TfLiteXNNPackDelegateOptions delegate_options =
|
||||||
|
TfLiteXNNPackDelegateOptionsDefault();
|
||||||
|
delegate_options.num_threads = 2;
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(&delegate_options),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto height = shape_rng();
|
||||||
|
const auto width = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, height, width, channels})
|
||||||
|
.Input2Shape({batch, height, width, channels})
|
||||||
|
.Test(BuiltinOperator_SQUARED_DIFFERENCE, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
} // namespace xnnpack
|
||||||
|
} // namespace tflite
|
840
tensorflow/lite/delegates/xnnpack/sub_test.cc
Normal file
840
tensorflow/lite/delegates/xnnpack/sub_test.cc
Normal file
@ -0,0 +1,840 @@
|
|||||||
|
/* Copyright 2020 The TensorFlow Authors. All Rights Reserved.
|
||||||
|
|
||||||
|
Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
|
you may not use this file except in compliance with the License.
|
||||||
|
You may obtain a copy of the License at
|
||||||
|
|
||||||
|
http://www.apache.org/licenses/LICENSE-2.0
|
||||||
|
|
||||||
|
Unless required by applicable law or agreed to in writing, software
|
||||||
|
distributed under the License is distributed on an "AS IS" BASIS,
|
||||||
|
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||||
|
See the License for the specific language governing permissions and
|
||||||
|
limitations under the License.
|
||||||
|
==============================================================================*/
|
||||||
|
|
||||||
|
#include <cstdint>
|
||||||
|
#include <functional>
|
||||||
|
#include <memory>
|
||||||
|
#include <random>
|
||||||
|
|
||||||
|
#include <gtest/gtest.h>
|
||||||
|
#include "tensorflow/lite/delegates/xnnpack/binary_elementwise_tester.h"
|
||||||
|
#include "tensorflow/lite/delegates/xnnpack/xnnpack_delegate.h"
|
||||||
|
|
||||||
|
namespace tflite {
|
||||||
|
namespace xnnpack {
|
||||||
|
|
||||||
|
TEST(Sub, 4DBy4D) {
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(nullptr),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto height = shape_rng();
|
||||||
|
const auto width = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, height, width, channels})
|
||||||
|
.Input2Shape({batch, height, width, channels})
|
||||||
|
.Test(BuiltinOperator_SUB, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(Sub, 4DBy4DBroadcastChannels) {
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(nullptr),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto height = shape_rng();
|
||||||
|
const auto width = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({1, 1, 1, channels})
|
||||||
|
.Input2Shape({batch, height, width, channels})
|
||||||
|
.Test(BuiltinOperator_SUB, xnnpack_delegate.get());
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, height, width, channels})
|
||||||
|
.Input2Shape({1, 1, 1, channels})
|
||||||
|
.Test(BuiltinOperator_SUB, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(Sub, 4DBy4DBroadcastWidth) {
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(nullptr),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto height = shape_rng();
|
||||||
|
const auto width = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({1, 1, width, 1})
|
||||||
|
.Input2Shape({batch, height, width, channels})
|
||||||
|
.Test(BuiltinOperator_SUB, xnnpack_delegate.get());
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, height, width, channels})
|
||||||
|
.Input2Shape({1, 1, width, 1})
|
||||||
|
.Test(BuiltinOperator_SUB, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(Sub, 4DBy4DBroadcastHeight) {
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(nullptr),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto height = shape_rng();
|
||||||
|
const auto width = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({1, height, 1, 1})
|
||||||
|
.Input2Shape({batch, height, width, channels})
|
||||||
|
.Test(BuiltinOperator_SUB, xnnpack_delegate.get());
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, height, width, channels})
|
||||||
|
.Input2Shape({1, height, 1, 1})
|
||||||
|
.Test(BuiltinOperator_SUB, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(Sub, 4DBy4DBroadcastBatch) {
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(nullptr),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto height = shape_rng();
|
||||||
|
const auto width = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, 1, 1, 1})
|
||||||
|
.Input2Shape({batch, height, width, channels})
|
||||||
|
.Test(BuiltinOperator_SUB, xnnpack_delegate.get());
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, height, width, channels})
|
||||||
|
.Input2Shape({batch, 1, 1, 1})
|
||||||
|
.Test(BuiltinOperator_SUB, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(Sub, 4DBy4DBroadcastHeightWidthChannels) {
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(nullptr),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto height = shape_rng();
|
||||||
|
const auto width = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({1, height, width, channels})
|
||||||
|
.Input2Shape({batch, height, width, channels})
|
||||||
|
.Test(BuiltinOperator_SUB, xnnpack_delegate.get());
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, height, width, channels})
|
||||||
|
.Input2Shape({1, height, width, channels})
|
||||||
|
.Test(BuiltinOperator_SUB, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(Sub, 4DBy3D) {
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(nullptr),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto height = shape_rng();
|
||||||
|
const auto width = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({height, width, channels})
|
||||||
|
.Input2Shape({batch, height, width, channels})
|
||||||
|
.Test(BuiltinOperator_SUB, xnnpack_delegate.get());
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, height, width, channels})
|
||||||
|
.Input2Shape({height, width, channels})
|
||||||
|
.Test(BuiltinOperator_SUB, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(Sub, 4DBy2D) {
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(nullptr),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto height = shape_rng();
|
||||||
|
const auto width = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({width, channels})
|
||||||
|
.Input2Shape({batch, height, width, channels})
|
||||||
|
.Test(BuiltinOperator_SUB, xnnpack_delegate.get());
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, height, width, channels})
|
||||||
|
.Input2Shape({width, channels})
|
||||||
|
.Test(BuiltinOperator_SUB, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(Sub, 4DBy1D) {
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(nullptr),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto height = shape_rng();
|
||||||
|
const auto width = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({channels})
|
||||||
|
.Input2Shape({batch, height, width, channels})
|
||||||
|
.Test(BuiltinOperator_SUB, xnnpack_delegate.get());
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, height, width, channels})
|
||||||
|
.Input2Shape({channels})
|
||||||
|
.Test(BuiltinOperator_SUB, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(Sub, 4DBy0D) {
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(nullptr),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto height = shape_rng();
|
||||||
|
const auto width = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({})
|
||||||
|
.Input2Shape({batch, height, width, channels})
|
||||||
|
.Test(BuiltinOperator_SUB, xnnpack_delegate.get());
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, height, width, channels})
|
||||||
|
.Input2Shape({})
|
||||||
|
.Test(BuiltinOperator_SUB, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(Sub, 2DBy2D) {
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(nullptr),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, channels})
|
||||||
|
.Input2Shape({batch, channels})
|
||||||
|
.Test(BuiltinOperator_SUB, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(Sub, 2DBy1D) {
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(nullptr),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({channels})
|
||||||
|
.Input2Shape({batch, channels})
|
||||||
|
.Test(BuiltinOperator_SUB, xnnpack_delegate.get());
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, channels})
|
||||||
|
.Input2Shape({channels})
|
||||||
|
.Test(BuiltinOperator_SUB, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(Sub, 2DBy0D) {
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(nullptr),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({})
|
||||||
|
.Input2Shape({batch, channels})
|
||||||
|
.Test(BuiltinOperator_SUB, xnnpack_delegate.get());
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, channels})
|
||||||
|
.Input2Shape({})
|
||||||
|
.Test(BuiltinOperator_SUB, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(Sub, 4DByStatic4D) {
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(nullptr),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto height = shape_rng();
|
||||||
|
const auto width = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, height, width, channels})
|
||||||
|
.Input2Shape({batch, height, width, channels})
|
||||||
|
.Input1Static(true)
|
||||||
|
.Test(BuiltinOperator_SUB, xnnpack_delegate.get());
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, height, width, channels})
|
||||||
|
.Input2Shape({batch, height, width, channels})
|
||||||
|
.Input2Static(true)
|
||||||
|
.Test(BuiltinOperator_SUB, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(Sub, 4DByStatic4DBroadcastChannels) {
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(nullptr),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto height = shape_rng();
|
||||||
|
const auto width = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({1, 1, 1, channels})
|
||||||
|
.Input2Shape({batch, height, width, channels})
|
||||||
|
.Input1Static(true)
|
||||||
|
.Test(BuiltinOperator_SUB, xnnpack_delegate.get());
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, height, width, channels})
|
||||||
|
.Input2Shape({1, 1, 1, channels})
|
||||||
|
.Input2Static(true)
|
||||||
|
.Test(BuiltinOperator_SUB, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(Sub, 4DByStatic4DBroadcastWidth) {
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(nullptr),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto height = shape_rng();
|
||||||
|
const auto width = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({1, 1, width, 1})
|
||||||
|
.Input2Shape({batch, height, width, channels})
|
||||||
|
.Input1Static(true)
|
||||||
|
.Test(BuiltinOperator_SUB, xnnpack_delegate.get());
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, height, width, channels})
|
||||||
|
.Input2Shape({1, 1, width, 1})
|
||||||
|
.Input2Static(true)
|
||||||
|
.Test(BuiltinOperator_SUB, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(Sub, 4DByStatic4DBroadcastHeight) {
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(nullptr),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto height = shape_rng();
|
||||||
|
const auto width = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({1, height, 1, 1})
|
||||||
|
.Input2Shape({batch, height, width, channels})
|
||||||
|
.Input1Static(true)
|
||||||
|
.Test(BuiltinOperator_SUB, xnnpack_delegate.get());
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, height, width, channels})
|
||||||
|
.Input2Shape({1, height, 1, 1})
|
||||||
|
.Input2Static(true)
|
||||||
|
.Test(BuiltinOperator_SUB, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(Sub, 4DByStatic4DBroadcastBatch) {
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(nullptr),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto height = shape_rng();
|
||||||
|
const auto width = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, 1, 1, 1})
|
||||||
|
.Input2Shape({batch, height, width, channels})
|
||||||
|
.Input1Static(true)
|
||||||
|
.Test(BuiltinOperator_SUB, xnnpack_delegate.get());
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, height, width, channels})
|
||||||
|
.Input2Shape({batch, 1, 1, 1})
|
||||||
|
.Input2Static(true)
|
||||||
|
.Test(BuiltinOperator_SUB, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(Sub, 4DByStatic4DBroadcastHeightWidthChannels) {
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(nullptr),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto height = shape_rng();
|
||||||
|
const auto width = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({1, height, width, channels})
|
||||||
|
.Input2Shape({batch, height, width, channels})
|
||||||
|
.Input1Static(true)
|
||||||
|
.Test(BuiltinOperator_SUB, xnnpack_delegate.get());
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, height, width, channels})
|
||||||
|
.Input2Shape({1, height, width, channels})
|
||||||
|
.Input2Static(true)
|
||||||
|
.Test(BuiltinOperator_SUB, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(Sub, 4DByStatic3D) {
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(nullptr),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto height = shape_rng();
|
||||||
|
const auto width = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({height, width, channels})
|
||||||
|
.Input2Shape({batch, height, width, channels})
|
||||||
|
.Input1Static(true)
|
||||||
|
.Test(BuiltinOperator_SUB, xnnpack_delegate.get());
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, height, width, channels})
|
||||||
|
.Input2Shape({height, width, channels})
|
||||||
|
.Input2Static(true)
|
||||||
|
.Test(BuiltinOperator_SUB, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(Sub, 4DByStatic2D) {
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(nullptr),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto height = shape_rng();
|
||||||
|
const auto width = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({width, channels})
|
||||||
|
.Input2Shape({batch, height, width, channels})
|
||||||
|
.Input1Static(true)
|
||||||
|
.Test(BuiltinOperator_SUB, xnnpack_delegate.get());
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, height, width, channels})
|
||||||
|
.Input2Shape({width, channels})
|
||||||
|
.Input2Static(true)
|
||||||
|
.Test(BuiltinOperator_SUB, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(Sub, 4DByStatic1D) {
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(nullptr),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto height = shape_rng();
|
||||||
|
const auto width = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({channels})
|
||||||
|
.Input2Shape({batch, height, width, channels})
|
||||||
|
.Input1Static(true)
|
||||||
|
.Test(BuiltinOperator_SUB, xnnpack_delegate.get());
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, height, width, channels})
|
||||||
|
.Input2Shape({channels})
|
||||||
|
.Input2Static(true)
|
||||||
|
.Test(BuiltinOperator_SUB, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(Sub, 4DByStatic0D) {
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(nullptr),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto height = shape_rng();
|
||||||
|
const auto width = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({})
|
||||||
|
.Input2Shape({batch, height, width, channels})
|
||||||
|
.Input1Static(true)
|
||||||
|
.Test(BuiltinOperator_SUB, xnnpack_delegate.get());
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, height, width, channels})
|
||||||
|
.Input2Shape({})
|
||||||
|
.Input2Static(true)
|
||||||
|
.Test(BuiltinOperator_SUB, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(Sub, 2DByStatic2D) {
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(nullptr),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, channels})
|
||||||
|
.Input2Shape({batch, channels})
|
||||||
|
.Input1Static(true)
|
||||||
|
.Test(BuiltinOperator_SUB, xnnpack_delegate.get());
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, channels})
|
||||||
|
.Input2Shape({batch, channels})
|
||||||
|
.Input2Static(true)
|
||||||
|
.Test(BuiltinOperator_SUB, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(Sub, 2DByStatic1D) {
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(nullptr),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({channels})
|
||||||
|
.Input2Shape({batch, channels})
|
||||||
|
.Input1Static(true)
|
||||||
|
.Test(BuiltinOperator_SUB, xnnpack_delegate.get());
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, channels})
|
||||||
|
.Input2Shape({channels})
|
||||||
|
.Input2Static(true)
|
||||||
|
.Test(BuiltinOperator_SUB, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(Sub, 2DByStatic0D) {
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(nullptr),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({})
|
||||||
|
.Input2Shape({batch, channels})
|
||||||
|
.Input1Static(true)
|
||||||
|
.Test(BuiltinOperator_SUB, xnnpack_delegate.get());
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, channels})
|
||||||
|
.Input2Shape({})
|
||||||
|
.Input2Static(true)
|
||||||
|
.Test(BuiltinOperator_SUB, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(Sub, FP16Weights) {
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(nullptr),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto height = shape_rng();
|
||||||
|
const auto width = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, height, width, channels})
|
||||||
|
.Input2Shape({batch, height, width, channels})
|
||||||
|
.Input1Static(true)
|
||||||
|
.FP16Weights()
|
||||||
|
.Test(BuiltinOperator_SUB, xnnpack_delegate.get());
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, height, width, channels})
|
||||||
|
.Input2Shape({batch, height, width, channels})
|
||||||
|
.Input2Static(true)
|
||||||
|
.FP16Weights()
|
||||||
|
.Test(BuiltinOperator_SUB, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(Sub, ReluActivation) {
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(nullptr),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto height = shape_rng();
|
||||||
|
const auto width = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, height, width, channels})
|
||||||
|
.Input2Shape({batch, height, width, channels})
|
||||||
|
.ReluActivation()
|
||||||
|
.Test(BuiltinOperator_SUB, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(Sub, Relu6Activation) {
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(nullptr),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto height = shape_rng();
|
||||||
|
const auto width = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, height, width, channels})
|
||||||
|
.Input2Shape({batch, height, width, channels})
|
||||||
|
.Relu6Activation()
|
||||||
|
.Test(BuiltinOperator_SUB, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(Sub, ReluMinus1To1Activation) {
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(nullptr),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto height = shape_rng();
|
||||||
|
const auto width = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, height, width, channels})
|
||||||
|
.Input2Shape({batch, height, width, channels})
|
||||||
|
.ReluMinus1To1Activation()
|
||||||
|
.Test(BuiltinOperator_SUB, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(Sub, DISABLED_TanhActivation) {
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(nullptr),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto height = shape_rng();
|
||||||
|
const auto width = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, height, width, channels})
|
||||||
|
.Input2Shape({batch, height, width, channels})
|
||||||
|
.TanhActivation()
|
||||||
|
.Test(BuiltinOperator_SUB, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(Sub, DISABLED_SignBitActivation) {
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(nullptr),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto height = shape_rng();
|
||||||
|
const auto width = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, height, width, channels})
|
||||||
|
.Input2Shape({batch, height, width, channels})
|
||||||
|
.SignBitActivation()
|
||||||
|
.Test(BuiltinOperator_SUB, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(Sub, MultiThreading) {
|
||||||
|
TfLiteXNNPackDelegateOptions delegate_options =
|
||||||
|
TfLiteXNNPackDelegateOptionsDefault();
|
||||||
|
delegate_options.num_threads = 2;
|
||||||
|
std::unique_ptr<TfLiteDelegate, decltype(&TfLiteXNNPackDelegateDelete)>
|
||||||
|
xnnpack_delegate(TfLiteXNNPackDelegateCreate(&delegate_options),
|
||||||
|
TfLiteXNNPackDelegateDelete);
|
||||||
|
|
||||||
|
std::random_device random_device;
|
||||||
|
auto rng = std::mt19937(random_device());
|
||||||
|
auto shape_rng =
|
||||||
|
std::bind(std::uniform_int_distribution<int32_t>(2, 5), std::ref(rng));
|
||||||
|
const auto batch = shape_rng();
|
||||||
|
const auto height = shape_rng();
|
||||||
|
const auto width = shape_rng();
|
||||||
|
const auto channels = shape_rng();
|
||||||
|
|
||||||
|
BinaryElementwiseTester()
|
||||||
|
.Input1Shape({batch, height, width, channels})
|
||||||
|
.Input2Shape({batch, height, width, channels})
|
||||||
|
.Test(BuiltinOperator_SUB, xnnpack_delegate.get());
|
||||||
|
}
|
||||||
|
|
||||||
|
} // namespace xnnpack
|
||||||
|
} // namespace tflite
|
@ -797,6 +797,13 @@ class Subgraph {
|
|||||||
node, context->tensors, dwconv_params,
|
node, context->tensors, dwconv_params,
|
||||||
quasi_static_tensors, xnnpack_tensors);
|
quasi_static_tensors, xnnpack_tensors);
|
||||||
}
|
}
|
||||||
|
case kTfLiteBuiltinDiv: {
|
||||||
|
const TfLiteDivParams* div_params =
|
||||||
|
static_cast<const TfLiteDivParams*>(node->builtin_data);
|
||||||
|
|
||||||
|
return VisitDivNode(subgraph, logging_context, node_index, node,
|
||||||
|
context->tensors, div_params, xnnpack_tensors);
|
||||||
|
}
|
||||||
case kTfLiteBuiltinFullyConnected: {
|
case kTfLiteBuiltinFullyConnected: {
|
||||||
const TfLiteFullyConnectedParams* fc_params =
|
const TfLiteFullyConnectedParams* fc_params =
|
||||||
static_cast<const TfLiteFullyConnectedParams*>(node->builtin_data);
|
static_cast<const TfLiteFullyConnectedParams*>(node->builtin_data);
|
||||||
@ -819,6 +826,12 @@ class Subgraph {
|
|||||||
context->tensors, pool_params,
|
context->tensors, pool_params,
|
||||||
xnnpack_tensors);
|
xnnpack_tensors);
|
||||||
}
|
}
|
||||||
|
case kTfLiteBuiltinMaximum:
|
||||||
|
return VisitMaximumNode(subgraph, logging_context, node_index, node,
|
||||||
|
context->tensors, xnnpack_tensors);
|
||||||
|
case kTfLiteBuiltinMinimum:
|
||||||
|
return VisitMinimumNode(subgraph, logging_context, node_index, node,
|
||||||
|
context->tensors, xnnpack_tensors);
|
||||||
case kTfLiteBuiltinMul: {
|
case kTfLiteBuiltinMul: {
|
||||||
const TfLiteMulParams* mul_params =
|
const TfLiteMulParams* mul_params =
|
||||||
static_cast<const TfLiteMulParams*>(node->builtin_data);
|
static_cast<const TfLiteMulParams*>(node->builtin_data);
|
||||||
@ -851,6 +864,17 @@ class Subgraph {
|
|||||||
context->tensors, softmax_params,
|
context->tensors, softmax_params,
|
||||||
xnnpack_tensors);
|
xnnpack_tensors);
|
||||||
}
|
}
|
||||||
|
case kTfLiteBuiltinSquaredDifference:
|
||||||
|
return VisitSquaredDifferenceNode(subgraph, logging_context, node_index,
|
||||||
|
node, context->tensors,
|
||||||
|
xnnpack_tensors);
|
||||||
|
case kTfLiteBuiltinSub: {
|
||||||
|
const TfLiteSubParams* sub_params =
|
||||||
|
static_cast<const TfLiteSubParams*>(node->builtin_data);
|
||||||
|
|
||||||
|
return VisitSubNode(subgraph, logging_context, node_index, node,
|
||||||
|
context->tensors, sub_params, xnnpack_tensors);
|
||||||
|
}
|
||||||
case kTfLiteBuiltinCustom: {
|
case kTfLiteBuiltinCustom: {
|
||||||
if (strcmp(registration->custom_name, "Convolution2DTransposeBias") ==
|
if (strcmp(registration->custom_name, "Convolution2DTransposeBias") ==
|
||||||
0) {
|
0) {
|
||||||
@ -1179,6 +1203,56 @@ class Subgraph {
|
|||||||
return kTfLiteOk;
|
return kTfLiteOk;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
static TfLiteStatus VisitDivNode(
|
||||||
|
xnn_subgraph_t subgraph, TfLiteContext* logging_context, int node_index,
|
||||||
|
TfLiteNode* node, const TfLiteTensor* tensors,
|
||||||
|
const TfLiteDivParams* div_params,
|
||||||
|
const std::vector<uint32_t>& xnnpack_tensors) {
|
||||||
|
TF_LITE_ENSURE_STATUS(
|
||||||
|
CheckNumInputsAndOutputs(logging_context, node, 2, 1, node_index));
|
||||||
|
|
||||||
|
const TfLiteTensor& input1_tensor = tensors[node->inputs->data[0]];
|
||||||
|
TF_LITE_ENSURE_STATUS(CheckTensorFloatType(
|
||||||
|
logging_context, input1_tensor, node->inputs->data[0], node_index));
|
||||||
|
TF_LITE_ENSURE_STATUS(CheckTensorNonDynamicAllocation(
|
||||||
|
logging_context, input1_tensor, node->inputs->data[0], node_index));
|
||||||
|
|
||||||
|
const TfLiteTensor& input2_tensor = tensors[node->inputs->data[1]];
|
||||||
|
TF_LITE_ENSURE_STATUS(CheckTensorFloatType(
|
||||||
|
logging_context, input2_tensor, node->inputs->data[1], node_index));
|
||||||
|
TF_LITE_ENSURE_STATUS(CheckTensorNonDynamicAllocation(
|
||||||
|
logging_context, input2_tensor, node->inputs->data[1], node_index));
|
||||||
|
|
||||||
|
const TfLiteTensor& output_tensor = tensors[node->outputs->data[0]];
|
||||||
|
TF_LITE_ENSURE_STATUS(CheckTensorFloatType(
|
||||||
|
logging_context, output_tensor, node->outputs->data[0], node_index));
|
||||||
|
TF_LITE_ENSURE_STATUS(CheckTensorNonDynamicAllocation(
|
||||||
|
logging_context, output_tensor, node->outputs->data[0], node_index));
|
||||||
|
|
||||||
|
float output_min = -std::numeric_limits<float>::infinity();
|
||||||
|
float output_max = +std::numeric_limits<float>::infinity();
|
||||||
|
if (div_params != nullptr) {
|
||||||
|
TF_LITE_ENSURE_STATUS(ConvertActivationToOutputRange(
|
||||||
|
logging_context, node_index, div_params->activation, &output_min,
|
||||||
|
&output_max));
|
||||||
|
}
|
||||||
|
|
||||||
|
if (subgraph != nullptr) {
|
||||||
|
const xnn_status status = xnn_define_divide(
|
||||||
|
subgraph, output_min, output_max,
|
||||||
|
/*input1_id=*/xnnpack_tensors[node->inputs->data[0]],
|
||||||
|
/*input2_id=*/xnnpack_tensors[node->inputs->data[1]],
|
||||||
|
/*output_id=*/xnnpack_tensors[node->outputs->data[0]], /*flags=*/0);
|
||||||
|
if (status != xnn_status_success) {
|
||||||
|
TF_LITE_KERNEL_LOG(logging_context, "failed to delegate DIV node #%d",
|
||||||
|
node_index);
|
||||||
|
return kTfLiteError;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return kTfLiteOk;
|
||||||
|
}
|
||||||
|
|
||||||
static TfLiteStatus VisitFullyConnectedNode(
|
static TfLiteStatus VisitFullyConnectedNode(
|
||||||
xnn_subgraph_t subgraph, TfLiteContext* logging_context, int node_index,
|
xnn_subgraph_t subgraph, TfLiteContext* logging_context, int node_index,
|
||||||
TfLiteNode* node, const TfLiteTensor* tensors,
|
TfLiteNode* node, const TfLiteTensor* tensors,
|
||||||
@ -1448,6 +1522,46 @@ class Subgraph {
|
|||||||
return kTfLiteOk;
|
return kTfLiteOk;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
static TfLiteStatus VisitMaximumNode(
|
||||||
|
xnn_subgraph_t subgraph, TfLiteContext* logging_context, int node_index,
|
||||||
|
TfLiteNode* node, const TfLiteTensor* tensors,
|
||||||
|
const std::vector<uint32_t>& xnnpack_tensors) {
|
||||||
|
TF_LITE_ENSURE_STATUS(
|
||||||
|
CheckNumInputsAndOutputs(logging_context, node, 2, 1, node_index));
|
||||||
|
|
||||||
|
const TfLiteTensor& input1_tensor = tensors[node->inputs->data[0]];
|
||||||
|
TF_LITE_ENSURE_STATUS(CheckTensorFloatType(
|
||||||
|
logging_context, input1_tensor, node->inputs->data[0], node_index));
|
||||||
|
TF_LITE_ENSURE_STATUS(CheckTensorNonDynamicAllocation(
|
||||||
|
logging_context, input1_tensor, node->inputs->data[0], node_index));
|
||||||
|
|
||||||
|
const TfLiteTensor& input2_tensor = tensors[node->inputs->data[1]];
|
||||||
|
TF_LITE_ENSURE_STATUS(CheckTensorFloatType(
|
||||||
|
logging_context, input2_tensor, node->inputs->data[1], node_index));
|
||||||
|
TF_LITE_ENSURE_STATUS(CheckTensorNonDynamicAllocation(
|
||||||
|
logging_context, input2_tensor, node->inputs->data[1], node_index));
|
||||||
|
|
||||||
|
const TfLiteTensor& output_tensor = tensors[node->outputs->data[0]];
|
||||||
|
TF_LITE_ENSURE_STATUS(CheckTensorFloatType(
|
||||||
|
logging_context, output_tensor, node->outputs->data[0], node_index));
|
||||||
|
TF_LITE_ENSURE_STATUS(CheckTensorNonDynamicAllocation(
|
||||||
|
logging_context, output_tensor, node->outputs->data[0], node_index));
|
||||||
|
|
||||||
|
if (subgraph != nullptr) {
|
||||||
|
const xnn_status status = xnn_define_maximum2(
|
||||||
|
subgraph, /*input1_id=*/xnnpack_tensors[node->inputs->data[0]],
|
||||||
|
/*input2_id=*/xnnpack_tensors[node->inputs->data[1]],
|
||||||
|
/*output_id=*/xnnpack_tensors[node->outputs->data[0]], /*flags=*/0);
|
||||||
|
if (status != xnn_status_success) {
|
||||||
|
TF_LITE_KERNEL_LOG(logging_context,
|
||||||
|
"failed to delegate MAXIMUM node #%d", node_index);
|
||||||
|
return kTfLiteError;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return kTfLiteOk;
|
||||||
|
}
|
||||||
|
|
||||||
static TfLiteStatus VisitMediaPipeDeconvolutionNode(
|
static TfLiteStatus VisitMediaPipeDeconvolutionNode(
|
||||||
xnn_subgraph_t subgraph, TfLiteContext* logging_context, int node_index,
|
xnn_subgraph_t subgraph, TfLiteContext* logging_context, int node_index,
|
||||||
TfLiteNode* node, const TfLiteTensor* tensors,
|
TfLiteNode* node, const TfLiteTensor* tensors,
|
||||||
@ -1672,6 +1786,46 @@ class Subgraph {
|
|||||||
return kTfLiteOk;
|
return kTfLiteOk;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
static TfLiteStatus VisitMinimumNode(
|
||||||
|
xnn_subgraph_t subgraph, TfLiteContext* logging_context, int node_index,
|
||||||
|
TfLiteNode* node, const TfLiteTensor* tensors,
|
||||||
|
const std::vector<uint32_t>& xnnpack_tensors) {
|
||||||
|
TF_LITE_ENSURE_STATUS(
|
||||||
|
CheckNumInputsAndOutputs(logging_context, node, 2, 1, node_index));
|
||||||
|
|
||||||
|
const TfLiteTensor& input1_tensor = tensors[node->inputs->data[0]];
|
||||||
|
TF_LITE_ENSURE_STATUS(CheckTensorFloatType(
|
||||||
|
logging_context, input1_tensor, node->inputs->data[0], node_index));
|
||||||
|
TF_LITE_ENSURE_STATUS(CheckTensorNonDynamicAllocation(
|
||||||
|
logging_context, input1_tensor, node->inputs->data[0], node_index));
|
||||||
|
|
||||||
|
const TfLiteTensor& input2_tensor = tensors[node->inputs->data[1]];
|
||||||
|
TF_LITE_ENSURE_STATUS(CheckTensorFloatType(
|
||||||
|
logging_context, input2_tensor, node->inputs->data[1], node_index));
|
||||||
|
TF_LITE_ENSURE_STATUS(CheckTensorNonDynamicAllocation(
|
||||||
|
logging_context, input2_tensor, node->inputs->data[1], node_index));
|
||||||
|
|
||||||
|
const TfLiteTensor& output_tensor = tensors[node->outputs->data[0]];
|
||||||
|
TF_LITE_ENSURE_STATUS(CheckTensorFloatType(
|
||||||
|
logging_context, output_tensor, node->outputs->data[0], node_index));
|
||||||
|
TF_LITE_ENSURE_STATUS(CheckTensorNonDynamicAllocation(
|
||||||
|
logging_context, output_tensor, node->outputs->data[0], node_index));
|
||||||
|
|
||||||
|
if (subgraph != nullptr) {
|
||||||
|
const xnn_status status = xnn_define_minimum2(
|
||||||
|
subgraph, /*input1_id=*/xnnpack_tensors[node->inputs->data[0]],
|
||||||
|
/*input2_id=*/xnnpack_tensors[node->inputs->data[1]],
|
||||||
|
/*output_id=*/xnnpack_tensors[node->outputs->data[0]], /*flags=*/0);
|
||||||
|
if (status != xnn_status_success) {
|
||||||
|
TF_LITE_KERNEL_LOG(logging_context,
|
||||||
|
"failed to delegate MINIMUM node #%d", node_index);
|
||||||
|
return kTfLiteError;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return kTfLiteOk;
|
||||||
|
}
|
||||||
|
|
||||||
static TfLiteStatus VisitMulNode(
|
static TfLiteStatus VisitMulNode(
|
||||||
xnn_subgraph_t subgraph, TfLiteContext* logging_context, int node_index,
|
xnn_subgraph_t subgraph, TfLiteContext* logging_context, int node_index,
|
||||||
TfLiteNode* node, const TfLiteTensor* tensors,
|
TfLiteNode* node, const TfLiteTensor* tensors,
|
||||||
@ -1928,6 +2082,97 @@ class Subgraph {
|
|||||||
return kTfLiteOk;
|
return kTfLiteOk;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
static TfLiteStatus VisitSquaredDifferenceNode(
|
||||||
|
xnn_subgraph_t subgraph, TfLiteContext* logging_context, int node_index,
|
||||||
|
TfLiteNode* node, const TfLiteTensor* tensors,
|
||||||
|
const std::vector<uint32_t>& xnnpack_tensors) {
|
||||||
|
TF_LITE_ENSURE_STATUS(
|
||||||
|
CheckNumInputsAndOutputs(logging_context, node, 2, 1, node_index));
|
||||||
|
|
||||||
|
const TfLiteTensor& input1_tensor = tensors[node->inputs->data[0]];
|
||||||
|
TF_LITE_ENSURE_STATUS(CheckTensorFloatType(
|
||||||
|
logging_context, input1_tensor, node->inputs->data[0], node_index));
|
||||||
|
TF_LITE_ENSURE_STATUS(CheckTensorNonDynamicAllocation(
|
||||||
|
logging_context, input1_tensor, node->inputs->data[0], node_index));
|
||||||
|
|
||||||
|
const TfLiteTensor& input2_tensor = tensors[node->inputs->data[1]];
|
||||||
|
TF_LITE_ENSURE_STATUS(CheckTensorFloatType(
|
||||||
|
logging_context, input2_tensor, node->inputs->data[1], node_index));
|
||||||
|
TF_LITE_ENSURE_STATUS(CheckTensorNonDynamicAllocation(
|
||||||
|
logging_context, input2_tensor, node->inputs->data[1], node_index));
|
||||||
|
|
||||||
|
const TfLiteTensor& output_tensor = tensors[node->outputs->data[0]];
|
||||||
|
TF_LITE_ENSURE_STATUS(CheckTensorFloatType(
|
||||||
|
logging_context, output_tensor, node->outputs->data[0], node_index));
|
||||||
|
TF_LITE_ENSURE_STATUS(CheckTensorNonDynamicAllocation(
|
||||||
|
logging_context, output_tensor, node->outputs->data[0], node_index));
|
||||||
|
|
||||||
|
if (subgraph != nullptr) {
|
||||||
|
const xnn_status status = xnn_define_squared_difference(
|
||||||
|
subgraph, /*input1_id=*/xnnpack_tensors[node->inputs->data[0]],
|
||||||
|
/*input2_id=*/xnnpack_tensors[node->inputs->data[1]],
|
||||||
|
/*output_id=*/xnnpack_tensors[node->outputs->data[0]], /*flags=*/0);
|
||||||
|
if (status != xnn_status_success) {
|
||||||
|
TF_LITE_KERNEL_LOG(logging_context,
|
||||||
|
"failed to delegate SQUARED_DIFFERENCE node #%d",
|
||||||
|
node_index);
|
||||||
|
return kTfLiteError;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return kTfLiteOk;
|
||||||
|
}
|
||||||
|
|
||||||
|
static TfLiteStatus VisitSubNode(
|
||||||
|
xnn_subgraph_t subgraph, TfLiteContext* logging_context, int node_index,
|
||||||
|
TfLiteNode* node, const TfLiteTensor* tensors,
|
||||||
|
const TfLiteSubParams* sub_params,
|
||||||
|
const std::vector<uint32_t>& xnnpack_tensors) {
|
||||||
|
TF_LITE_ENSURE_STATUS(
|
||||||
|
CheckNumInputsAndOutputs(logging_context, node, 2, 1, node_index));
|
||||||
|
|
||||||
|
const TfLiteTensor& input1_tensor = tensors[node->inputs->data[0]];
|
||||||
|
TF_LITE_ENSURE_STATUS(CheckTensorFloatType(
|
||||||
|
logging_context, input1_tensor, node->inputs->data[0], node_index));
|
||||||
|
TF_LITE_ENSURE_STATUS(CheckTensorNonDynamicAllocation(
|
||||||
|
logging_context, input1_tensor, node->inputs->data[0], node_index));
|
||||||
|
|
||||||
|
const TfLiteTensor& input2_tensor = tensors[node->inputs->data[1]];
|
||||||
|
TF_LITE_ENSURE_STATUS(CheckTensorFloatType(
|
||||||
|
logging_context, input2_tensor, node->inputs->data[1], node_index));
|
||||||
|
TF_LITE_ENSURE_STATUS(CheckTensorNonDynamicAllocation(
|
||||||
|
logging_context, input2_tensor, node->inputs->data[1], node_index));
|
||||||
|
|
||||||
|
const TfLiteTensor& output_tensor = tensors[node->outputs->data[0]];
|
||||||
|
TF_LITE_ENSURE_STATUS(CheckTensorFloatType(
|
||||||
|
logging_context, output_tensor, node->outputs->data[0], node_index));
|
||||||
|
TF_LITE_ENSURE_STATUS(CheckTensorNonDynamicAllocation(
|
||||||
|
logging_context, output_tensor, node->outputs->data[0], node_index));
|
||||||
|
|
||||||
|
float output_min = -std::numeric_limits<float>::infinity();
|
||||||
|
float output_max = +std::numeric_limits<float>::infinity();
|
||||||
|
if (sub_params != nullptr) {
|
||||||
|
TF_LITE_ENSURE_STATUS(ConvertActivationToOutputRange(
|
||||||
|
logging_context, node_index, sub_params->activation, &output_min,
|
||||||
|
&output_max));
|
||||||
|
}
|
||||||
|
|
||||||
|
if (subgraph != nullptr) {
|
||||||
|
const xnn_status status = xnn_define_subtract(
|
||||||
|
subgraph, output_min, output_max,
|
||||||
|
/*input1_id=*/xnnpack_tensors[node->inputs->data[0]],
|
||||||
|
/*input2_id=*/xnnpack_tensors[node->inputs->data[1]],
|
||||||
|
/*output_id=*/xnnpack_tensors[node->outputs->data[0]], /*flags=*/0);
|
||||||
|
if (status != xnn_status_success) {
|
||||||
|
TF_LITE_KERNEL_LOG(logging_context, "failed to delegate SUB node #%d",
|
||||||
|
node_index);
|
||||||
|
return kTfLiteError;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return kTfLiteOk;
|
||||||
|
}
|
||||||
|
|
||||||
private:
|
private:
|
||||||
Subgraph(xnn_runtime_t runtime, std::unordered_set<int>&& externals)
|
Subgraph(xnn_runtime_t runtime, std::unordered_set<int>&& externals)
|
||||||
: runtime_(runtime, &xnn_delete_runtime), externals_(externals) {}
|
: runtime_(runtime, &xnn_delete_runtime), externals_(externals) {}
|
||||||
|
@ -164,11 +164,11 @@ def tf_repositories(path_prefix = "", tf_repo_name = ""):
|
|||||||
|
|
||||||
tf_http_archive(
|
tf_http_archive(
|
||||||
name = "XNNPACK",
|
name = "XNNPACK",
|
||||||
sha256 = "dfa6181e238f0ca88a641952678cd7f3e38da541d8b731ce3fea1d0eeffb6101",
|
sha256 = "30b468db7d85b5f4afb3fd60947d690bc1c29d4eccca8fffeabe5b5328621c0e",
|
||||||
strip_prefix = "XNNPACK-b2217ddb5fa74db09d9da1326902269ae18e41ad",
|
strip_prefix = "XNNPACK-9d3a459441c272d82be14b579656b961066eba2c",
|
||||||
urls = [
|
urls = [
|
||||||
"https://storage.googleapis.com/mirror.tensorflow.org/github.com/google/XNNPACK/archive/b2217ddb5fa74db09d9da1326902269ae18e41ad.zip",
|
"https://storage.googleapis.com/mirror.tensorflow.org/github.com/google/XNNPACK/archive/9d3a459441c272d82be14b579656b961066eba2c.zip",
|
||||||
"https://github.com/google/XNNPACK/archive/b2217ddb5fa74db09d9da1326902269ae18e41ad.zip",
|
"https://github.com/google/XNNPACK/archive/9d3a459441c272d82be14b579656b961066eba2c.zip",
|
||||||
],
|
],
|
||||||
)
|
)
|
||||||
|
|
||||||
|
Loading…
Reference in New Issue
Block a user