Home
last modified time | relevance | path

Searched refs:set_requested_device (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/compiler/jit/
Dbuild_xla_ops_pass_test.cc141 call->set_requested_device(kXlaDeviceName); in TEST_F()
216 call->set_requested_device(kXlaDeviceName); in TEST_F()
Dencapsulate_xla_computations_pass_test.cc111 node->set_requested_device("/gpu:0"); in MakeBodyGraph()
220 node->set_requested_device("/gpu:0"); in TEST()
/external/tensorflow/tensorflow/c/
Dpython_api.cc55 op->node.set_requested_device(device); in SetRequestedDevice()
/external/tensorflow/tensorflow/core/graph/
Dgraph.h107 void set_requested_device(const string& device);
Dgraph.cc223 void Node::set_requested_device(const string& device) { in set_requested_device() function in tensorflow::Node
Dmkl_layout_pass.cc3409 new_node->set_requested_device(mklop->requested_device()); in FuseTransposeMklOpTranspose()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dfunction_optimizer.cc1515 func_body_node->set_requested_device(output_port.node->device()); in PlaceInlinedFunctionBody()
1533 func_body_node->set_requested_device(func_node.device()); in PlaceInlinedFunctionBody()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dtf2xla_util.cc523 n->set_requested_device(matching_node->requested_device()); in SetNodeShardingFromNeighbors()
/external/tensorflow/tensorflow/core/common_runtime/
Dplacer_test.cc926 ->set_requested_device("/job:a/replica:0/task:0/device:fakecpu:0"); in TEST_F()
928 ->set_requested_device("/job:a/replica:0/task:0/device:fakegpu:0"); in TEST_F()
Dfunction.cc1606 node->set_requested_device(caller->def().device()); in InlineFunctionBody()