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 | 490 instruction_to_profile_idx, in CreateHloProfilingArtifacts() argument
499 *instruction_to_profile_idx, in CreateHloProfilingArtifacts()
502 (*hlo_profile_index_map)->instruction_to_profile_idx())); in CreateHloProfilingArtifacts()
618 std::unordered_map<const HloInstruction*, int64> instruction_to_profile_idx; in RunBackend() local
624 *module, &instruction_to_profile_idx, &computation_to_profile_idx, in RunBackend()
665 std::move(instruction_to_profile_idx), in RunBackend()
843 std::unordered_map<const HloInstruction*, int64> instruction_to_profile_idx; in CompileAheadOfTime() local
850 *module, &instruction_to_profile_idx, &computation_to_profile_idx, in CompileAheadOfTime()
856 std::move(instruction_to_profile_idx), in CompileAheadOfTime()
|
| D | ir_emitter.h | 79 instruction_to_profile_idx,
|
| D | ir_emitter.cc | 86 std::unordered_map<const HloInstruction*, int64> instruction_to_profile_idx, in IrEmitter() argument
94 instruction_to_profile_idx_(std::move(instruction_to_profile_idx)), in IrEmitter()
|