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

Explicitly convert StringRef to string_view

Browse Source 
PiperOrigin-RevId: 300487444
Change-Id: I4da515a62085935fed643764e00b8fedbd5a67a9
This commit is contained in:
Smit Hinsu 2020-03-11 23:51:29 -07:00 committed by TensorFlower Gardener
parent 373069d559
commit 78a2adc697
1 changed files with 7 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
tensorflow/compiler/mlir/mlir_graph_optimization_pass.cc
View File

@ -31,6 +31,10 @@ limitations under the License.
namespace tensorflow { namespace tensorflow {
static inline absl::string_view StringRefToView(llvm::StringRef ref) {
return {ref.data(), ref.size()};
}
// Dumps the MLIR module to disk. // Dumps the MLIR module to disk.
// This require the TF_DUMP_GRAPH_PREFIX to be set to a path that exist (or can // This require the TF_DUMP_GRAPH_PREFIX to be set to a path that exist (or can
// be created). // be created).
@ -115,7 +119,7 @@ Status MlirFunctionOptimizationPass::Run(
for (auto& pass_registration : registry_->passes()) { for (auto& pass_registration : registry_->passes()) {
llvm::StringRef name = pass_registration.pass->name(); llvm::StringRef name = pass_registration.pass->name();
VLOG(2) << "Run MLIR graph optimization pass: " << absl::string_view(name); VLOG(2) << "Run MLIR graph optimization pass: " << StringRefToView(name);
if (VLOG_IS_ON(1)) { if (VLOG_IS_ON(1)) {
DumpModule(*module_ref, llvm::formatv("mlir_{0}_before_", name)); DumpModule(*module_ref, llvm::formatv("mlir_{0}_before_", name));
@ -185,8 +189,8 @@ Status MlirV1CompatGraphOptimizationPass::Run(
AddDevicesToOp(*module_ref, options.device_set); AddDevicesToOp(*module_ref, options.device_set);
for (auto& pass_registration : registry_->passes()) { for (auto& pass_registration : registry_->passes()) {
absl::string_view name = pass_registration.pass->name(); llvm::StringRef name = pass_registration.pass->name();
VLOG(2) << "Run MLIR graph optimization pass: " << name; VLOG(2) << "Run MLIR graph optimization pass: " << StringRefToView(name);
if (VLOG_IS_ON(1)) { if (VLOG_IS_ON(1)) {
DumpModule(*module_ref, llvm::formatv("mlir_{0}_before_", name)); DumpModule(*module_ref, llvm::formatv("mlir_{0}_before_", name));