| /external/tensorflow/tensorflow/compiler/xla/service/ |
| hlo_execution_profile.h | 54 return FindOrDie(computation_to_profile_idx(), &computation);
66 return computation_to_profile_idx().size();
83 computation_to_profile_idx() const { function in class:xla::HloProfileIndexMap
|
| hlo_execution_profile.cc | 65 hlo_profile_index_map.computation_to_profile_idx();
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| cpu_compiler.cc | 492 computation_to_profile_idx,
516 *computation_to_profile_idx =
517 (*hlo_profile_index_map)->computation_to_profile_idx();
619 std::unordered_map<const HloComputation*, int64> computation_to_profile_idx;
624 *module, &instruction_to_profile_idx, &computation_to_profile_idx,
666 std::move(computation_to_profile_idx),
[all...] |
| ir_emitter.h | 73 // computation_to_profile_idx: the mapping from HLO computations to their
81 computation_to_profile_idx,
|
| ir_emitter.cc | 87 std::unordered_map<const HloComputation*, int64> computation_to_profile_idx,
95 computation_to_profile_idx_(std::move(computation_to_profile_idx)),
[all...] |
