Searched refs:computation_placer (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | backend.cc | 113 TF_ASSIGN_OR_RETURN(auto computation_placer, in CreateBackend()
117 computation_placer, options.intra_op_parallelism_threads())); in CreateBackend()
146 ComputationPlacer* computation_placer, in Backend() argument
151 computation_placer_(computation_placer) { in Backend()
|
| D | backend.h | 95 ComputationPlacer* computation_placer() const { return computation_placer_; } in computation_placer() function
162 ComputationPlacer* computation_placer,
|
| D | local_service.cc | 184 return backend().computation_placer()->DeviceId( in ReplicaNumberToDeviceOrdinal()
|
| D | BUILD | 692 ":computation_placer",
722 ":computation_placer",
937 ":computation_placer",
2119 name = "computation_placer",
2120 srcs = ["computation_placer.cc"],
2121 hdrs = ["computation_placer.h"],
3230 ":computation_placer",
3483 ":computation_placer",
|
| D | hlo_runner.cc | 421 backend().computation_placer()->AssignDevices(options.num_replicas, 1)); in ExecuteReplicated()
|
| D | service.cc | 1169 backend.computation_placer()->DeviceId(replica, device_handle.handle(), in Replicas()
|
| /external/tensorflow/tensorflow/compiler/xrt/kernels/ |
| D | BUILD | 22 "//tensorflow/compiler/xla/service:computation_placer",
52 "//tensorflow/compiler/xla/service:computation_placer",
|
| /external/tensorflow/tensorflow/compiler/xla/service/interpreter/ |
| D | BUILD | 37 "//tensorflow/compiler/xla/service:computation_placer",
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | BUILD | 76 "//tensorflow/compiler/xla/service:computation_placer",
|
| D | local_computation_builder.cc | 357 client.client()->backend().computation_placer()->AssignDevices( in Compile()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | BUILD | 261 "//tensorflow/compiler/xla/service:computation_placer",
1092 "//tensorflow/compiler/xla/service:computation_placer",
|