Searched refs:x_config (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | matrix.cc | 118 Status ValidateEinsumNumericDimensions(absl::Span<const int64> x_config, in ValidateEinsumNumericDimensions() argument
122 if (absl::c_linear_search(x_config, dim) || in ValidateEinsumNumericDimensions()
132 for (auto dim : x_config) { in ValidateEinsumNumericDimensions()
135 if (absl::c_count(x_config, dim) > 1) { in ValidateEinsumNumericDimensions()
145 if (absl::c_linear_search(x_config, dim) || in ValidateEinsumNumericDimensions()
159 xla::XlaOp Einsum(xla::XlaOp x, absl::Span<const int64> x_config, xla::XlaOp y, in Einsum() argument
166 ValidateEinsumNumericDimensions(x_config, y_config, output_config)); in Einsum()
167 const int64 x_rank = x_config.size(); in Einsum()
183 for (auto d : x_config) { in Einsum()
207 auto dim_name = x_config[i]; in Einsum()
[all …]
|
| D | matrix.h | 92 Status ValidateEinsumNumericDimensions(absl::Span<const int64> x_config,
107 xla::XlaOp x, absl::Span<const int64> x_config, xla::XlaOp y,
|