Searched refs:ReplicatedExecuteOptions (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_runner.h | 106 const ReplicatedExecuteOptions& options) override;
111 const ReplicatedExecuteOptions& options,
120 Executable* executable, const ReplicatedExecuteOptions& options,
132 const ReplicatedExecuteOptions& options,
166 const ReplicatedExecuteOptions& options,
|
| D | hlo_runner_interface.h | 45 struct ReplicatedExecuteOptions { struct
157 const ReplicatedExecuteOptions& options) = 0;
162 const ReplicatedExecuteOptions& options,
169 const ReplicatedExecuteOptions& options,
|
| D | hlo_runner.cc | 195 std::unique_ptr<HloModule> module, const ReplicatedExecuteOptions& options, in ExecuteReplicated()
210 const ReplicatedExecuteOptions& options, in ExecuteReplicatedImpl()
342 Executable* executable, const ReplicatedExecuteOptions& options, in ExecuteReplicated()
394 const ReplicatedExecuteOptions& options, in ExecuteReplicated()
447 const ReplicatedExecuteOptions& options) { in ExecuteReplicated()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | replicated_io_feed_test.cc | 66 HloRunner::ReplicatedExecuteOptions opts; in XLA_TEST_F()
|
| D | hlo_test_base.cc | 251 HloRunner::ReplicatedExecuteOptions options; in ExecuteReplicated()
265 HloRunner::ReplicatedExecuteOptions options; in ExecuteReplicated()
282 HloRunner::ReplicatedExecuteOptions options; in ExecuteReplicated()
592 HloRunner::ReplicatedExecuteOptions options; in RunReplicated()
|
| D | collective_ops_test.cc | 379 HloRunner::ReplicatedExecuteOptions opts; in XLA_TEST_F()
|