Searched refs:instruction_to_profile_idx (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_execution_profile.h | 50 return FindOrDie(instruction_to_profile_idx(), &instruction); in GetProfileIndexFor()
62 return instruction_to_profile_idx().size(); in instruction_count()
78 instruction_to_profile_idx() const { in instruction_to_profile_idx() function
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_compiler.cc | 540 instruction_to_profile_idx, in CreateHloProfilingArtifacts() argument
549 *instruction_to_profile_idx, in CreateHloProfilingArtifacts()
552 (*hlo_profile_index_map)->instruction_to_profile_idx())); in CreateHloProfilingArtifacts()
697 std::unordered_map<const HloInstruction*, int64> instruction_to_profile_idx; in RunBackend() local
703 *module, &instruction_to_profile_idx, &computation_to_profile_idx, in RunBackend()
738 std::move(instruction_to_profile_idx), in RunBackend()
933 std::unordered_map<const HloInstruction*, int64> instruction_to_profile_idx; in CompileAheadOfTime() local
940 *module, &instruction_to_profile_idx, &computation_to_profile_idx, in CompileAheadOfTime()
946 std::move(instruction_to_profile_idx), in CompileAheadOfTime()
|
| D | ir_emitter.h | 87 instruction_to_profile_idx,
|
| D | ir_emitter.cc | 94 std::unordered_map<const HloInstruction*, int64> instruction_to_profile_idx, in IrEmitter() argument
103 instruction_to_profile_idx_(std::move(instruction_to_profile_idx)), in IrEmitter()
|