 |
Caffe2 - Python API
A deep learning, cross platform ML framework
|
|
Caffe2 - Python API
A deep learning, cross platform ML framework
|
Public Member Functions | |
| def | setUp (self) |
| def | test_extension_function (self) |
| def | test_extension_module (self) |
| def | test_backward (self) |
| def | test_jit_compile_extension (self) |
| def | test_cuda_extension (self) |
| def | test_jit_cuda_extension (self) |
| def | test_jit_cudnn_extension (self) |
| def | test_optional (self) |
| def | test_inline_jit_compile_extension_with_functions_as_list (self) |
| def | test_inline_jit_compile_extension_with_functions_as_dict (self) |
| def | test_inline_jit_compile_extension_multiple_sources_and_no_functions (self) |
| def | test_inline_jit_compile_extension_cuda (self) |
| def | test_inline_jit_compile_extension_throws_when_functions_is_bad (self) |
| def | test_lenient_flag_handling_in_jit_extensions (self) |
| def | test_complex_registration (self) |
| def | test_half_support (self) |
| def | test_reload_jit_extension (self) |
| def | test_cpp_frontend_module_has_same_output_as_python (self) |
| def | test_cpp_frontend_module_python_inter_op (self) |
| def | test_cpp_frontend_module_has_up_to_date_attributes (self) |
| def | test_cpp_frontend_module_python_inter_op_with_cuda (self) |
| def | test_returns_shared_library_path_when_is_python_module_is_true (self) |
| def | test_no_python_abi_suffix_sets_the_correct_library_name (self) |
| def | test_set_default_type_also_changes_aten_default_type (self) |
Public Attributes | |
| x | |
| net | |
Definition at line 41 of file test_cpp_extensions.py.
| def test_cpp_extensions.TestCppExtension.test_half_support | ( | self | ) |
Checks for an issue with operator< ambiguity for half when certain THC headers are included. See https://github.com/pytorch/pytorch/pull/10301#issuecomment-416773333 for the corresponding issue.
Definition at line 321 of file test_cpp_extensions.py.
1.8.11
