Home	Sort by relevance Sort by last modified time

Full Search Definition Symbol File Path History   | | Help

    Searched full: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/core/graph/
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...]
subgraph.cc	104 recv_node->set_assigned_device_name(device_info.name()); 252 send_node->set_assigned_device_name(device_info.name());
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);
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());
mkl_layout_pass.cc	[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...]
mkl_layout_pass_test.cc	59 node->set_assigned_device_name(device);     [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);

Completed in 467 milliseconds