Searched refs:einsum_config (Results 1 – 2 of 2) sorted by relevance
86 absl::string_view einsum_config);98 xla::XlaOp x, xla::XlaOp y, absl::string_view einsum_config,
283 absl::string_view einsum_config) { in ParseEinsumString() argument286 absl::StrSplit(einsum_config, ','); in ParseEinsumString()328 XlaOp Einsum(XlaOp x, XlaOp y, absl::string_view einsum_config, in Einsum() argument333 ParseEinsumString(einsum_config)); in Einsum()