Searched defs:x_config (Results 1 – 1 of 1) sorted by relevance
118 Status ValidateEinsumNumericDimensions(absl::Span<const int64> x_config, in ValidateEinsumNumericDimensions()159 xla::XlaOp Einsum(xla::XlaOp x, absl::Span<const int64> x_config, xla::XlaOp y, in Einsum()269 std::vector<int64> x_config = batch_dimension_numbers; in BatchDot() local302 auto& x_config = einsum_config_numeric[0]; in ParseEinsumString() local