Home
last modified time | relevance | path

Searched defs:assigned_device (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/
Dplacer.cc120 Status AssignAndLog(int assigned_device, Node* node, in AssignAndLog()
223 int assigned_device = -1; in Run() local
262 int assigned_device = -1; in Run() local
Dcolocation_graph.cc1398 const Device* assigned_device = in InitializeMemberWithAssignedDevice() local
/external/tensorflow/tensorflow/compiler/jit/
Dnode_matchers.h91 const absl::optional<string>& assigned_device() const { in assigned_device() function
115 void set_assigned_device(string assigned_device) { in set_assigned_device()
Dnode_matchers.cc332 absl::optional<string> assigned_device; member
449 impl::NodeMatcherProperties AssignedDevice(string assigned_device) { in AssignedDevice()
/external/tensorflow/tensorflow/cc/framework/
Dscope.cc235 const string& assigned_device) in Impl()
/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Ddistributed_tpu_rewrite_pass.cc1346 const string& assigned_device = unpadded_input->assigned_device_name(); in SetPaddingNodesDevices() local