Searched refs:XlaResourceOpKind (Results 1 – 4 of 4) sorted by relevance
29 enum class XlaResourceOpKind { enum43 explicit XlaResourceOpInfo(XlaResourceOpKind op_kind, in XlaResourceOpInfo()47 XlaResourceOpKind kind() const { return op_kind_; } in kind()50 static absl::string_view XlaResourceOpKindToString(XlaResourceOpKind op_kind);53 XlaResourceOpKind op_kind_;
22 XlaResourceOpKind op_kind) { in XlaResourceOpKindToString()24 case XlaResourceOpKind::kRead: in XlaResourceOpKindToString()26 case XlaResourceOpKind::kWrite: in XlaResourceOpKindToString()28 case XlaResourceOpKind::kReadWrite: in XlaResourceOpKindToString()37 auto add = [&](absl::string_view op, XlaResourceOpKind op_kind, in CreateResourceOpInfoMap()44 auto kRead = XlaResourceOpKind::kRead; in CreateResourceOpInfoMap()45 auto kWrite = XlaResourceOpKind::kWrite; in CreateResourceOpInfoMap()46 auto kReadWrite = XlaResourceOpKind::kReadWrite; in CreateResourceOpInfoMap()
103 absl::optional<XlaResourceOpKind>* out_resource_op_kind) { in XlaResourceOpKindForNode()123 *out_resource_op_kind = XlaResourceOpKind::kReadWrite; in XlaResourceOpKindForNode()135 bool IsEdgeSafe(XlaResourceOpKind from, XlaResourceOpKind to) { in IsEdgeSafe()141 return from == XlaResourceOpKind::kRead || to == XlaResourceOpKind::kWrite; in IsEdgeSafe()144 using ResourceOp = std::pair<int, XlaResourceOpKind>;242 string NodeToString(const Node& n, XlaResourceOpKind resource_op_kind) { in NodeToString()267 absl::optional<XlaResourceOpKind> op_kind; in ComputeIncompatibleResourceOperationPairs()304 if (op_kind != XlaResourceOpKind::kRead) { in ComputeIncompatibleResourceOperationPairs()
2616 if (op_info && op_info->kind() != XlaResourceOpKind::kRead) { in ContainsResourceWriteOp()2625 if (op_info && op_info->kind() != XlaResourceOpKind::kRead) { in ContainsResourceWriteOp()