Home
last modified time | relevance | path

Searched refs:async_wrapped_instruction (Results 1 – 8 of 8) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dhlo_dataflow_analysis_test.cc1096 const HloInstruction* async_wrapped_instruction = in TEST_P() local
1097 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()
H A Dasync_collective_creator_test.cc204 ASSERT_THAT(start->async_wrapped_instruction(), NotNull()); in TEST_F()
H A Dhlo_instructions.cc296 HloInstruction* HloAsyncInstruction::async_wrapped_instruction() const { in async_wrapped_instruction() function in xla::HloAsyncInstruction
302 return async_wrapped_instruction()->opcode(); in async_wrapped_opcode()
317 async_wrapped_instruction()->ExtraAttributesToString(options); in ExtraAttributesToStringImpl()
H A Dhlo_verifier.cc1371 async_op->async_wrapped_instruction()->ToString(), in CheckAsyncOpOperand()
1372 operand->async_wrapped_instruction()->ToString(), in CheckAsyncOpOperand()
H A Dhlo_graph_dumper.cc1110 return GetInstructionColor(instr->async_wrapped_instruction()); in GetInstructionColor()
H A Dhlo_instruction.h2127 HloInstruction* async_wrapped_instruction() const;
H A Dhlo_instructions.h230 HloInstruction* async_wrapped_instruction() const;
H A Dhlo_instruction.cc4854 HloInstruction* HloInstruction::async_wrapped_instruction() const { in async_wrapped_instruction() function in xla::HloInstruction
4855 return Cast<HloAsyncInstruction>(this)->async_wrapped_instruction(); in async_wrapped_instruction()