Home
last modified time | relevance | path

Searched refs:C10_API (Results 1 – 25 of 86) sorted by relevance

1234

/aosp_15_r20/external/pytorch/c10/util/
H A DException.h30 class C10_API Error : public std::exception {
117 class C10_API Warning {
119 class C10_API UserWarning {};
120 class C10_API DeprecationWarning {};
167 void C10_API warn(const Warning& warning);
169 class C10_API WarningHandler {
196 C10_API void set_warning_handler(WarningHandler* handler) noexcept(true);
198 C10_API WarningHandler* get_warning_handler() noexcept(true);
200 class C10_API WarningHandlerGuard {
216 C10_API void set_warnAlways(bool) noexcept(true);
[all …]
H A DLogging.h63 C10_API bool InitCaffeLogging(int* argc, char** argv);
64 C10_API void UpdateLoggingLevelsFromFlags();
66 [[noreturn]] C10_API void ThrowEnforceNotMet(
73 [[noreturn]] C10_API void ThrowEnforceNotMet(
80 [[noreturn]] C10_API inline void ThrowEnforceNotMet(
89 [[noreturn]] C10_API void ThrowEnforceFiniteNotMet(
96 [[noreturn]] C10_API void ThrowEnforceFiniteNotMet(
103 [[noreturn]] C10_API inline void ThrowEnforceFiniteNotMet(
128 C10_API void ShowLogInfoToStderr();
130 C10_API void SetStackTraceFetcher(std::function<::c10::Backtrace()> fetcher);
[all …]
H A Dint128.h55 class C10_API uint128 {
94 C10_API friend std::ostream& operator<<(std::ostream& o, const uint128& b);
128 C10_API extern const uint128_pod kuint128max;
131 C10_API extern std::ostream& operator<<(std::ostream& o, const uint128& b);
157 C10_API inline UINT128_CONSTEXPR uint128::uint128() : lo_(0), hi_(0) {} in uint128()
158 C10_API inline UINT128_CONSTEXPR uint128::uint128(uint64_t top, uint64_t bottom) in uint128()
160 C10_API inline UINT128_CONSTEXPR uint128::uint128(const uint128_pod& v) in uint128()
162 C10_API inline UINT128_CONSTEXPR uint128::uint128(uint64_t bottom) in uint128()
165 C10_API inline UINT128_CONSTEXPR uint128::uint128(uint32_t bottom) in uint128()
167 C10_API inline UINT128_CONSTEXPR uint128::uint128(int bottom) in uint128()
[all …]
H A Dnuma.h14 C10_API bool IsNUMAEnabled();
19 C10_API void NUMABind(int numa_node_id);
24 C10_API int GetNUMANode(const void* ptr);
29 C10_API int GetNumNUMANodes();
34 C10_API void NUMAMove(void* ptr, size_t size, int numa_node_id);
39 C10_API int GetCurrentNUMANode();
H A Dtempfile.h10 struct C10_API TempFile {
36 struct C10_API TempDir {
66 C10_API std::optional<TempFile> try_make_tempfile(
71 C10_API TempFile make_tempfile(std::string_view name_prefix = "torch-file-");
83 C10_API std::optional<TempDir> try_make_tempdir(
88 C10_API TempDir make_tempdir(std::string_view name_prefix = "torch-dir-");
H A DStringUtil.h23 C10_API std::string StripBasename(const std::string& full_path);
25 C10_API std::string ExcludeFileExtension(const std::string& full_path);
60 C10_API std::ostream& _str(std::ostream& ss, const wchar_t* wCStr);
61 C10_API std::ostream& _str(std::ostream& ss, const wchar_t& wChar);
62 C10_API std::ostream& _str(std::ostream& ss, const std::wstring& wString);
132 size_t C10_API
136 struct C10_API SourceLocation {
H A DFlags.h42 C10_API void SetUsageMessage(const std::string& str);
47 C10_API const char* UsageMessage();
57 C10_API bool ParseCommandLineFlags(int* pargc, char*** pargv);
62 C10_API bool CommandLineFlagsHasBeenParsed();
159 class C10_API C10FlagParser {
211 #define C10_DECLARE_typed_var(type, name) C10_API extern type FLAGS_##name
H A DDeadlockDetection.h26 C10_API bool check_python_gil();
28 struct C10_API PythonGILHooks {
35 C10_API void SetPythonGILHooks(PythonGILHooks* factory);
39 struct C10_API PythonGILHooksRegisterer {
H A Dsignal_handler.h24 class C10_API SignalHandler {
44 class C10_API FatalSignalHandler {
49 C10_API void setPrintStackTracesOnFatalSignal(bool print);
50 C10_API bool printStackTracesOnFatalSignal();
H A DThreadLocalDebugInfo.h21 class C10_API DebugInfoBase {
33 class C10_API ThreadLocalDebugInfo {
67 class C10_API DebugInfoGuard {
/aosp_15_r20/external/pytorch/c10/core/
H A DSymInt.h35 class C10_API SymInt {
321 C10_API RetTy operator%(const SymInt& a, scalar_t b); \
322 C10_API RetTy operator%(scalar_t a, const SymInt& b);
325 C10_API RetTy operator+(const SymInt& a, scalar_t b); \
326 C10_API RetTy operator-(const SymInt& a, scalar_t b); \
327 C10_API RetTy operator*(const SymInt& a, scalar_t b); \
328 C10_API RetTy operator/(const SymInt& a, scalar_t b); \
329 C10_API RetTy operator+(scalar_t a, const SymInt& b); \
330 C10_API RetTy operator-(scalar_t a, const SymInt& b); \
331 C10_API RetTy operator*(scalar_t a, const SymInt& b); \
[all …]
H A DCPUAllocator.h20 C10_API void NoDelete(void*);
24 class C10_API ProfiledCPUMemoryReporter {
38 C10_API ProfiledCPUMemoryReporter& profiledCPUMemoryReporter();
41 C10_API at::Allocator* GetCPUAllocator();
44 C10_API void SetCPUAllocator(at::Allocator* alloc, uint8_t priority = 0);
47 C10_API at::Allocator* GetDefaultCPUAllocator();
50 C10_API at::Allocator* GetDefaultMobileCPUAllocator();
55 C10_API void SetCPUCachingAllocator(Allocator* alloc, uint8_t priority = 0);
57 C10_API Allocator* GetCPUCachingAllocator();
H A DAllocator.h26 class C10_API DataPtr {
160 struct C10_API Allocator {
225 struct C10_API InefficientStdFunctionContext {
255 C10_API void SetAllocator(DeviceType t, Allocator* alloc, uint8_t priority = 0);
256 C10_API Allocator* GetAllocator(const DeviceType& t);
272 struct C10_API MemoryReportingInfoBase : public c10::DebugInfoBase {
300 C10_API bool memoryProfilingEnabled();
301 C10_API void reportMemoryUsageToProfiler(
308 C10_API void reportOutOfMemoryToProfiler(
H A DDeviceType.h98 C10_API std::string DeviceTypeName(DeviceType d, bool lower_case = false);
100 C10_API bool isValidDeviceType(DeviceType d);
102 C10_API std::ostream& operator<<(std::ostream& stream, DeviceType type);
104 C10_API void register_privateuse1_backend(const std::string& backend_name);
105 C10_API std::string get_privateuse1_backend(bool lower_case = true);
107 C10_API bool is_privateuse1_backend_registered();
H A DStorageImpl.h19 C10_API void throwNullDataPtrError();
20 C10_API void warnDeprecatedDataPtr();
45 struct C10_API StorageImpl : public c10::intrusive_ptr_target {
318 C10_API void SetStorageImplCreate(DeviceType t, StorageImplCreateHelper fptr);
320 C10_API StorageImplCreateHelper GetStorageImplCreate(DeviceType t);
322 C10_API c10::intrusive_ptr<c10::StorageImpl> make_storage_impl(
H A DDispatchKeySet.h38 C10_API std::array<FunctionalityOffsetAndMask, num_functionality_keys>
561 C10_API self_type& operator++();
630 C10_API std::string toString(DispatchKeySet);
631 C10_API std::ostream& operator<<(std::ostream&, DispatchKeySet);
633 C10_API inline int getDispatchTableIndexForDispatchKey(DispatchKey k) { in getDispatchTableIndexForDispatchKey()
807 C10_API bool isBackendDispatchKey(DispatchKey t);
810 C10_API DispatchKeySet getRuntimeDispatchKeySet(DispatchKey t);
814 C10_API bool runtimeDispatchKeySetHas(DispatchKey t, DispatchKey k);
818 C10_API DispatchKeySet getBackendKeySetFromAutograd(DispatchKey t);
901 C10_API bool isIncludedInAlias(DispatchKey k, DispatchKey alias);
H A DDefaultDtype.h11 C10_API void set_default_dtype(caffe2::TypeMeta dtype);
12 C10_API const caffe2::TypeMeta get_default_dtype();
13 C10_API ScalarType get_default_dtype_as_scalartype();
14 C10_API const caffe2::TypeMeta get_default_complex_dtype();
H A DRefcountedDeleter.h26 struct C10_API RefcountedDeleterContext {
40 C10_API void refcounted_deleter(void* ctx_);
44 C10_API void maybeApplyRefcountedDeleter(const c10::Storage& storage);
49 C10_API c10::Storage newStorageImplFromRefcountedDataPtr(
H A DGradMode.h8 struct C10_API GradMode {
15 struct C10_API AutoGradMode {
27 struct C10_API NoGradGuard : public AutoGradMode {
33 struct C10_API AutoFwGradMode {
H A DDispatchKey.h597 C10_API const char* toString(DispatchKey);
598 C10_API const char* toString(BackendComponent);
599 C10_API std::ostream& operator<<(std::ostream&, DispatchKey);
600 C10_API std::ostream& operator<<(std::ostream&, BackendComponent);
602 C10_API DispatchKey getAutogradKeyFromBackend(BackendComponent k);
606 C10_API c10::DispatchKey parseDispatchKey(const std::string& k);
/aosp_15_r20/external/pytorch/c10/core/impl/
H A DLocalDispatchKeySet.h31 struct C10_API PODLocalDispatchKeySet {
56 struct C10_API LocalDispatchKeySet {
66 C10_API LocalDispatchKeySet tls_local_dispatch_key_set();
68 extern C10_API thread_local PODLocalDispatchKeySet raw_local_dispatch_key_set;
70 inline C10_API LocalDispatchKeySet tls_local_dispatch_key_set() { in tls_local_dispatch_key_set()
78 C10_API void _force_tls_local_dispatch_key_set(LocalDispatchKeySet key_set);
82 class C10_API IncludeDispatchKeyGuard {
100 class C10_API ExcludeDispatchKeyGuard {
118 struct C10_API ForceDispatchKeyGuard {
157 C10_API bool tls_is_dispatch_key_excluded(DispatchKey x);
[all …]
H A DCOW.h20 C10_API c10::intrusive_ptr<StorageImpl> lazy_clone_storage(
24 C10_API bool has_simple_data_ptr(const c10::StorageImpl& storage);
27 C10_API bool is_cow_data_ptr(const c10::DataPtr& data_ptr);
30 C10_API void materialize_cow_storage(StorageImpl& storage);
/aosp_15_r20/external/pytorch/c10/cuda/
H A DCUDAStream.h212 C10_API CUDAStream
215 C10_API CUDAStream
225 C10_API CUDAStream
234 C10_API CUDAStream getDefaultCUDAStream(DeviceIndex device_index = -1);
243 C10_API CUDAStream getCurrentCUDAStream(DeviceIndex device_index = -1);
255 C10_API void setCurrentCUDAStream(CUDAStream stream);
257 C10_API std::ostream& operator<<(std::ostream& stream, const CUDAStream& s);
/aosp_15_r20/external/pytorch/c10/mobile/
H A DCPUProfilingAllocator.h20 class C10_API AllocationPlan {
46 class C10_API AllocationPlanner {
72 class C10_API CPUProfilingAllocator {
97 class C10_API WithProfileAllocationsGuard {
118 class C10_API WithValidateAllocationPlanGuard {
142 class C10_API WithProfilingAllocatorGuard {
/aosp_15_r20/external/pytorch/c10/macros/
H A DExport.h89 #define C10_API C10_EXPORT macro
91 #define C10_API C10_IMPORT macro
155 #define C10_API_ENUM C10_API

1234