| /external/pytorch/torch/csrc/ |
| D | Module.cpp | 233 TORCH_CHECK( in THPModule_crashIfCsrcASAN() local
247 TORCH_CHECK( in THPModule_crashIfCsrcUBSAN() local
278 TORCH_CHECK( in THPModule_crashIfATenASAN() local
296 TORCH_CHECK( in THPModule_crashIfDebugAssertsFail() local
313 TORCH_CHECK( in THPModule_setNumThreads() local
334 TORCH_CHECK( in THPModule_setNumInteropThreads() local
496 TORCH_CHECK( in THPModule_setBackcompatBroadcastWarn() local
519 TORCH_CHECK( in THPModule_setBackcompatKeepdimWarn() local
594 TORCH_CHECK(THPVariable_Check(data), "data must be a Tensor"); in THPModule_toDLPack() local
625 TORCH_CHECK( in THModule_rename_privateuse1_backend() local
[all …]
|
| D | StorageSharing.cpp | 159 TORCH_CHECK(PyTuple_GET_SIZE(args) == 3, "tuple of 3 items expected"); in THPStorage_newSharedFilename() local
251 TORCH_CHECK(PyTuple_GET_SIZE(args) == 2, "tuple of 2 items expected"); in THPStorage_newSharedFd() local
379 TORCH_CHECK(PyTuple_GET_SIZE(args) == 2, "tuple of 2 items expected"); in THPStorage_releaseIPCCounter() local
432 TORCH_CHECK(PyTuple_GET_SIZE(args) == 8, "tuple of 8 items expected"); in THPStorage_newSharedCuda() local
586 TORCH_CHECK( in THPStorage_newWithWeakPtr() local
613 TORCH_CHECK(THPUtils_checkLong(arg), "_expired(): arg must be an 'int'"); in THPStorage_expired() local
|
| D | StorageMethods.cpp | 463 TORCH_CHECK( in THPStorage_newWithFile() local
569 TORCH_CHECK(PyTuple_GET_SIZE(args) == 1, "tuple of 1 item expected"); in THPStorage_byteswap() local
|
| D | Storage.cpp | 299 TORCH_CHECK( in THPStorage_pynew() local
|
| /external/pytorch/torch/csrc/xpu/ |
| D | Module.cpp | 50 TORCH_CHECK(THPUtils_checkLong(arg), "invalid argument to set_device"); in THXPModule_setDevice_wrap() local
61 TORCH_CHECK(THPUtils_checkLong(arg), "invalid argument to exchange_device"); in THXPModule_exchangeDevice_wrap() local
77 TORCH_CHECK( in THXPModule_maybeExchangeDevice_wrap() local
112 TORCH_CHECK( in THXPModule_getCurrentStream_wrap() local
133 TORCH_CHECK( in THXPModule_getCurrentStream_raw() local
181 TORCH_CHECK(THPUtils_checkLong(arg), "invalid argument to synchronize"); in THXPModule_xpuSynchronize() local
202 TORCH_CHECK(THPUtils_checkLong(arg), "invalid argument to memory_stats"); in THXPModule_memoryStats() local
245 TORCH_CHECK( in THXPModule_resetPeakMemoryStats() local
257 TORCH_CHECK( in THXPModule_resetAccumulatedMemoryStats() local
|
| /external/pytorch/torch/csrc/cuda/ |
| D | Module.cpp | 87 TORCH_CHECK(THPUtils_checkLong(arg), "invalid argument to setDevice"); in THCPModule_setDevice_wrap() local
99 TORCH_CHECK(THPUtils_checkLong(arg), "invalid argument to exchangeDevice"); in THCPModule_exchangeDevice() local
114 TORCH_CHECK(THPUtils_checkLong(arg), "invalid argument to exchangeDevice"); in THCPModule_maybeExchangeDevice() local
191 TORCH_CHECK( in THCPModule_getCurrentStream_wrap() local
212 TORCH_CHECK( in THCPModule_getCurrentStream_raw() local
224 TORCH_CHECK( in THCPModule_getDefaultStream_wrap() local
465 TORCH_CHECK( in THCPModule_cudaSleep() local
513 TORCH_CHECK( in THCPModule_hasPrimaryContext() local
565 TORCH_CHECK(THPUtils_checkLong(arg), "invalid argument to memory_allocated"); in THCPModule_memoryStats() local
623 TORCH_CHECK( in THCPModule_resetAccumulatedMemoryStats() local
[all …]
|
| /external/pytorch/c10/util/ |
| D | Exception.h | 524 #define TORCH_CHECK(cond, ...) \ macro
533 #define TORCH_CHECK(cond, ...) \ macro
|
| /external/pytorch/torch/csrc/mps/ |
| D | Module.cpp | 97 TORCH_CHECK( in MPSModule_setMemoryFraction() local
|
| /external/pytorch/torch/csrc/autograd/ |
| D | init.cpp | 1204 TORCH_CHECK(mode != Py_None); in set_dispatch_mode() local
1224 TORCH_CHECK(arg != Py_None); in get_dispatch_mode() local
1240 TORCH_CHECK(arg != Py_None); in unset_dispatch_mode() local
|
| D | python_function.cpp | 79 TORCH_CHECK(!self->has_freed_buffers, ERR_BACKWARD_TWICE); in unpack_saved_variables() local
1400 TORCH_CHECK(THPVariable_Check(_var), "_register_hook_dict expected a Tensor"); in THPFunction__register_hook_dict() local
1578 TORCH_CHECK(!self->has_freed_buffers, ERR_BACKWARD_TWICE); in THPFunction_raw_saved_tensors() local
|
| D | python_variable.cpp | 1881 TORCH_CHECK( in THPVariable_pynew() local
|