Searched refs:hlo_snapshot (Results 1 – 7 of 7) sorted by relevance
181 executable_->hlo_snapshot()->set_execution_platform( in ExecuteAndDump()183 TF_RETURN_IF_ERROR(RecordArguments(arguments, executable_->hlo_snapshot())); in ExecuteAndDump()188 TF_RETURN_IF_ERROR(RecordResult(&result, executable_->hlo_snapshot())); in ExecuteAndDump()189 DumpHloSnapshotIfEnabled(executable_->module(), *executable_->hlo_snapshot()); in ExecuteAndDump()195 HloSnapshot* hlo_snapshot) { in RecordArguments() argument196 hlo_snapshot->clear_arguments(); in RecordArguments()199 *hlo_snapshot->add_arguments() = literal.ToProto(); in RecordArguments()205 HloSnapshot* hlo_snapshot) { in RecordResult() argument206 hlo_snapshot->clear_result(); in RecordResult()208 *hlo_snapshot->mutable_result() = literal.ToProto(); in RecordResult()
81 HloSnapshot* hlo_snapshot);84 Status RecordResult(const ShapedBuffer* result, HloSnapshot* hlo_snapshot);
182 void set_hlo_snapshot(std::unique_ptr<xla::HloSnapshot> hlo_snapshot) { in set_hlo_snapshot() argument183 hlo_snapshot_ = std::move(hlo_snapshot); in set_hlo_snapshot()186 HloSnapshot* hlo_snapshot() const { return hlo_snapshot_.get(); } in hlo_snapshot() function
356 auto hlo_snapshot = absl::make_unique<HloSnapshot>(); in BuildExecutables() local357 *hlo_snapshot->mutable_hlo()->mutable_hlo_module() = *module_protos[i]; in BuildExecutables()358 hlo_snapshots.push_back(std::move(hlo_snapshot)); in BuildExecutables()702 executable_ptrs[i]->hlo_snapshot())); in ExecuteGraphParallel()749 executable->hlo_snapshot())); in ExecuteGraphParallel()751 *executable->hlo_snapshot()); in ExecuteGraphParallel()806 auto hlo_snapshot = absl::make_unique<HloSnapshot>(); in BuildExecutable() local807 *hlo_snapshot->mutable_hlo()->mutable_hlo_module() = module_proto; in BuildExecutable()808 executable->set_hlo_snapshot(std::move(hlo_snapshot)); in BuildExecutable()895 executable->hlo_snapshot()->set_execution_platform( in Execute()[all …]
68 string hlo_snapshot; member665 if (!opts.hlo_snapshot.empty()) { in RealMain()668 opts.hlo_snapshot, &snapshot)) in RealMain()670 << opts.hlo_snapshot; in RealMain()719 tensorflow::Flag("hlo_snapshot", &opts.hlo_snapshot, in main()
112 client->LoadSnapshot(computation_proto.hlo_snapshot())); in Compile()
57 xla.HloSnapshot hlo_snapshot = 2; field