HomeSort by relevance Sort by last modified time
    Searched refs:DeviceAssignment (Results 1 - 7 of 7) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/service/
computation_placer.h 38 class DeviceAssignment : public Array2D<int> {
40 DeviceAssignment() {}
41 DeviceAssignment(int replica_count, int computation_count)
53 // Return a std::unique_ptr<DeviceAssignment> instead of a DeviceAssignment
55 // due to a StatusOr of an incomplete type (DeviceAssignment).
56 static StatusOr<std::unique_ptr<DeviceAssignment>> Deserialize(
75 virtual StatusOr<DeviceAssignment> AssignDevices(int replica_count,
computation_placer.cc 39 Status DeviceAssignment::Serialize(DeviceAssignmentProto* proto) const {
52 /* static */ StatusOr<std::unique_ptr<DeviceAssignment>>
53 DeviceAssignment::Deserialize(const DeviceAssignmentProto& proto) {
61 auto assignment = MakeUnique<DeviceAssignment>(proto.replica_count(),
85 StatusOr<DeviceAssignment> ComputationPlacer::AssignDevices(
87 DeviceAssignment assignment(replica_count, computation_count);
service.cc 509 TF_ASSIGN_OR_RETURN(DeviceAssignment device_assignment,
645 TF_ASSIGN_OR_RETURN(DeviceAssignment device_assignment,
    [all...]
  /external/tensorflow/tensorflow/compiler/xla/
executable_run_options.h 43 class DeviceAssignment;
84 DeviceAssignment* device_assignment);
85 const DeviceAssignment* device_assignment() const;
90 DeviceAssignment* device_assignment_ = nullptr;
executable_run_options.cc 81 DeviceAssignment* device_assignment) {
86 const DeviceAssignment* ExecutableRunOptions::device_assignment() const {
  /external/tensorflow/tensorflow/contrib/tpu/python/tpu/
device_assignment.py 44 class DeviceAssignment(object):
48 `DeviceAssignment`; it is easier if less flexible than constructing a
49 `DeviceAssignment` directly.
53 """Constructs a `DeviceAssignment` object.
229 Returns a `DeviceAssignment` that describes the cores in the topology assigned
251 A DeviceAssignment object, which describes the mapping between the logical
354 return DeviceAssignment(topology, core_assignment=assignment)
  /external/tensorflow/tensorflow/compiler/xla/python/
local_computation_builder.cc 191 DeviceAssignment device_assignment =

Completed in 177 milliseconds