Home
last modified time | relevance | path

Searched refs:op_or_val (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/
Dop_or_arg_name_mapper.cc78 llvm::StringRef OpOrArgNameMapper::GetUniqueName(OpOrVal op_or_val) { in GetUniqueName() argument
79 auto& name = op_or_val_to_name_[op_or_val]; in GetUniqueName()
82 llvm::StringRef ref = GetUniqueName(GetName(op_or_val)); in GetUniqueName()
87 absl::string_view OpOrArgNameMapper::GetUniqueNameView(OpOrVal op_or_val) { in GetUniqueNameView() argument
88 auto& name = op_or_val_to_name_[op_or_val]; in GetUniqueNameView()
91 name = StringRefToView(GetUniqueName(GetName(op_or_val))); in GetUniqueNameView()
95 int OpOrArgNameMapper::InitOpName(OpOrVal op_or_val, llvm::StringRef name) { in InitOpName() argument
98 op_or_val, StringRefToView(it.first->first())); in InitOpName()
107 std::string OpOrArgLocNameMapper::GetName(OpOrVal op_or_val) { in GetName() argument
108 if (auto* op = op_or_val.dyn_cast<mlir::Operation*>()) { in GetName()
[all …]
Dop_or_arg_name_mapper.h42 llvm::StringRef GetUniqueName(OpOrVal op_or_val);
45 absl::string_view GetUniqueNameView(OpOrVal op_or_val);
53 int InitOpName(OpOrVal op_or_val, llvm::StringRef name);
72 virtual std::string GetName(OpOrVal op_or_val) = 0;
87 std::string GetName(OpOrVal op_or_val) override;
94 std::string GetName(OpOrVal op_or_val) override;
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/
Dexport_graphdef.cc86 std::string GetName(OpOrVal op_or_val) override { in GetName() argument
87 std::string name = OpOrArgLocNameMapper::GetName(op_or_val); in GetName()
Dimport_model.cc173 std::string GetName(OpOrVal op_or_val) override { in GetName() argument