Searched defs:einsum_config (Results 1 – 2 of 2) sorted by relevance
559 absl::string_view einsum_config, int64 x_rank, int64 y_rank) { in ParseEinsumString()649 std::string NormalizeEinsumString(absl::string_view einsum_config) { in NormalizeEinsumString()673 XlaOp Einsum(XlaOp x, XlaOp y, absl::string_view einsum_config, in Einsum()691 XlaOp Einsum(XlaOp x, absl::string_view einsum_config, in Einsum()
594 -> StatusOr<HloInstruction*> { in PartitionBaseCase()