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

  /external/tensorflow/tensorflow/core/common_runtime/
direct_session_with_tracking_alloc_test.cc 54 a->set_assigned_device_name("/job:localhost/replica:0/task:0/cpu:0");
59 x->set_assigned_device_name("/job:localhost/replica:0/task:0/cpu:1");
63 y->set_assigned_device_name("/job:localhost/replica:0/task:0/cpu:0");
66 y_neg->set_assigned_device_name("/job:localhost/replica:0/task:0/cpu:1");
156 a->set_assigned_device_name("/job:localhost/replica:0/task:0/cpu:0");
161 x->set_assigned_device_name("/job:localhost/replica:0/task:0/device:GPU:0");
163 x->set_assigned_device_name("/job:localhost/replica:0/task:0/device:SYCL:0");
168 y->set_assigned_device_name("/job:localhost/replica:0/task:0/device:GPU:0");
170 y->set_assigned_device_name("/job:localhost/replica:0/task:0/device:SYCL:0");
174 y_neg->set_assigned_device_name("/job:localhost/replica:0/task:0/cpu:0")
    [all...]
direct_session_test.cc 65 a->set_assigned_device_name("/job:localhost/replica:0/task:0/cpu:0");
70 x->set_assigned_device_name("/job:localhost/replica:0/task:0/cpu:1");
75 y->set_assigned_device_name("/job:localhost/replica:0/task:0/cpu:0");
80 y_neg->set_assigned_device_name("/job:localhost/replica:0/task:0/cpu:1");
237 a->set_assigned_device_name("/job:localhost/replica:0/task:0/cpu:0");
241 x->set_assigned_device_name("/job:localhost/replica:0/task:0/cpu:1");
244 y->set_assigned_device_name("/job:localhost/replica:0/task:0/cpu:2");
257 y->set_assigned_device_name("/job:localhost/replica:0/task:0/cpu:1");
303 var->set_assigned_device_name("/job:localhost/replica:0/task:0/cpu:0");
311 twenty_node->set_assigned_device_name(
    [all...]
placer_test.cc 446 GetNodeByName(g, "in")->set_assigned_device_name(
486 GetNodeByName(g, "in")->set_assigned_device_name(
    [all...]
graph_execution_state.cc 229 n->set_assigned_device_name(iter->second);
368 node->set_assigned_device_name(node->requested_device());
constant_folding.cc 546 constant_node->set_assigned_device_name(partition_device->name());
  /external/tensorflow/tensorflow/core/platform/
device_tracer_test.cc 61 a->set_assigned_device_name("/job:localhost/replica:0/task:0/cpu:0");
66 x->set_assigned_device_name("/job:localhost/replica:0/task:0/device:GPU:0");
71 y->set_assigned_device_name("/job:localhost/replica:0/task:0/device:GPU:0");
76 i->set_assigned_device_name("/job:localhost/replica:0/task:0/cpu:0");
80 y_neg->set_assigned_device_name(
  /external/tensorflow/tensorflow/core/debug/
debug_gateway_test.cc 67 a->set_assigned_device_name(kDeviceName);
73 x->set_assigned_device_name(kDeviceName);
78 y->set_assigned_device_name(kDeviceName);
83 y_neg->set_assigned_device_name(kDeviceName);
525 a->set_assigned_device_name(kDeviceName);
528 c->set_assigned_device_name(kDeviceName);
535 y->set_assigned_device_name(kDeviceName);
628 var->set_assigned_device_name(kDeviceName);
639 init_val->set_assigned_device_name(kDeviceName);
643 init->set_assigned_device_name(kDeviceName)
    [all...]
  /external/tensorflow/tensorflow/compiler/tf2xla/
tf2xla_util_test.cc 230 a_node->set_assigned_device_name("foo");
234 a_node->set_assigned_device_name("/device:TPU_REPLICATED_CORE:2");
242 b_node->set_assigned_device_name("/device:TPU_REPLICATED_CORE:1");
sharding_util.cc 105 dst->set_assigned_device_name(device_name);
tf2xla_util.cc 278 n->set_assigned_device_name(matching_node->assigned_device_name());
tf2xla.cc 257 node->set_assigned_device_name(
  /external/tensorflow/tensorflow/core/distributed_runtime/rpc/
grpc_session_test.cc 535 a->set_assigned_device_name(dev_a);
537 a_err->set_assigned_device_name(dev_a);
539 a2->set_assigned_device_name(dev_a);
549 b->set_assigned_device_name(dev_b);
551 b_delay->set_assigned_device_name(dev_b);
553 b2->set_assigned_device_name(dev_b);
592 a->set_assigned_device_name(dev_a);
597 a_err->set_assigned_device_name(dev_a);
599 a2->set_assigned_device_name(dev_a);
609 b->set_assigned_device_name(dev_b)
    [all...]
  /external/tensorflow/tensorflow/compiler/jit/
build_xla_launch_ops_pass.cc 88 launch_node->set_assigned_device_name(node->assigned_device_name());
encapsulate_subgraphs_pass.cc     [all...]
mark_for_compilation_pass_test.cc 43 n->set_assigned_device_name(kCpuDevice);
  /external/tensorflow/tensorflow/core/graph/
graph.h 114 void set_assigned_device_name(const string& device_name);
561 void set_assigned_device_name(Node* node, const string& device_name) { function in class:tensorflow::Graph
819 inline void Node::set_assigned_device_name(const string& device_name) { function in class:tensorflow::Node
820 graph_->set_assigned_device_name(this, device_name);
graph_partition.cc 334 node->set_assigned_device_name(
343 node->set_assigned_device_name(
360 node->set_assigned_device_name(
389 res_node->set_assigned_device_name(device_name);
402 res_node->set_assigned_device_name(device_name);
413 res_node->set_assigned_device_name(device_name);
422 res_node->set_assigned_device_name(device_name);
444 res_node->set_assigned_device_name(device_name);
    [all...]
mkl_tfconversion_pass.cc 185 conversion_node->set_assigned_device_name(src->assigned_device_name());
266 conversion_node->set_assigned_device_name(n->assigned_device_name());
subgraph.cc 104 recv_node->set_assigned_device_name(device_info.name());
252 send_node->set_assigned_device_name(device_info.name());
mkl_layout_pass.cc     [all...]
mkl_layout_pass_test.cc 59 node->set_assigned_device_name(device);
    [all...]
graph.cc 346 copy->set_assigned_device_name(node->assigned_device_name());
graph_partition_test.cc 87 node->set_assigned_device_name(device_name);
graph_constructor.cc 553 (*node)->set_assigned_device_name(node_def.device());
    [all...]

Completed in 906 milliseconds