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.cc65 hlo_profile_index_map.computation_to_profile_idx(); in CreateHloProfilePrinterData()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_compiler.cc492 computation_to_profile_idx, in CreateHloProfilingArtifacts() argument
516 *computation_to_profile_idx = in CreateHloProfilingArtifacts()
517 (*hlo_profile_index_map)->computation_to_profile_idx(); in CreateHloProfilingArtifacts()
619 std::unordered_map<const HloComputation*, int64> computation_to_profile_idx; in RunBackend() local
624 *module, &instruction_to_profile_idx, &computation_to_profile_idx, in RunBackend()
666 std::move(computation_to_profile_idx), in RunBackend()
844 std::unordered_map<const HloComputation*, int64> computation_to_profile_idx; in CompileAheadOfTime() local
850 *module, &instruction_to_profile_idx, &computation_to_profile_idx, in CompileAheadOfTime()
857 std::move(computation_to_profile_idx), in CompileAheadOfTime()
Dir_emitter.h81 computation_to_profile_idx,
Dir_emitter.cc87 std::unordered_map<const HloComputation*, int64> computation_to_profile_idx, in IrEmitter() argument
95 computation_to_profile_idx_(std::move(computation_to_profile_idx)), in IrEmitter()