Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
5 changes: 5 additions & 0 deletions python/tflite_micro/BUILD
Original file line number Diff line number Diff line change
Expand Up @@ -91,6 +91,10 @@ py_test(
"nomsan", # Python doesn't like these symbols in _runtime.so
"noubsan",
],
target_compatible_with = select({
"@bazel_tools//src/conditions:windows": ["@platforms//:incompatible"],
"//conditions:default": [],
}),
deps = [
":runtime",
requirement("numpy"),
Expand All @@ -110,6 +114,7 @@ py_test(
],
# Only compatible when compression is NOT enabled
target_compatible_with = select({
"@bazel_tools//src/conditions:windows": ["@platforms//:incompatible"],
"//:with_compression_enabled": ["@platforms//:incompatible"],
"//conditions:default": [],
}),
Expand Down
24 changes: 24 additions & 0 deletions tensorflow/lite/micro/compression/BUILD
Original file line number Diff line number Diff line change
Expand Up @@ -105,6 +105,10 @@ py_test(
size = "small",
srcs = ["metadata_test.py"],
main = "metadata_test.py",
target_compatible_with = select({
"@bazel_tools//src/conditions:windows": ["@platforms//:incompatible"],
"//conditions:default": [],
}),
deps = [
"metadata_py",
"@flatbuffers//:runtime_py",
Expand Down Expand Up @@ -150,6 +154,10 @@ py_test(
"nomsan", # Sanitizer symbols don't work with Python extension libs
"noubsan",
],
target_compatible_with = select({
"@bazel_tools//src/conditions:windows": ["@platforms//:incompatible"],
"//conditions:default": [],
}),
deps = [
":compress",
":metadata_py",
Expand All @@ -176,6 +184,10 @@ py_test(
name = "model_facade_test",
size = "small",
srcs = ["model_facade_test.py"],
target_compatible_with = select({
"@bazel_tools//src/conditions:windows": ["@platforms//:incompatible"],
"//conditions:default": [],
}),
deps = [
":model_facade",
":test_models",
Expand All @@ -195,6 +207,10 @@ py_test(
name = "spec_test",
size = "small",
srcs = ["spec_test.py"],
target_compatible_with = select({
"@bazel_tools//src/conditions:windows": ["@platforms//:incompatible"],
"//conditions:default": [],
}),
deps = [
":spec",
requirement("tensorflow"),
Expand All @@ -213,6 +229,10 @@ py_test(
name = "spec_builder_test",
size = "small",
srcs = ["spec_builder_test.py"],
target_compatible_with = select({
"@bazel_tools//src/conditions:windows": ["@platforms//:incompatible"],
"//conditions:default": [],
}),
deps = [
":spec",
":spec_builder",
Expand All @@ -234,6 +254,10 @@ py_test(
name = "test_models_test",
size = "small",
srcs = ["test_models_test.py"],
target_compatible_with = select({
"@bazel_tools//src/conditions:windows": ["@platforms//:incompatible"],
"//conditions:default": [],
}),
deps = [
":test_models",
"//tensorflow/lite/python:schema_py",
Expand Down
4 changes: 4 additions & 0 deletions tensorflow/lite/micro/examples/mnist_lstm/BUILD
Original file line number Diff line number Diff line change
Expand Up @@ -39,6 +39,10 @@ py_test(
"nomsan", # Python doesn't like these symbols
"noubsan",
],
target_compatible_with = select({
"@bazel_tools//src/conditions:windows": ["@platforms//:incompatible"],
"//conditions:default": [],
}),
deps = [
":evaluate",
":train",
Expand Down
4 changes: 4 additions & 0 deletions tensorflow/lite/micro/examples/recipes/BUILD
Original file line number Diff line number Diff line change
Expand Up @@ -33,6 +33,10 @@ py_test(
"nomsan", # Python doesn't like these symbols
"noubsan",
],
target_compatible_with = select({
"@bazel_tools//src/conditions:windows": ["@platforms//:incompatible"],
"//conditions:default": [],
}),
deps = [
":resource_variables_lib",
"//python/tflite_micro:runtime",
Expand Down
24 changes: 24 additions & 0 deletions tensorflow/lite/micro/kernels/BUILD
Original file line number Diff line number Diff line change
Expand Up @@ -540,6 +540,10 @@ tflm_cc_test(
"//tensorflow/lite/micro:test_helpers",
"//tensorflow/lite/micro/testing:micro_test",
],
target_compatible_with = select({
"@bazel_tools//src/conditions:windows": ["@platforms//:incompatible"],
"//conditions:default": [],
}),
)

tflm_cc_test(
Expand All @@ -554,6 +558,10 @@ tflm_cc_test(
"//tensorflow/lite/micro:test_helpers",
"//tensorflow/lite/micro/testing:micro_test",
],
target_compatible_with = select({
"@bazel_tools//src/conditions:windows": ["@platforms//:incompatible"],
"//conditions:default": [],
}),
)

tflm_cc_test(
Expand Down Expand Up @@ -908,6 +916,10 @@ tflm_cc_test(
"//tensorflow/lite/micro:test_helpers",
"//tensorflow/lite/micro/testing:micro_test",
],
target_compatible_with = select({
"@bazel_tools//src/conditions:windows": ["@platforms//:incompatible"],
"//conditions:default": [],
}),
)

tflm_cc_test(
Expand All @@ -922,6 +934,10 @@ tflm_cc_test(
"//tensorflow/lite/micro:test_helpers",
"//tensorflow/lite/micro/testing:micro_test",
],
target_compatible_with = select({
"@bazel_tools//src/conditions:windows": ["@platforms//:incompatible"],
"//conditions:default": [],
}),
)

tflm_cc_test(
Expand Down Expand Up @@ -1132,6 +1148,10 @@ tflm_cc_test(
"//tensorflow/lite/micro:debug_log",
"//tensorflow/lite/micro/testing:micro_test",
],
target_compatible_with = select({
"@bazel_tools//src/conditions:windows": ["@platforms//:incompatible"],
"//conditions:default": [],
}),
)

tflm_cc_test(
Expand Down Expand Up @@ -1586,6 +1606,10 @@ tflm_cc_test(
"//tensorflow/lite/micro:test_helpers",
"//tensorflow/lite/micro/testing:micro_test",
],
target_compatible_with = select({
"@bazel_tools//src/conditions:windows": ["@platforms//:incompatible"],
"//conditions:default": [],
}),
)

tflm_cc_test(
Expand Down
4 changes: 4 additions & 0 deletions tensorflow/lite/micro/python/tflite_size/tests/BUILD
Original file line number Diff line number Diff line change
Expand Up @@ -24,6 +24,10 @@ py_test(
"nomsan", # Python doesn't like these symbols from flatbuffer_size_wrapper_pybind.so
"noubsan",
],
target_compatible_with = select({
"@bazel_tools//src/conditions:windows": ["@platforms//:incompatible"],
"//conditions:default": [],
}),
deps = [
requirement("tensorflow"),
"//tensorflow/lite/micro/python/tflite_size/src:flatbuffer_size",
Expand Down
4 changes: 2 additions & 2 deletions tensorflow/lite/micro/test_helpers.cc
Original file line number Diff line number Diff line change
Expand Up @@ -451,11 +451,11 @@ const Model* BuildModelWithUnusedOperatorOutputs() {
TensorType_INT8, 0,
builder->CreateString("test_unused_output_tensor"), 0, false)};
constexpr size_t inputs_size = 0;
const int32_t inputs[inputs_size] = {};
const int32_t inputs[] = {0};
constexpr size_t outputs_size = 1;
const int32_t outputs[outputs_size] = {0};
constexpr size_t operator_inputs_size = 0;
const int32_t operator_inputs[operator_inputs_size] = {};
const int32_t operator_inputs[] = {0};
constexpr size_t operator_outputs_size = 2;
const int32_t operator_outputs[operator_outputs_size] = {0, 1};
constexpr size_t operators_size = 1;
Expand Down
4 changes: 4 additions & 0 deletions tensorflow/lite/micro/tools/BUILD
Original file line number Diff line number Diff line change
Expand Up @@ -69,6 +69,10 @@ py_test(
"nomsan", # Python doesn't like these symbols
"noubsan",
],
target_compatible_with = select({
"@bazel_tools//src/conditions:windows": ["@platforms//:incompatible"],
"//conditions:default": [],
}),
deps = [
":requantize_flatbuffer",
requirement("numpy"),
Expand Down
Loading