mirror of
https://github.com/zebrajr/tensorflow.git
synced 2025-12-07 12:20:24 +01:00
Miscellaneous cleanups
PiperOrigin-RevId: 158012131
This commit is contained in:
parent
7ffc357325
commit
48a4853ebe
|
|
@ -68,7 +68,7 @@ cc_library(
|
||||||
"//tensorflow/compiler/jit/kernels:xla_device_launch_op",
|
"//tensorflow/compiler/jit/kernels:xla_device_launch_op",
|
||||||
"//tensorflow/compiler/tf2xla:xla_compiler",
|
"//tensorflow/compiler/tf2xla:xla_compiler",
|
||||||
"//tensorflow/compiler/tf2xla/kernels:xla_ops",
|
"//tensorflow/compiler/tf2xla/kernels:xla_ops",
|
||||||
"//tensorflow/compiler/xla/service:cpu_plugin",
|
"//tensorflow/compiler/xla/service:cpu_plugin", # buildcleaner: keep
|
||||||
"//tensorflow/core:core_cpu_internal",
|
"//tensorflow/core:core_cpu_internal",
|
||||||
"//tensorflow/core:lib",
|
"//tensorflow/core:lib",
|
||||||
],
|
],
|
||||||
|
|
@ -85,7 +85,7 @@ cc_library(
|
||||||
"//tensorflow/compiler/jit/kernels:xla_device_launch_op",
|
"//tensorflow/compiler/jit/kernels:xla_device_launch_op",
|
||||||
"//tensorflow/compiler/tf2xla:xla_compiler",
|
"//tensorflow/compiler/tf2xla:xla_compiler",
|
||||||
"//tensorflow/compiler/tf2xla/kernels:xla_ops",
|
"//tensorflow/compiler/tf2xla/kernels:xla_ops",
|
||||||
"//tensorflow/compiler/xla/service:gpu_plugin",
|
"//tensorflow/compiler/xla/service:gpu_plugin", # buildcleaner: keep
|
||||||
"//tensorflow/core:core_cpu_internal",
|
"//tensorflow/core:core_cpu_internal",
|
||||||
"//tensorflow/core:lib",
|
"//tensorflow/core:lib",
|
||||||
],
|
],
|
||||||
|
|
|
||||||
|
|
@ -681,7 +681,6 @@ cc_test(
|
||||||
srcs = ["buffer_liveness_test.cc"],
|
srcs = ["buffer_liveness_test.cc"],
|
||||||
deps = [
|
deps = [
|
||||||
":buffer_liveness",
|
":buffer_liveness",
|
||||||
":cpu_plugin",
|
|
||||||
":hlo",
|
":hlo",
|
||||||
"//tensorflow/compiler/xla:shape_util",
|
"//tensorflow/compiler/xla:shape_util",
|
||||||
"//tensorflow/compiler/xla:types",
|
"//tensorflow/compiler/xla:types",
|
||||||
|
|
@ -727,7 +726,6 @@ cc_test(
|
||||||
":call_graph",
|
":call_graph",
|
||||||
":computation_tracker",
|
":computation_tracker",
|
||||||
":copy_insertion",
|
":copy_insertion",
|
||||||
":cpu_plugin",
|
|
||||||
":flatten_call_graph",
|
":flatten_call_graph",
|
||||||
":hlo",
|
":hlo",
|
||||||
":hlo_ordering",
|
":hlo_ordering",
|
||||||
|
|
@ -792,7 +790,6 @@ cc_test(
|
||||||
name = "hlo_ordering_test",
|
name = "hlo_ordering_test",
|
||||||
srcs = ["hlo_ordering_test.cc"],
|
srcs = ["hlo_ordering_test.cc"],
|
||||||
deps = [
|
deps = [
|
||||||
":cpu_plugin",
|
|
||||||
":hlo",
|
":hlo",
|
||||||
":hlo_ordering",
|
":hlo_ordering",
|
||||||
"//tensorflow/compiler/xla:shape_util",
|
"//tensorflow/compiler/xla:shape_util",
|
||||||
|
|
@ -862,7 +859,6 @@ cc_test(
|
||||||
srcs = ["algebraic_simplifier_test.cc"],
|
srcs = ["algebraic_simplifier_test.cc"],
|
||||||
deps = [
|
deps = [
|
||||||
":algebraic_simplifier",
|
":algebraic_simplifier",
|
||||||
":cpu_plugin",
|
|
||||||
":hlo",
|
":hlo",
|
||||||
":hlo_matchers",
|
":hlo_matchers",
|
||||||
":hlo_pass",
|
":hlo_pass",
|
||||||
|
|
@ -930,7 +926,6 @@ cc_test(
|
||||||
name = "inliner_test",
|
name = "inliner_test",
|
||||||
srcs = ["inliner_test.cc"],
|
srcs = ["inliner_test.cc"],
|
||||||
deps = [
|
deps = [
|
||||||
":cpu_plugin",
|
|
||||||
":hlo",
|
":hlo",
|
||||||
":hlo_matchers",
|
":hlo_matchers",
|
||||||
":inliner",
|
":inliner",
|
||||||
|
|
@ -1087,7 +1082,6 @@ cc_test(
|
||||||
name = "hlo_computation_test",
|
name = "hlo_computation_test",
|
||||||
srcs = ["hlo_computation_test.cc"],
|
srcs = ["hlo_computation_test.cc"],
|
||||||
deps = [
|
deps = [
|
||||||
":cpu_plugin",
|
|
||||||
":hlo",
|
":hlo",
|
||||||
":hlo_matchers",
|
":hlo_matchers",
|
||||||
"//tensorflow/compiler/xla:literal_util",
|
"//tensorflow/compiler/xla:literal_util",
|
||||||
|
|
@ -1118,7 +1112,6 @@ cc_test(
|
||||||
name = "hlo_module_test",
|
name = "hlo_module_test",
|
||||||
srcs = ["hlo_module_test.cc"],
|
srcs = ["hlo_module_test.cc"],
|
||||||
deps = [
|
deps = [
|
||||||
":cpu_plugin",
|
|
||||||
":hlo",
|
":hlo",
|
||||||
"//tensorflow/compiler/xla:literal_util",
|
"//tensorflow/compiler/xla:literal_util",
|
||||||
"//tensorflow/compiler/xla:shape_util",
|
"//tensorflow/compiler/xla:shape_util",
|
||||||
|
|
@ -1258,7 +1251,6 @@ cc_test(
|
||||||
srcs = ["copy_insertion_test.cc"],
|
srcs = ["copy_insertion_test.cc"],
|
||||||
deps = [
|
deps = [
|
||||||
":copy_insertion",
|
":copy_insertion",
|
||||||
":cpu_plugin",
|
|
||||||
":hlo",
|
":hlo",
|
||||||
":hlo_matchers",
|
":hlo_matchers",
|
||||||
":tuple_points_to_analysis",
|
":tuple_points_to_analysis",
|
||||||
|
|
@ -1323,7 +1315,6 @@ cc_test(
|
||||||
name = "hlo_rematerialization_test",
|
name = "hlo_rematerialization_test",
|
||||||
srcs = ["hlo_rematerialization_test.cc"],
|
srcs = ["hlo_rematerialization_test.cc"],
|
||||||
deps = [
|
deps = [
|
||||||
":cpu_plugin",
|
|
||||||
":hlo",
|
":hlo",
|
||||||
":hlo_matchers",
|
":hlo_matchers",
|
||||||
":hlo_ordering",
|
":hlo_ordering",
|
||||||
|
|
@ -1340,7 +1331,6 @@ cc_test(
|
||||||
name = "hlo_dce_test",
|
name = "hlo_dce_test",
|
||||||
srcs = ["hlo_dce_test.cc"],
|
srcs = ["hlo_dce_test.cc"],
|
||||||
deps = [
|
deps = [
|
||||||
":cpu_plugin",
|
|
||||||
":hlo",
|
":hlo",
|
||||||
":hlo_dce",
|
":hlo_dce",
|
||||||
"//tensorflow/compiler/xla:literal_util",
|
"//tensorflow/compiler/xla:literal_util",
|
||||||
|
|
@ -1363,7 +1353,6 @@ cc_test(
|
||||||
deps = [
|
deps = [
|
||||||
":algebraic_simplifier",
|
":algebraic_simplifier",
|
||||||
":computation_layout",
|
":computation_layout",
|
||||||
":cpu_plugin",
|
|
||||||
":hlo",
|
":hlo",
|
||||||
":hlo_matchers",
|
":hlo_matchers",
|
||||||
":layout_assignment",
|
":layout_assignment",
|
||||||
|
|
@ -1436,7 +1425,6 @@ cc_test(
|
||||||
name = "hlo_cse_test",
|
name = "hlo_cse_test",
|
||||||
srcs = ["hlo_cse_test.cc"],
|
srcs = ["hlo_cse_test.cc"],
|
||||||
deps = [
|
deps = [
|
||||||
":cpu_plugin",
|
|
||||||
":hlo",
|
":hlo",
|
||||||
":hlo_cse",
|
":hlo_cse",
|
||||||
":hlo_matchers",
|
":hlo_matchers",
|
||||||
|
|
@ -1472,7 +1460,6 @@ cc_test(
|
||||||
name = "hlo_constant_folding_test",
|
name = "hlo_constant_folding_test",
|
||||||
srcs = ["hlo_constant_folding_test.cc"],
|
srcs = ["hlo_constant_folding_test.cc"],
|
||||||
deps = [
|
deps = [
|
||||||
":cpu_plugin",
|
|
||||||
":hlo",
|
":hlo",
|
||||||
":hlo_constant_folding",
|
":hlo_constant_folding",
|
||||||
":hlo_matchers",
|
":hlo_matchers",
|
||||||
|
|
|
||||||
|
|
@ -1891,7 +1891,6 @@ cc_library(
|
||||||
":framework",
|
":framework",
|
||||||
":lib",
|
":lib",
|
||||||
":lib_internal",
|
":lib_internal",
|
||||||
":test",
|
|
||||||
],
|
],
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
@ -1906,7 +1905,7 @@ cc_library(
|
||||||
deps = [
|
deps = [
|
||||||
":lib",
|
":lib",
|
||||||
":lib_internal",
|
":lib_internal",
|
||||||
":test",
|
":test", # buildcleaner: keep
|
||||||
"//tensorflow/core/platform/default/build_config:test_main",
|
"//tensorflow/core/platform/default/build_config:test_main",
|
||||||
],
|
],
|
||||||
alwayslink = 1,
|
alwayslink = 1,
|
||||||
|
|
|
||||||
|
|
@ -185,21 +185,13 @@ cc_test(
|
||||||
name = "virtual_scheduler_test",
|
name = "virtual_scheduler_test",
|
||||||
srcs = ["virtual_scheduler_test.cc"],
|
srcs = ["virtual_scheduler_test.cc"],
|
||||||
deps = [
|
deps = [
|
||||||
":graph_properties",
|
|
||||||
":utils",
|
|
||||||
":virtual_placer",
|
":virtual_placer",
|
||||||
":virtual_scheduler",
|
":virtual_scheduler",
|
||||||
"//tensorflow/cc:cc_ops",
|
"//tensorflow/cc:cc_ops",
|
||||||
"//tensorflow/core:framework",
|
|
||||||
"//tensorflow/core:protos_all_cc",
|
|
||||||
"//tensorflow/core:tensorflow",
|
"//tensorflow/core:tensorflow",
|
||||||
"//tensorflow/core:test",
|
"//tensorflow/core:test",
|
||||||
"//tensorflow/core:test_main",
|
"//tensorflow/core:test_main",
|
||||||
"//tensorflow/core/grappler:grappler_item",
|
|
||||||
"//tensorflow/core/grappler:utils",
|
|
||||||
"//tensorflow/core/grappler/clusters:utils",
|
|
||||||
"//tensorflow/core/grappler/clusters:virtual_cluster",
|
"//tensorflow/core/grappler/clusters:virtual_cluster",
|
||||||
"//tensorflow/core/grappler/costs:cost_estimator",
|
|
||||||
],
|
],
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -48,4 +48,4 @@ class IGraphTransferOpsDefinitions {
|
||||||
|
|
||||||
} // namespace tensorflow
|
} // namespace tensorflow
|
||||||
|
|
||||||
#endif // THIRD_PARTY_TENSORFLOW_CORE_KERNELS_HEXAGON_I_GRAPH_TRANSFER_OPS_DEFINITIONS_H
|
#endif // THIRD_PARTY_TENSORFLOW_CORE_KERNELS_HEXAGON_I_GRAPH_TRANSFER_OPS_DEFINITIONS_H_
|
||||||
|
|
|
||||||
|
|
@ -298,9 +298,9 @@ TEST(InputBuffer, ReadVarint32) {
|
||||||
// Generates data.
|
// Generates data.
|
||||||
std::vector<uint32> data;
|
std::vector<uint32> data;
|
||||||
uint32 i = 0;
|
uint32 i = 0;
|
||||||
for (; i < (1 << 10); i += 1) data.push_back(i);
|
for (; i < (1U << 10); i += 1) data.push_back(i);
|
||||||
for (; i < (1 << 15); i += 5) data.push_back(i);
|
for (; i < (1U << 15); i += 5) data.push_back(i);
|
||||||
for (; i < (1 << 31); i += 132817) data.push_back(i);
|
for (; i < (1U << 31); i += 132817) data.push_back(i);
|
||||||
data.push_back(std::numeric_limits<uint32>::max());
|
data.push_back(std::numeric_limits<uint32>::max());
|
||||||
|
|
||||||
// Writes the varints.
|
// Writes the varints.
|
||||||
|
|
|
||||||
|
|
@ -58,17 +58,15 @@ void ZlibInputStream::InitZlibBuffer() {
|
||||||
z_stream_->avail_in = 0;
|
z_stream_->avail_in = 0;
|
||||||
|
|
||||||
int status = inflateInit2(z_stream_.get(), zlib_options_.window_bits);
|
int status = inflateInit2(z_stream_.get(), zlib_options_.window_bits);
|
||||||
if (status != Z_OK) {
|
|
||||||
LOG(FATAL) << "inflateInit failed with status " << status;
|
CHECK_EQ(status, Z_OK) << "inflateInit failed with status " << status;
|
||||||
z_stream_.reset(nullptr);
|
|
||||||
} else {
|
|
||||||
z_stream_->next_in = z_stream_input_.get();
|
z_stream_->next_in = z_stream_input_.get();
|
||||||
z_stream_->next_out = z_stream_output_.get();
|
z_stream_->next_out = z_stream_output_.get();
|
||||||
next_unread_byte_ = reinterpret_cast<char*>(z_stream_output_.get());
|
next_unread_byte_ = reinterpret_cast<char*>(z_stream_output_.get());
|
||||||
z_stream_->avail_in = 0;
|
z_stream_->avail_in = 0;
|
||||||
z_stream_->avail_out = output_buffer_capacity_;
|
z_stream_->avail_out = output_buffer_capacity_;
|
||||||
}
|
}
|
||||||
}
|
|
||||||
|
|
||||||
Status ZlibInputStream::ReadFromStream() {
|
Status ZlibInputStream::ReadFromStream() {
|
||||||
int bytes_to_read = input_buffer_capacity_;
|
int bytes_to_read = input_buffer_capacity_;
|
||||||
|
|
|
||||||
|
|
@ -48,27 +48,9 @@ limitations under the License.
|
||||||
namespace perftools {
|
namespace perftools {
|
||||||
namespace gputools {
|
namespace gputools {
|
||||||
|
|
||||||
class KernelBase;
|
|
||||||
class Stream;
|
class Stream;
|
||||||
class Timer;
|
class Timer;
|
||||||
|
|
||||||
namespace blas {
|
|
||||||
class BlasSupport;
|
|
||||||
} // namespace blas
|
|
||||||
|
|
||||||
namespace fft {
|
|
||||||
class Support;
|
|
||||||
} // namespace fft
|
|
||||||
|
|
||||||
namespace rng {
|
|
||||||
class RngSupport;
|
|
||||||
} // namespace rng
|
|
||||||
|
|
||||||
} // namespace gputools
|
|
||||||
} // namespace perftools
|
|
||||||
|
|
||||||
namespace perftools {
|
|
||||||
namespace gputools {
|
|
||||||
namespace internal {
|
namespace internal {
|
||||||
|
|
||||||
// Platform-dependent interface class for the generic Events interface, in
|
// Platform-dependent interface class for the generic Events interface, in
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue
Block a user