Searched refs:HloSnapshot (Results 1 – 22 of 22) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | dump.h | 34 class HloSnapshot; variable
85 const HloSnapshot& snapshot);
86 void DumpHloSnapshotIfEnabled(const HloSnapshot& snapshot,
|
| D | executable.h | 182 void set_hlo_snapshot(std::unique_ptr<xla::HloSnapshot> hlo_snapshot) { in set_hlo_snapshot()
186 HloSnapshot* hlo_snapshot() const { return hlo_snapshot_.get(); } in hlo_snapshot()
200 std::unique_ptr<HloSnapshot> hlo_snapshot_;
|
| D | dump.cc | 348 const HloSnapshot& snapshot) { in DumpHloSnapshotIfEnabled()
375 void DumpHloSnapshotIfEnabled(const HloSnapshot& snapshot, in DumpHloSnapshotIfEnabled()
|
| D | service.cc | 75 HloSnapshot* module) { in RecordArguments()
88 TransferManager* transfer_manager, HloSnapshot* module) { in RecordResult()
354 std::vector<std::unique_ptr<HloSnapshot>> hlo_snapshots; in BuildExecutables()
356 auto hlo_snapshot = absl::make_unique<HloSnapshot>(); in BuildExecutables()
806 auto hlo_snapshot = absl::make_unique<HloSnapshot>(); in BuildExecutable()
|
| D | hlo.proto | 460 message HloSnapshot { message
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_computation.cc | 31 StatusOr<std::unique_ptr<HloSnapshot>> XlaComputation::Snapshot() const { in Snapshot()
35 auto session = absl::make_unique<HloSnapshot>(); in Snapshot()
|
| D | local_client.h | 81 HloSnapshot* hlo_snapshot);
84 Status RecordResult(const ShapedBuffer* result, HloSnapshot* hlo_snapshot);
|
| D | xla_computation.h | 52 StatusOr<std::unique_ptr<HloSnapshot>> Snapshot() const;
|
| D | local_client.cc | 195 HloSnapshot* hlo_snapshot) { in RecordArguments()
205 HloSnapshot* hlo_snapshot) { in RecordResult()
|
| D | client.h | 225 StatusOr<XlaComputation> LoadSnapshot(const HloSnapshot& module);
|
| D | client.cc | 212 StatusOr<XlaComputation> Client::LoadSnapshot(const HloSnapshot& module) { in LoadSnapshot()
|
| /external/tensorflow/tensorflow/compiler/xla/tools/ |
| D | replay_computation.cc | 94 const HloSnapshot& module, LocalClient* client) { in CompileExecutable()
196 StatusOr<Literal> ReplayComputation(const HloSnapshot& module, in ReplayComputation()
309 StatusOr<HloSnapshot> ParseInputFile(const string& filename, in ParseInputFile()
312 HloSnapshot snapshot; in ParseInputFile()
350 std::vector<HloSnapshot> snapshots; in RealMain()
352 StatusOr<HloSnapshot> maybe_snapshot = ParseInputFile(arg, opts); in RealMain()
|
| D | show_signature.cc | 50 HloSnapshot module; in RealMain()
|
| D | convert_computation.cc | 36 HloSnapshot module; in RealMain()
|
| D | dumped_computation_to_operation_list.cc | 66 HloSnapshot snapshot; in RealMain()
|
| D | dumped_computation_to_text.cc | 41 HloSnapshot snapshot; in RealMain()
|
| D | interactive_graphviz.cc | 666 HloSnapshot snapshot; in RealMain()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | replay_test.cc | 47 std::unique_ptr<HloSnapshot> module = in TEST_F()
80 std::unique_ptr<HloSnapshot> module = in XLA_TEST_F()
127 std::unique_ptr<HloSnapshot> module = in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/aot/ |
| D | compile.cc | 107 TF_ASSIGN_OR_RETURN(std::unique_ptr<xla::HloSnapshot> module, in CompileGraph()
|
| /external/tensorflow/tensorflow/compiler/xrt/ |
| D | xrt.proto | 57 xla.HloSnapshot hlo_snapshot = 2;
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | xla.proto | 255 // If true, every time an HLO module is run, we will dump an HloSnapshot
|
| /external/tensorflow/tensorflow/compiler/xrt/tests/ |
| D | raw_api_test.cc | 219 xla::HloSnapshot* dst) { in StoreComputationSnapshot()
|