Searched refs:ExecuteReplicated (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | collective_ops_test.cc | 101 ExecuteReplicated(std::move(module), {&input_value}, in TestTwoReplicasOneOperand()
246 ExecuteReplicated(std::move(module), {&input_literal}, in XLA_TEST_F()
297 .ExecuteReplicated(e.executable.get(), e.opts, &e.device_assn) in XLA_TEST_F()
363 test_runner_.ExecuteReplicated(executable.get(), opts, &device_assn) in XLA_TEST_F()
390 ExecuteReplicated(std::move(module), {&input_literal}, /*num_replicas=*/4, in XLA_TEST_F()
423 ExecuteReplicated(std::move(module), {}, kNumReplicas, in XLA_TEST_F()
451 ExecuteReplicated(std::move(module), {}, kNumReplicas, in XLA_TEST_F()
|
| D | hlo_test_base.cc | 193 StatusOr<std::vector<Literal>> HloTestBase::ExecuteReplicated( in ExecuteReplicated() function in xla::HloTestBase
203 return test_runner_.ExecuteReplicated(std::move(module), options); in ExecuteReplicated()
206 StatusOr<std::vector<Literal>> HloTestBase::ExecuteReplicated( in ExecuteReplicated() function in xla::HloTestBase
217 return test_runner_.ExecuteReplicated(std::move(module), options, in ExecuteReplicated()
|
| D | hlo_test_base.h | 156 StatusOr<std::vector<Literal>> ExecuteReplicated(
161 StatusOr<std::vector<Literal>> ExecuteReplicated(
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_runner.h | 176 StatusOr<std::vector<Literal>> ExecuteReplicated(
181 StatusOr<std::vector<Literal>> ExecuteReplicated(
191 StatusOr<std::vector<Literal>> ExecuteReplicated(
|
| D | hlo_runner.cc | 270 StatusOr<std::vector<Literal>> HloRunner::ExecuteReplicated( in ExecuteReplicated() function in xla::HloRunner
276 return ExecuteReplicated(executable.get(), options, device_assignment); in ExecuteReplicated()
279 StatusOr<std::vector<Literal>> HloRunner::ExecuteReplicated( in ExecuteReplicated() function in xla::HloRunner
421 StatusOr<std::vector<Literal>> HloRunner::ExecuteReplicated( in ExecuteReplicated() function in xla::HloRunner
427 return ExecuteReplicated(std::move(module), options, &device_assignment); in ExecuteReplicated()
|