Searched refs:hlo_snapshot (Results 1 – 4 of 4) sorted by relevance
704 std::shared_ptr<xla::HloSnapshot> hlo_snapshot; in DoWork() local706 hlo_snapshot = in DoWork()712 [hlo_snapshot, literal](Status status) { in DoWork()719 *hlo_snapshot->add_arguments() = literal->ToProto(); in DoWork()772 if (hlo_snapshot) { in DoWork()777 [hlo_snapshot, literal](Status status) { in DoWork()779 *hlo_snapshot->mutable_result() = literal->ToProto(); in DoWork()785 DumpHloSnapshotIfEnabled(*hlo_snapshot, in DoWork()
73 string hlo_snapshot; member675 "'--hlo_snapshot', '--hlo_text', '--hlo_module_proto' flags.";746 tensorflow::Flag("hlo_snapshot", &opts.hlo_snapshot,
150 client->LoadSnapshot(computation_proto.hlo_snapshot())); in Compile()
85 xla.HloSnapshot hlo_snapshot = 2; field