List of all items
Structs
- CudaDevicePropWrapper
- CudaRuntimeFlagFlipper
- CudnnFilterDescWrapper
- CudnnTensorDescWrapper
- NetMutator
- SimpleArray
- WorkspaceMutator
Enums
Traits
Macros
- caffe2_build_strings
- caffe2_version
- caffe_sdt
- caffe_sdt_arg
- caffe_sdt_arg_constraint
- caffe_sdt_arg_template_0
- caffe_sdt_arg_template_1
- caffe_sdt_arg_template_2
- caffe_sdt_arg_template_3
- caffe_sdt_arg_template_4
- caffe_sdt_arg_template_5
- caffe_sdt_arg_template_6
- caffe_sdt_arg_template_7
- caffe_sdt_arg_template_8
- caffe_sdt_argfmt
- caffe_sdt_argsize
- caffe_sdt_asm_1
- caffe_sdt_asm_2
- caffe_sdt_asm_3
- caffe_sdt_asm_addr
- caffe_sdt_asm_string
- caffe_sdt_isarray
- caffe_sdt_narg
- caffe_sdt_narg_
- caffe_sdt_nop
- caffe_sdt_note_content
- caffe_sdt_note_name
- caffe_sdt_note_type
- caffe_sdt_operands_0
- caffe_sdt_operands_1
- caffe_sdt_operands_2
- caffe_sdt_operands_3
- caffe_sdt_operands_4
- caffe_sdt_operands_5
- caffe_sdt_operands_6
- caffe_sdt_operands_7
- caffe_sdt_operands_8
- caffe_sdt_probe
- caffe_sdt_probe_n
- caffe_sdt_s
- cublas_check
- cublas_enforce
- cuda_1d_kernel_loop
- cuda_2d_kernel_loop
- cuda_check
- cuda_driverapi_check
- cuda_driverapi_enforce
- cuda_enforce
- cudnn_check
- cudnn_enforce
- cudnn_version_min
- curand_check
- curand_enforce
- dispatch_function_by_value_with_type_1
- dispatch_function_by_value_with_type_2
- dispatch_function_by_value_with_type_3
- impl_cudnn_type_wrapper
Functions
- assert_near
- assert_tensor
- assert_tensor_equals
- assert_tensor_equals_data
- assert_tensor_equals_with_type
- assert_tensor_equals_with_type_f32
- assert_tensor_list_equals
- caffe_cuda_get_device
- caffe_cuda_set_device
- caffe_get_blocks
- caffe_get_blocks_2d
- check_cudnn_versions
- constant_fill_tensor
- create_tensor
- create_tensor_and_constant_fill
- create_tensor_and_fill
- create_tensor_in_workspace_and_fill
- cublas_get_error_string
- cuda_version
- cudnn_compiled_version
- cudnn_get_error_string
- cudnn_runtime_version
- curand_get_error_string
- data_type_to_type_meta
- device_query
- dynamic_cast_if_rtti
- fill_tensor
- get_build_options
- get_cuda_peer_access_pattern
- get_cudnn_tensor_format
- get_defaultGPUID
- get_device_property
- get_dim_from_order_string
- get_tensor
- has_cuda_gpu
- has_cuda_runtime
- has_hip_runtime
- name_scope_separator
- num_cuda_devices
- print_cudnn_info
- random_fill
- set_cuda_runtime_flag
- set_defaultGPUID
- set_hip_runtime_flag
- string_to_storage_order
- tensor_core_available
- type_eq
- type_meta_to_data_type
Type Definitions
Constants
- CAFFE2_CUDA_MAX_PEER_SIZE
- CAFFE_CUDA_NUM_THREADS
- CAFFE_CUDA_NUM_THREADS_2D_DIMX
- CAFFE_CUDA_NUM_THREADS_2D_DIMY
- CAFFE_MAXIMUM_NUM_BLOCKS
- CAFFE_MAXIMUM_NUM_BLOCKS_2D_DIMX
- CAFFE_MAXIMUM_NUM_BLOCKS_2D_DIMY
- CUDNN_VERSION
- gDefaultGPUID
- kCUDAGridDimMaxX
- kCUDAGridDimMaxY
- kCUDAGridDimMaxZ
- kCUDATensorMaxDims
- kFp16CUDADevicePropMajor
- kIsLittleEndian