Searched refs:ExecuteLocallyOrDie (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | local_client_execute_test.cc | 60 ExecuteLocallyOrDie(builder.Build().ValueOrDie(), {}); in XLA_TEST_F()
73 ExecuteLocallyOrDie(builder.Build().ValueOrDie(), {&x_value}); in XLA_TEST_F()
86 ExecuteLocallyOrDie(builder.Build().ValueOrDie(), {&x_array}); in XLA_TEST_F()
100 ExecuteLocallyOrDie(builder.Build().ValueOrDie(), {&x_array}); in XLA_TEST_F()
114 ScopedShapedBuffer result = ExecuteLocallyOrDie( in XLA_TEST_F()
143 ExecuteLocallyOrDie(computation, {&x_array, &y_array}); in XLA_TEST_F()
150 ExecuteLocallyOrDie(computation, {&y_array, &x_array}); in XLA_TEST_F()
169 ScopedShapedBuffer result_colmaj = ExecuteLocallyOrDie( in XLA_TEST_F()
181 ScopedShapedBuffer result_rowmaj = ExecuteLocallyOrDie( in XLA_TEST_F()
206 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 | 102 ScopedShapedBuffer ExecuteLocallyOrDie(
105 ScopedShapedBuffer ExecuteLocallyOrDie(
|
| D | local_client_test_base.cc | 157 ScopedShapedBuffer LocalClientTestBase::ExecuteLocallyOrDie( in ExecuteLocallyOrDie() function in xla::LocalClientTestBase
165 ScopedShapedBuffer LocalClientTestBase::ExecuteLocallyOrDie( in ExecuteLocallyOrDie() function in xla::LocalClientTestBase
|