| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | name_uniquer_test.cc | 32 EXPECT_EQ("foo", uniquer.GetUniqueName("foo")); in TEST_F()
33 EXPECT_EQ("foo__1", uniquer.GetUniqueName("foo")); in TEST_F()
34 EXPECT_EQ("foo__2", uniquer.GetUniqueName("foo")); in TEST_F()
35 EXPECT_EQ("bar", uniquer.GetUniqueName("bar")); in TEST_F()
36 EXPECT_EQ("foo__3", uniquer.GetUniqueName("foo")); in TEST_F()
37 EXPECT_EQ("bar__1", uniquer.GetUniqueName("bar")); in TEST_F()
38 EXPECT_EQ("qux", uniquer.GetUniqueName("qux")); in TEST_F()
44 EXPECT_EQ("foo", uniquer.GetUniqueName("foo")); in TEST_F()
45 EXPECT_EQ("foo.1", uniquer.GetUniqueName("foo")); in TEST_F()
46 EXPECT_EQ("foo.2", uniquer.GetUniqueName("foo")); in TEST_F()
[all …]
|
| D | name_uniquer.h | 41 string GetUniqueName(absl::string_view prefix = "");
|
| D | name_uniquer.cc | 77 string NameUniquer::GetUniqueName(absl::string_view prefix) { in GetUniqueName() function in xla::NameUniquer
|
| D | hlo_module.cc | 107 computation_name_uniquer_.GetUniqueName(computation->name()); in AddComputationInternal()
109 instruction_name_uniquer_.GetUniqueName(instruction->name()); in AddComputationInternal()
|
| D | hlo_computation.cc | 1143 name_ = name_uniquer->GetUniqueName(name_); in UniquifyName()
|
| D | hlo_instruction.cc | 3885 name_ = name_uniquer->GetUniqueName(name_); in UniquifyName()
|
| /external/tensorflow/tensorflow/compiler/mlir/ |
| D | op_or_arg_name_mapper.cc | 45 llvm::StringRef OpOrArgNameMapper::GetUniqueName(llvm::StringRef prefix) { in GetUniqueName() function in tensorflow::OpOrArgNameMapper
78 llvm::StringRef OpOrArgNameMapper::GetUniqueName(OpOrVal op_or_val) { in GetUniqueName() function in tensorflow::OpOrArgNameMapper
82 llvm::StringRef ref = GetUniqueName(GetName(op_or_val)); in GetUniqueName()
91 name = StringRefToView(GetUniqueName(GetName(op_or_val))); in GetUniqueNameView()
|
| D | op_or_arg_name_mapper.h | 39 llvm::StringRef GetUniqueName(llvm::StringRef prefix);
42 llvm::StringRef GetUniqueName(OpOrVal op_or_val);
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | region_control_flow_to_functional.cc | 70 return (mapper.GetUniqueName(op) + suffix).str(); in GetName()
316 mapper.GetUniqueName(if_region._then_func_nameAttr().getValue()) in ConvertIfOp()
327 mapper.GetUniqueName(if_region._else_func_nameAttr().getValue()) in ConvertIfOp()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/ |
| D | export_graphdef.cc | 219 std::string(op_to_name_.GetUniqueName(func.getName().str()))); in GetArgumentNode()
260 std::string(op_to_name_.GetUniqueName(function.getName().str()))); in GetReturnNode()
379 auto name = op_to_name_.GetUniqueName(inst); in AddInstructionNode()
512 (void)exporter.op_to_name_.GetUniqueName(tensor_id_node); in Convert()
529 (void)exporter.op_to_name_.GetUniqueName(name); in Convert()
|
| D | import_model.cc | 1237 function_name_uniquifier_->GetUniqueName(func_name)); in ConvertLibFunction()
|
| /external/tensorflow/tensorflow/cc/framework/ |
| D | scope_internal.h | 88 string GetUniqueName(const string& prefix, bool check_single_use) const;
|
| D | scope.cc | 360 string Scope::Impl::GetUniqueName(const string& prefix, in GetUniqueName() function in tensorflow::Scope::Impl
386 GetUniqueName(default_name, true /* check_single_use */); in GetNameForOp()
412 impl()->GetUniqueName(child_scope_name, false /* check_single_use */); in NewSubScope()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter_nested.cc | 86 ir_emitter_context_->name_uniquer()->GetUniqueName( in CodegenNestedComputation()
|
| D | ir_emitter_unnested.cc | 562 string kernel_name = ir_emitter_context_->name_uniquer()->GetUniqueName( in BuildKernelPrototype()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | while_loop_outline.cc | 60 return (mapper_.GetUniqueName(op) + suffix).str(); in GetName()
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | xla_compiler.cc | 70 return uniquer->name_uniquer.GetUniqueName(name); in UniquifyName()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ |
| D | flatbuffer_export.cc | 648 return std::string(name_mapper_.GetUniqueName(val)); in UniqueName()
1390 if (has_input_attr) name = std::string(name_mapper_.GetUniqueName(arg)); in BuildSubGraph()
1417 qtype, name_mapper_.GetUniqueName(intermediate).str()); in BuildSubGraph()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf_with_tf2xla.cc | 421 op_, name_mapper_.GetUniqueName(op_), /*ignore_unregistered_attrs=*/true); in LegalizeOp()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | ir_emitter.cc | 158 string function_name = name_uniquer_.GetUniqueName(function_name_prefix); in EmitComputation()
|