blob: b91a7f43e5c03e933d10572e54e0c8c914c55f71 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
|
# Copyright 2017 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.
# ==============================================================================
########################################################
# tf_core_profiler library
########################################################
file(GLOB_RECURSE tf_core_profiler_srcs
"${tensorflow_source_dir}/tensorflow/core/profiler/*.proto"
"${tensorflow_source_dir}/tensorflow/core/profiler/tfprof_options.h"
"${tensorflow_source_dir}/tensorflow/core/profiler/tfprof_options.cc"
"${tensorflow_source_dir}/tensorflow/core/profiler/internal/*.h"
"${tensorflow_source_dir}/tensorflow/core/profiler/internal/*.cc"
"${tensorflow_source_dir}/tensorflow/core/profiler/internal/advisor/*.h"
"${tensorflow_source_dir}/tensorflow/core/profiler/internal/advisor/*.cc"
"${tensorflow_source_dir}/tensorflow/core/platform/regexp.h"
)
file(GLOB_RECURSE tf_core_profiler_exclude_srcs
"${tensorflow_source_dir}/tensorflow/core/profiler/internal/*test.cc"
"${tensorflow_source_dir}/tensorflow/core/profiler/internal/advisor/*test.cc"
"${tensorflow_source_dir}/tensorflow/core/profiler/internal/print_model_analysis.cc"
"${tensorflow_source_dir}/tensorflow/core/profiler/internal/print_model_analysis.h"
)
list(REMOVE_ITEM tf_core_profiler_srcs ${tf_core_profiler_exclude_srcs})
add_library(tf_core_profiler OBJECT ${tf_core_profiler_srcs})
add_dependencies(tf_core_profiler tf_core_lib)
|
