Searched refs:x_slice (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | dot_operation_test.cc | 609 auto x_slice = Slice(x_flat, {i, 0, 0}, {i + 1, 2, 2}, {1, 1, 1}); in XLA_TYPED_TEST() local
610 x_slice = Reshape(x_slice, {0, 1, 2}, {2, 2}); in XLA_TYPED_TEST()
614 auto out = Dot(x_slice, y_slice); in XLA_TYPED_TEST()
|