Searched refs:kHloOpcodeIsVariadic (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_opcode.h | 50 V(kAfterAll, "after-all", kHloOpcodeIsVariadic) \
52 V(kAllReduce, "all-reduce", kHloOpcodeIsVariadic) \
53 V(kAllToAll, "all-to-all", kHloOpcodeIsVariadic) \
61 V(kCall, "call", kHloOpcodeIsVariadic) \
71 V(kConcatenate, "concatenate", kHloOpcodeIsVariadic) \
72 V(kConditional, "conditional", kHloOpcodeIsVariadic) \
80 V(kCustomCall, "custom-call", kHloOpcodeIsVariadic) \
84 V(kDynamicSlice, "dynamic-slice", kHloOpcodeIsVariadic) \
85 V(kDynamicUpdateSlice, "dynamic-update-slice", kHloOpcodeIsVariadic) \
90 V(kFusion, "fusion", kHloOpcodeIsVariadic) \
[all …]
|
| D | hlo_opcode.cc | 55 return arity == kHloOpcodeIsVariadic; in HloOpcodeIsVariadic()
65 return arity == kHloOpcodeIsVariadic ? absl::nullopt \ in HloOpcodeArity()
|