Searched refs:Einsum (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | matrix.h | 97 xla::XlaOp Einsum(
106 xla::XlaOp Einsum(
|
| D | matrix.cc | 159 xla::XlaOp Einsum(xla::XlaOp x, absl::Span<const int64> x_config, xla::XlaOp y, in Einsum() function
278 return Einsum(x, x_config, y, y_config, output_config, precision); in BatchDot()
328 XlaOp Einsum(XlaOp x, XlaOp y, absl::string_view einsum_config, in Einsum() function
334 return Einsum(x, einsum_config_numeric[0], y, einsum_config_numeric[1], in Einsum()
|
| D | matrix_test.cc | 118 XLA_TEST_F(MatrixTest, Einsum) { in XLA_TEST_F() argument
133 Einsum(l_index, row, "abc,adc->abd"); in XLA_TEST_F()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | einsum_op.cc | 43 ctx->SetOutput(0, xla::Einsum(lhs, rhs, equation_)); in Compile()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | dot_operation_test.cc | 1175 Einsum(x, y, std::get<2>(GetParam())); in XLA_TEST_P()
1205 INSTANTIATE_TEST_CASE_P(Einsum, EinsumTest,
1255 XLA_TEST_F(DotOperationTextTest, Einsum) { in XLA_TEST_F() argument
|