Searched refs:y_config (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | matrix.cc | 119 absl::Span<const int64> y_config, in ValidateEinsumNumericDimensions() argument
123 absl::c_linear_search(y_config, dim)) { in ValidateEinsumNumericDimensions()
133 if (absl::c_linear_search(y_config, dim) || in ValidateEinsumNumericDimensions()
144 for (auto dim : y_config) { in ValidateEinsumNumericDimensions()
147 if (absl::c_count(y_config, dim) > 1) { in ValidateEinsumNumericDimensions()
160 absl::Span<const int64> y_config, in Einsum() argument
166 ValidateEinsumNumericDimensions(x_config, y_config, output_config)); in Einsum()
168 const int64 y_rank = y_config.size(); in Einsum()
187 for (auto d : y_config) { in Einsum()
204 return absl::c_find(y_config, d) - y_config.begin(); in Einsum()
[all …]
|
| D | matrix.h | 93 absl::Span<const int64> y_config,
108 absl::Span<const int64> y_config, absl::Span<const int64> output_config,
|