Searched refs:x_array (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | local_client_execute_test.cc | 85 auto x_array = LiteralToShapedBuffer(LiteralUtil::CreateR1<float>({})); in XLA_TEST_F() local
87 ExecuteLocallyOrDie(builder.Build().ValueOrDie(), {&x_array}); in XLA_TEST_F()
98 auto x_array = in XLA_TEST_F() local
101 ExecuteLocallyOrDie(builder.Build().ValueOrDie(), {&x_array}); in XLA_TEST_F()
112 auto x_array = in XLA_TEST_F() local
116 builder.Build().ValueOrDie(), {&x_array}, DefaultExecutableBuildOptions(), in XLA_TEST_F()
132 auto x_array = LiteralToShapedBuffer(LiteralUtil::CreateR2WithLayout( in XLA_TEST_F() local
135 x_array.on_device_shape().layout(), LayoutUtil::MakeLayout({0, 1}))); 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()
[all …]
|
| D | local_client_allocation_test.cc | 47 auto x_array = in XLA_TEST_F() local
|