Searched defs:hlo_instruction (Results 1 – 17 of 17) sorted by relevance
27 const HloInstruction* hlo_instruction) in HostToDeviceCopyThunk()45 const HloInstruction* hlo_instruction) in DeviceToDeviceCopyThunk()
101 const HloInstruction* hlo_instruction) { in FinishHloInstruction()112 const HloInstruction* hlo_instruction) { in MakeScopedInstructionProfiler()
39 const HloInstruction* hlo_instruction) in TupleThunk()
73 explicit Thunk(Kind kind, const HloInstruction* hlo_instruction) in Thunk()80 const HloInstruction* hlo_instruction() const { return hlo_instruction_; } in hlo_instruction() function
90 const HloInstruction* hlo_instruction) in ScopedInstructionProfiler()
27 const HloInstruction* hlo_instruction) in OutfeedThunk()
38 const HloInstruction* hlo_instruction, in KernelThunk()
27 const HloInstruction* hlo_instruction) in InfeedThunk()
40 const HloInstruction *hlo_instruction, in GemmThunk()
95 void EmissionContext::addError(const HloInstruction* hlo_instruction, in addError()
100 Status LogicalBufferAnalysis::DefaultAction(HloInstruction* hlo_instruction) { in DefaultAction()
323 Status DefaultAction(HloInstructionPtr hlo_instruction) override { in DefaultAction()
199 Status TuplePointsToAnalysis::DefaultAction(HloInstruction* hlo_instruction) { in DefaultAction()
234 Status DefaultAction(HloInstruction* hlo_instruction) override { in TEST_F()
94 const HloInstruction* hlo_instruction) { in HandleElementwiseOp()
55 Status DefaultAction(HloInstruction* hlo_instruction) override { in DefaultAction()577 Status DefaultAction(HloInstruction* hlo_instruction) override { in DefaultAction()
194 Status DefaultAction(HloInstruction* hlo_instruction) override { in DefaultAction()