Searched refs:set_requested_device (Results 1 – 13 of 13) sorted by relevance
143 call->set_requested_device(kXlaDeviceName); in TEST_F()221 call->set_requested_device(kXlaDeviceName); in TEST_F()289 call->set_requested_device(kXlaDeviceName); in TEST_F()
111 node->set_requested_device("/gpu:0"); in MakeBodyGraph()219 node->set_requested_device("/gpu:0"); in TEST()
54 op->node.set_requested_device(device); in SetRequestedDevice()
517 if (device.has_value()) node->set_requested_device(*device); in InlineFunctionBody()526 if (device.has_value()) node->set_requested_device(*device); in InlineFunctionBody()541 if (device.has_value()) node->set_requested_device(*device); in InlineFunctionBody()
186 node->set_requested_device(input.node->def().device()); in RemoveListArrayConverter()
1059 ->set_requested_device("/job:a/replica:0/task:0/device:FakeCPU:0"); in TEST_F()1061 ->set_requested_device("/job:a/replica:0/task:0/device:FakeGPU:0"); in TEST_F()1571 GetNodeByName(g, "var")->set_requested_device(gpu_device); in TEST_F()1599 GetNodeByName(g, "var")->set_requested_device(remote_device); in TEST_F()
1214 if (node->name() == "b") node->set_requested_device(call_device); in TEST_F()
3841 new_node->set_requested_device(mklop->requested_device()); in FuseTransposeMklOpTranspose()
1093 out->set_requested_device(requested_device); in SetPaddingNodesDevices()1101 out->set_requested_device(requested_device); in SetPaddingNodesDevices()1173 node->set_requested_device(src_node->requested_device()); in PlaceOpsOnTPU()2602 read_node->set_requested_device(variables[i].node->requested_device()); in BuildVariableReads()3498 image->set_requested_device(DeviceNameUtils::ParsedNameToString(device)); in CopyOutsideCompilationNodes()3745 n->set_requested_device(""); in LowerOutsideCompilationFunctionalNodes()4193 node->set_requested_device(tf_device_assignment[replica_id][0]); in UpdateHeadTailOutsideCompilation()4200 node->set_requested_device(cpu_device); in UpdateHeadTailOutsideCompilation()
117 void set_requested_device(const std::string& device);
229 void Node::set_requested_device(const string& device) { in set_requested_device() function in tensorflow::Node
427 node->set_requested_device(device); in AssignDevicesToArgsAndRetvals()
522 n->set_requested_device(matching_node->requested_device()); in SetNodeShardingFromNeighbors()