experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Fix rocm build

Browse Source 
PiperOrigin-RevId: 279856250
Change-Id: I32dbb045d621cdbc31c6b3f906877c50770c3ad5
This commit is contained in:
Jose Baiocchi 2019-11-11 17:47:52 -08:00 committed by TensorFlower Gardener
parent 9b00382f63
commit 09d58c6012
2 changed files with 6 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/core/profiler/utils/xplane_builder.cc
View File

@ -14,6 +14,8 @@ limitations under the License.
==============================================================================*/
#include "tensorflow/core/profiler/utils/xplane_builder.h"
#include "absl/strings/numbers.h"
namespace tensorflow {
namespace profiler {

8
tensorflow/core/profiler/utils/xplane_builder.h
View File

@ -74,7 +74,7 @@ class XEventBuilder {
AddStat(metadata)->set_double_value(value);
}
void AddStatValue(const XStatMetadata& metadata, absl::string_view value) {
AddStat(metadata)->set_str_value(value);
AddStat(metadata)->set_str_value(string(value));
}
void AddStatValue(const XStatMetadata& metadata, string&& value) {
AddStat(metadata)->set_str_value(std::move(value));
@ -104,10 +104,10 @@ class XLineBuilder {
void SetId(int64 id) { line_->set_id(id); }
void SetName(absl::string_view name) { line_->set_name(name); }
void SetName(absl::string_view name) { line_->set_name(string(name)); }
void SetNameIfEmpty(absl::string_view name) {
if (line_->name().empty()) line_->set_name(name);
if (line_->name().empty()) SetName(name);
}
void SetTimestampNs(int64 timestamp_ns) {
@ -133,7 +133,7 @@ class XPlaneBuilder {
void SetId(int64 id) { plane_->set_id(id); }
void SetName(absl::string_view name) { plane_->set_name(name); }
void SetName(absl::string_view name) { plane_->set_name(string(name)); }
void ReserveLines(size_t num_lines) {
plane_->mutable_lines()->Reserve(num_lines);