Searched refs:async_wrapped_instruction (Results 1 – 8 of 8) sorted by relevance
1096 const HloInstruction* async_wrapped_instruction = in TEST_P() local1097 async_start->async_wrapped_instruction(); in TEST_P()1104 &analysis.GetValueDefinedAt(async_wrapped_instruction, {}))); in TEST_P()1108 EXPECT_TRUE(analysis.GetValueDefinedAt(async_wrapped_instruction, {}) in TEST_P()1119 &analysis.GetValueDefinedAt(async_wrapped_instruction, {}))); in TEST_P()1124 &analysis.GetValueDefinedAt(async_wrapped_instruction, {}))); in TEST_P()1167 async_op->async_wrapped_instruction()->called_computations()[0]; in TEST_P()
204 ASSERT_THAT(start->async_wrapped_instruction(), NotNull()); in TEST_F()
296 HloInstruction* HloAsyncInstruction::async_wrapped_instruction() const { in async_wrapped_instruction() function in xla::HloAsyncInstruction302 return async_wrapped_instruction()->opcode(); in async_wrapped_opcode()317 async_wrapped_instruction()->ExtraAttributesToString(options); in ExtraAttributesToStringImpl()
1371 async_op->async_wrapped_instruction()->ToString(), in CheckAsyncOpOperand()1372 operand->async_wrapped_instruction()->ToString(), in CheckAsyncOpOperand()
1110 return GetInstructionColor(instr->async_wrapped_instruction()); in GetInstructionColor()
2127 HloInstruction* async_wrapped_instruction() const;
230 HloInstruction* async_wrapped_instruction() const;
4854 HloInstruction* HloInstruction::async_wrapped_instruction() const { in async_wrapped_instruction() function in xla::HloInstruction4855 return Cast<HloAsyncInstruction>(this)->async_wrapped_instruction(); in async_wrapped_instruction()