Home
last modified time | relevance | path

Searched refs:DumpHloSnapshotIfEnabled (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Ddump.h98 void DumpHloSnapshotIfEnabled(const HloModule& module,
100 void DumpHloSnapshotIfEnabled(const HloSnapshot& snapshot,
Ddump.cc486 void DumpHloSnapshotIfEnabled(const HloModule& module, in DumpHloSnapshotIfEnabled() function
519 void DumpHloSnapshotIfEnabled(const HloSnapshot& snapshot, in DumpHloSnapshotIfEnabled() function
Dservice.cc701 DumpHloSnapshotIfEnabled(executable_ptrs[i]->module(), snapshots[i]); in ExecuteGraphParallel()
724 DumpHloSnapshotIfEnabled(executable->module(), snapshots[i]); in ExecuteGraphParallel()
894 DumpHloSnapshotIfEnabled(executable->module(), snapshot); in Execute()
/external/tensorflow/tensorflow/compiler/xla/client/
Dlocal_client.cc236 DumpHloSnapshotIfEnabled(*snapshot, GetDebugOptionsFromFlags()); in DumpOutputsAndSaveSnapshot()
/external/tensorflow/tensorflow/core/tpu/kernels/
Dtpu_execute_op.cc785 DumpHloSnapshotIfEnabled(*hlo_snapshot, in DoWork()