HomeSort by relevance Sort by last modified time
    Searched refs:hlo_snapshot (Results 1 - 6 of 6) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/client/
local_client.cc 181 executable_->hlo_snapshot()->set_execution_platform(
183 TF_RETURN_IF_ERROR(RecordArguments(arguments, executable_->hlo_snapshot()));
188 TF_RETURN_IF_ERROR(RecordResult(&result, executable_->hlo_snapshot()));
189 DumpHloSnapshotIfEnabled(executable_->module(), *executable_->hlo_snapshot());
195 HloSnapshot* hlo_snapshot) {
196 hlo_snapshot->clear_arguments();
199 *hlo_snapshot->add_arguments() = literal.ToProto();
205 HloSnapshot* hlo_snapshot) {
206 hlo_snapshot->clear_result();
208 *hlo_snapshot->mutable_result() = literal.ToProto()
    [all...]
local_client.h 81 HloSnapshot* hlo_snapshot);
84 Status RecordResult(const ShapedBuffer* result, HloSnapshot* hlo_snapshot);
  /external/tensorflow/tensorflow/compiler/xla/service/
executable.h 182 void set_hlo_snapshot(std::unique_ptr<xla::HloSnapshot> hlo_snapshot) {
183 hlo_snapshot_ = std::move(hlo_snapshot);
186 HloSnapshot* hlo_snapshot() const { return hlo_snapshot_.get(); } function in class:xla::Executable
service.cc 356 auto hlo_snapshot = absl::make_unique<HloSnapshot>(); local
357 *hlo_snapshot->mutable_hlo()->mutable_hlo_module() = *module_protos[i];
358 hlo_snapshots.push_back(std::move(hlo_snapshot));
702 executable_ptrs[i]->hlo_snapshot()));
749 executable->hlo_snapshot()));
751 *executable->hlo_snapshot());
806 auto hlo_snapshot = absl::make_unique<HloSnapshot>(); local
    [all...]
  /external/tensorflow/tensorflow/compiler/xla/tools/
interactive_graphviz.cc 68 string hlo_snapshot; member in struct:xla::tools::__anon44679::Options
95 --{hlo_snapshot,hlo_proto,hlo_text}=path/to/binary_proto
636 if (!opts.hlo_snapshot.empty()) {
637 nonempty_proto_flags.push_back("--hlo_snapshot");
665 if (!opts.hlo_snapshot.empty()) {
668 opts.hlo_snapshot, &snapshot))
670 << opts.hlo_snapshot;
719 tensorflow::Flag("hlo_snapshot", &opts.hlo_snapshot,
  /external/tensorflow/tensorflow/compiler/xrt/kernels/
xrt_compile_ops.cc 112 client->LoadSnapshot(computation_proto.hlo_snapshot()));

Completed in 76 milliseconds