Searched defs:requested_device (Results 1 – 6 of 6) sorted by relevance
262 std::string requested_device; member379 const std::string& requested_device, in AddRecvDeduplicationDataNode()513 absl::string_view tpu_replicate_attr, const std::string& requested_device, in RewriteGraphForTpuReplicateAttrAndDevice()554 const std::string& requested_device = activations_node->requested_device(); in InsertActivationsNodeIntoMap() local594 const std::string& requested_device = gradients_node->requested_device(); in InsertGradientsNodeIntoMap() local655 const std::string& requested_device = in RewriteGraph() local
1333 const string& requested_device = unpadded_input->requested_device(); in SetPaddingNodesDevices() local3636 DeviceNameUtils::ParsedName requested_device; in BuildExecuteNodes() local
90 const auto& requested_device = op.DeviceName(); in SelectFromNodeDef() local
208 const std::string& Node::requested_device() const { return def().device(); } in requested_device() function in tensorflow::Node
919 std::optional<string> requested_device) { in BuildIdentityNode()
166 const std::string& requested_device() const { in requested_device() function