Searched refs:computation_placer (Results 1 - 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | backend.cc | 86 TF_ASSIGN_OR_RETURN(auto computation_placer,
90 computation_placer, options.intra_op_parallelism_threads()));
126 TransferManager* transfer_manager, ComputationPlacer* computation_placer,
131 computation_placer_(computation_placer) {
123 Backend( perftools::gputools::Platform* platform, Compiler* compiler, tensorflow::gtl::ArraySlice<se::StreamExecutor*> stream_executors, TransferManager* transfer_manager, ComputationPlacer* computation_placer, int intra_op_parallelism_threads) argument
|
| H A D | backend.h | 25 #include "tensorflow/compiler/xla/service/computation_placer.h"
88 ComputationPlacer* computation_placer() const { return computation_placer_; } function in class:xla::Backend
164 ComputationPlacer* computation_placer,
|
| H A D | local_service.cc | 150 return backend().computation_placer()->DeviceId(
|
| H A D | service.cc | 510 backend->computation_placer()->AssignDevices(
646 backend->computation_placer()->AssignDevices(
1620 backend.computation_placer()->DeviceId(replica, device_handle.handle(),
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| H A D | local_computation_builder.cc | 193 .computation_placer()
|
Completed in 260 milliseconds