C++ API
torch_npu::init_npu
torch_npu::finalize_npu
torch::npu::synchronize
c10::npu::current_device
at::Device
struct c10_npu::NPUEvent
class at_npu::NPUGeneratorImpl
at_npu::detail::getDefaultNPUGenerator
at_npu::detail::createNPUGenerator
class c10_npu::NPUStream
c10_npu::getNPUStreamFromPool
c10_npu::getDefaultNPUStream
c10_npu::getCurrentNPUStream
c10_npu::setCurrentNPUStream
class at_npu::native::OpCommand
struct c10_npu::NPUHooksInterface
struct c10_npu::NPUHooksArgs
c10_npu::device_count
c10_npu::GetDevice
c10_npu::SetDevice
c10_npu::current_device
c10_npu::set_device
c10_npu::warning_state
c10_npu::warn_or_error_on_sync
at_npu::native::get_npu_format
at_npu::native::get_npu_storage_sizes
at_npu::native::npu_format_cast
at_npu::native::empty_with_format
c10_npu::c10_npu_get_error_message
父主题:
PyTorch Adapter自定义API