Searched defs:Einsum (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | matrix.cc | 352 xla::XlaOp Einsum(xla::XlaOp x, absl::Span<const int64> x_config, xla::XlaOp y, in Einsum() function
673 XlaOp Einsum(XlaOp x, XlaOp y, absl::string_view einsum_config, in Einsum() function
691 XlaOp Einsum(XlaOp x, absl::string_view einsum_config, in Einsum() function
|
| D | matrix_test.cc | 198 XLA_TEST_F(MatrixTest, Einsum) { in XLA_TEST_F() argument
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | op_level_cost_estimator_test.cc | 1768 TEST_F(OpLevelCostEstimatorTest, Einsum) { in TEST_F() argument
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | dot_operation_test.cc | 1376 XLA_TEST_F(DotOperationTextTest, Einsum) { in XLA_TEST_F() argument
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 18191 func Einsum(scope *Scope, inputs []tf.Output, equation string) (output tf.Output) { func
|