Home
last modified time | relevance | path

Searched refs:computation_to_profile_idx (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_execution_profile.h54 return FindOrDie(computation_to_profile_idx(), &computation); in GetProfileIndexFor()
66 return computation_to_profile_idx().size(); in computation_count()
83 computation_to_profile_idx() const { in computation_to_profile_idx() function
Dhlo_execution_profile.cc66 hlo_profile_index_map.computation_to_profile_idx(); in CreateHloProfilePrinterData()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_compiler.cc542 computation_to_profile_idx, in CreateHloProfilingArtifacts() argument
566 *computation_to_profile_idx = in CreateHloProfilingArtifacts()
567 (*hlo_profile_index_map)->computation_to_profile_idx(); in CreateHloProfilingArtifacts()
698 std::unordered_map<const HloComputation*, int64> computation_to_profile_idx; in RunBackend() local
703 *module, &instruction_to_profile_idx, &computation_to_profile_idx, in RunBackend()
739 std::move(computation_to_profile_idx), in RunBackend()
934 std::unordered_map<const HloComputation*, int64> computation_to_profile_idx; in CompileAheadOfTime() local
940 *module, &instruction_to_profile_idx, &computation_to_profile_idx, in CompileAheadOfTime()
947 std::move(computation_to_profile_idx), in CompileAheadOfTime()
Dir_emitter.h89 computation_to_profile_idx,
Dir_emitter.cc95 std::unordered_map<const HloComputation*, int64> computation_to_profile_idx, in IrEmitter() argument
104 computation_to_profile_idx_(std::move(computation_to_profile_idx)), in IrEmitter()