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
|