Home
last modified time | relevance | path

Searched refs:async_wrapped_opcode (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dasync_collective_creator_test.cc205 EXPECT_THAT(start->async_wrapped_opcode(), HloOpcode::kAllToAll); in TEST_F()
H A Dhlo_parser.cc325 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 Dhlo_instruction.cc3096 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 Dhlo_instruction.h2130 HloOpcode async_wrapped_opcode() const;
H A Dhlo_instructions.h231 HloOpcode async_wrapped_opcode() const;
H A Dhlo_instructions.cc301 HloOpcode HloAsyncInstruction::async_wrapped_opcode() const { in async_wrapped_opcode() function in xla::HloAsyncInstruction