Home | History | Annotate | Download | only in util

Lines Matching refs:str_util

19 #include "tensorflow/core/lib/strings/str_util.h"
80 if (str_util::ConsumeLeadingDigits(in, &tmp)) {
110 return DeviceName(job, replica, task, "/", str_util::Lowercase(type), id);
121 if (str_util::ConsumePrefix(&fullname, "/job:")) {
122 p->has_job = !str_util::ConsumePrefix(&fullname, "*");
128 if (str_util::ConsumePrefix(&fullname, "/replica:")) {
129 p->has_replica = !str_util::ConsumePrefix(&fullname, "*");
135 if (str_util::ConsumePrefix(&fullname, "/task:")) {
136 p->has_task = !str_util::ConsumePrefix(&fullname, "*");
142 if (str_util::ConsumePrefix(&fullname, "/device:")) {
143 p->has_type = !str_util::ConsumePrefix(&fullname, "*");
147 if (!str_util::ConsumePrefix(&fullname, ":")) {
150 p->has_id = !str_util::ConsumePrefix(&fullname, "*");
159 if (str_util::ConsumePrefix(&fullname, "/cpu:") ||
160 str_util::ConsumePrefix(&fullname, "/CPU:")) {
163 p->has_id = !str_util::ConsumePrefix(&fullname, "*");
169 if (str_util::ConsumePrefix(&fullname, "/gpu:") ||
170 str_util::ConsumePrefix(&fullname, "/GPU:")) {
173 p->has_id = !str_util::ConsumePrefix(&fullname, "*");
377 if (!str_util::ConsumePrefix(&name, ":")) {