Searched refs:x_config (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | matrix.cc | 352 xla::XlaOp Einsum(xla::XlaOp x, absl::Span<const int64> x_config, xla::XlaOp y, in Einsum() argument
358 auto x_diagonal_labels = EinsumDiagonalLabels(x_config); in Einsum()
360 return Einsum(EinsumDiagonal(x, x_config), x_diagonal_labels->at(0), y, in Einsum()
365 return Einsum(x, x_config, EinsumDiagonal(y, y_config), in Einsum()
371 Einsum(x, x_config, y, y_config, output_diagonal_labels->at(0), in Einsum()
378 const int64 x_rank = x_config.size(); in Einsum()
385 for (auto d : x_config) { in Einsum()
414 auto dim_name = x_config[i]; in Einsum()
463 output_transpose_dims.push_back(*output_dimension_number(x_config[d])); in Einsum()
467 if (auto output_dim = output_dimension_number(x_config[d])) { in Einsum()
|
| D | matrix.h | 130 xla::XlaOp x, absl::Span<const int64> x_config, xla::XlaOp y,
|