Searched refs:transpose_y (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | transpose_folding_test.cc | 228 HloInstruction* transpose_y = in TEST_F() local
241 transpose_y->shape().dimensions(dnums.kernel_spatial_dimensions(i))); in TEST_F()
244 x->shape(), transpose_y->shape(), /*feature_group_count=*/1, in TEST_F()
249 conv_shape.ValueOrDie(), x, transpose_y, in TEST_F()
285 HloInstruction* transpose_y = in TEST_F() local
298 transpose_y->shape().dimensions(dnums.kernel_spatial_dimensions(i))); in TEST_F()
301 x->shape(), transpose_y->shape(), /*feature_group_count=*/1, in TEST_F()
306 conv_shape.ValueOrDie(), x, transpose_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
551 if (transpose_y) { in BatchDot()
|