Searched refs:kHloOpcodeIsVariadic (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | hlo_opcode.h | 54 V(kAfterAll, "after-all", kHloOpcodeIsVariadic) \
55 V(kAllGather, "all-gather", kHloOpcodeIsVariadic) \
56 V(kAllGatherStart, "all-gather-start", kHloOpcodeIsVariadic) \
58 V(kAllReduce, "all-reduce", kHloOpcodeIsVariadic) \
59 V(kAllReduceStart, "all-reduce-start", kHloOpcodeIsVariadic) \
61 V(kAllToAll, "all-to-all", kHloOpcodeIsVariadic) \
62 V(kAsyncStart, "async-start", kHloOpcodeIsVariadic) \
72 V(kCall, "call", kHloOpcodeIsVariadic) \
76 V(kCollectivePermute, "collective-permute", kHloOpcodeIsVariadic) \
77 V(kCollectivePermuteStart, "collective-permute-start", kHloOpcodeIsVariadic) \
[all …]
|
| H A D | hlo_opcode.cc | 56 return arity == kHloOpcodeIsVariadic; in HloOpcodeIsVariadic()
66 return arity == kHloOpcodeIsVariadic ? std::nullopt \ in HloOpcodeArity()
|