Searched refs:transpose_x (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | transpose_folding_test.cc | 348 HloInstruction* transpose_x = in TEST_F() local
363 transpose_x->shape(), y->shape(), /*feature_group_count=*/1, in TEST_F()
368 conv_shape.ValueOrDie(), transpose_x, y, in TEST_F()
416 HloInstruction* transpose_x = in TEST_F() local
431 transpose_x->shape(), y->shape(), /*feature_group_count=*/1, in TEST_F()
436 conv_shape.ValueOrDie(), transpose_x, y, in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | matrix.h | 87 xla::XlaOp x, bool transpose_x, xla::XlaOp y, bool transpose_y,
|
| D | matrix.cc | 543 XlaOp BatchDot(XlaOp x, bool transpose_x, XlaOp y, bool transpose_y, in BatchDot() argument
548 if (transpose_x) { in BatchDot()
|