Searched refs:y_config (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | matrix.cc | 353 absl::Span<const int64> y_config, in Einsum() argument
361 y_config, output_config, precision); in Einsum()
363 auto y_diagonal_labels = EinsumDiagonalLabels(y_config); 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()
379 const int64 y_rank = y_config.size(); in Einsum()
389 for (auto d : y_config) { in Einsum()
406 return absl::c_find(y_config, d) - y_config.begin(); in Einsum()
445 auto dim_name = y_config[i]; in Einsum()
475 if (auto output_dim = output_dimension_number(y_config[d])) { in Einsum()
|
| D | matrix.h | 131 absl::Span<const int64> y_config, absl::Span<const int64> output_config,
|