e2e3a943c0
END_PUBLIC --- Commit1e1b3d902authored by Pete Warden<pete@petewarden.com> Committed by gunan<gunan@google.com>: Changed output directory for Pi CI build to fix permissions problem with nightlies (#13257) * Fix for RTLD_GLOBAL breakage of Pi builds, and removed Eigen version change for Pi that's no longer needed * Fixed Pi Zero OpenBLAS build problems and tidied up directories used * More robust checks in Pi build script * Changed output directory for Pi CI build to fix permissions problem --- Commitfe3a2e65cauthored by Yan Facai (???)<facai.yan@gmail.com> Committed by drpngx<drpngx@users.noreply.github.com>: check invalid string type for dest_nodes in extract_sub_graph (#13057) * BUG: check str type * TST: add unit test * CLN: remove list check * CLN: use warning * CLN: 2 indent * CLN: raise TypeError if not list * CLN: check string only --- Commit225ab7629authored by Jean Wanka<jm.wanka@gmail.com> Committed by Jean Wanka<jm.wanka@gmail.com>: Fix polynomial decay with cycle for global step=0 For polynomial decay with cycle=True the learning rate at step 0 becomes NaN, because in the process of calculating it we devide by 0. This change should fix it, by setting the multiplier for the decay steps to one for global_step=0. --- Commit286f57061authored by Bjarke Hammersholt Roune<broune@google.com> Committed by TensorFlower Gardener<gardener@tensorflow.org>: Make Service::TransferToClient not attempt to manipulate the literal when the transfer failed, preventing a crash and allowing the caller to see the reason for the failed transfer. PiperOrigin-RevId: 169770126 --- Commite0501bc4dauthored by Yong Tang<yong.tang.github@outlook.com> Committed by Shanqing Cai<cais@google.com>: Fix GRUBlockCell parameter naming inconsistency (#13153) * Fix GRUBlockCell parameter naming inconsistency This fix tries to fix the issue in 13137 where parameter `cell_size` is used instead of `num_units`. This is inconsistent with other RNN cells. This fix adds support of `num_units` while at the same time maintains backward compatiblility for `cell_size`. This fix fixes 13137. Signed-off-by: Yong Tang <yong.tang.github@outlook.com> * Add `@deprecated_args` for 'cell_size' in `GRUBlockCell` This commit adds `@deprecated_args` for 'cell_size' in `GRUBlockCell` Signed-off-by: Yong Tang <yong.tang.github@outlook.com> * Address review comment Signed-off-by: Yong Tang <yong.tang.github@outlook.com> --- Commit02a2eba05authored by Pete Warden<pete@petewarden.com> Committed by gunan<gunan@google.com>: Fix for RTLD_GLOBAL breakage of Pi builds, and removed Eigen version change that's no longer needed (#13251) * Fix for RTLD_GLOBAL breakage of Pi builds, and removed Eigen version change for Pi that's no longer needed * Fixed Pi Zero OpenBLAS build problems and tidied up directories used * More robust checks in Pi build script --- Commit8ef722253authored by Sanjoy Das<sanjoy@google.com> Committed by TensorFlower Gardener<gardener@tensorflow.org>: Remove a redundant setName. The EmitComputation should have emitted a function with the right name, so use a CHECK instead. PiperOrigin-RevId: 169764856 --- Commit1b94147dcauthored by Neal Wu<wun@google.com> Committed by TensorFlower Gardener<gardener@tensorflow.org>: Fix broken GitHub links in tensorflow and tensorflow_models resulting from The Great Models Move (a.k.a. the research subfolder) PiperOrigin-RevId: 169763373 --- Commitb1ada5f0cauthored by Justine Tunney<jart@google.com> Committed by TensorFlower Gardener<gardener@tensorflow.org>: Fix TensorBoard python -m invoke in docs PiperOrigin-RevId: 169758752 --- Commit2957cd894authored by Mustafa Ispir<ispir@google.com> Committed by TensorFlower Gardener<gardener@tensorflow.org>: Local run option of estimator training. PiperOrigin-RevId: 169756384 --- Commit1dc2fe7acauthored by Gunhan Gulsoy<gunan@google.com> Committed by TensorFlower Gardener<gardener@tensorflow.org>: BEGIN_PUBLIC Automated g4 rollback of changelist 166264198 PiperOrigin-RevId: 169998124
81 lines
1.5 KiB
Python
81 lines
1.5 KiB
Python
# Ops that communicate with other processes via MPI.
|
|
|
|
package(default_visibility = [
|
|
"//tensorflow:__subpackages__",
|
|
])
|
|
|
|
licenses(["notice"]) # Apache 2.0
|
|
|
|
filegroup(
|
|
name = "all_files",
|
|
srcs = glob(
|
|
["**/*"],
|
|
exclude = [
|
|
"**/METADATA",
|
|
"**/OWNERS",
|
|
],
|
|
),
|
|
visibility = ["//tensorflow:__subpackages__"],
|
|
)
|
|
|
|
load(
|
|
"//tensorflow/core:platform/default/build_config.bzl",
|
|
"tf_proto_library_cc",
|
|
)
|
|
|
|
tf_proto_library_cc(
|
|
name = "mpi_message_proto",
|
|
srcs = ["mpi_message.proto"],
|
|
cc_api_version = 2,
|
|
protodeps = ["//tensorflow/core:protos_all"],
|
|
visibility = [
|
|
"//tensorflow:__subpackages__",
|
|
],
|
|
)
|
|
|
|
load("//tensorflow:tensorflow.bzl", "tf_custom_op_library")
|
|
load("//tensorflow:tensorflow.bzl", "tf_py_test")
|
|
|
|
tf_custom_op_library(
|
|
name = "mpi_collectives.so",
|
|
srcs = [
|
|
"mpi_ops.cc",
|
|
"ring.cc",
|
|
"ring.h",
|
|
],
|
|
gpu_srcs = [
|
|
"ring.cu.cc",
|
|
"ring.h",
|
|
],
|
|
deps = [
|
|
":mpi_message_proto_cc",
|
|
"//third_party/mpi",
|
|
],
|
|
)
|
|
|
|
tf_py_test(
|
|
name = "mpi_ops_test",
|
|
srcs = ["mpi_ops_test.py"],
|
|
additional_deps = [
|
|
"//tensorflow:tensorflow_py",
|
|
"//tensorflow/python:platform",
|
|
],
|
|
data = [
|
|
":mpi_collectives.so",
|
|
],
|
|
tags = ["manual"],
|
|
)
|
|
|
|
py_library(
|
|
name = "mpi_ops_py",
|
|
srcs = [
|
|
"__init__.py",
|
|
"mpi_ops.py",
|
|
],
|
|
data = [
|
|
":mpi_collectives.so",
|
|
],
|
|
srcs_version = "PY2AND3",
|
|
visibility = ["//visibility:public"],
|
|
)
|