Searched defs:hlo_instruction (Results 1 – 17 of 17) sorted by relevance
27 const HloInstruction* hlo_instruction) in HostToDeviceCopyThunk()46 const HloInstruction* hlo_instruction) in DeviceToDeviceCopyThunk()
101 const HloInstruction* hlo_instruction) { in FinishHloInstruction()112 const HloInstruction* hlo_instruction) { in MakeScopedInstructionProfiler()
69 explicit Thunk(Kind kind, const HloInstruction* hlo_instruction) in Thunk()76 const HloInstruction* hlo_instruction() const { return hlo_instruction_; } in hlo_instruction() function
39 const HloInstruction* hlo_instruction) in TupleThunk()
90 const HloInstruction* hlo_instruction) in ScopedInstructionProfiler()
27 const HloInstruction* hlo_instruction) in OutfeedThunk()
32 const HloInstruction* hlo_instruction, in KernelThunk()
27 const HloInstruction* hlo_instruction) in InfeedThunk()
271 DotDimensionNumbers GetDimensionNumbers(const HloInstruction& hlo_instruction) { in GetDimensionNumbers()298 const HloInstruction* hlo_instruction, in GemmThunk()
100 Status LogicalBufferAnalysis::DefaultAction(HloInstruction* hlo_instruction) { in DefaultAction()
251 Status DefaultAction(HloInstructionPtr hlo_instruction) override { in DefaultAction()
198 Status TuplePointsToAnalysis::DefaultAction(HloInstruction* hlo_instruction) { in DefaultAction()
86 const HloInstruction* hlo_instruction) { in HandleElementwiseOp()
235 Status DefaultAction(HloInstruction* hlo_instruction) override { in TEST_F()
53 Status DefaultAction(HloInstruction* hlo_instruction) override { in DefaultAction()575 Status DefaultAction(HloInstruction* hlo_instruction) override { in DefaultAction()
147 Status DefaultAction(HloInstruction* hlo_instruction) override { in DefaultAction()
206 Status DefaultAction(HloInstruction* hlo_instruction) override { in DefaultAction()