Searched refs:async_wrapped_opcode (Results 1 – 6 of 6) sorted by relevance
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
H A D | async_collective_creator_test.cc | 205 EXPECT_THAT(start->async_wrapped_opcode(), HloOpcode::kAllToAll); in TEST_F()
|
H A D | hlo_parser.cc | 325 std::optional<HloOpcode> async_wrapped_opcode, 492 std::optional<HloOpcode>* async_wrapped_opcode); 1136 std::optional<HloOpcode> async_wrapped_opcode; in ParseInstructionRhs() local 1141 !ParseOpcode(&opcode, &async_wrapped_opcode)) { in ParseInstructionRhs() 1181 async_wrapped_opcode, attrs, allow_attributes); in ParseInstructionRhs() 1248 std::optional<HloOpcode> async_wrapped_opcode, in CreateInstruction() argument 1625 if (async_wrapped_opcode) { in CreateInstruction() 1667 async_wrapped_root_shape, *async_wrapped_opcode, in CreateInstruction() 5481 HloOpcode* opcode, std::optional<HloOpcode>* async_wrapped_opcode) { in ParseOpcode() argument 5508 *async_wrapped_opcode = status_or_result.ValueOrDie(); in ParseOpcode()
|
H A D | hlo_instruction.cc | 3096 StrAppend(&result, HloOpcodeString(async_wrapped_opcode()), suffix); in ToStringWithCanonicalNameMap() 4858 HloOpcode HloInstruction::async_wrapped_opcode() const { in async_wrapped_opcode() function in xla::HloInstruction 4859 return Cast<HloAsyncInstruction>(this)->async_wrapped_opcode(); in async_wrapped_opcode()
|
H A D | hlo_instruction.h | 2130 HloOpcode async_wrapped_opcode() const;
|
H A D | hlo_instructions.h | 231 HloOpcode async_wrapped_opcode() const;
|
H A D | hlo_instructions.cc | 301 HloOpcode HloAsyncInstruction::async_wrapped_opcode() const { in async_wrapped_opcode() function in xla::HloAsyncInstruction
|