OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
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