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