Searched refs:y_array (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | local_client_execute_test.cc | 138 auto y_array = LiteralToShapedBuffer(LiteralUtil::CreateR2WithLayout( in XLA_TEST_F() local
141 y_array.on_device_shape().layout(), LayoutUtil::MakeLayout({1, 0}))); in XLA_TEST_F()
144 ExecuteLocallyOrDie(computation, {&x_array, &y_array}); in XLA_TEST_F()
151 ExecuteLocallyOrDie(computation, {&y_array, &x_array}); in XLA_TEST_F()
166 auto y_array = LiteralToShapedBuffer( in XLA_TEST_F() local
171 computation, {&x_array, &y_array}, in XLA_TEST_F()
184 computation, {&x_array, &y_array}, in XLA_TEST_F()
205 auto y_array = LiteralToShapedBuffer( in XLA_TEST_F() local
209 ExecuteLocallyOrDie(computation, {&x_array, &y_array}); in XLA_TEST_F()
233 auto y_array = LiteralToShapedBuffer( in XLA_TEST_F() local
[all …]
|