Home
last modified time | relevance | path

Searched refs:kHloOpcodeIsVariadic (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dhlo_opcode.h54 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 Dhlo_opcode.cc56 return arity == kHloOpcodeIsVariadic; in HloOpcodeIsVariadic()
66 return arity == kHloOpcodeIsVariadic ? std::nullopt \ in HloOpcodeArity()