Searched refs:ExecuteLocallyOrDie (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | local_client_execute_test.cc | 61 ExecuteLocallyOrDie(builder.Build().ValueOrDie(), {}); in XLA_TEST_F()
74 ExecuteLocallyOrDie(builder.Build().ValueOrDie(), {&x_value}); in XLA_TEST_F()
87 ExecuteLocallyOrDie(builder.Build().ValueOrDie(), {&x_array}); in XLA_TEST_F()
101 ExecuteLocallyOrDie(builder.Build().ValueOrDie(), {&x_array}); in XLA_TEST_F()
115 ScopedShapedBuffer result = ExecuteLocallyOrDie( 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()
170 ScopedShapedBuffer result_colmaj = ExecuteLocallyOrDie( in XLA_TEST_F()
183 ScopedShapedBuffer result_rowmaj = ExecuteLocallyOrDie( in XLA_TEST_F()
209 ExecuteLocallyOrDie(computation, {&x_array, &y_array}); in XLA_TEST_F()
[all …]
|
| D | local_client_allocation_test.cc | 57 ExecuteLocallyOrDie(builder.Build().ValueOrDie(), {}, in XLA_TEST_F()
91 auto result = ExecuteLocallyOrDie( in XLA_TEST_F()
|
| D | local_client_test_base.h | 107 ScopedShapedBuffer ExecuteLocallyOrDie(
110 ScopedShapedBuffer ExecuteLocallyOrDie(
|
| D | local_client_test_base.cc | 163 ScopedShapedBuffer LocalClientTestBase::ExecuteLocallyOrDie( in ExecuteLocallyOrDie() function in xla::LocalClientTestBase
171 ScopedShapedBuffer LocalClientTestBase::ExecuteLocallyOrDie( in ExecuteLocallyOrDie() function in xla::LocalClientTestBase
|