| /external/tensorflow/tensorflow/compiler/aot/tests/ |
| D | BUILD | 128 cpp_class = "AddComp",
145 cpp_class = "AddWithCkptComp",
158 cpp_class = "AddWithCkptSaverComp",
172 cpp_class = "AssertComp",
184 cpp_class = "CondComp",
196 cpp_class = "FunctionComp",
208 cpp_class = "GatherComp",
220 cpp_class = "foo::bar::MatMulComp",
232 cpp_class = "::foo::bar::MatMulAndAddComp",
245 cpp_class = "MatMulAndAddCompWithProfiling",
[all …]
|
| /external/tensorflow/tensorflow/compiler/aot/ |
| D | codegen_test.cc | 79 void ExpectOK(const string& cpp_class, const string& want_class_name, in ExpectOK() argument
83 TF_EXPECT_OK(ParseCppClass(cpp_class, &class_name, &namespaces)); in ExpectOK()
88 void ExpectFail(const string& cpp_class) { in ExpectFail() argument
91 EXPECT_NE(ParseCppClass(cpp_class, &class_name, &namespaces), Status::OK()) in ExpectFail()
92 << cpp_class; in ExpectFail()
|
| D | BUILD | 161 cpp_class = "AddComp",
173 cpp_class = "AddComp",
188 cpp_class = "UnknownOpAddComp",
200 cpp_class = "UnknownOpAddComp",
215 cpp_class = "UnknownOpAddComp",
227 cpp_class = "UnknownOpAddComp",
241 cpp_class = "UnknownOpAddComp",
253 cpp_class = "UnknownOpAddComp",
|
| D | tfcompile.bzl | 12 cpp_class = "MatMulComp",
32 cpp_class = None,
72 cpp_class: The name of the generated C++ class, wrapping the generated
103 if not cpp_class:
104 fail("cpp_class must be specified")
238 " --cpp_class=" + cpp_class +
279 " --cpp_class=" + cpp_class +
325 cpp_class_split = cpp_class.rsplit("::", 2)
332 "-e \"s|{{TFCOMPILE_CPP_CLASS}}|" + cpp_class + "|g\" " +
|
| D | codegen.cc | 790 Status ParseCppClass(const string& cpp_class, string* class_name, in ParseCppClass() argument
794 if (cpp_class.empty()) { in ParseCppClass()
795 return errors::InvalidArgument("empty cpp_class: " + cpp_class); in ParseCppClass()
797 std::vector<string> parts = absl::StrSplit(cpp_class, "::"); in ParseCppClass()
805 parts[i], "in namespace component of cpp_class: " + cpp_class)); in ParseCppClass()
809 parts[i], "in class name of cpp_class: " + cpp_class)); in ParseCppClass()
|
| D | flags.h | 39 string cpp_class; member
|
| D | flags.cc | 62 {"cpp_class", &flags->cpp_class, in AppendMainFlags()
|
| D | codegen.h | 94 Status ParseCppClass(const string& cpp_class, string* class_name,
|
| D | compile.cc | 253 if (flags.cpp_class.empty()) { in Main()
258 TF_RETURN_IF_ERROR(ParseCppClass(flags.cpp_class, &codegen_opts.class_name, in Main()
|
| /external/tensorflow/tensorflow/python/ |
| D | tfcompile_wrapper.cc | 39 std::string entry_point, std::string cpp_class, in PYBIND11_MODULE()
51 flags.cpp_class = std::move(cpp_class); in PYBIND11_MODULE()
|
| /external/tensorflow/tensorflow/python/tools/ |
| D | tools.bzl | 17 cpp_class,
43 cpp_class = "tensorflow::CompiledModel",
78 cpp_class: The name of the C++ class that will be generated, including
141 "--cpp_class {} ".format(cpp_class) +
|
| D | BUILD | 410 cpp_class = "XMatmulYLarge",
419 cpp_class = "XMatmulYLargeMultithreaded",
429 cpp_class = "XMatmulYSmall",
438 cpp_class = "XPlusY",
448 cpp_class = "VarsAndArithmeticFrozen",
458 cpp_class = "VarsAndArithmetic",
|
| D | saved_model_aot_compile.py | 217 cpp_class, argument
356 '__xla_' + output_prefix + '__' + cpp_class)
369 cpp_class=cpp_class,
|
| D | saved_model_cli.py | 841 cpp_class=args.cpp_class,
|
| /external/tensorflow/tensorflow/tools/proto_text/ |
| D | gen_proto_text_functions_lib.cc | 226 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/ |
| D | tfcompile.md | 97 # cpp_class specifies the name of the generated C++ class, with namespaces allowed.
100 cpp_class = "foo::bar::MatMulComp",
195 because that was the `cpp_class` specified in the `tf_library` macro. All
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | BUILD | 1853 cpp_class = "LSTMLayerInference",
|