Searched refs:set_requested_device (Results 1 – 10 of 10) sorted by relevance
141 call->set_requested_device(kXlaDeviceName); in TEST_F()216 call->set_requested_device(kXlaDeviceName); in TEST_F()
111 node->set_requested_device("/gpu:0"); in MakeBodyGraph()220 node->set_requested_device("/gpu:0"); in TEST()
55 op->node.set_requested_device(device); in SetRequestedDevice()
107 void set_requested_device(const string& device);
223 void Node::set_requested_device(const string& device) { in set_requested_device() function in tensorflow::Node
3409 new_node->set_requested_device(mklop->requested_device()); in FuseTransposeMklOpTranspose()
1515 func_body_node->set_requested_device(output_port.node->device()); in PlaceInlinedFunctionBody()1533 func_body_node->set_requested_device(func_node.device()); in PlaceInlinedFunctionBody()
523 n->set_requested_device(matching_node->requested_device()); in SetNodeShardingFromNeighbors()
926 ->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()
1606 node->set_requested_device(caller->def().device()); in InlineFunctionBody()