mirror of https://github.com/tensorflow/tfjs.git
80 lines
1.9 KiB
Python
80 lines
1.9 KiB
Python
# Copyright 2021 Google LLC. 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.
|
|
# =============================================================================
|
|
|
|
load("@npm//@bazel/concatjs:index.bzl", "ts_config")
|
|
load("//tools:tfjs_web_test.bzl", "grep_flag", "headless_flag")
|
|
|
|
package(default_visibility = ["//visibility:public"])
|
|
|
|
exports_files([
|
|
"babel.config.json",
|
|
"tsconfig.json",
|
|
"tsconfig_es5.json",
|
|
"tsconfig.test.json",
|
|
"tsconfig_base.json",
|
|
"tsconfig_ts_library.json",
|
|
])
|
|
|
|
ts_config(
|
|
name = "tsconfig_ts_library",
|
|
src = "tsconfig_ts_library.json",
|
|
deps = [
|
|
":tsconfig.json",
|
|
":tsconfig_base.json",
|
|
],
|
|
)
|
|
|
|
grep_flag(
|
|
name = "grep",
|
|
build_setting_default = "",
|
|
visibility = ["//visibility:public"],
|
|
)
|
|
|
|
headless_flag(
|
|
name = "headless",
|
|
build_setting_default = True,
|
|
visibility = ["//visibility:public"],
|
|
)
|
|
|
|
test_suite(
|
|
name = "tests_cpu",
|
|
tests = [
|
|
"//tfjs-backend-cpu:tests",
|
|
"//tfjs-backend-wasm:tests",
|
|
"//tfjs-converter:tests",
|
|
"//tfjs-core:tests",
|
|
"//tfjs-data:tests",
|
|
"//tfjs-tfdf:tests",
|
|
"//tfjs-tflite:tests",
|
|
],
|
|
)
|
|
|
|
test_suite(
|
|
name = "tests_gpu",
|
|
tests = [
|
|
"//tfjs-backend-webgl:tests",
|
|
"//tfjs-backend-webgpu:tests",
|
|
"//tfjs-layers:tests",
|
|
],
|
|
)
|
|
|
|
test_suite(
|
|
name = "tests",
|
|
tests = [
|
|
":tests_cpu",
|
|
":tests_gpu",
|
|
],
|
|
)
|