Searched defs:op_kind (Results 1 – 6 of 6) sorted by relevance
22 XlaResourceOpKind op_kind) { in XlaResourceOpKindToString()38 XlaResourceKind resource_kind) { in CreateResourceOpInfoMap()
43 explicit XlaResourceOpInfo(XlaResourceOpKind op_kind, in XlaResourceOpInfo()
35 const HloAllReduceInstruction *ar1, ReductionKind op_kind) { in AreCompatible()
267 absl::optional<XlaResourceOpKind> op_kind; in ComputeIncompatibleResourceOperationPairs() local
124 int op_kind = (flags & AF_PRED) ? 1 : in optimize_cc_op2() local
624 xla::RendezvousKey::CollectiveOpKind op_kind = in GetRendezvousKey() local