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

  /external/tensorflow/tensorflow/compiler/xla/service/
hlo_execution_profile.cc 50 const HloProfileIndexMap& hlo_profile_index_map,
56 size_t profile_counters_size = hlo_profile_index_map.total_count();
62 hlo_profile_index_map.computation_count());
65 hlo_profile_index_map.computation_to_profile_idx();
109 hlo_profile_index_map.GetProfileIndexFor(*hlo));
114 for (const auto& pair : hlo_profile_index_map.extra_metric_to_profile_idx()) {
126 const HloProfileIndexMap* hlo_profile_index_map)
128 hlo_profile_index_map_(*hlo_profile_index_map),
executable.h 66 std::unique_ptr<HloProfileIndexMap> hlo_profile_index_map)
69 hlo_profile_index_map_(std::move(hlo_profile_index_map)) {
153 const HloProfileIndexMap& hlo_profile_index_map() const { function in class:xla::Executable
hlo_execution_profile.h 99 const HloProfileIndexMap& hlo_profile_index_map,
111 const HloProfileIndexMap* hlo_profile_index_map);
executable.cc 80 &hlo_profile_index_map())
  /external/tensorflow/tensorflow/compiler/xla/service/cpu/
cpu_executable.h 55 std::unique_ptr<HloProfileIndexMap> hlo_profile_index_map);
cpu_compiler.cc 493 std::unique_ptr<HloProfileIndexMap>* hlo_profile_index_map,
495 *hlo_profile_index_map = absl::make_unique<HloProfileIndexMap>(module);
502 (*hlo_profile_index_map)->instruction_to_profile_idx()));
515 **hlo_profile_index_map, cost_analysis, entry_computation.name());
517 (*hlo_profile_index_map)->computation_to_profile_idx();
620 std::unique_ptr<HloProfileIndexMap> hlo_profile_index_map;
625 &hlo_profile_index_map, &hlo_profile_printer_data));
717 std::move(hlo_profile_printer_data), std::move(hlo_profile_index_map)));
    [all...]
cpu_executable.cc 56 std::unique_ptr<HloProfileIndexMap> hlo_profile_index_map)
58 std::move(hlo_profile_index_map)),
  /external/tensorflow/tensorflow/compiler/xla/service/gpu/
gpu_executable.h 60 std::unique_ptr<HloProfileIndexMap> hlo_profile_index_map);
gpu_executable.cc 57 std::unique_ptr<HloProfileIndexMap> hlo_profile_index_map)
59 std::move(hlo_profile_index_map)),
  /external/tensorflow/tensorflow/compiler/xla/tests/
xla_hlo_profile_test.cc 168 &executable->hlo_profile_index_map());

Completed in 405 milliseconds