Home
last modified time | relevance | path

Searched refs:computation_placer (Results 1 – 18 of 18) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dbackend.cc91 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()
Dbackend.h95 ComputationPlacer* computation_placer() const { return computation_placer_; } in computation_placer() function
162 ComputationPlacer* computation_placer,
Dlocal_service.cc191 return backend().computation_placer()->DeviceId( in ReplicaNumberToDeviceOrdinal()
Dhlo_runner.cc375 backend().computation_placer()->AssignDevices(options.num_replicas, 1)); in ExecuteReplicated()
422 backend().computation_placer()->AssignDevices(options.num_replicas, 1)); in ExecuteReplicated()
DBUILD912 ":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 …]
Dservice.cc1146 backend.computation_placer()->DeviceId(replica, device_handle.handle(), in Replicas()
/external/tensorflow/tensorflow/compiler/xrt/kernels/
DBUILD35 "//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/
DBUILD38 "//tensorflow/compiler/xla/service:computation_placer",
67 "//tensorflow/compiler/xla/service:computation_placer",
/external/tensorflow/tensorflow/compiler/xla/pjrt/
DBUILD166 "//tensorflow/compiler/xla/service:computation_placer",
197 "//tensorflow/compiler/xla/service:computation_placer",
246 "//tensorflow/compiler/xla/service:computation_placer",
Dpjrt_stream_executor_client.cc256 return client_->backend().computation_placer()->AssignDevices(num_replicas, in GetDefaultDeviceAssignment()
/external/tensorflow/tensorflow/compiler/xla/service/interpreter/
DBUILD40 "//tensorflow/compiler/xla/service:computation_placer",
/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
DBUILD144 "//tensorflow/compiler/xla/service:computation_placer",
/external/tensorflow/tensorflow/compiler/xla/client/
DBUILD103 "//tensorflow/compiler/xla/service:computation_placer",
/external/tensorflow/tensorflow/core/tpu/
DBUILD231 "//tensorflow/compiler/xla/service:computation_placer",
/external/tensorflow/tensorflow/core/tpu/kernels/
DBUILD195 "//tensorflow/compiler/xla/service:computation_placer",
304 "//tensorflow/compiler/xla/service:computation_placer",
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
DBUILD603 "//tensorflow/compiler/xla/service:computation_placer",
/external/tensorflow/tensorflow/compiler/xla/tests/
DBUILD306 "//tensorflow/compiler/xla/service:computation_placer",
1444 "//tensorflow/compiler/xla/service:computation_placer",
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/
DBUILD1870 "//tensorflow/compiler/xla/service:computation_placer",