Home
last modified time | relevance | path

Searched defs:cpp_class (Results 1 – 10 of 10) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/
H A Dpython.h53 PyModuleClass<ModuleType> cpp_class, in bind_cpp_module_wrapper()
220 auto cpp_class = in bind_module() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/transformers/cuda/mem_eff_attention/kernels/
H A Dgenerate_kernels.py87 def cpp_class(self) -> str: member in FwdKernel
193 def cpp_class(self) -> str: member in BwdKernel
/aosp_15_r20/external/tensorflow/tensorflow/compiler/aot/
H A Dcodegen_test.cc79 void ExpectOK(const string& cpp_class, const string& want_class_name, in ExpectOK()
88 void ExpectFail(const string& cpp_class) { in ExpectFail()
H A Dflags.h39 string cpp_class; member
H A Dcodegen.cc791 Status ParseCppClass(const string& cpp_class, string* class_name, in ParseCppClass()
/aosp_15_r20/external/tensorflow/tensorflow/python/
H A Dtfcompile_wrapper.cc43 bool gen_program_shape) { in PYBIND11_MODULE()
/aosp_15_r20/external/tensorflow/tensorflow/python/tools/
H A Dsaved_model_aot_compile.py315 cpp_class, argument
/aosp_15_r20/external/pytorch/torch/jit/
H A D_recursive.py1005 def wrap_cpp_class(cpp_class): argument
H A D_script.py446 def __init__(self, cpp_class): argument
/aosp_15_r20/external/tensorflow/tensorflow/tools/proto_text/
H A Dgen_proto_text_functions_lib.cc226 string cpp_class = d.cpp_type() == FieldDescriptor::CPPTYPE_MESSAGE in GetCppClass() local