experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

don't trace arguments (include tensor shapes and op attributes used for cost analysis) for send/recv ops.

Browse Source 
PiperOrigin-RevId: 301612015
Change-Id: I1c761c25afaa12435a4e98d847a6ad44b6f2c25f
This commit is contained in:
A. Unique TensorFlower 2020-03-18 10:07:46 -07:00 committed by TensorFlower Gardener
parent b16bed28bf
commit 78c0e5b189
1 changed files with 4 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
tensorflow/core/kernels/sendrecv_ops.cc
View File

@ -115,14 +115,8 @@ string SendOp::TraceString(OpKernelContext* ctx, bool verbose) {
auto dst_it = attr.find("_dst");
const string& src = src_it != attr.end() ? src_it->second.s() : "";
const string& dst = dst_it != attr.end() ? dst_it->second.s() : "";
if (!verbose) {
return strings::StrCat(name_view(), ":", type_string_view(), "#from=", src,
",to=", dst, "#");
} else {
string trace_args = GetTraceArgument(ctx);
return strings::StrCat(name_view(), ":", type_string_view(), "#from=", src,
",to=", dst, ",", trace_args, "#");
}
return strings::StrCat(name_view(), ":", type_string_view(), "#from=", src,
",to=", dst, "#");
}
REGISTER_KERNEL_BUILDER(Name("_Send").Device(DEVICE_CPU), SendOp);
@ -163,14 +157,8 @@ string RecvOp::TraceString(OpKernelContext* ctx, bool verbose) {
auto dst_it = attr.find("_dst");
const string& src = src_it != attr.end() ? src_it->second.s() : "";
const string& dst = dst_it != attr.end() ? dst_it->second.s() : "";
if (!verbose) {
return strings::StrCat(name_view(), ":", type_string_view(), "#from=", src,
",to=", dst, "#");
} else {
string trace_args = GetTraceArgument(ctx);
return strings::StrCat(name_view(), ":", type_string_view(), "#from=", src,
",to=", dst, ",", trace_args, "#");
}
return strings::StrCat(name_view(), ":", type_string_view(), "#from=", src,
",to=", dst, "#");
}
namespace {