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

Add a TFE_Py_Execute traceme, which is the entrance to TF c++

Browse Source 
PiperOrigin-RevId: 311817887
Change-Id: If924b3f3273096c961e6cc24459a620ce3889963
This commit is contained in:
Yanhua Sun 2020-05-15 16:02:58 -07:00 committed by TensorFlower Gardener
parent 55b36215c8
commit eb07fd848a
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/python/eager/pywrap_tfe_src.cc
View File

@ -852,6 +852,8 @@ void TFE_Py_ExecuteCancelable(TFE_Context* ctx, const char* device_name,
TFE_CancellationManager* cancellation_manager,
TFE_OutputTensorHandles* outputs,
TF_Status* out_status) {
tensorflow::profiler::TraceMe activity(
"TFE_Py_ExecuteCancelable", tensorflow::profiler::TraceMeLevel::kInfo);
TFE_Op* op = GetOp(ctx, op_name, device_name, out_status);
auto cleaner = tensorflow::gtl::MakeCleanup([ctx, op] { ReturnOp(ctx, op); });
if (!out_status->status.ok()) return;