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