Searched refs:x_config (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | matrix.cc | 383 xla::XlaOp Einsum(xla::XlaOp x, absl::Span<const int64_t> x_config, in Einsum() argument
390 auto x_diagonal_labels = EinsumDiagonalLabels(x_config); in Einsum()
392 return Einsum(EinsumDiagonal(x, x_config), x_diagonal_labels->at(0), y, in Einsum()
397 return Einsum(x, x_config, EinsumDiagonal(y, y_config), in Einsum()
404 Einsum(x, x_config, y, y_config, output_diagonal_labels->at(0), in Einsum()
411 const int64_t x_rank = x_config.size(); in Einsum()
418 for (auto d : x_config) { in Einsum()
447 auto dim_name = x_config[i]; in Einsum()
496 output_transpose_dims.push_back(*output_dimension_number(x_config[d])); in Einsum()
500 if (auto output_dim = output_dimension_number(x_config[d])) { in Einsum()
|
| D | matrix.h | 142 xla::XlaOp x, absl::Span<const int64_t> x_config, xla::XlaOp y,
|