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",
|