| /aosp_15_r20/external/tensorflow/tensorflow/compiler/aot/tests/ |
| H A D | BUILD | 140 cpp_class = "AddComp",
157 cpp_class = "AddWithCkptComp",
170 cpp_class = "AddWithCkptSaverComp",
184 cpp_class = "AssertComp",
196 cpp_class = "CondComp",
208 cpp_class = "FunctionComp",
220 cpp_class = "GatherComp",
232 cpp_class = "foo::bar::MatMulComp",
244 cpp_class = "::foo::bar::MatMulAndAddComp",
257 cpp_class = "SplitsComp",
[all …]
|
| /aosp_15_r20/external/tensorflow/tensorflow/tools/pip_package/xla_build/pip_test/ |
| H A D | run_xla_aot_test.sh | 56 --cpp_class VarsAndArithmetic --signature_def_key serving_default --tag_set serve
61 --cpp_class VarsAndArithmeticFrozen --signature_def_key serving_default --tag_set serve
66 --cpp_class XMatmulYSmall --signature_def_key serving_default --tag_set serve
71 --cpp_class XMatmulYLarge --signature_def_key serving_default --tag_set serve
77 --cpp_class XMatmulYLargeMultithreaded --signature_def_key serving_default --tag_set serve
82 --cpp_class XPlusY --signature_def_key serving_default --tag_set serve
|
| /aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/ |
| H A D | python.h | 53 PyModuleClass<ModuleType> cpp_class, in bind_cpp_module_wrapper() argument
85 [cpp_module, cpp_class]( in bind_cpp_module_wrapper()
87 cpp_module.attr("__init__")(self, cpp_class(*args, **kwargs)); in bind_cpp_module_wrapper()
220 auto cpp_class = in bind_module() local
222 detail::bind_cpp_module_wrapper(module, cpp_class, name); in bind_module()
223 return cpp_class; in bind_module()
|
| /aosp_15_r20/external/cronet/third_party/jni_zero/codegen/ |
| H A D | natives_header.py | 72 cpp_class = native.first_param_cpp_type
73 if cpp_class:
81 if not cpp_class:
87 if cpp_class:
103 if cpp_class:
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/aot/ |
| H A D | BUILD | 163 cpp_class = "AddComp",
175 cpp_class = "AddComp",
190 cpp_class = "UnknownOpAddComp",
202 cpp_class = "UnknownOpAddComp",
217 cpp_class = "UnknownOpAddComp",
229 cpp_class = "UnknownOpAddComp",
243 cpp_class = "UnknownOpAddComp",
255 cpp_class = "UnknownOpAddComp",
|
| H A D | tfcompile.bzl | 12 cpp_class = "MatMulComp",
78 "--cpp_class=" + ctx.attr.cpp_class,
116 "cpp_class": attr.string(),
137 cpp_class = None,
177 cpp_class: The name of the generated C++ class, wrapping the generated
208 if not cpp_class:
209 fail("cpp_class must be specified")
322 cpp_class = cpp_class,
387 cpp_class_split = cpp_class.rsplit("::", 2)
394 "-e \"s|{{TFCOMPILE_CPP_CLASS}}|" + cpp_class + "|g\" " +
|
| H A 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), OkStatus()) in ExpectFail()
92 << cpp_class; in ExpectFail()
|
| H A D | codegen.cc | 791 Status ParseCppClass(const string& cpp_class, string* class_name, in ParseCppClass() argument
795 if (cpp_class.empty()) { in ParseCppClass()
796 return errors::InvalidArgument("empty cpp_class: " + cpp_class); in ParseCppClass()
798 std::vector<string> parts = absl::StrSplit(cpp_class, "::"); in ParseCppClass()
806 parts[i], "in namespace component of cpp_class: " + cpp_class)); in ParseCppClass()
810 parts[i], "in class name of cpp_class: " + cpp_class)); in ParseCppClass()
|
| H A D | compile.cc | 272 if (flags.cpp_class.empty()) { in Main()
277 TF_RETURN_IF_ERROR(ParseCppClass(flags.cpp_class, &codegen_opts.class_name, in Main()
|
| H A D | flags.h | 39 string cpp_class; member
|
| H A D | flags.cc | 62 {"cpp_class", &flags->cpp_class, in AppendMainFlags()
|
| H A D | codegen.h | 94 Status ParseCppClass(const string& cpp_class, string* class_name,
|
| /aosp_15_r20/prebuilts/clang/host/linux-x86/clang-r522817/lib/cmake/llvm/ |
| D | TensorFlowCompile.cmake | 46 # called ${cpp_class} - which may be a namespace-qualified class name.
47 function(tf_compile model tag_set signature_def_key fname cpp_class hdr_file obj_file)
57 --cpp_class ${cpp_class}
73 …pile model default_url default_path test_model_generator tag_set signature_def_key fname cpp_class)
111 …tf_compile(${LLVM_ML_MODELS_ABSOLUTE} ${tag_set} ${signature_def_key} ${fname} ${cpp_class} ${hdr_…
|
| /aosp_15_r20/prebuilts/clang/host/linux-x86/clang-r530567b/lib/cmake/llvm/ |
| D | TensorFlowCompile.cmake | 46 # called ${cpp_class} - which may be a namespace-qualified class name.
47 function(tf_compile model tag_set signature_def_key fname cpp_class hdr_file obj_file)
57 --cpp_class ${cpp_class}
73 …pile model default_url default_path test_model_generator tag_set signature_def_key fname cpp_class)
111 …tf_compile(${LLVM_ML_MODELS_ABSOLUTE} ${tag_set} ${signature_def_key} ${fname} ${cpp_class} ${hdr_…
|
| /aosp_15_r20/prebuilts/clang/host/linux-x86/clang-r530567/lib/cmake/llvm/ |
| D | TensorFlowCompile.cmake | 46 # called ${cpp_class} - which may be a namespace-qualified class name.
47 function(tf_compile model tag_set signature_def_key fname cpp_class hdr_file obj_file)
57 --cpp_class ${cpp_class}
73 …pile model default_url default_path test_model_generator tag_set signature_def_key fname cpp_class)
111 …tf_compile(${LLVM_ML_MODELS_ABSOLUTE} ${tag_set} ${signature_def_key} ${fname} ${cpp_class} ${hdr_…
|
| /aosp_15_r20/prebuilts/clang/host/linux-x86/clang-r536225/lib/cmake/llvm/ |
| D | TensorFlowCompile.cmake | 46 # called ${cpp_class} - which may be a namespace-qualified class name.
47 function(tf_compile model tag_set signature_def_key fname cpp_class hdr_file obj_file)
57 --cpp_class ${cpp_class}
73 …pile model default_url default_path test_model_generator tag_set signature_def_key fname cpp_class)
111 …tf_compile(${LLVM_ML_MODELS_ABSOLUTE} ${tag_set} ${signature_def_key} ${fname} ${cpp_class} ${hdr_…
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/tools/ |
| H A 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) +
|
| H A D | BUILD | 413 cpp_class = "XMatmulYLarge",
422 cpp_class = "XMatmulYLargeMultithreaded",
432 cpp_class = "XMatmulYSmall",
441 cpp_class = "XPlusY",
451 cpp_class = "VarsAndArithmeticFrozen",
461 cpp_class = "VarsAndArithmetic",
|
| H A D | saved_model_aot_compile.py | 315 cpp_class, argument
388 '__xla_' + output_prefix + '__' + cpp_class)
401 cpp_class=cpp_class,
|
| H A D | saved_model_cli.py | 893 cpp_class=args.cpp_class,
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ |
| H A 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()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/transformers/cuda/mem_eff_attention/kernels/ |
| H A D | generate_kernels.py | 87 def cpp_class(self) -> str: member in FwdKernel
110 CPP_CLASS=self.cpp_class,
193 def cpp_class(self) -> str: member in BwdKernel
219 CPP_CLASS=self.cpp_class,
|
| /aosp_15_r20/external/tensorflow/tensorflow/tools/proto_text/ |
| H A 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()
|
| /aosp_15_r20/tools/asuite/atest/test_finders/ |
| H A D | test_finder_utils_unittest.py | 594 cpp_class = os.path.join(uc.FIND_PATH, uc.FIND_PATH_FILENAME_CC + '.cpp')
607 self.assertTrue(cpp_class in cc_tmp_test_result)
614 cpp_class = os.path.join(uc.FIND_PATH, uc.FIND_PATH_FILENAME_CC + '.cpp')
627 self.assertTrue(cpp_class in cc_tmp_test_result)
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/g3doc/ |
| H A 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
|