Searched refs:computation_placer (Results 1 – 18 of 18) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | backend.cc | 91 TF_ASSIGN_OR_RETURN(auto computation_placer, in CreateBackend()
95 computation_placer, options.intra_op_parallelism_threads())); in CreateBackend()
124 ComputationPlacer* computation_placer, in Backend() argument
129 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 | 191 return backend().computation_placer()->DeviceId( in ReplicaNumberToDeviceOrdinal()
|
| D | hlo_runner.cc | 375 backend().computation_placer()->AssignDevices(options.num_replicas, 1)); in ExecuteReplicated()
422 backend().computation_placer()->AssignDevices(options.num_replicas, 1)); in ExecuteReplicated()
|
| D | BUILD | 912 ":computation_placer",
942 ":computation_placer",
1167 ":computation_placer",
2906 name = "computation_placer",
2907 srcs = ["computation_placer.cc"],
2908 hdrs = ["computation_placer.h"],
2935 hdrs = ["computation_placer.h"],
4339 ":computation_placer",
4558 ":computation_placer",
4579 ":computation_placer",
[all …]
|
| D | service.cc | 1146 backend.computation_placer()->DeviceId(replica, device_handle.handle(), in Replicas()
|
| /external/tensorflow/tensorflow/compiler/xrt/kernels/ |
| D | BUILD | 35 "//tensorflow/compiler/xla/service:computation_placer",
70 "//tensorflow/compiler/xla/service:computation_placer",
127 "//tensorflow/compiler/xla/service:computation_placer",
|
| /external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/client/ |
| D | BUILD | 38 "//tensorflow/compiler/xla/service:computation_placer",
67 "//tensorflow/compiler/xla/service:computation_placer",
|
| /external/tensorflow/tensorflow/compiler/xla/pjrt/ |
| D | BUILD | 166 "//tensorflow/compiler/xla/service:computation_placer",
197 "//tensorflow/compiler/xla/service:computation_placer",
246 "//tensorflow/compiler/xla/service:computation_placer",
|
| D | pjrt_stream_executor_client.cc | 256 return client_->backend().computation_placer()->AssignDevices(num_replicas, in GetDefaultDeviceAssignment()
|
| /external/tensorflow/tensorflow/compiler/xla/service/interpreter/ |
| D | BUILD | 40 "//tensorflow/compiler/xla/service:computation_placer",
|
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | BUILD | 144 "//tensorflow/compiler/xla/service:computation_placer",
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | BUILD | 103 "//tensorflow/compiler/xla/service:computation_placer",
|
| /external/tensorflow/tensorflow/core/tpu/ |
| D | BUILD | 231 "//tensorflow/compiler/xla/service:computation_placer",
|
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | BUILD | 195 "//tensorflow/compiler/xla/service:computation_placer",
304 "//tensorflow/compiler/xla/service:computation_placer",
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | BUILD | 603 "//tensorflow/compiler/xla/service:computation_placer",
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | BUILD | 306 "//tensorflow/compiler/xla/service:computation_placer",
1444 "//tensorflow/compiler/xla/service:computation_placer",
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ |
| D | BUILD | 1870 "//tensorflow/compiler/xla/service:computation_placer",
|