Searched defs:BatchDot (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | elemental_ir_emitter_test.cc | 99 XLA_TEST_F(ElementalIrEmitterExecutionTest, BatchDot) { in XLA_TEST_F() argument
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | matrix.cc | 573 XlaOp BatchDot(XlaOp x, XlaOp y, PrecisionConfig::Precision precision, in BatchDot() function
578 XlaOp BatchDot(XlaOp x, bool transpose_x, XlaOp y, bool transpose_y, in BatchDot() function
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 441 Value BatchDot(Location loc, Value lhs, bool transpose_lhs, Value rhs, in BatchDot() function
|