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