Searched defs:op_kind (Results 1 – 7 of 7) sorted by relevance
23 XlaResourceOpKind op_kind) { in XlaResourceOpKindToString()39 XlaResourceKind resource_kind) { in CreateResourceOpInfoMap()
43 explicit XlaResourceOpInfo(XlaResourceOpKind op_kind, in XlaResourceOpInfo()
36 const HloAllReduceInstruction *ar1, ReductionKind op_kind) { in AreCompatible()
267 std::optional<XlaResourceOpKind> op_kind; in ComputeIncompatibleResourceOperationPairs() local
105 inline std::string GetTfgOpName(OpKind op_kind) { in GetTfgOpName()
700 xla::RendezvousKey::CollectiveOpKind op_kind = in GetRendezvousKey() local
1089 int op_kind = Choose<int>({matmul, batchmatmul, dot, outer}); in ChooseEinsumArguments() local