Searched defs:einsum_config (Results 1 – 2 of 2) sorted by relevance
595 absl::string_view einsum_config, int64_t x_rank, int64_t y_rank) { in ParseEinsumString()685 std::string NormalizeEinsumString(absl::string_view einsum_config) { in NormalizeEinsumString()709 XlaOp Einsum(XlaOp x, XlaOp y, absl::string_view einsum_config, in Einsum()728 XlaOp Einsum(XlaOp x, absl::string_view einsum_config, in Einsum()
884 -> StatusOr<HloInstruction*> { in PartitionBaseCase()