HomeSort by relevance Sort by last modified time
    Searched full:cpp_class (Results 1 - 14 of 14) sorted by null

  /external/tensorflow/tensorflow/compiler/aot/tests/
BUILD 70 cpp_class = "AddComp",
86 cpp_class = "AddWithCkptComp",
98 cpp_class = "AddWithCkptSaverComp",
111 cpp_class = "FunctionComp",
122 cpp_class = "GatherComp",
133 cpp_class = "foo::bar::MatMulComp",
144 cpp_class = "MatMulAndAddComp",
156 cpp_class = "SplitsComp",
  /external/tensorflow/tensorflow/compiler/aot/
test.cc 44 #define CPP_CLASS {{TFCOMPILE_CPP_CLASS}} // NOLINT(whitespace/braces)
66 CPP_CLASS computation;
68 zero_buffers(computation.args(), CPP_CLASS::ArgSizes(), CPP_CLASS::kNumArgs);
80 CPP_CLASS computation;
82 zero_buffers(computation.args(), CPP_CLASS::ArgSizes(), CPP_CLASS::kNumArgs);
benchmark_main.template 26 #define CPP_CLASS {{TFCOMPILE_CPP_CLASS}} // NOLINT(whitespace/braces)
36 CPP_CLASS computation;
tfcompile.bzl 14 cpp_class = "MatMulComp",
24 cpp_class=None, gen_test=True, gen_benchmark=True,
52 cpp_class: The name of the generated C++ class, wrapping the generated
77 if not cpp_class:
78 fail("cpp_class must be specified")
155 " --cpp_class=" + cpp_class +
194 " --cpp_class=" + cpp_class +
238 cpp_class_split = cpp_class.rsplit("::", maxsplit=2
    [all...]
BUILD 133 cpp_class = "AddComp",
146 cpp_class = "UnknownOpAddComp",
160 cpp_class = "UnknownOpAddComp",
173 cpp_class = "UnknownOpAddComp",
tfcompile_main.cc 54 "--cpp_class=\"mynamespace::MyComputation\"\n"
100 if (flags.cpp_class.empty()) {
101 return errors::InvalidArgument("Must specify --cpp_class");
103 TF_RETURN_IF_ERROR(ParseCppClass(flags.cpp_class, &codegen_opts.class_name,
codegen.h 80 // ParseCppClass parses `cpp_class` into its `class_name` and `namespaces`
84 Status ParseCppClass(const string& cpp_class, string* class_name,
codegen_test.cc 71 void ExpectOK(const string& cpp_class, const string& want_class_name,
75 TF_EXPECT_OK(ParseCppClass(cpp_class, &class_name, &namespaces));
80 void ExpectFail(const string& cpp_class) {
83 EXPECT_NE(ParseCppClass(cpp_class, &class_name, &namespaces), Status::OK());
flags.h 36 string cpp_class; member in struct:tensorflow::tfcompile::MainFlags
codegen.cc 568 Status ParseCppClass(const string& cpp_class, string* class_name,
574 while ((end = cpp_class.find("::", begin)) != string::npos) {
575 const string ns = cpp_class.substr(begin, end - begin);
577 ns, "in namespace component of cpp_class: " + cpp_class));
581 const string name = cpp_class.substr(begin);
583 ValidateCppIdent(name, "in class name of cpp_class: " + cpp_class));
flags.cc 55 {"cpp_class", &flags->cpp_class,
  /external/tensorflow/tensorflow/docs_src/performance/xla/
tfcompile.md 99 # 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/tools/proto_text/
gen_proto_text_functions_lib.cc 226 string cpp_class = d.cpp_type() == FieldDescriptor::CPPTYPE_MESSAGE local
233 if (cpp_class == "int64") {
234 cpp_class = kProtobufInt64Typename;
237 return cpp_class;
  /external/tensorflow/tensorflow/compiler/tests/
BUILD 813 cpp_class = "LSTMLayerInference",

Completed in 111 milliseconds