Home
last modified time | relevance | path

Searched refs:cpp_class (Results 1 – 12 of 12) sorted by relevance

/external/tensorflow/tensorflow/compiler/aot/tests/
DBUILD91 cpp_class = "AddComp",
107 cpp_class = "AddWithCkptComp",
119 cpp_class = "AddWithCkptSaverComp",
132 cpp_class = "AssertComp",
143 cpp_class = "CondComp",
154 cpp_class = "FunctionComp",
165 cpp_class = "GatherComp",
176 cpp_class = "foo::bar::MatMulComp",
187 cpp_class = "MatMulAndAddComp",
199 cpp_class = "MatMulAndAddCompWithProfiling",
[all …]
/external/tensorflow/tensorflow/compiler/aot/
Dtfcompile.bzl14 cpp_class = "MatMulComp",
32 cpp_class = None,
69 cpp_class: The name of the generated C++ class, wrapping the generated
95 if not cpp_class:
96 fail("cpp_class must be specified")
195 " --cpp_class=" + cpp_class +
237 " --cpp_class=" + cpp_class +
286 cpp_class_split = cpp_class.rsplit("::", 2)
293 "-e \"s|{{TFCOMPILE_CPP_CLASS}}|" + cpp_class + "|g\" " +
Dcodegen_test.cc76 void ExpectOK(const string& cpp_class, const string& want_class_name, in ExpectOK() argument
80 TF_EXPECT_OK(ParseCppClass(cpp_class, &class_name, &namespaces)); in ExpectOK()
85 void ExpectFail(const string& cpp_class) { in ExpectFail() argument
88 EXPECT_NE(ParseCppClass(cpp_class, &class_name, &namespaces), Status::OK()); in ExpectFail()
DBUILD121 cpp_class = "AddComp",
134 cpp_class = "UnknownOpAddComp",
148 cpp_class = "UnknownOpAddComp",
161 cpp_class = "UnknownOpAddComp",
Dcodegen.cc737 Status ParseCppClass(const string& cpp_class, string* class_name, in ParseCppClass() argument
743 while ((end = cpp_class.find("::", begin)) != string::npos) { in ParseCppClass()
744 const string ns = cpp_class.substr(begin, end - begin); in ParseCppClass()
746 ns, "in namespace component of cpp_class: " + cpp_class)); in ParseCppClass()
750 const string name = cpp_class.substr(begin); in ParseCppClass()
752 ValidateCppIdent(name, "in class name of cpp_class: " + cpp_class)); in ParseCppClass()
Dtfcompile_main.cc102 if (flags.cpp_class.empty()) { in Main()
107 TF_RETURN_IF_ERROR(ParseCppClass(flags.cpp_class, &codegen_opts.class_name, in Main()
Dflags.h36 string cpp_class; member
Dflags.cc55 {"cpp_class", &flags->cpp_class, in AppendMainFlags()
Dcodegen.h94 Status ParseCppClass(const string& cpp_class, string* class_name,
/external/tensorflow/tensorflow/tools/proto_text/
Dgen_proto_text_functions_lib.cc226 string cpp_class = d.cpp_type() == FieldDescriptor::CPPTYPE_MESSAGE in GetCppClass() local
233 if (cpp_class == "int64") { in GetCppClass()
234 cpp_class = kProtobufInt64Typename; in GetCppClass()
237 return cpp_class; in GetCppClass()
/external/tensorflow/tensorflow/compiler/xla/g3doc/
Dtfcompile.md99 # cpp_class specifies the name of the generated C++ class, with namespaces allowed.
102 cpp_class = "foo::bar::MatMulComp",
197 because that was the `cpp_class` specified in the `tf_library` macro. All
/external/tensorflow/tensorflow/compiler/tests/
DBUILD1220 cpp_class = "LSTMLayerInference",