Directory listing for /PyTorch/pytorch/test/compiled_autograd_skips/
FakeTensorDispatchCache.test__upsample_bilinear2d_aa_backward_dynamic_shapes
MiniOpTest.test_aot_dispatch_dynamic__test_mm
MiniOpTest.test_aot_dispatch_static__test_mm
PackedSequenceTest.test_pack_padded_sequence
TestAllowMutationOnSaved.test_backward_out_of_context
TestAllowMutationOnSaved.test_basic
TestAllowMutationOnSaved.test_double_backward
TestAllowMutationOnSaved.test_inplace_foreach
TestAllowMutationOnSaved.test_save_base_and_modify_view
TestAllowMutationOnSaved.test_save_view_modify_base
TestAllowMutationOnSaved.test_saved_but_not_anymore
TestAllowMutationOnSaved.test_views
TestAllowMutationOnSaved.test_with_math_views
TestAllowMutationOnSaved.test_with_out_variant
TestAutodiffSubgraphSlicing.test_bias_as_arg
TestAutodiffSubgraphSlicing.test_bias_as_module_attr
TestAutodiffSubgraphSlicing.test_constructed_bias
TestAutograd.test_accumulate_grad_tensor_reference
TestAutograd.test_anomaly_assign_parent_cleanup
TestAutograd.test_anomaly_detect_nan
TestAutograd.test_anomaly_grad_warnings
TestAutograd.test_autograd_inplace_views_cross_dtype
TestAutograd.test_autograd_node_isinstance
TestAutograd.test_callback_adds_callback
TestAutograd.test_callback_propagates_errors_from_device_thread
TestAutograd.test_create_graph_and_full_backward_hook_cycle
TestAutograd.test_current_graph_task_execution_order
TestAutograd.test_current_node
TestAutograd.test_custom_autograd_no_early_free
TestAutograd.test_custom_function_cycle
TestAutograd.test_custom_function_error
TestAutograd.test_custom_function_forward_mode_forward_is_no_op
TestAutograd.test_custom_function_forward_mode_wrong_formula
TestAutograd.test_custom_function_save_for_forward
TestAutograd.test_default_saved_tensors_hooks_double_backward
TestAutograd.test_grad_batched_grad
TestAutograd.test_grad_nonleaf_register_hook
TestAutograd.test_gradcheck_backward_mul_by_grad_output
TestAutograd.test_gradcheck_check_batched_grad
TestAutograd.test_gradcheck_check_forward_or_backward_only
TestAutograd.test_gradcheck_check_no_differentiable_outputs
TestAutograd.test_gradcheck_complex_non_complex_outputs
TestAutograd.test_gradcheck_dense_and_sparse_inputs
TestAutograd.test_gradcheck_forward_ad
TestAutograd.test_gradcheck_forward_ad_batched_grad
TestAutograd.test_gradcheck_multiple_mkldnn_inputs
TestAutograd.test_gradcheck_undefined_grad
TestAutograd.test_gradcheck_validates_input_mkldnn
TestAutograd.test_gradcheck_validates_inputs
TestAutograd.test_graph_save_on_cpu
TestAutograd.test_graph_save_on_cpu_cuda
TestAutograd.test_hooks
TestAutograd.test_input_buffer_accum
TestAutograd.test_mark_non_differentiable_none
TestAutograd.test_naughty_autograd_function_stashing_ctx
TestAutograd.test_no_grad_copy
TestAutograd.test_no_grad_copy_sparse
TestAutograd.test_node_post_hook_registered_during_unpack_hook
TestAutograd.test_post_accumulate_grad_hook_ordering
TestAutograd.test_profiler_seq_nr
TestAutograd.test_reentrant_with_callbacks_both_depths
TestAutograd.test_reentrant_with_callbacks_depth_0
TestAutograd.test_reentrant_with_callbacks_depth_1
TestAutograd.test_reentrant_with_non_leaf_variable_hook
TestAutograd.test_save_output_nr
TestAutograd.test_saved_variable_packing_unpacking_saved_original_with_hooks
TestAutograd.test_setup_context_when_forward_has_default_args
TestAutograd.test_sparse_gather_both_scalar
TestAutograd.test_sparse_gather_dim0
TestAutograd.test_sparse_gather_dim1
TestAutograd.test_sparse_gather_dim_neg
TestAutograd.test_sparse_gather_ind_scalar
TestAutograd.test_sparse_gather_x_scalar
TestAutograd.test_sparse_mm_backward
TestAutograd.test_to_sparse_backward
TestAutograd.test_unrelated_inputs
TestAutogradComplex.test_view_func_for_complex_views
TestAutogradDeviceTypeCPU.test_inplace_on_view_gradcheck_cpu
TestAutogradDeviceTypeCPU.test_sparse_backward_cpu_complex128
TestAutogradDeviceTypeCPU.test_sparse_backward_cpu_float64
TestAutogradDeviceTypeCPU.test_sparse_ctor_getter_backward_cpu_complex128
TestAutogradDeviceTypeCPU.test_sparse_ctor_getter_backward_cpu_float64
TestAutogradDeviceTypeCPU.test_sparse_mask_autograd_cpu
TestAutogradDeviceTypeCPU.test_where_functional_cpu
TestAutogradDeviceTypeCUDA.test_inplace_on_view_python_cuda
TestAutogradDeviceTypeCUDA.test_pin_memory_cuda
TestAutogradDeviceTypeCUDA.test_reentrant_parent_error_on_cpu_cuda
TestAutogradDeviceTypeCUDA.test_sparse_backward_cuda_complex128
TestAutogradDeviceTypeCUDA.test_sparse_backward_cuda_float64
TestAutogradDeviceTypeCUDA.test_sparse_ctor_getter_backward_cuda_complex128
TestAutogradDeviceTypeCUDA.test_sparse_ctor_getter_backward_cuda_float64
TestAutogradDeviceTypeCUDA.test_sparse_mask_autograd_cuda
TestAutogradDeviceTypeCUDA.test_where_functional_cuda
TestAutogradFallback.test_autograd_function_registered_to_cpu_mode_warn
TestAutogradFallback.test_base_does_not_require_grad_mode_warn
TestAutogradFallback.test_composite_registered_to_cpu_mode_warn
TestAutogradFallback.test_cpu_return_self_mode_warn
TestAutogradFallback.test_inplace_autograd_function_registered_to_cpu_mode_warn
TestAutogradFallback.test_no_autograd_kernel_inplace_mode_warn
TestAutogradFallback.test_no_autograd_kernel_mode_warn
TestAutogradFallback.test_post_autograd_returns_leaf_mode_warn
TestAutogradFallback.test_post_autograd_returns_mix_of_requires_grad_tensors_mode_warn
TestAutogradFallback.test_supports_tensor_lists_mode_warn
TestAutogradFallback.test_undefined_grads_mode_warn
TestAutogradFallback.test_undefined_inputs_outputs_mode_warn
TestAutogradForwardMode.test_advanced_packing_unpacking
TestAutogradFunctional.test_hessian_create_graph_vectorize_False_base_tensor
TestAutogradFunctional.test_hessian_create_graph_vectorize_False_logging_tensor
TestAutogradFunctional.test_hessian_create_graph_vectorize_True_base_tensor
TestAutogradFunctional.test_hessian_create_graph_vectorize_True_logging_tensor
TestAutogradFunctional.test_hessian_err_check_vectorize_True_base_tensor
TestAutogradFunctional.test_hessian_err_check_vectorize_True_logging_tensor
TestAutogradFunctional.test_hessian_output_vectorized_base_tensor
TestAutogradFunctional.test_hessian_output_vectorized_logging_tensor
TestAutogradFunctional.test_hessian_vectorize_correctness_multi_input_base_tensor
TestAutogradFunctional.test_hessian_vectorize_correctness_multi_input_logging_tensor
TestAutogradFunctional.test_hessian_vectorize_correctness_simple_base_tensor
TestAutogradFunctional.test_hessian_vectorize_correctness_simple_logging_tensor
TestAutogradFunctional.test_hessian_vectorize_correctness_unrelated_outputs_base_tensor
TestAutogradFunctional.test_hessian_vectorize_correctness_unrelated_outputs_logging_tensor
TestAutogradFunctional.test_hessian_vectorize_raises_no_warnings_base_tensor
TestAutogradFunctional.test_hessian_vectorize_raises_no_warnings_logging_tensor
TestAutogradFunctional.test_hvp_create_graph_base_tensor
TestAutogradFunctional.test_hvp_create_graph_logging_tensor
TestAutogradFunctional.test_jacobian_create_graph_vectorize_False_base_tensor
TestAutogradFunctional.test_jacobian_create_graph_vectorize_False_logging_tensor
TestAutogradFunctional.test_jacobian_create_graph_vectorize_True_base_tensor
TestAutogradFunctional.test_jacobian_create_graph_vectorize_True_logging_tensor
TestAutogradFunctional.test_jacobian_err_check_vectorize_True_base_tensor
TestAutogradFunctional.test_jacobian_err_check_vectorize_True_logging_tensor
TestAutogradFunctional.test_jacobian_output_vectorized_base_tensor
TestAutogradFunctional.test_jacobian_output_vectorized_logging_tensor
TestAutogradFunctional.test_jacobian_scalar_vectorized_base_tensor
TestAutogradFunctional.test_jacobian_scalar_vectorized_logging_tensor
TestAutogradFunctional.test_jacobian_vectorize_correctness_different_devices_base_tensor
TestAutogradFunctional.test_jacobian_vectorize_correctness_different_devices_logging_tensor
TestAutogradFunctional.test_jacobian_vectorize_correctness_different_dtype_base_tensor
TestAutogradFunctional.test_jacobian_vectorize_correctness_different_dtype_logging_tensor
TestAutogradFunctional.test_jacobian_vectorize_correctness_multi_input_base_tensor
TestAutogradFunctional.test_jacobian_vectorize_correctness_multi_input_logging_tensor
TestAutogradFunctional.test_jacobian_vectorize_correctness_multi_input_multi_output_base_tensor
TestAutogradFunctional.test_jacobian_vectorize_correctness_multi_input_multi_output_logging_tensor
TestAutogradFunctional.test_jacobian_vectorize_correctness_simple_base_tensor
TestAutogradFunctional.test_jacobian_vectorize_correctness_simple_logging_tensor
TestAutogradFunctional.test_jacobian_vectorize_correctness_unrelated_outputs_base_tensor
TestAutogradFunctional.test_jacobian_vectorize_correctness_unrelated_outputs_logging_tensor
TestAutogradFunctional.test_jacobian_vectorize_correctness_zero_dim_base_tensor
TestAutogradFunctional.test_jacobian_vectorize_correctness_zero_dim_logging_tensor
TestAutogradFunctional.test_jacobian_vectorize_raises_no_warnings_base_tensor
TestAutogradFunctional.test_jacobian_vectorize_raises_no_warnings_logging_tensor
TestAutogradFunctional.test_jvp_create_graph_base_tensor
TestAutogradFunctional.test_jvp_create_graph_logging_tensor
TestAutogradFunctional.test_vhp_create_graph_base_tensor
TestAutogradFunctional.test_vhp_create_graph_logging_tensor
TestAutogradFunctional.test_vjp_create_graph_base_tensor
TestAutogradFunctional.test_vjp_create_graph_logging_tensor
TestAutogradMultipleDispatchCPU.test_autograd_composite_implicit_and_dispatch_registration_cpu
TestAutogradMultipleDispatchCPU.test_autograd_multiple_dispatch_registrations_cpu
TestAutogradMultipleDispatchCPU.test_per_dispatch_key_input_saving_cpu
TestAutogradMultipleDispatchCUDA.test_autograd_composite_implicit_and_dispatch_registration_cuda
TestAutogradMultipleDispatchCUDA.test_autograd_multiple_dispatch_registrations_cuda
TestAutogradMultipleDispatchCUDA.test_backward_single_threaded_cuda
TestAutogradStreamSynchronization.test_side_stream_backward_overlap
TestBasicsCPU.test_nn_unfold_cpu
TestBasicsCPU.test_softmax_cpu
TestBasicsCPU.test_stack_cpu
TestBasicsCPU.test_to_dense_and_sparse_coo_cpu
TestBasicsCPU.test_to_dense_and_sparse_csr_cpu
TestBasicsCPU.test_to_dense_cpu
TestBasicsCPU.test_to_sparse_cpu
TestBasicsCPU.test_unfold_cpu
TestBasicsCPU.test_where_cpu
TestControlFlowTraced.test_tracing_map_autograd_aot_functionalized
TestControlFlowTraced.test_tracing_map_autograd_symbolic_dict
TestControlFlowTraced.test_tracing_map_autograd_symbolic_list
TestControlFlowTraced.test_tracing_map_autograd_symbolic_simple
TestConvolutionNNDeviceTypeCPU.test_Conv2d_backward_depthwise_cpu_complex128
TestConvolutionNNDeviceTypeCPU.test_conv3d_same_padding_backward_cpu_complex128
TestConvolutionNNDeviceTypeCPU.test_conv3d_valid_padding_backward_cpu_complex128
TestConvolutionNNDeviceTypeCPU.test_conv_backend_mkldnn1d_has_bias_False_strided_False_contiguous_False_cpu
TestConvolutionNNDeviceTypeCPU.test_conv_backend_mkldnn1d_has_bias_False_strided_False_contiguous_True_cpu
TestConvolutionNNDeviceTypeCPU.test_conv_backend_mkldnn1d_has_bias_False_strided_True_contiguous_False_cpu
TestConvolutionNNDeviceTypeCPU.test_conv_backend_mkldnn1d_has_bias_False_strided_True_contiguous_True_cpu
TestConvolutionNNDeviceTypeCPU.test_conv_backend_mkldnn1d_has_bias_True_strided_False_contiguous_False_cpu
TestConvolutionNNDeviceTypeCPU.test_conv_backend_mkldnn1d_has_bias_True_strided_False_contiguous_True_cpu
TestConvolutionNNDeviceTypeCPU.test_conv_backend_mkldnn1d_has_bias_True_strided_True_contiguous_False_cpu
TestConvolutionNNDeviceTypeCPU.test_conv_backend_mkldnn1d_has_bias_True_strided_True_contiguous_True_cpu
TestConvolutionNNDeviceTypeCPU.test_conv_backend_mkldnn2d_has_bias_False_strided_False_contiguous_False_cpu
TestConvolutionNNDeviceTypeCPU.test_conv_backend_mkldnn2d_has_bias_False_strided_False_contiguous_True_cpu
TestConvolutionNNDeviceTypeCPU.test_conv_backend_mkldnn2d_has_bias_False_strided_True_contiguous_False_cpu
TestConvolutionNNDeviceTypeCPU.test_conv_backend_mkldnn2d_has_bias_False_strided_True_contiguous_True_cpu
TestConvolutionNNDeviceTypeCPU.test_conv_backend_mkldnn2d_has_bias_True_strided_False_contiguous_False_cpu
TestConvolutionNNDeviceTypeCPU.test_conv_backend_mkldnn2d_has_bias_True_strided_False_contiguous_True_cpu
TestConvolutionNNDeviceTypeCPU.test_conv_backend_mkldnn2d_has_bias_True_strided_True_contiguous_False_cpu
TestConvolutionNNDeviceTypeCPU.test_conv_backend_mkldnn2d_has_bias_True_strided_True_contiguous_True_cpu
TestConvolutionNNDeviceTypeCPU.test_conv_backend_mkldnn3d_has_bias_False_strided_False_contiguous_False_cpu
TestConvolutionNNDeviceTypeCPU.test_conv_backend_mkldnn3d_has_bias_False_strided_False_contiguous_True_cpu
TestConvolutionNNDeviceTypeCPU.test_conv_backend_mkldnn3d_has_bias_False_strided_True_contiguous_False_cpu
TestConvolutionNNDeviceTypeCPU.test_conv_backend_mkldnn3d_has_bias_False_strided_True_contiguous_True_cpu
TestConvolutionNNDeviceTypeCPU.test_conv_backend_mkldnn3d_has_bias_True_strided_False_contiguous_False_cpu
TestConvolutionNNDeviceTypeCPU.test_conv_backend_mkldnn3d_has_bias_True_strided_False_contiguous_True_cpu
TestConvolutionNNDeviceTypeCPU.test_conv_backend_mkldnn3d_has_bias_True_strided_True_contiguous_False_cpu
TestConvolutionNNDeviceTypeCPU.test_conv_backend_mkldnn3d_has_bias_True_strided_True_contiguous_True_cpu
TestConvolutionNNDeviceTypeCPU.test_conv_backend_mkldnn_empty_batch1d_has_bias_False_strided_False_contiguous_False_cpu
TestConvolutionNNDeviceTypeCPU.test_conv_backend_mkldnn_empty_batch1d_has_bias_False_strided_False_contiguous_True_cpu
TestConvolutionNNDeviceTypeCPU.test_conv_backend_mkldnn_empty_batch1d_has_bias_False_strided_True_contiguous_False_cpu
TestConvolutionNNDeviceTypeCPU.test_conv_backend_mkldnn_empty_batch1d_has_bias_False_strided_True_contiguous_True_cpu
TestConvolutionNNDeviceTypeCPU.test_conv_backend_mkldnn_empty_batch1d_has_bias_True_strided_False_contiguous_False_cpu
TestConvolutionNNDeviceTypeCPU.test_conv_backend_mkldnn_empty_batch1d_has_bias_True_strided_False_contiguous_True_cpu
TestConvolutionNNDeviceTypeCPU.test_conv_backend_mkldnn_empty_batch1d_has_bias_True_strided_True_contiguous_False_cpu
TestConvolutionNNDeviceTypeCPU.test_conv_backend_mkldnn_empty_batch1d_has_bias_True_strided_True_contiguous_True_cpu
TestConvolutionNNDeviceTypeCPU.test_conv_backend_mkldnn_empty_batch2d_has_bias_False_strided_False_contiguous_False_cpu
TestConvolutionNNDeviceTypeCPU.test_conv_backend_mkldnn_empty_batch2d_has_bias_False_strided_False_contiguous_True_cpu
TestConvolutionNNDeviceTypeCPU.test_conv_backend_mkldnn_empty_batch2d_has_bias_False_strided_True_contiguous_False_cpu
TestConvolutionNNDeviceTypeCPU.test_conv_backend_mkldnn_empty_batch2d_has_bias_False_strided_True_contiguous_True_cpu
TestConvolutionNNDeviceTypeCPU.test_conv_backend_mkldnn_empty_batch2d_has_bias_True_strided_False_contiguous_False_cpu
TestConvolutionNNDeviceTypeCPU.test_conv_backend_mkldnn_empty_batch2d_has_bias_True_strided_False_contiguous_True_cpu
TestConvolutionNNDeviceTypeCPU.test_conv_backend_mkldnn_empty_batch2d_has_bias_True_strided_True_contiguous_False_cpu
TestConvolutionNNDeviceTypeCPU.test_conv_backend_mkldnn_empty_batch2d_has_bias_True_strided_True_contiguous_True_cpu
TestConvolutionNNDeviceTypeCPU.test_conv_backend_mkldnn_empty_batch3d_has_bias_False_strided_False_contiguous_False_cpu
TestConvolutionNNDeviceTypeCPU.test_conv_backend_mkldnn_empty_batch3d_has_bias_False_strided_False_contiguous_True_cpu
TestConvolutionNNDeviceTypeCPU.test_conv_backend_mkldnn_empty_batch3d_has_bias_False_strided_True_contiguous_False_cpu
TestConvolutionNNDeviceTypeCPU.test_conv_backend_mkldnn_empty_batch3d_has_bias_False_strided_True_contiguous_True_cpu
TestConvolutionNNDeviceTypeCPU.test_conv_backend_mkldnn_empty_batch3d_has_bias_True_strided_False_contiguous_False_cpu
TestConvolutionNNDeviceTypeCPU.test_conv_backend_mkldnn_empty_batch3d_has_bias_True_strided_False_contiguous_True_cpu
TestConvolutionNNDeviceTypeCPU.test_conv_backend_mkldnn_empty_batch3d_has_bias_True_strided_True_contiguous_False_cpu
TestConvolutionNNDeviceTypeCPU.test_conv_backend_mkldnn_empty_batch3d_has_bias_True_strided_True_contiguous_True_cpu
TestConvolutionNNDeviceTypeCPU.test_conv_backend_mkldnn_empty_batch_channel1d_has_bias_False_strided_False_contiguous_False_cpu
TestConvolutionNNDeviceTypeCPU.test_conv_backend_mkldnn_empty_batch_channel1d_has_bias_False_strided_False_contiguous_True_cpu
TestConvolutionNNDeviceTypeCPU.test_conv_backend_mkldnn_empty_batch_channel1d_has_bias_False_strided_True_contiguous_False_cpu
TestConvolutionNNDeviceTypeCPU.test_conv_backend_mkldnn_empty_batch_channel1d_has_bias_False_strided_True_contiguous_True_cpu
TestConvolutionNNDeviceTypeCPU.test_conv_backend_mkldnn_empty_batch_channel1d_has_bias_True_strided_False_contiguous_False_cpu
TestConvolutionNNDeviceTypeCPU.test_conv_backend_mkldnn_empty_batch_channel1d_has_bias_True_strided_False_contiguous_True_cpu
TestConvolutionNNDeviceTypeCPU.test_conv_backend_mkldnn_empty_batch_channel1d_has_bias_True_strided_True_contiguous_False_cpu
TestConvolutionNNDeviceTypeCPU.test_conv_backend_mkldnn_empty_batch_channel1d_has_bias_True_strided_True_contiguous_True_cpu
TestConvolutionNNDeviceTypeCPU.test_conv_backend_mkldnn_empty_batch_channel2d_has_bias_False_strided_False_contiguous_False_cpu
TestConvolutionNNDeviceTypeCPU.test_conv_backend_mkldnn_empty_batch_channel2d_has_bias_False_strided_False_contiguous_True_cpu
TestConvolutionNNDeviceTypeCPU.test_conv_backend_mkldnn_empty_batch_channel2d_has_bias_False_strided_True_contiguous_False_cpu
TestConvolutionNNDeviceTypeCPU.test_conv_backend_mkldnn_empty_batch_channel2d_has_bias_False_strided_True_contiguous_True_cpu
TestConvolutionNNDeviceTypeCPU.test_conv_backend_mkldnn_empty_batch_channel2d_has_bias_True_strided_False_contiguous_False_cpu
TestConvolutionNNDeviceTypeCPU.test_conv_backend_mkldnn_empty_batch_channel2d_has_bias_True_strided_False_contiguous_True_cpu
TestConvolutionNNDeviceTypeCPU.test_conv_backend_mkldnn_empty_batch_channel2d_has_bias_True_strided_True_contiguous_False_cpu
TestConvolutionNNDeviceTypeCPU.test_conv_backend_mkldnn_empty_batch_channel2d_has_bias_True_strided_True_contiguous_True_cpu
TestConvolutionNNDeviceTypeCPU.test_conv_backend_mkldnn_empty_batch_channel3d_has_bias_False_strided_False_contiguous_False_cpu
TestConvolutionNNDeviceTypeCPU.test_conv_backend_mkldnn_empty_batch_channel3d_has_bias_False_strided_False_contiguous_True_cpu
TestConvolutionNNDeviceTypeCPU.test_conv_backend_mkldnn_empty_batch_channel3d_has_bias_False_strided_True_contiguous_False_cpu
TestConvolutionNNDeviceTypeCPU.test_conv_backend_mkldnn_empty_batch_channel3d_has_bias_False_strided_True_contiguous_True_cpu
TestConvolutionNNDeviceTypeCPU.test_conv_backend_mkldnn_empty_batch_channel3d_has_bias_True_strided_False_contiguous_False_cpu
TestConvolutionNNDeviceTypeCPU.test_conv_backend_mkldnn_empty_batch_channel3d_has_bias_True_strided_False_contiguous_True_cpu
TestConvolutionNNDeviceTypeCPU.test_conv_backend_mkldnn_empty_batch_channel3d_has_bias_True_strided_True_contiguous_False_cpu
TestConvolutionNNDeviceTypeCPU.test_conv_backend_mkldnn_empty_batch_channel3d_has_bias_True_strided_True_contiguous_True_cpu
TestConvolutionNNDeviceTypeCPU.test_conv_backend_mkldnn_empty_channel1d_has_bias_False_strided_False_contiguous_False_cpu
TestConvolutionNNDeviceTypeCPU.test_conv_backend_mkldnn_empty_channel1d_has_bias_False_strided_False_contiguous_True_cpu
TestConvolutionNNDeviceTypeCPU.test_conv_backend_mkldnn_empty_channel1d_has_bias_False_strided_True_contiguous_False_cpu
TestConvolutionNNDeviceTypeCPU.test_conv_backend_mkldnn_empty_channel1d_has_bias_False_strided_True_contiguous_True_cpu
TestConvolutionNNDeviceTypeCPU.test_conv_backend_mkldnn_empty_channel1d_has_bias_True_strided_False_contiguous_False_cpu
TestConvolutionNNDeviceTypeCPU.test_conv_backend_mkldnn_empty_channel1d_has_bias_True_strided_False_contiguous_True_cpu
TestConvolutionNNDeviceTypeCPU.test_conv_backend_mkldnn_empty_channel1d_has_bias_True_strided_True_contiguous_False_cpu
TestConvolutionNNDeviceTypeCPU.test_conv_backend_mkldnn_empty_channel1d_has_bias_True_strided_True_contiguous_True_cpu
TestConvolutionNNDeviceTypeCPU.test_conv_backend_mkldnn_empty_channel2d_has_bias_False_strided_False_contiguous_False_cpu
TestConvolutionNNDeviceTypeCPU.test_conv_backend_mkldnn_empty_channel2d_has_bias_False_strided_False_contiguous_True_cpu
TestConvolutionNNDeviceTypeCPU.test_conv_backend_mkldnn_empty_channel2d_has_bias_False_strided_True_contiguous_False_cpu
TestConvolutionNNDeviceTypeCPU.test_conv_backend_mkldnn_empty_channel2d_has_bias_False_strided_True_contiguous_True_cpu
TestConvolutionNNDeviceTypeCPU.test_conv_backend_mkldnn_empty_channel2d_has_bias_True_strided_False_contiguous_False_cpu
TestConvolutionNNDeviceTypeCPU.test_conv_backend_mkldnn_empty_channel2d_has_bias_True_strided_False_contiguous_True_cpu
TestConvolutionNNDeviceTypeCPU.test_conv_backend_mkldnn_empty_channel2d_has_bias_True_strided_True_contiguous_False_cpu
TestConvolutionNNDeviceTypeCPU.test_conv_backend_mkldnn_empty_channel2d_has_bias_True_strided_True_contiguous_True_cpu
TestConvolutionNNDeviceTypeCPU.test_conv_backend_mkldnn_empty_channel3d_has_bias_False_strided_False_contiguous_False_cpu
TestConvolutionNNDeviceTypeCPU.test_conv_backend_mkldnn_empty_channel3d_has_bias_False_strided_False_contiguous_True_cpu
TestConvolutionNNDeviceTypeCPU.test_conv_backend_mkldnn_empty_channel3d_has_bias_False_strided_True_contiguous_False_cpu
TestConvolutionNNDeviceTypeCPU.test_conv_backend_mkldnn_empty_channel3d_has_bias_False_strided_True_contiguous_True_cpu
TestConvolutionNNDeviceTypeCPU.test_conv_backend_mkldnn_empty_channel3d_has_bias_True_strided_False_contiguous_False_cpu
TestConvolutionNNDeviceTypeCPU.test_conv_backend_mkldnn_empty_channel3d_has_bias_True_strided_False_contiguous_True_cpu
TestConvolutionNNDeviceTypeCPU.test_conv_backend_mkldnn_empty_channel3d_has_bias_True_strided_True_contiguous_False_cpu
TestConvolutionNNDeviceTypeCPU.test_conv_backend_mkldnn_empty_channel3d_has_bias_True_strided_True_contiguous_True_cpu
TestConvolutionNNDeviceTypeCPU.test_conv_double_backward_stride_cpu
TestCppApiParity.test_torch_nn_Embedding_sparse
TestCppApiParity.test_torch_nn_EmbeddingBag_sparse
TestCustomOpTestingCPU.test_aot_autograd_check_degenerate_cases_check_gradients_auto_dynamic_False_cpu
TestCustomOpTestingCPU.test_aot_autograd_check_degenerate_cases_check_gradients_auto_dynamic_True_cpu
TestCustomOpTestingCPU.test_aot_autograd_check_degenerate_cases_check_gradients_False_dynamic_False_cpu
TestCustomOpTestingCPU.test_aot_autograd_check_degenerate_cases_check_gradients_False_dynamic_True_cpu
TestCustomOpTestingCPU.test_global_state_mutation_cpu
TestEmbeddingNN.test_embedding_sparse_basic
TestEmbeddingNN.test_embedding_sparse_empty_tensor
TestEmbeddingNNDeviceTypeCPU.test_embedding_backward_cpu_float64
TestEmbeddingNNDeviceTypeCPU.test_embedding_bag_2D_padding_idx_cpu_float32
TestEmbeddingNNDeviceTypeCPU.test_embedding_bag_2D_padding_idx_cpu_float64
TestEmbeddingNNDeviceTypeCPU.test_embedding_bag_bfloat16_cpu_int32_int32
TestEmbeddingNNDeviceTypeCPU.test_embedding_bag_bfloat16_cpu_int32_int64
TestEmbeddingNNDeviceTypeCPU.test_embedding_bag_bfloat16_cpu_int64_int32
TestEmbeddingNNDeviceTypeCPU.test_embedding_bag_bfloat16_cpu_int64_int64
TestEmbeddingNNDeviceTypeCPU.test_embedding_bag_device_cpu_int32_int32_bfloat16
TestEmbeddingNNDeviceTypeCPU.test_embedding_bag_device_cpu_int32_int32_float16
TestEmbeddingNNDeviceTypeCPU.test_embedding_bag_device_cpu_int32_int32_float32
TestEmbeddingNNDeviceTypeCPU.test_embedding_bag_device_cpu_int32_int32_float64
TestEmbeddingNNDeviceTypeCPU.test_embedding_bag_device_cpu_int32_int64_bfloat16
TestEmbeddingNNDeviceTypeCPU.test_embedding_bag_device_cpu_int32_int64_float16
TestEmbeddingNNDeviceTypeCPU.test_embedding_bag_device_cpu_int32_int64_float32
TestEmbeddingNNDeviceTypeCPU.test_embedding_bag_device_cpu_int32_int64_float64
TestEmbeddingNNDeviceTypeCPU.test_embedding_bag_device_cpu_int64_int32_bfloat16
TestEmbeddingNNDeviceTypeCPU.test_embedding_bag_device_cpu_int64_int32_float16
TestEmbeddingNNDeviceTypeCPU.test_embedding_bag_device_cpu_int64_int32_float32
TestEmbeddingNNDeviceTypeCPU.test_embedding_bag_device_cpu_int64_int32_float64
TestEmbeddingNNDeviceTypeCPU.test_embedding_bag_device_cpu_int64_int64_bfloat16
TestEmbeddingNNDeviceTypeCPU.test_embedding_bag_device_cpu_int64_int64_float16
TestEmbeddingNNDeviceTypeCPU.test_embedding_bag_device_cpu_int64_int64_float32
TestEmbeddingNNDeviceTypeCPU.test_embedding_bag_device_cpu_int64_int64_float64
TestEmbeddingNNDeviceTypeCPU.test_embedding_bag_half_cpu_int32_int32
TestEmbeddingNNDeviceTypeCPU.test_embedding_bag_half_cpu_int32_int64
TestEmbeddingNNDeviceTypeCPU.test_embedding_bag_half_cpu_int64_int32
TestEmbeddingNNDeviceTypeCPU.test_embedding_bag_half_cpu_int64_int64
TestEmbeddingNNDeviceTypeCPU.test_EmbeddingBag_per_sample_weights_and_no_offsets_cpu_int32_float32
TestEmbeddingNNDeviceTypeCPU.test_EmbeddingBag_per_sample_weights_and_no_offsets_cpu_int32_float64
TestEmbeddingNNDeviceTypeCPU.test_EmbeddingBag_per_sample_weights_and_no_offsets_cpu_int64_float32
TestEmbeddingNNDeviceTypeCPU.test_EmbeddingBag_per_sample_weights_and_no_offsets_cpu_int64_float64
TestExamplesCorrectnessCPU.test_maml_omniglot_mechanism_functional_call_cpu
TestExamplesCorrectnessCPU.test_maml_omniglot_mechanism_make_functional_cpu
TestExpandedWeightFunctionalCPU.test_cnn_model_mean_cpu
TestExpandedWeightFunctionalCPU.test_cnn_model_sum_cpu
TestExpandedWeightFunctionalCPU.test_embedding_model_cpu
TestExpandedWeightFunctionalCPU.test_group_norm_model_num_dim_1_cpu
TestExpandedWeightFunctionalCPU.test_group_norm_model_num_dim_2_cpu
TestExpandedWeightFunctionalCPU.test_group_norm_model_num_dim_3_cpu
TestExpandedWeightFunctionalCPU.test_instance_norm_model_num_dim_1_cpu
TestExpandedWeightFunctionalCPU.test_instance_norm_model_num_dim_2_cpu
TestExpandedWeightFunctionalCPU.test_instance_norm_model_num_dim_3_cpu
TestExpandedWeightFunctionalCPU.test_layer_norm_model_num_dim_1_cpu
TestExpandedWeightFunctionalCPU.test_layer_norm_model_num_dim_2_cpu
TestExpandedWeightFunctionalCPU.test_layer_norm_model_num_dim_3_cpu
TestExpandedWeightModuleCPU.test_Conv1d_circular_stride2_pad2_cpu
TestExpandedWeightModuleCPU.test_Conv1d_circular_stride2_pad2_multiple_inputs_cpu
TestExpandedWeightModuleCPU.test_Conv1d_cpu
TestExpandedWeightModuleCPU.test_Conv1d_multiple_inputs_cpu
TestExpandedWeightModuleCPU.test_Conv1d_pad1_cpu
TestExpandedWeightModuleCPU.test_Conv1d_pad1_multiple_inputs_cpu
TestExpandedWeightModuleCPU.test_Conv1d_pad1size1_cpu
TestExpandedWeightModuleCPU.test_Conv1d_pad1size1_multiple_inputs_cpu
TestExpandedWeightModuleCPU.test_Conv1d_pad2_cpu
TestExpandedWeightModuleCPU.test_Conv1d_pad2_multiple_inputs_cpu
TestExpandedWeightModuleCPU.test_Conv1d_pad2size1_cpu
TestExpandedWeightModuleCPU.test_Conv1d_pad2size1_multiple_inputs_cpu
TestExpandedWeightModuleCPU.test_Conv1d_reflect_stride2_pad2_cpu
TestExpandedWeightModuleCPU.test_Conv1d_reflect_stride2_pad2_multiple_inputs_cpu
TestExpandedWeightModuleCPU.test_Conv1d_replicate_stride2_pad2_cpu
TestExpandedWeightModuleCPU.test_Conv1d_replicate_stride2_pad2_multiple_inputs_cpu
TestExpandedWeightModuleCPU.test_Conv1d_stride_cpu
TestExpandedWeightModuleCPU.test_Conv1d_stride_multiple_inputs_cpu
TestExpandedWeightModuleCPU.test_Conv1d_zeros_stride2_pad2_cpu
TestExpandedWeightModuleCPU.test_Conv1d_zeros_stride2_pad2_multiple_inputs_cpu
TestExpandedWeightModuleCPU.test_Conv2d_circular_stride2_pad2_cpu
TestExpandedWeightModuleCPU.test_Conv2d_circular_stride2_pad2_multiple_inputs_cpu
TestExpandedWeightModuleCPU.test_Conv2d_cpu
TestExpandedWeightModuleCPU.test_Conv2d_dilated_cpu
TestExpandedWeightModuleCPU.test_Conv2d_dilated_multiple_inputs_cpu
TestExpandedWeightModuleCPU.test_Conv2d_multiple_inputs_cpu
TestExpandedWeightModuleCPU.test_Conv2d_no_bias_cpu
TestExpandedWeightModuleCPU.test_Conv2d_no_bias_multiple_inputs_cpu
TestExpandedWeightModuleCPU.test_Conv2d_padding_cpu
TestExpandedWeightModuleCPU.test_Conv2d_padding_multiple_inputs_cpu
TestExpandedWeightModuleCPU.test_Conv2d_reflect_stride2_pad2_cpu
TestExpandedWeightModuleCPU.test_Conv2d_reflect_stride2_pad2_multiple_inputs_cpu
TestExpandedWeightModuleCPU.test_Conv2d_replicate_stride2_pad2_cpu
TestExpandedWeightModuleCPU.test_Conv2d_replicate_stride2_pad2_multiple_inputs_cpu
TestExpandedWeightModuleCPU.test_Conv2d_strided_cpu
TestExpandedWeightModuleCPU.test_Conv2d_strided_multiple_inputs_cpu
TestExpandedWeightModuleCPU.test_Conv2d_zeros_stride2_pad2_cpu
TestExpandedWeightModuleCPU.test_Conv2d_zeros_stride2_pad2_multiple_inputs_cpu
TestExpandedWeightModuleCPU.test_Conv3d_1x1x1_no_bias_cpu
TestExpandedWeightModuleCPU.test_Conv3d_1x1x1_no_bias_multiple_inputs_cpu
TestExpandedWeightModuleCPU.test_Conv3d_circular_stride2_pad2_cpu
TestExpandedWeightModuleCPU.test_Conv3d_circular_stride2_pad2_multiple_inputs_cpu
TestExpandedWeightModuleCPU.test_Conv3d_cpu
TestExpandedWeightModuleCPU.test_Conv3d_multiple_inputs_cpu
TestExpandedWeightModuleCPU.test_Conv3d_no_bias_cpu
TestExpandedWeightModuleCPU.test_Conv3d_no_bias_multiple_inputs_cpu
TestExpandedWeightModuleCPU.test_Conv3d_replicate_stride2_pad2_cpu
TestExpandedWeightModuleCPU.test_Conv3d_replicate_stride2_pad2_multiple_inputs_cpu
TestExpandedWeightModuleCPU.test_Conv3d_stride_cpu
TestExpandedWeightModuleCPU.test_Conv3d_stride_multiple_inputs_cpu
TestExpandedWeightModuleCPU.test_Conv3d_stride_padding_cpu
TestExpandedWeightModuleCPU.test_Conv3d_stride_padding_multiple_inputs_cpu
TestExpandedWeightModuleCPU.test_Conv3d_zeros_stride2_pad2_cpu
TestExpandedWeightModuleCPU.test_Conv3d_zeros_stride2_pad2_multiple_inputs_cpu
TestExpandedWeightModuleCPU.test_Linear_cpu
TestExpandedWeightModuleCPU.test_Linear_multiple_inputs_cpu
TestExpandedWeightModuleCPU.test_Linear_no_bias_cpu
TestExpandedWeightModuleCPU.test_Linear_no_bias_multiple_inputs_cpu
TestExpandedWeightModuleCPU.test_per_sample_api_failing_cpu
TestFakeQuantize.test_fq_module_per_channel
TestGenerateOpcheckTests.test_opcheck
TestGenerateOpcheckTests.test_opcheck_customopdef
TestGenericProxyTensorFake.test_make_fx_model_fwd_bwd
TestGenericProxyTensorFake.test_make_fx_model_fwd_bwd_wgtupdate
TestGenericProxyTensorFake.test_proxy_tensor
TestGenericProxyTensorFake.test_resnet18_backward_trace
TestGenericProxyTensorReal.test_make_fx_model_fwd_bwd
TestGenericProxyTensorReal.test_make_fx_model_fwd_bwd_wgtupdate
TestGenericProxyTensorReal.test_proxy_tensor
TestGenericProxyTensorReal.test_resnet18_backward_trace
TestGenericProxyTensorSymbolic.test_make_fx_model_fwd_bwd
TestGenericProxyTensorSymbolic.test_make_fx_model_fwd_bwd_wgtupdate
TestGenericProxyTensorSymbolic.test_proxy_tensor
TestGenericProxyTensorSymbolic.test_resnet18_backward_trace
TestJit.test_script_autograd_grad
TestJit.test_script_backward_twice
TestJitGeneratedModule.test_nn_Conv2d_no_bias
TestJitGeneratedModule.test_nn_Conv3d_1x1x1_no_bias
TestJitGeneratedModule.test_nn_Conv3d_no_bias
TestJitGeneratedModule.test_nn_ConvTranspose1d_no_bias
TestJitGeneratedModule.test_nn_ConvTranspose2d_dilated
TestJitGeneratedModule.test_nn_ConvTranspose2d_no_bias
TestJitGeneratedModule.test_nn_Softsign_no_batch_dim
TestJitGeneratedModule.test_nn_Tanhshrink_no_batch_dim
TestLinalgCPU.test_invariance_error_spectral_decompositions_cpu_complex128
TestMkldnnCPU.test_adaptive_avg_pool2d_cpu
TestMkldnnCPU.test_autograd_from_mkldnn_cpu
TestMkldnnCPU.test_autograd_to_mkldnn_cpu
TestMkldnnCPU.test_avg_pool2d_cpu
TestMkldnnCPU.test_avg_pool3d_cpu
TestMkldnnCPU.test_batch_norm_2d_cpu
TestMkldnnCPU.test_conv2d_cpu
TestMkldnnCPU.test_conv3d_cpu
TestMkldnnCPU.test_gelu_bf16_cpu
TestMkldnnCPU.test_gelu_cpu
TestMkldnnCPU.test_linear_backward_cpu
TestMkldnnCPU.test_linear_non_contiguous_weight_cpu
TestMkldnnCPU.test_max_pool2d_cpu
TestMkldnnCPU.test_max_pool3d_cpu
TestMkldnnCPU.test_prelu_bf16_cpu
TestMkldnnCPU.test_prelu_cpu
TestMkldnnCPU.test_relu__cpu
TestMkldnnCPU.test_relu_cpu
TestMkldnnCPU.test_reshape_backward_cpu
TestModels.test_dcgan_models
TestModels.test_mnist
TestModels.test_reinforcement_learning
TestModels.test_vae
TestModuleGlobalHooks.test_module_backward_global_hook_writeable
TestModuleGlobalHooks.test_module_global_hook_invalid_outputs
TestModuleHookNN.test_hook_invalid_outputs
TestModuleHookNN.test_hooks
TestMultithreadAutograd.test_fork_join_in_middle
TestMultithreadAutograd.test_multi_grad_all_hooks
TestMultithreadAutograd.test_multi_grad_any_hooks
TestMultithreadAutograd.test_python_thread_in_middle
TestMultithreadAutograd.test_set_multithreading_enabled_as_context_manager_and_function
TestMultithreadAutograd.test_simple_backward
TestMultithreadAutograd.test_simple_backward_same_input
TestNamedTensor.test_autograd_ignores_names
TestNamedTensor.test_autograd_smoke
TestNamedTensor.test_autograd_warns_named_grad
TestNamedTensor.test_tensor_grad_is_unnamed
TestNestedCheckpoint.test_nested_checkpoint_early_stop_False
TestNestedCheckpoint.test_nested_checkpoint_early_stop_True
TestNestedCheckpoint.test_nested_checkpoint_set_early_stop_no_recompution_needed
TestNestedCheckpoint.test_nested_checkpoint_two_children_early_stop_False
TestNestedCheckpoint.test_nested_checkpoint_two_children_early_stop_True
TestNestedTensorAutogradCPU.test_abs_backward_cpu
TestNestedTensorAutogradCPU.test_accumulate_grad_different_strides_cpu
TestNestedTensorAutogradCPU.test_as_nested_tensor_propagates_gradients_cpu
TestNestedTensorAutogradCPU.test_backward_add_strided_cpu
TestNestedTensorAutogradCPU.test_backward_for_add_op_cpu
TestNestedTensorAutogradCPU.test_backward_for_sub_op_cpu
TestNestedTensorAutogradCPU.test_backward_sub_strided_cpu
TestNestedTensorAutogradCPU.test_dropout_backward_jagged_cpu
TestNestedTensorAutogradCPU.test_dropout_backward_strided_cpu
TestNestedTensorAutogradCPU.test_gelu_backward_cpu
TestNestedTensorAutogradCPU.test_indexing_backward_cpu
TestNestedTensorAutogradCPU.test_layer_norm_backward_5d_size_128_cpu
TestNestedTensorAutogradCPU.test_layer_norm_backward_5d_size_2_cpu
TestNestedTensorAutogradCPU.test_layer_norm_backward_5d_size_32_cpu
TestNestedTensorAutogradCPU.test_layer_norm_backward_5d_size_4_cpu
TestNestedTensorAutogradCPU.test_layer_norm_backward_edge_case_cpu
TestNestedTensorAutogradCPU.test_layer_norm_backward_size_1023_cpu
TestNestedTensorAutogradCPU.test_layer_norm_backward_size_1024_cpu
TestNestedTensorAutogradCPU.test_layer_norm_backward_size_128_cpu
TestNestedTensorAutogradCPU.test_layer_norm_backward_size_256_cpu
TestNestedTensorAutogradCPU.test_layer_norm_backward_size_2_cpu
TestNestedTensorAutogradCPU.test_layer_norm_backward_size_32_cpu
TestNestedTensorAutogradCPU.test_layer_norm_backward_size_4_cpu
TestNestedTensorAutogradCPU.test_layer_norm_backward_size_512_cpu
TestNestedTensorAutogradCPU.test_layer_norm_backward_size_513_cpu
TestNestedTensorAutogradCPU.test_masked_fill_backward_cpu
TestNestedTensorAutogradCPU.test_nested_tensor_bmm_backward_cpu
TestNestedTensorAutogradCPU.test_nested_tensor_bmm_gradcheck_cpu
TestNestedTensorAutogradCPU.test_nested_tensor_from_list_cpu
TestNestedTensorAutogradCPU.test_nested_tensor_from_mask_and_to_padded_cpu
TestNestedTensorAutogradCPU.test_nested_tensor_from_padded_cpu
TestNestedTensorAutogradCPU.test_nested_tensor_from_padded_fused_cpu
TestNestedTensorAutogradCPU.test_nested_tensor_generates_leaf_cpu
TestNestedTensorAutogradCPU.test_nested_tensor_linear_backward_cpu
TestNestedTensorAutogradCPU.test_nested_tensor_linear_cpu
TestNestedTensorAutogradCPU.test_nested_tensor_linear_plus_transpose_cpu
TestNestedTensorAutogradCPU.test_nested_tensor_matmul_backward_cpu
TestNestedTensorAutogradCPU.test_nested_tensor_matmul_gradcheck_cpu
TestNestedTensorAutogradCPU.test_nested_tensor_reshape_backward_cpu
TestNestedTensorAutogradCPU.test_nested_tensor_reshape_gradcheck_cpu
TestNestedTensorAutogradCPU.test_nested_tensor_softmax_cpu
TestNestedTensorAutogradCPU.test_nested_tensor_squeeze_backward_cpu
TestNestedTensorAutogradCPU.test_nested_tensor_squeeze_gradcheck_cpu
TestNestedTensorAutogradCPU.test_nested_tensor_to_padded_tensor_cpu
TestNestedTensorAutogradCPU.test_nested_tensor_transpose_backward_cpu
TestNestedTensorAutogradCPU.test_nested_tensor_transpose_gradcheck_cpu
TestNestedTensorAutogradCPU.test_nested_tensor_unsqueeze_backward_cpu
TestNestedTensorAutogradCPU.test_nested_tensor_unsqueeze_gradcheck_cpu
TestNestedTensorAutogradCPU.test_relu_backward_cpu
TestNestedTensorAutogradCPU.test_selu_backward_cpu
TestNestedTensorAutogradCPU.test_split_with_sizes_flow_through_cpu
TestNestedTensorAutogradCPU.test_to_buffer_series_ops_grad_with_broadcast_cpu
TestNestedTensorAutogradCPU.test_unbind_flow_through_cpu
TestNestedTensorAutogradCPU.test_values_grad_with_broadcast_cpu
TestNestedTensorAutogradCUDA.test_abs_backward_cuda
TestNestedTensorAutogradCUDA.test_accumulate_grad_different_strides_cuda
TestNestedTensorAutogradCUDA.test_as_nested_tensor_propagates_gradients_cuda
TestNestedTensorAutogradCUDA.test_backward_add_strided_cuda
TestNestedTensorAutogradCUDA.test_backward_for_add_op_cuda
TestNestedTensorAutogradCUDA.test_backward_for_sub_op_cuda
TestNestedTensorAutogradCUDA.test_backward_sub_strided_cuda
TestNestedTensorAutogradCUDA.test_dropout_backward_jagged_cuda
TestNestedTensorAutogradCUDA.test_dropout_backward_strided_cuda
TestNestedTensorAutogradCUDA.test_gelu_backward_cuda
TestNestedTensorAutogradCUDA.test_indexing_backward_cuda
TestNestedTensorAutogradCUDA.test_layer_norm_backward_5d_size_128_cuda
TestNestedTensorAutogradCUDA.test_layer_norm_backward_5d_size_2_cuda
TestNestedTensorAutogradCUDA.test_layer_norm_backward_5d_size_32_cuda
TestNestedTensorAutogradCUDA.test_layer_norm_backward_5d_size_4_cuda
TestNestedTensorAutogradCUDA.test_layer_norm_backward_edge_case_cuda
TestNestedTensorAutogradCUDA.test_layer_norm_backward_size_1023_cuda
TestNestedTensorAutogradCUDA.test_layer_norm_backward_size_1024_cuda
TestNestedTensorAutogradCUDA.test_layer_norm_backward_size_128_cuda
TestNestedTensorAutogradCUDA.test_layer_norm_backward_size_256_cuda
TestNestedTensorAutogradCUDA.test_layer_norm_backward_size_2_cuda
TestNestedTensorAutogradCUDA.test_layer_norm_backward_size_32_cuda
TestNestedTensorAutogradCUDA.test_layer_norm_backward_size_4_cuda
TestNestedTensorAutogradCUDA.test_layer_norm_backward_size_512_cuda
TestNestedTensorAutogradCUDA.test_layer_norm_backward_size_513_cuda
TestNestedTensorAutogradCUDA.test_masked_fill_backward_cuda
TestNestedTensorAutogradCUDA.test_nested_tensor_bmm_backward_cuda
TestNestedTensorAutogradCUDA.test_nested_tensor_bmm_gradcheck_cuda
TestNestedTensorAutogradCUDA.test_nested_tensor_from_list_cuda
TestNestedTensorAutogradCUDA.test_nested_tensor_from_mask_and_to_padded_cuda
TestNestedTensorAutogradCUDA.test_nested_tensor_from_padded_cuda
TestNestedTensorAutogradCUDA.test_nested_tensor_from_padded_fused_cuda
TestNestedTensorAutogradCUDA.test_nested_tensor_generates_leaf_cuda
TestNestedTensorAutogradCUDA.test_nested_tensor_linear_backward_cuda
TestNestedTensorAutogradCUDA.test_nested_tensor_linear_cuda
TestNestedTensorAutogradCUDA.test_nested_tensor_linear_plus_transpose_cuda
TestNestedTensorAutogradCUDA.test_nested_tensor_matmul_backward_cuda
TestNestedTensorAutogradCUDA.test_nested_tensor_matmul_gradcheck_cuda
TestNestedTensorAutogradCUDA.test_nested_tensor_reshape_backward_cuda
TestNestedTensorAutogradCUDA.test_nested_tensor_reshape_gradcheck_cuda
TestNestedTensorAutogradCUDA.test_nested_tensor_softmax_cuda
TestNestedTensorAutogradCUDA.test_nested_tensor_squeeze_backward_cuda
TestNestedTensorAutogradCUDA.test_nested_tensor_squeeze_gradcheck_cuda
TestNestedTensorAutogradCUDA.test_nested_tensor_to_padded_tensor_cuda
TestNestedTensorAutogradCUDA.test_nested_tensor_transpose_backward_cuda
TestNestedTensorAutogradCUDA.test_nested_tensor_transpose_gradcheck_cuda
TestNestedTensorAutogradCUDA.test_nested_tensor_unsqueeze_backward_cuda
TestNestedTensorAutogradCUDA.test_nested_tensor_unsqueeze_gradcheck_cuda
TestNestedTensorAutogradCUDA.test_relu_backward_cuda
TestNestedTensorAutogradCUDA.test_selu_backward_cuda
TestNestedTensorAutogradCUDA.test_split_with_sizes_flow_through_cuda
TestNestedTensorAutogradCUDA.test_to_buffer_series_ops_grad_with_broadcast_cuda
TestNestedTensorAutogradCUDA.test_unbind_flow_through_cuda
TestNestedTensorAutogradCUDA.test_values_grad_with_broadcast_cuda
TestNestedTensorDeviceTypeCPU.test_detach_cpu_float16
TestNestedTensorDeviceTypeCPU.test_detach_cpu_float32
TestNestedTensorDeviceTypeCPU.test_detach_cpu_float64
TestNestedTensorDeviceTypeCPU.test_embedding_jagged_cpu
TestNestedTensorDeviceTypeCPU.test_nested_tensor_chunk_cpu_float16
TestNestedTensorDeviceTypeCPU.test_nested_tensor_chunk_cpu_float32
TestNestedTensorDeviceTypeCPU.test_nested_tensor_chunk_cpu_float64
TestNestedTensorDeviceTypeCPU.test_nested_tensor_indexing_cpu_float16
TestNestedTensorDeviceTypeCPU.test_nested_tensor_indexing_cpu_float32
TestNestedTensorDeviceTypeCPU.test_nested_tensor_indexing_cpu_float64
TestNestedTensorDeviceTypeCPU.test_nested_tensor_sum_dim_cpu_float32
TestNestedTensorDeviceTypeCUDA.test_detach_cuda_float16
TestNestedTensorDeviceTypeCUDA.test_detach_cuda_float32
TestNestedTensorDeviceTypeCUDA.test_detach_cuda_float64
TestNestedTensorDeviceTypeCUDA.test_embedding_jagged_cuda
TestNestedTensorDeviceTypeCUDA.test_nested_tensor_chunk_cuda_float16
TestNestedTensorDeviceTypeCUDA.test_nested_tensor_chunk_cuda_float32
TestNestedTensorDeviceTypeCUDA.test_nested_tensor_chunk_cuda_float64
TestNestedTensorDeviceTypeCUDA.test_nested_tensor_indexing_cuda_float16
TestNestedTensorDeviceTypeCUDA.test_nested_tensor_indexing_cuda_float32
TestNestedTensorDeviceTypeCUDA.test_nested_tensor_indexing_cuda_float64
TestNestedTensorOpInfoCPU.test_nested_tensor_input_mutation_backward_cpu
TestNestedTensorOpInfoCUDA.test_nested_tensor_input_mutation_backward_cuda
TestNestedTensorSubclassCPU.test_apply__cpu_float32
TestNestedTensorSubclassCPU.test_as_nested_tensor_from_tensor_dim_2_layout_jagged_requires_grad_True_contiguous_False_cpu_float16
TestNestedTensorSubclassCPU.test_as_nested_tensor_from_tensor_dim_2_layout_jagged_requires_grad_True_contiguous_False_cpu_float32
TestNestedTensorSubclassCPU.test_as_nested_tensor_from_tensor_dim_2_layout_jagged_requires_grad_True_contiguous_False_cpu_float64
TestNestedTensorSubclassCPU.test_as_nested_tensor_from_tensor_dim_2_layout_jagged_requires_grad_True_contiguous_True_cpu_float16
TestNestedTensorSubclassCPU.test_as_nested_tensor_from_tensor_dim_2_layout_jagged_requires_grad_True_contiguous_True_cpu_float32
TestNestedTensorSubclassCPU.test_as_nested_tensor_from_tensor_dim_2_layout_jagged_requires_grad_True_contiguous_True_cpu_float64
TestNestedTensorSubclassCPU.test_as_nested_tensor_from_tensor_dim_2_layout_strided_requires_grad_True_contiguous_False_cpu_float16
TestNestedTensorSubclassCPU.test_as_nested_tensor_from_tensor_dim_2_layout_strided_requires_grad_True_contiguous_False_cpu_float32
TestNestedTensorSubclassCPU.test_as_nested_tensor_from_tensor_dim_2_layout_strided_requires_grad_True_contiguous_False_cpu_float64
TestNestedTensorSubclassCPU.test_as_nested_tensor_from_tensor_dim_2_layout_strided_requires_grad_True_contiguous_True_cpu_float16
TestNestedTensorSubclassCPU.test_as_nested_tensor_from_tensor_dim_2_layout_strided_requires_grad_True_contiguous_True_cpu_float32
TestNestedTensorSubclassCPU.test_as_nested_tensor_from_tensor_dim_2_layout_strided_requires_grad_True_contiguous_True_cpu_float64
TestNestedTensorSubclassCPU.test_as_nested_tensor_from_tensor_dim_3_layout_jagged_requires_grad_True_contiguous_False_cpu_float16
TestNestedTensorSubclassCPU.test_as_nested_tensor_from_tensor_dim_3_layout_jagged_requires_grad_True_contiguous_False_cpu_float32
TestNestedTensorSubclassCPU.test_as_nested_tensor_from_tensor_dim_3_layout_jagged_requires_grad_True_contiguous_False_cpu_float64
TestNestedTensorSubclassCPU.test_as_nested_tensor_from_tensor_dim_3_layout_jagged_requires_grad_True_contiguous_True_cpu_float16
TestNestedTensorSubclassCPU.test_as_nested_tensor_from_tensor_dim_3_layout_jagged_requires_grad_True_contiguous_True_cpu_float32
TestNestedTensorSubclassCPU.test_as_nested_tensor_from_tensor_dim_3_layout_jagged_requires_grad_True_contiguous_True_cpu_float64
TestNestedTensorSubclassCPU.test_as_nested_tensor_from_tensor_dim_3_layout_strided_requires_grad_True_contiguous_False_cpu_float16
TestNestedTensorSubclassCPU.test_as_nested_tensor_from_tensor_dim_3_layout_strided_requires_grad_True_contiguous_False_cpu_float32
TestNestedTensorSubclassCPU.test_as_nested_tensor_from_tensor_dim_3_layout_strided_requires_grad_True_contiguous_False_cpu_float64
TestNestedTensorSubclassCPU.test_as_nested_tensor_from_tensor_dim_3_layout_strided_requires_grad_True_contiguous_True_cpu_float16
TestNestedTensorSubclassCPU.test_as_nested_tensor_from_tensor_dim_3_layout_strided_requires_grad_True_contiguous_True_cpu_float32
TestNestedTensorSubclassCPU.test_as_nested_tensor_from_tensor_dim_3_layout_strided_requires_grad_True_contiguous_True_cpu_float64
TestNestedTensorSubclassCPU.test_as_nested_tensor_from_tensor_dim_4_layout_jagged_requires_grad_True_contiguous_False_cpu_float16
TestNestedTensorSubclassCPU.test_as_nested_tensor_from_tensor_dim_4_layout_jagged_requires_grad_True_contiguous_False_cpu_float32
TestNestedTensorSubclassCPU.test_as_nested_tensor_from_tensor_dim_4_layout_jagged_requires_grad_True_contiguous_False_cpu_float64
TestNestedTensorSubclassCPU.test_as_nested_tensor_from_tensor_dim_4_layout_jagged_requires_grad_True_contiguous_True_cpu_float16
TestNestedTensorSubclassCPU.test_as_nested_tensor_from_tensor_dim_4_layout_jagged_requires_grad_True_contiguous_True_cpu_float32
TestNestedTensorSubclassCPU.test_as_nested_tensor_from_tensor_dim_4_layout_jagged_requires_grad_True_contiguous_True_cpu_float64
TestNestedTensorSubclassCPU.test_as_nested_tensor_from_tensor_dim_4_layout_strided_requires_grad_True_contiguous_False_cpu_float16
TestNestedTensorSubclassCPU.test_as_nested_tensor_from_tensor_dim_4_layout_strided_requires_grad_True_contiguous_False_cpu_float32
TestNestedTensorSubclassCPU.test_as_nested_tensor_from_tensor_dim_4_layout_strided_requires_grad_True_contiguous_False_cpu_float64
TestNestedTensorSubclassCPU.test_as_nested_tensor_from_tensor_dim_4_layout_strided_requires_grad_True_contiguous_True_cpu_float16
TestNestedTensorSubclassCPU.test_as_nested_tensor_from_tensor_dim_4_layout_strided_requires_grad_True_contiguous_True_cpu_float32
TestNestedTensorSubclassCPU.test_as_nested_tensor_from_tensor_dim_4_layout_strided_requires_grad_True_contiguous_True_cpu_float64
TestNestedTensorSubclassCPU.test_autograd_function_with_None_grad_cpu_float32
TestNestedTensorSubclassCPU.test_binary_pointwise_broadcasting_cpu
TestNestedTensorSubclassCPU.test_binary_pointwise_cpu
TestNestedTensorSubclassCPU.test_binary_pointwise_transposed_cpu
TestNestedTensorSubclassCPU.test_composite_op_in_inference_mode_cpu_float32
TestNestedTensorSubclassCPU.test_construction_from_list_cpu_float32
TestNestedTensorSubclassCPU.test_is_contiguous_cpu
TestNestedTensorSubclassCPU.test_jagged_layout_construction_as_nested_tensor_components_require_grad_True_cpu_float16
TestNestedTensorSubclassCPU.test_jagged_layout_construction_as_nested_tensor_components_require_grad_True_cpu_float32
TestNestedTensorSubclassCPU.test_jagged_layout_construction_as_nested_tensor_components_require_grad_True_cpu_float64
TestNestedTensorSubclassCPU.test_jagged_layout_construction_nested_tensor_requires_grad_False_components_require_grad_False_cpu_float16
TestNestedTensorSubclassCPU.test_jagged_layout_construction_nested_tensor_requires_grad_False_components_require_grad_False_cpu_float32
TestNestedTensorSubclassCPU.test_jagged_layout_construction_nested_tensor_requires_grad_False_components_require_grad_False_cpu_float64
TestNestedTensorSubclassCPU.test_jagged_layout_construction_nested_tensor_requires_grad_False_components_require_grad_True_cpu_float16
TestNestedTensorSubclassCPU.test_jagged_layout_construction_nested_tensor_requires_grad_False_components_require_grad_True_cpu_float32
TestNestedTensorSubclassCPU.test_jagged_layout_construction_nested_tensor_requires_grad_False_components_require_grad_True_cpu_float64
TestNestedTensorSubclassCPU.test_jagged_layout_construction_nested_tensor_requires_grad_True_components_require_grad_False_cpu_float16
TestNestedTensorSubclassCPU.test_jagged_layout_construction_nested_tensor_requires_grad_True_components_require_grad_False_cpu_float32
TestNestedTensorSubclassCPU.test_jagged_layout_construction_nested_tensor_requires_grad_True_components_require_grad_False_cpu_float64
TestNestedTensorSubclassCPU.test_jagged_layout_construction_nested_tensor_requires_grad_True_components_require_grad_True_cpu_float16
TestNestedTensorSubclassCPU.test_jagged_layout_construction_nested_tensor_requires_grad_True_components_require_grad_True_cpu_float32
TestNestedTensorSubclassCPU.test_jagged_layout_construction_nested_tensor_requires_grad_True_components_require_grad_True_cpu_float64
TestNestedTensorSubclassCPU.test_jagged_op_different_output_shape_dim_mean_keepdim_True_requires_grad_False_components_require_grad_False_cpu_float32
TestNestedTensorSubclassCPU.test_jagged_op_different_output_shape_dim_mean_keepdim_True_requires_grad_False_components_require_grad_True_cpu_float32
TestNestedTensorSubclassCPU.test_jagged_op_different_output_shape_dim_mean_keepdim_True_requires_grad_True_components_require_grad_False_cpu_float32
TestNestedTensorSubclassCPU.test_jagged_op_different_output_shape_dim_mean_keepdim_True_requires_grad_True_components_require_grad_True_cpu_float32
TestNestedTensorSubclassCPU.test_jagged_op_different_output_shape_dim_sum_keepdim_False_requires_grad_False_components_require_grad_False_cpu_float32
TestNestedTensorSubclassCPU.test_jagged_op_different_output_shape_dim_sum_keepdim_False_requires_grad_False_components_require_grad_True_cpu_float32
TestNestedTensorSubclassCPU.test_jagged_op_different_output_shape_dim_sum_keepdim_False_requires_grad_True_components_require_grad_False_cpu_float32
TestNestedTensorSubclassCPU.test_jagged_op_different_output_shape_dim_sum_keepdim_False_requires_grad_True_components_require_grad_True_cpu_float32
TestNestedTensorSubclassCPU.test_jagged_op_different_output_shape_dim_sum_keepdim_True_requires_grad_False_components_require_grad_False_cpu_float32
TestNestedTensorSubclassCPU.test_jagged_op_different_output_shape_dim_sum_keepdim_True_requires_grad_False_components_require_grad_True_cpu_float32
TestNestedTensorSubclassCPU.test_jagged_op_different_output_shape_dim_sum_keepdim_True_requires_grad_True_components_require_grad_False_cpu_float32
TestNestedTensorSubclassCPU.test_jagged_op_different_output_shape_dim_sum_keepdim_True_requires_grad_True_components_require_grad_True_cpu_float32
TestNestedTensorSubclassCPU.test_jagged_view_from_values_offsets_requires_grad_True_values_is_view_False_cpu_float16
TestNestedTensorSubclassCPU.test_jagged_view_from_values_offsets_requires_grad_True_values_is_view_False_cpu_float32
TestNestedTensorSubclassCPU.test_jagged_view_from_values_offsets_requires_grad_True_values_is_view_False_cpu_float64
TestNestedTensorSubclassCPU.test_jagged_view_from_values_offsets_requires_grad_True_values_is_view_True_cpu_float16
TestNestedTensorSubclassCPU.test_jagged_view_from_values_offsets_requires_grad_True_values_is_view_True_cpu_float32
TestNestedTensorSubclassCPU.test_jagged_view_from_values_offsets_requires_grad_True_values_is_view_True_cpu_float64
TestNestedTensorSubclassCPU.test_layer_norm_2d_input_requires_grad_False_components_require_grad_False_cpu_float32
TestNestedTensorSubclassCPU.test_layer_norm_2d_input_requires_grad_False_components_require_grad_True_cpu_float32
TestNestedTensorSubclassCPU.test_layer_norm_2d_input_requires_grad_True_components_require_grad_False_cpu_float32
TestNestedTensorSubclassCPU.test_layer_norm_2d_input_requires_grad_True_components_require_grad_True_cpu_float32
TestNestedTensorSubclassCPU.test_layer_norm_operate_on_batch_dim_requires_grad_False_components_require_grad_False_cpu_float32
TestNestedTensorSubclassCPU.test_layer_norm_operate_on_batch_dim_requires_grad_False_components_require_grad_True_cpu_float32
TestNestedTensorSubclassCPU.test_layer_norm_operate_on_batch_dim_requires_grad_True_components_require_grad_False_cpu_float32
TestNestedTensorSubclassCPU.test_layer_norm_operate_on_batch_dim_requires_grad_True_components_require_grad_True_cpu_float32
TestNestedTensorSubclassCPU.test_layer_norm_reduce_ragged_idx_1_requires_grad_True_components_require_grad_False_cpu_float32
TestNestedTensorSubclassCPU.test_layer_norm_reduce_ragged_idx_1_requires_grad_True_components_require_grad_True_cpu_float32
TestNestedTensorSubclassCPU.test_linear_nt_dim_3_cpu
TestNestedTensorSubclassCPU.test_linear_nt_dim_4_cpu
TestNestedTensorSubclassCPU.test_linear_nt_dim_5_cpu
TestNestedTensorSubclassCPU.test_njt_cat_cpu
TestNestedTensorSubclassCPU.test_op_dim_reduce_batch_only_different_output_shape_mean_keepdim_True_requires_grad_False_components_require_grad_False_cpu_float32
TestNestedTensorSubclassCPU.test_op_dim_reduce_batch_only_different_output_shape_mean_keepdim_True_requires_grad_False_components_require_grad_True_cpu_float32
TestNestedTensorSubclassCPU.test_op_dim_reduce_batch_only_different_output_shape_mean_keepdim_True_requires_grad_True_components_require_grad_False_cpu_float32
TestNestedTensorSubclassCPU.test_op_dim_reduce_batch_only_different_output_shape_mean_keepdim_True_requires_grad_True_components_require_grad_True_cpu_float32
TestNestedTensorSubclassCPU.test_op_dim_reduce_batch_only_different_output_shape_sum_keepdim_False_requires_grad_False_components_require_grad_False_cpu_float32
TestNestedTensorSubclassCPU.test_op_dim_reduce_batch_only_different_output_shape_sum_keepdim_False_requires_grad_False_components_require_grad_True_cpu_float32
TestNestedTensorSubclassCPU.test_op_dim_reduce_batch_only_different_output_shape_sum_keepdim_False_requires_grad_True_components_require_grad_False_cpu_float32
TestNestedTensorSubclassCPU.test_op_dim_reduce_batch_only_different_output_shape_sum_keepdim_False_requires_grad_True_components_require_grad_True_cpu_float32
TestNestedTensorSubclassCPU.test_op_dim_reduce_batch_only_different_output_shape_sum_keepdim_True_requires_grad_False_components_require_grad_False_cpu_float32
TestNestedTensorSubclassCPU.test_op_dim_reduce_batch_only_different_output_shape_sum_keepdim_True_requires_grad_False_components_require_grad_True_cpu_float32
TestNestedTensorSubclassCPU.test_op_dim_reduce_batch_only_different_output_shape_sum_keepdim_True_requires_grad_True_components_require_grad_False_cpu_float32
TestNestedTensorSubclassCPU.test_op_dim_reduce_batch_only_different_output_shape_sum_keepdim_True_requires_grad_True_components_require_grad_True_cpu_float32
TestNestedTensorSubclassCPU.test_op_dim_reduce_ragged_idx_1_different_output_shape_mean_keepdim_True_requires_grad_False_components_require_grad_False_cpu_float32
TestNestedTensorSubclassCPU.test_op_dim_reduce_ragged_idx_1_different_output_shape_mean_keepdim_True_requires_grad_False_components_require_grad_True_cpu_float32
TestNestedTensorSubclassCPU.test_op_dim_reduce_ragged_idx_1_different_output_shape_mean_keepdim_True_requires_grad_True_components_require_grad_False_cpu_float32
TestNestedTensorSubclassCPU.test_op_dim_reduce_ragged_idx_1_different_output_shape_mean_keepdim_True_requires_grad_True_components_require_grad_True_cpu_float32
TestNestedTensorSubclassCPU.test_op_dim_reduce_ragged_idx_1_different_output_shape_sum_keepdim_False_requires_grad_False_components_require_grad_False_cpu_float32
TestNestedTensorSubclassCPU.test_op_dim_reduce_ragged_idx_1_different_output_shape_sum_keepdim_False_requires_grad_False_components_require_grad_True_cpu_float32
TestNestedTensorSubclassCPU.test_op_dim_reduce_ragged_idx_1_different_output_shape_sum_keepdim_False_requires_grad_True_components_require_grad_False_cpu_float32
TestNestedTensorSubclassCPU.test_op_dim_reduce_ragged_idx_1_different_output_shape_sum_keepdim_False_requires_grad_True_components_require_grad_True_cpu_float32
TestNestedTensorSubclassCPU.test_op_dim_reduce_ragged_idx_1_different_output_shape_sum_keepdim_True_requires_grad_False_components_require_grad_False_cpu_float32
TestNestedTensorSubclassCPU.test_op_dim_reduce_ragged_idx_1_different_output_shape_sum_keepdim_True_requires_grad_False_components_require_grad_True_cpu_float32
TestNestedTensorSubclassCPU.test_op_dim_reduce_ragged_idx_1_different_output_shape_sum_keepdim_True_requires_grad_True_components_require_grad_False_cpu_float32
TestNestedTensorSubclassCPU.test_op_dim_reduce_ragged_idx_1_different_output_shape_sum_keepdim_True_requires_grad_True_components_require_grad_True_cpu_float32
TestNestedTensorSubclassCPU.test_op_dim_transpose_non_ragged_dim_different_output_shape_mean_keepdim_True_requires_grad_False_components_require_grad_False_cpu_float32
TestNestedTensorSubclassCPU.test_op_dim_transpose_non_ragged_dim_different_output_shape_mean_keepdim_True_requires_grad_False_components_require_grad_True_cpu_float32
TestNestedTensorSubclassCPU.test_op_dim_transpose_non_ragged_dim_different_output_shape_mean_keepdim_True_requires_grad_True_components_require_grad_False_cpu_float32
TestNestedTensorSubclassCPU.test_op_dim_transpose_non_ragged_dim_different_output_shape_mean_keepdim_True_requires_grad_True_components_require_grad_True_cpu_float32
TestNestedTensorSubclassCPU.test_op_dim_transpose_non_ragged_dim_different_output_shape_sum_keepdim_False_requires_grad_False_components_require_grad_False_cpu_float32
TestNestedTensorSubclassCPU.test_op_dim_transpose_non_ragged_dim_different_output_shape_sum_keepdim_False_requires_grad_False_components_require_grad_True_cpu_float32
TestNestedTensorSubclassCPU.test_op_dim_transpose_non_ragged_dim_different_output_shape_sum_keepdim_False_requires_grad_True_components_require_grad_False_cpu_float32
TestNestedTensorSubclassCPU.test_op_dim_transpose_non_ragged_dim_different_output_shape_sum_keepdim_False_requires_grad_True_components_require_grad_True_cpu_float32
TestNestedTensorSubclassCPU.test_op_dim_transpose_non_ragged_dim_different_output_shape_sum_keepdim_True_requires_grad_False_components_require_grad_False_cpu_float32
TestNestedTensorSubclassCPU.test_op_dim_transpose_non_ragged_dim_different_output_shape_sum_keepdim_True_requires_grad_False_components_require_grad_True_cpu_float32
TestNestedTensorSubclassCPU.test_op_dim_transpose_non_ragged_dim_different_output_shape_sum_keepdim_True_requires_grad_True_components_require_grad_False_cpu_float32
TestNestedTensorSubclassCPU.test_op_dim_transpose_non_ragged_dim_different_output_shape_sum_keepdim_True_requires_grad_True_components_require_grad_True_cpu_float32
TestNestedTensorSubclassCPU.test_op_dim_with_lengths_different_output_shape_mean_keepdim_True_requires_grad_False_components_require_grad_False_cpu_float32
TestNestedTensorSubclassCPU.test_op_dim_with_lengths_different_output_shape_mean_keepdim_True_requires_grad_False_components_require_grad_True_cpu_float32
TestNestedTensorSubclassCPU.test_op_dim_with_lengths_different_output_shape_mean_keepdim_True_requires_grad_True_components_require_grad_False_cpu_float32
TestNestedTensorSubclassCPU.test_op_dim_with_lengths_different_output_shape_mean_keepdim_True_requires_grad_True_components_require_grad_True_cpu_float32
TestNestedTensorSubclassCPU.test_op_dim_with_lengths_different_output_shape_sum_keepdim_False_requires_grad_False_components_require_grad_False_cpu_float32
TestNestedTensorSubclassCPU.test_op_dim_with_lengths_different_output_shape_sum_keepdim_False_requires_grad_False_components_require_grad_True_cpu_float32
TestNestedTensorSubclassCPU.test_op_dim_with_lengths_different_output_shape_sum_keepdim_False_requires_grad_True_components_require_grad_False_cpu_float32
TestNestedTensorSubclassCPU.test_op_dim_with_lengths_different_output_shape_sum_keepdim_False_requires_grad_True_components_require_grad_True_cpu_float32
TestNestedTensorSubclassCPU.test_op_dim_with_lengths_different_output_shape_sum_keepdim_True_requires_grad_False_components_require_grad_False_cpu_float32
TestNestedTensorSubclassCPU.test_op_dim_with_lengths_different_output_shape_sum_keepdim_True_requires_grad_False_components_require_grad_True_cpu_float32
TestNestedTensorSubclassCPU.test_op_dim_with_lengths_different_output_shape_sum_keepdim_True_requires_grad_True_components_require_grad_False_cpu_float32
TestNestedTensorSubclassCPU.test_op_dim_with_lengths_different_output_shape_sum_keepdim_True_requires_grad_True_components_require_grad_True_cpu_float32
TestNestedTensorSubclassCPU.test_reshape_decomp_requires_grad_True_cpu
TestNestedTensorSubclassCPU.test_sdpa_cpu_bfloat16
TestNestedTensorSubclassCPU.test_sdpa_cpu_float16
TestNestedTensorSubclassCPU.test_sdpa_cpu_float32
TestNestedTensorSubclassCPU.test_sdpa_with_constant_sequence_length_cpu_float16
TestNestedTensorSubclassCPU.test_sdpa_with_constant_sequence_length_cpu_float32
TestNestedTensorSubclassCPU.test_sdpa_with_constant_sequence_length_cpu_float64
TestNestedTensorSubclassCPU.test_softmax_cpu
TestNestedTensorSubclassCPU.test_softmax_dim_reduce_ragged_idx_1_requires_grad_False_components_require_grad_False_cpu_float32
TestNestedTensorSubclassCPU.test_softmax_dim_reduce_ragged_idx_1_requires_grad_False_components_require_grad_True_cpu_float32
TestNestedTensorSubclassCPU.test_softmax_dim_reduce_ragged_idx_1_requires_grad_True_components_require_grad_False_cpu_float32
TestNestedTensorSubclassCPU.test_softmax_dim_reduce_ragged_idx_1_requires_grad_True_components_require_grad_True_cpu_float32
TestNestedTensorSubclassCPU.test_softmax_dim_reduce_ragged_idx_greater_than_1_same_output_shape_transpose_offset_1_requires_grad_False_components_require_grad_False_cpu_float32
TestNestedTensorSubclassCPU.test_softmax_dim_reduce_ragged_idx_greater_than_1_same_output_shape_transpose_offset_1_requires_grad_False_components_require_grad_True_cpu_float32
TestNestedTensorSubclassCPU.test_softmax_dim_reduce_ragged_idx_greater_than_1_same_output_shape_transpose_offset_1_requires_grad_True_components_require_grad_False_cpu_float32
TestNestedTensorSubclassCPU.test_softmax_dim_reduce_ragged_idx_greater_than_1_same_output_shape_transpose_offset_1_requires_grad_True_components_require_grad_True_cpu_float32
TestNestedTensorSubclassCPU.test_softmax_dim_reduce_ragged_idx_greater_than_1_same_output_shape_transpose_offset_2_requires_grad_False_components_require_grad_False_cpu_float32
TestNestedTensorSubclassCPU.test_softmax_dim_reduce_ragged_idx_greater_than_1_same_output_shape_transpose_offset_2_requires_grad_False_components_require_grad_True_cpu_float32
TestNestedTensorSubclassCPU.test_softmax_dim_reduce_ragged_idx_greater_than_1_same_output_shape_transpose_offset_2_requires_grad_True_components_require_grad_False_cpu_float32
TestNestedTensorSubclassCPU.test_softmax_dim_reduce_ragged_idx_greater_than_1_same_output_shape_transpose_offset_2_requires_grad_True_components_require_grad_True_cpu_float32
TestNestedTensorSubclassCPU.test_softmax_dim_requires_grad_False_components_require_grad_False_cpu_float32
TestNestedTensorSubclassCPU.test_softmax_dim_requires_grad_False_components_require_grad_True_cpu_float32
TestNestedTensorSubclassCPU.test_softmax_dim_requires_grad_True_components_require_grad_False_cpu_float32
TestNestedTensorSubclassCPU.test_softmax_dim_requires_grad_True_components_require_grad_True_cpu_float32
TestNestedTensorSubclassCPU.test_softmax_dim_transpose_non_ragged_dim_requires_grad_False_components_require_grad_False_cpu_float32
TestNestedTensorSubclassCPU.test_softmax_dim_transpose_non_ragged_dim_requires_grad_False_components_require_grad_True_cpu_float32
TestNestedTensorSubclassCPU.test_softmax_dim_transpose_non_ragged_dim_requires_grad_True_components_require_grad_False_cpu_float32
TestNestedTensorSubclassCPU.test_softmax_dim_transpose_non_ragged_dim_requires_grad_True_components_require_grad_True_cpu_float32
TestNestedTensorSubclassCPU.test_softmax_dim_with_lengths_requires_grad_False_components_require_grad_False_cpu_float32
TestNestedTensorSubclassCPU.test_softmax_dim_with_lengths_requires_grad_False_components_require_grad_True_cpu_float32
TestNestedTensorSubclassCPU.test_softmax_dim_with_lengths_requires_grad_True_components_require_grad_False_cpu_float32
TestNestedTensorSubclassCPU.test_softmax_dim_with_lengths_requires_grad_True_components_require_grad_True_cpu_float32
TestNestedTensorSubclassCPU.test_softmax_reduce_batch_dim_requires_grad_False_components_require_grad_False_cpu_float32
TestNestedTensorSubclassCPU.test_softmax_reduce_batch_dim_requires_grad_False_components_require_grad_True_cpu_float32
TestNestedTensorSubclassCPU.test_softmax_reduce_batch_dim_requires_grad_True_components_require_grad_False_cpu_float32
TestNestedTensorSubclassCPU.test_softmax_reduce_batch_dim_requires_grad_True_components_require_grad_True_cpu_float32
TestNestedTensorSubclassCPU.test_squeeze_cpu
TestNestedTensorSubclassCPU.test_sum_dim_reduce_batch_and_non_batch_keepdim_False_requires_grad_False_components_require_grad_False_cpu_float32
TestNestedTensorSubclassCPU.test_sum_dim_reduce_batch_and_non_batch_keepdim_False_requires_grad_False_components_require_grad_True_cpu_float32
TestNestedTensorSubclassCPU.test_sum_dim_reduce_batch_and_non_batch_keepdim_False_requires_grad_True_components_require_grad_False_cpu_float32
TestNestedTensorSubclassCPU.test_sum_dim_reduce_batch_and_non_batch_keepdim_False_requires_grad_True_components_require_grad_True_cpu_float32
TestNestedTensorSubclassCPU.test_sum_dim_reduce_batch_and_non_batch_keepdim_True_requires_grad_False_components_require_grad_False_cpu_float32
TestNestedTensorSubclassCPU.test_sum_dim_reduce_batch_and_non_batch_keepdim_True_requires_grad_False_components_require_grad_True_cpu_float32
TestNestedTensorSubclassCPU.test_sum_dim_reduce_batch_and_non_batch_keepdim_True_requires_grad_True_components_require_grad_False_cpu_float32
TestNestedTensorSubclassCPU.test_sum_dim_reduce_batch_and_non_batch_keepdim_True_requires_grad_True_components_require_grad_True_cpu_float32
TestNestedTensorSubclassCPU.test_sum_dim_reduce_ragged_and_non_batch_keepdim_False_requires_grad_False_components_require_grad_False_cpu_float32
TestNestedTensorSubclassCPU.test_sum_dim_reduce_ragged_and_non_batch_keepdim_False_requires_grad_False_components_require_grad_True_cpu_float32
TestNestedTensorSubclassCPU.test_sum_dim_reduce_ragged_and_non_batch_keepdim_False_requires_grad_True_components_require_grad_False_cpu_float32
TestNestedTensorSubclassCPU.test_sum_dim_reduce_ragged_and_non_batch_keepdim_False_requires_grad_True_components_require_grad_True_cpu_float32
TestNestedTensorSubclassCPU.test_sum_dim_reduce_ragged_and_non_batch_keepdim_True_requires_grad_False_components_require_grad_False_cpu_float32
TestNestedTensorSubclassCPU.test_sum_dim_reduce_ragged_and_non_batch_keepdim_True_requires_grad_False_components_require_grad_True_cpu_float32
TestNestedTensorSubclassCPU.test_sum_dim_reduce_ragged_and_non_batch_keepdim_True_requires_grad_True_components_require_grad_False_cpu_float32
TestNestedTensorSubclassCPU.test_sum_dim_reduce_ragged_and_non_batch_keepdim_True_requires_grad_True_components_require_grad_True_cpu_float32
TestNestedTensorSubclassCPU.test_tensor_attributes_cpu
TestNestedTensorSubclassCPU.test_to_copy_cpu
TestNestedTensorSubclassCPU.test_to_padded_tensor_nt_dim_2_requires_grad_True_cpu_float16
TestNestedTensorSubclassCPU.test_to_padded_tensor_nt_dim_2_requires_grad_True_cpu_float32
TestNestedTensorSubclassCPU.test_to_padded_tensor_nt_dim_2_requires_grad_True_cpu_float64
TestNestedTensorSubclassCPU.test_to_padded_tensor_nt_dim_3_requires_grad_True_cpu_float16
TestNestedTensorSubclassCPU.test_to_padded_tensor_nt_dim_3_requires_grad_True_cpu_float32
TestNestedTensorSubclassCPU.test_to_padded_tensor_nt_dim_3_requires_grad_True_cpu_float64
TestNestedTensorSubclassCPU.test_to_padded_tensor_nt_dim_4_requires_grad_True_cpu_float16
TestNestedTensorSubclassCPU.test_to_padded_tensor_nt_dim_4_requires_grad_True_cpu_float32
TestNestedTensorSubclassCPU.test_to_padded_tensor_nt_dim_4_requires_grad_True_cpu_float64
TestNestedTensorSubclassCPU.test_unary_pointwise_cpu
TestNestedTensorSubclassCPU.test_unary_pointwise_transposed_inputs_cpu
TestNestedTensorSubclassCPU.test_unbind_backward_cpu_float16
TestNestedTensorSubclassCPU.test_unbind_backward_cpu_float32
TestNestedTensorSubclassCPU.test_unbind_backward_cpu_float64
TestNestedTensorSubclassCPU.test_unbind_cpu
TestNestedTensorSubclassCPU.test_unbind_transpose_ragged_idx_2_cpu
TestNestedTensorSubclassCPU.test_unbind_transpose_ragged_idx_3_cpu
TestNestedTensorSubclassCPU.test_unbind_transpose_ragged_idx_last_dim_cpu
TestNestedTensorSubclassCUDA.test_as_nested_tensor_from_tensor_dim_2_layout_jagged_requires_grad_True_contiguous_False_cuda_float16
TestNestedTensorSubclassCUDA.test_as_nested_tensor_from_tensor_dim_2_layout_jagged_requires_grad_True_contiguous_False_cuda_float32
TestNestedTensorSubclassCUDA.test_as_nested_tensor_from_tensor_dim_2_layout_jagged_requires_grad_True_contiguous_False_cuda_float64
TestNestedTensorSubclassCUDA.test_as_nested_tensor_from_tensor_dim_2_layout_jagged_requires_grad_True_contiguous_True_cuda_float16
TestNestedTensorSubclassCUDA.test_as_nested_tensor_from_tensor_dim_2_layout_jagged_requires_grad_True_contiguous_True_cuda_float32
TestNestedTensorSubclassCUDA.test_as_nested_tensor_from_tensor_dim_2_layout_jagged_requires_grad_True_contiguous_True_cuda_float64
TestNestedTensorSubclassCUDA.test_as_nested_tensor_from_tensor_dim_2_layout_strided_requires_grad_True_contiguous_False_cuda_float16
TestNestedTensorSubclassCUDA.test_as_nested_tensor_from_tensor_dim_2_layout_strided_requires_grad_True_contiguous_False_cuda_float32
TestNestedTensorSubclassCUDA.test_as_nested_tensor_from_tensor_dim_2_layout_strided_requires_grad_True_contiguous_False_cuda_float64
TestNestedTensorSubclassCUDA.test_as_nested_tensor_from_tensor_dim_2_layout_strided_requires_grad_True_contiguous_True_cuda_float16
TestNestedTensorSubclassCUDA.test_as_nested_tensor_from_tensor_dim_2_layout_strided_requires_grad_True_contiguous_True_cuda_float32
TestNestedTensorSubclassCUDA.test_as_nested_tensor_from_tensor_dim_2_layout_strided_requires_grad_True_contiguous_True_cuda_float64
TestNestedTensorSubclassCUDA.test_as_nested_tensor_from_tensor_dim_3_layout_jagged_requires_grad_True_contiguous_False_cuda_float16
TestNestedTensorSubclassCUDA.test_as_nested_tensor_from_tensor_dim_3_layout_jagged_requires_grad_True_contiguous_False_cuda_float32
TestNestedTensorSubclassCUDA.test_as_nested_tensor_from_tensor_dim_3_layout_jagged_requires_grad_True_contiguous_False_cuda_float64
TestNestedTensorSubclassCUDA.test_as_nested_tensor_from_tensor_dim_3_layout_jagged_requires_grad_True_contiguous_True_cuda_float16
TestNestedTensorSubclassCUDA.test_as_nested_tensor_from_tensor_dim_3_layout_jagged_requires_grad_True_contiguous_True_cuda_float32
TestNestedTensorSubclassCUDA.test_as_nested_tensor_from_tensor_dim_3_layout_jagged_requires_grad_True_contiguous_True_cuda_float64
TestNestedTensorSubclassCUDA.test_as_nested_tensor_from_tensor_dim_3_layout_strided_requires_grad_True_contiguous_False_cuda_float16
TestNestedTensorSubclassCUDA.test_as_nested_tensor_from_tensor_dim_3_layout_strided_requires_grad_True_contiguous_False_cuda_float32
TestNestedTensorSubclassCUDA.test_as_nested_tensor_from_tensor_dim_3_layout_strided_requires_grad_True_contiguous_False_cuda_float64
TestNestedTensorSubclassCUDA.test_as_nested_tensor_from_tensor_dim_3_layout_strided_requires_grad_True_contiguous_True_cuda_float16
TestNestedTensorSubclassCUDA.test_as_nested_tensor_from_tensor_dim_3_layout_strided_requires_grad_True_contiguous_True_cuda_float32
TestNestedTensorSubclassCUDA.test_as_nested_tensor_from_tensor_dim_3_layout_strided_requires_grad_True_contiguous_True_cuda_float64
TestNestedTensorSubclassCUDA.test_as_nested_tensor_from_tensor_dim_4_layout_jagged_requires_grad_True_contiguous_False_cuda_float16
TestNestedTensorSubclassCUDA.test_as_nested_tensor_from_tensor_dim_4_layout_jagged_requires_grad_True_contiguous_False_cuda_float32
TestNestedTensorSubclassCUDA.test_as_nested_tensor_from_tensor_dim_4_layout_jagged_requires_grad_True_contiguous_False_cuda_float64
TestNestedTensorSubclassCUDA.test_as_nested_tensor_from_tensor_dim_4_layout_jagged_requires_grad_True_contiguous_True_cuda_float16
TestNestedTensorSubclassCUDA.test_as_nested_tensor_from_tensor_dim_4_layout_jagged_requires_grad_True_contiguous_True_cuda_float32
TestNestedTensorSubclassCUDA.test_as_nested_tensor_from_tensor_dim_4_layout_jagged_requires_grad_True_contiguous_True_cuda_float64
TestNestedTensorSubclassCUDA.test_as_nested_tensor_from_tensor_dim_4_layout_strided_requires_grad_True_contiguous_False_cuda_float16
TestNestedTensorSubclassCUDA.test_as_nested_tensor_from_tensor_dim_4_layout_strided_requires_grad_True_contiguous_False_cuda_float32
TestNestedTensorSubclassCUDA.test_as_nested_tensor_from_tensor_dim_4_layout_strided_requires_grad_True_contiguous_False_cuda_float64
TestNestedTensorSubclassCUDA.test_as_nested_tensor_from_tensor_dim_4_layout_strided_requires_grad_True_contiguous_True_cuda_float16
TestNestedTensorSubclassCUDA.test_as_nested_tensor_from_tensor_dim_4_layout_strided_requires_grad_True_contiguous_True_cuda_float32
TestNestedTensorSubclassCUDA.test_as_nested_tensor_from_tensor_dim_4_layout_strided_requires_grad_True_contiguous_True_cuda_float64
TestNestedTensorSubclassCUDA.test_autograd_function_with_None_grad_cuda_float32
TestNestedTensorSubclassCUDA.test_binary_pointwise_broadcasting_cuda
TestNestedTensorSubclassCUDA.test_binary_pointwise_cuda
TestNestedTensorSubclassCUDA.test_binary_pointwise_transposed_cuda
TestNestedTensorSubclassCUDA.test_chunk_cuda
TestNestedTensorSubclassCUDA.test_flex_attention_converts_stacked_seq_indices_cuda_float32
TestNestedTensorSubclassCUDA.test_flex_attention_noncontig_with_holes_False_cross_attention_False_cuda_float32
TestNestedTensorSubclassCUDA.test_flex_attention_noncontig_with_holes_False_cross_attention_True_cuda_float32
TestNestedTensorSubclassCUDA.test_jagged_layout_construction_as_nested_tensor_components_require_grad_True_cuda_float16
TestNestedTensorSubclassCUDA.test_jagged_layout_construction_as_nested_tensor_components_require_grad_True_cuda_float32
TestNestedTensorSubclassCUDA.test_jagged_layout_construction_as_nested_tensor_components_require_grad_True_cuda_float64
TestNestedTensorSubclassCUDA.test_jagged_layout_construction_nested_tensor_requires_grad_True_components_require_grad_False_cuda_float16
TestNestedTensorSubclassCUDA.test_jagged_layout_construction_nested_tensor_requires_grad_True_components_require_grad_False_cuda_float32
TestNestedTensorSubclassCUDA.test_jagged_layout_construction_nested_tensor_requires_grad_True_components_require_grad_False_cuda_float64
TestNestedTensorSubclassCUDA.test_jagged_layout_construction_nested_tensor_requires_grad_True_components_require_grad_True_cuda_float16
TestNestedTensorSubclassCUDA.test_jagged_layout_construction_nested_tensor_requires_grad_True_components_require_grad_True_cuda_float32
TestNestedTensorSubclassCUDA.test_jagged_layout_construction_nested_tensor_requires_grad_True_components_require_grad_True_cuda_float64
TestNestedTensorSubclassCUDA.test_jagged_view_from_values_offsets_requires_grad_True_values_is_view_False_cuda_float16
TestNestedTensorSubclassCUDA.test_jagged_view_from_values_offsets_requires_grad_True_values_is_view_False_cuda_float32
TestNestedTensorSubclassCUDA.test_jagged_view_from_values_offsets_requires_grad_True_values_is_view_False_cuda_float64
TestNestedTensorSubclassCUDA.test_jagged_view_from_values_offsets_requires_grad_True_values_is_view_True_cuda_float16
TestNestedTensorSubclassCUDA.test_jagged_view_from_values_offsets_requires_grad_True_values_is_view_True_cuda_float32
TestNestedTensorSubclassCUDA.test_jagged_view_from_values_offsets_requires_grad_True_values_is_view_True_cuda_float64
TestNestedTensorSubclassCUDA.test_linear_backward_memory_usage_cuda_float32
TestNestedTensorSubclassCUDA.test_linear_nt_dim_3_cuda
TestNestedTensorSubclassCUDA.test_linear_nt_dim_4_cuda
TestNestedTensorSubclassCUDA.test_linear_nt_dim_5_cuda
TestNestedTensorSubclassCUDA.test_njt_cat_cuda
TestNestedTensorSubclassCUDA.test_reshape_decomp_requires_grad_True_cuda
TestNestedTensorSubclassCUDA.test_sdpa_cuda_bfloat16
TestNestedTensorSubclassCUDA.test_sdpa_cuda_float16
TestNestedTensorSubclassCUDA.test_sdpa_cuda_float32
TestNestedTensorSubclassCUDA.test_sdpa_with_constant_sequence_length_cuda_float16
TestNestedTensorSubclassCUDA.test_sdpa_with_constant_sequence_length_cuda_float32
TestNestedTensorSubclassCUDA.test_sdpa_with_constant_sequence_length_cuda_float64
TestNestedTensorSubclassCUDA.test_softmax_cuda
TestNestedTensorSubclassCUDA.test_to_padded_tensor_nt_dim_2_requires_grad_True_cuda_float16
TestNestedTensorSubclassCUDA.test_to_padded_tensor_nt_dim_2_requires_grad_True_cuda_float32
TestNestedTensorSubclassCUDA.test_to_padded_tensor_nt_dim_2_requires_grad_True_cuda_float64
TestNestedTensorSubclassCUDA.test_to_padded_tensor_nt_dim_3_requires_grad_True_cuda_float16
TestNestedTensorSubclassCUDA.test_to_padded_tensor_nt_dim_3_requires_grad_True_cuda_float32
TestNestedTensorSubclassCUDA.test_to_padded_tensor_nt_dim_3_requires_grad_True_cuda_float64
TestNestedTensorSubclassCUDA.test_to_padded_tensor_nt_dim_4_requires_grad_True_cuda_float16
TestNestedTensorSubclassCUDA.test_to_padded_tensor_nt_dim_4_requires_grad_True_cuda_float32
TestNestedTensorSubclassCUDA.test_to_padded_tensor_nt_dim_4_requires_grad_True_cuda_float64
TestNestedTensorSubclassCUDA.test_unary_pointwise_cuda
TestNestedTensorSubclassCUDA.test_unary_pointwise_transposed_inputs_cuda
TestNestedTensorSubclassCUDA.test_unbind_backward_cuda_float16
TestNestedTensorSubclassCUDA.test_unbind_backward_cuda_float32
TestNestedTensorSubclassCUDA.test_unbind_backward_cuda_float64
TestNN.test_AdaptiveLogSoftmax
TestNN.test_affine_grid
TestNN.test_affine_grid_3d
TestNN.test_bilinear_no_bias
TestNN.test_Conv1d_circular_stride2_pad2
TestNN.test_Conv1d_reflect_stride2_pad2
TestNN.test_Conv1d_replicate_stride2_pad2
TestNN.test_Conv2d_circular_stride2_pad2
TestNN.test_Conv2d_reflect_stride2_pad2
TestNN.test_Conv2d_replicate_stride2_pad2
TestNN.test_Conv3d_circular_stride2_pad2
TestNN.test_Conv3d_replicate_stride2_pad2
TestNN.test_cosine_embedding_loss_margin_no_reduce
TestNN.test_cosine_embedding_loss_no_reduce
TestNN.test_CosineEmbeddingLoss_no_batch_dim_mean
TestNN.test_CosineEmbeddingLoss_no_batch_dim_none
TestNN.test_CosineEmbeddingLoss_no_batch_dim_sum
TestNN.test_elu_inplace_on_view
TestNN.test_Embedding_sparse
TestNN.test_EmbeddingBag_sparse
TestNN.test_grid_sample
TestNN.test_HingeEmbeddingLoss_margin_no_reduce
TestNN.test_HingeEmbeddingLoss_no_batch_dim_mean
TestNN.test_HingeEmbeddingLoss_no_batch_dim_none
TestNN.test_HingeEmbeddingLoss_no_batch_dim_sum
TestNN.test_HingeEmbeddingLoss_no_reduce
TestNN.test_interpolate
TestNN.test_Linear
TestNN.test_linear_autograd_device_cpu_bias_weightCOO
TestNN.test_linear_autograd_device_cpu_bias_weightCSC
TestNN.test_linear_autograd_device_cpu_bias_weightCSR
TestNN.test_linear_autograd_device_cpu_nobias_weightCOO
TestNN.test_linear_autograd_device_cpu_nobias_weightCSC
TestNN.test_linear_autograd_device_cpu_nobias_weightCSR
TestNN.test_Linear_no_batch_dim
TestNN.test_Linear_no_bias
TestNN.test_MarginRankingLoss_no_batch_dim_mean
TestNN.test_MarginRankingLoss_no_batch_dim_none
TestNN.test_MarginRankingLoss_no_batch_dim_sum
TestNN.test_Mish_no_batch_dim
TestNN.test_MultiLabelSoftMarginLoss_no_batch_dim_mean
TestNN.test_MultiLabelSoftMarginLoss_no_batch_dim_none
TestNN.test_MultiLabelSoftMarginLoss_no_batch_dim_sum
TestNN.test_MultiLabelSoftMarginLoss_no_reduce
TestNN.test_MultiLabelSoftMarginLoss_weights_no_reduce
TestNN.test_normalize
TestNN.test_PairwiseDistance
TestNN.test_PairwiseDistance_broadcast_lhs
TestNN.test_PairwiseDistance_broadcast_rhs
TestNN.test_PairwiseDistance_no_batch_dim
TestNN.test_PairwiseDistance_with_non_default_args
TestNN.test_PoissonNLLLoss_no_batch_dim_mean
TestNN.test_PoissonNLLLoss_no_batch_dim_none
TestNN.test_PoissonNLLLoss_no_batch_dim_sum
TestNN.test_PoissonNLLLoss_no_reduce
TestNN.test_relu_inplace_on_view
TestNN.test_SiLU_no_batch_dim
TestNN.test_Softsign_no_batch_dim
TestNN.test_swap_module_params_poisons_acc_grad
TestNN.test_Tanhshrink_no_batch_dim
TestNN.test_Transformer_multilayer_coder
TestNN.test_TransformerDecoderLayer_gelu_activation
TestNN.test_TransformerDecoderLayer_relu_activation
TestNN.test_TransformerEncoderLayer_gelu_activation
TestNN.test_TransformerEncoderLayer_relu_activation
TestNN.test_triplet_margin_loss
TestNN.test_triplet_margin_loss_no_reduce
TestNN.test_triplet_margin_loss_swap
TestNN.test_triplet_margin_loss_swap_no_reduce
TestNN.test_TripletMarginLoss_no_batch_dim_mean
TestNN.test_TripletMarginLoss_no_batch_dim_none
TestNN.test_TripletMarginLoss_no_batch_dim_sum
TestNN.test_upsampling_bfloat16
TestNN.test_upsampling_not_recompute_scale_factor
TestNN.test_upsamplingLinear1d
TestNNDeviceTypeCPU.test_elu_inplace_with_neg_alpha_cpu
TestNNDeviceTypeCPU.test_GroupNorm_general_cpu
TestNNDeviceTypeCPU.test_groupnorm_nhwc_cpu_bfloat16
TestNNDeviceTypeCPU.test_groupnorm_nhwc_cpu_float16
TestNNDeviceTypeCPU.test_groupnorm_nhwc_cpu_float32
TestNNDeviceTypeCPU.test_groupnorm_nhwc_cpu_float64
TestNNDeviceTypeCPU.test_GRU_grad_and_gradgrad_cpu_float64
TestNNDeviceTypeCPU.test_leaky_relu_inplace_with_neg_slope_cpu
TestNNDeviceTypeCPU.test_LSTM_differentiable_backward_using_oneDNN_cpu_bfloat16
TestNNDeviceTypeCPU.test_LSTM_differentiable_backward_using_oneDNN_cpu_float32
TestNNDeviceTypeCPU.test_LSTM_grad_and_gradgrad_cpu_float64
TestNNDeviceTypeCPU.test_triplet_margin_with_distance_loss_cpu
TestNNDeviceTypeCPU.test_triplet_margin_with_distance_loss_default_parity_cpu
TestNNDeviceTypeCPU.test_upsamplingBiMode2d_antialias_False_align_corners_False_mode_bicubic_memory_format0_cpu
TestNNDeviceTypeCPU.test_upsamplingBiMode2d_antialias_False_align_corners_False_mode_bicubic_memory_format1_cpu
TestNNDeviceTypeCPU.test_upsamplingBiMode2d_antialias_False_align_corners_False_mode_bilinear_memory_format0_cpu
TestNNDeviceTypeCPU.test_upsamplingBiMode2d_antialias_False_align_corners_False_mode_bilinear_memory_format1_cpu
TestNNDeviceTypeCPU.test_upsamplingBiMode2d_antialias_False_align_corners_True_mode_bicubic_memory_format0_cpu
TestNNDeviceTypeCPU.test_upsamplingBiMode2d_antialias_False_align_corners_True_mode_bicubic_memory_format1_cpu
TestNNDeviceTypeCPU.test_upsamplingBiMode2d_antialias_False_align_corners_True_mode_bilinear_memory_format0_cpu
TestNNDeviceTypeCPU.test_upsamplingBiMode2d_antialias_False_align_corners_True_mode_bilinear_memory_format1_cpu
TestNNDeviceTypeCPU.test_upsamplingBiMode2d_antialias_True_align_corners_False_mode_bicubic_memory_format0_cpu
TestNNDeviceTypeCPU.test_upsamplingBiMode2d_antialias_True_align_corners_False_mode_bicubic_memory_format1_cpu
TestNNDeviceTypeCPU.test_upsamplingBiMode2d_antialias_True_align_corners_True_mode_bicubic_memory_format0_cpu
TestNNDeviceTypeCPU.test_upsamplingBiMode2d_antialias_True_align_corners_True_mode_bicubic_memory_format1_cpu
TestNNDeviceTypeCPU.test_upsamplingNearest1d_mode_nearest-exact_cpu
TestNNDeviceTypeCPU.test_upsamplingNearest1d_mode_nearest_cpu
TestNNDeviceTypeCPU.test_upsamplingTrilinear3d_align_corners_False_memory_format0_cpu
TestNNDeviceTypeCPU.test_upsamplingTrilinear3d_align_corners_False_memory_format1_cpu
TestNNDeviceTypeCPU.test_upsamplingTrilinear3d_align_corners_True_memory_format0_cpu
TestNNDeviceTypeCPU.test_upsamplingTrilinear3d_align_corners_True_memory_format1_cpu
TestNNParametrization.test_new_spectral_norm_swap_False
TestOpenReg.test_autograd_init
TestPoolingNNDeviceTypeCPU.test_adaptive_pooling_empty_output_size_cpu_float32
TestPoolingNNDeviceTypeCPU.test_adaptive_pooling_empty_output_size_cpu_float64
TestPoolingNNDeviceTypeCPU.test_max_pool3d_ndhwc_cpu_bfloat16
TestPoolingNNDeviceTypeCPU.test_max_pool3d_ndhwc_cpu_float16
TestPoolingNNDeviceTypeCPU.test_max_pool3d_ndhwc_cpu_float32
TestPoolingNNDeviceTypeCPU.test_max_pool3d_ndhwc_cpu_float64
TestProfiler.test_profiler_fwd_bwd_link
TestProfiler.test_source
TestPythonDispatch.test_custom_autograd
TestPythonDispatch.test_shallow_copy_and_detach
TestPythonDispatch.test_subclass_autograd_device_check
TestReductions.test_amax_grad
TestReductions.test_amin_grad
TestReductions.test_backward
TestReductions.test_mean_dim_grad
TestReductions.test_mean_grad_case_1a
TestReductions.test_mean_grad_case_1b
TestReductions.test_mean_grad_case_1e
TestReductions.test_prod_grad
TestReductions.test_sum_grad
TestScatterGatherCPU.test_gather_backward_with_empty_index_tensor_sparse_grad_True_cpu_float32
TestScatterGatherCPU.test_gather_backward_with_empty_index_tensor_sparse_grad_True_cpu_float64
TestScript.test_cat
TestScript.test_linear_grad
TestScript.test_mm_batching
TestScript.test_stack
TestSelectiveActivationCheckpoint.test_flops_and_mem
TestSparseAnyCPU.test_as_sparse_gradcheck_SparseBSC_masked_fast_cpu
TestSparseAnyCPU.test_as_sparse_gradcheck_SparseBSC_masked_slow_cpu
TestSparseAnyCPU.test_as_sparse_gradcheck_SparseBSC_nonmasked_fast_cpu
TestSparseAnyCPU.test_as_sparse_gradcheck_SparseBSC_nonmasked_slow_cpu
TestSparseAnyCPU.test_as_sparse_gradcheck_SparseBSR_masked_fast_cpu
TestSparseAnyCPU.test_as_sparse_gradcheck_SparseBSR_masked_slow_cpu
TestSparseAnyCPU.test_as_sparse_gradcheck_SparseBSR_nonmasked_fast_cpu
TestSparseAnyCPU.test_as_sparse_gradcheck_SparseBSR_nonmasked_slow_cpu
TestSparseAnyCPU.test_as_sparse_gradcheck_SparseCOO_masked_fast_cpu
TestSparseAnyCPU.test_as_sparse_gradcheck_SparseCOO_masked_slow_cpu
TestSparseAnyCPU.test_as_sparse_gradcheck_SparseCOO_nonmasked_fast_cpu
TestSparseAnyCPU.test_as_sparse_gradcheck_SparseCOO_nonmasked_slow_cpu
TestSparseAnyCPU.test_as_sparse_gradcheck_SparseCSC_masked_fast_cpu
TestSparseAnyCPU.test_as_sparse_gradcheck_SparseCSC_masked_slow_cpu
TestSparseAnyCPU.test_as_sparse_gradcheck_SparseCSC_nonmasked_fast_cpu
TestSparseAnyCPU.test_as_sparse_gradcheck_SparseCSC_nonmasked_slow_cpu
TestSparseAnyCPU.test_as_sparse_gradcheck_SparseCSR_masked_fast_cpu
TestSparseAnyCPU.test_as_sparse_gradcheck_SparseCSR_masked_slow_cpu
TestSparseAnyCPU.test_as_sparse_gradcheck_SparseCSR_nonmasked_fast_cpu
TestSparseAnyCPU.test_as_sparse_gradcheck_SparseCSR_nonmasked_slow_cpu
TestSparseAnyCPU.test_gradcheck_mm_SparseBSR_masked_fast_cpu_complex128
TestSparseAnyCPU.test_gradcheck_mm_SparseBSR_masked_fast_cpu_float64
TestSparseAnyCPU.test_gradcheck_mm_SparseBSR_masked_slow_cpu_complex128
TestSparseAnyCPU.test_gradcheck_mm_SparseBSR_masked_slow_cpu_float64
TestSparseAnyCPU.test_gradcheck_mm_SparseBSR_sparse_fast_cpu_complex128
TestSparseAnyCPU.test_gradcheck_mm_SparseBSR_sparse_fast_cpu_float64
TestSparseAnyCPU.test_gradcheck_mm_SparseBSR_sparse_slow_cpu_complex128
TestSparseAnyCPU.test_gradcheck_mm_SparseBSR_sparse_slow_cpu_float64
TestSparseAnyCPU.test_gradcheck_mm_SparseCSC_masked_fast_cpu_complex128
TestSparseAnyCPU.test_gradcheck_mm_SparseCSC_masked_fast_cpu_float64
TestSparseAnyCPU.test_gradcheck_mm_SparseCSC_masked_slow_cpu_complex128
TestSparseAnyCPU.test_gradcheck_mm_SparseCSC_masked_slow_cpu_float64
TestSparseAnyCPU.test_gradcheck_mm_SparseCSC_sparse_fast_cpu_complex128
TestSparseAnyCPU.test_gradcheck_mm_SparseCSC_sparse_fast_cpu_float64
TestSparseAnyCPU.test_gradcheck_mm_SparseCSC_sparse_slow_cpu_complex128
TestSparseAnyCPU.test_gradcheck_mm_SparseCSC_sparse_slow_cpu_float64
TestSparseAnyCPU.test_gradcheck_mm_SparseCSR_masked_fast_cpu_complex128
TestSparseAnyCPU.test_gradcheck_mm_SparseCSR_masked_fast_cpu_float64
TestSparseAnyCPU.test_gradcheck_mm_SparseCSR_masked_slow_cpu_complex128
TestSparseAnyCPU.test_gradcheck_mm_SparseCSR_masked_slow_cpu_float64
TestSparseAnyCPU.test_gradcheck_mm_SparseCSR_sparse_fast_cpu_complex128
TestSparseAnyCPU.test_gradcheck_mm_SparseCSR_sparse_fast_cpu_float64
TestSparseAnyCPU.test_gradcheck_mm_SparseCSR_sparse_slow_cpu_complex128
TestSparseAnyCPU.test_gradcheck_mm_SparseCSR_sparse_slow_cpu_float64
TestSparseAnyCPU.test_gradcheck_to_dense_SparseBSC_int64_masked_cpu_complex128
TestSparseAnyCPU.test_gradcheck_to_dense_SparseBSC_int64_masked_cpu_float64
TestSparseAnyCPU.test_gradcheck_to_dense_SparseBSC_int64_sparse_cpu_complex128
TestSparseAnyCPU.test_gradcheck_to_dense_SparseBSC_int64_sparse_cpu_float64
TestSparseAnyCPU.test_gradcheck_to_dense_SparseBSR_int64_masked_cpu_complex128
TestSparseAnyCPU.test_gradcheck_to_dense_SparseBSR_int64_masked_cpu_float64
TestSparseAnyCPU.test_gradcheck_to_dense_SparseBSR_int64_sparse_cpu_complex128
TestSparseAnyCPU.test_gradcheck_to_dense_SparseBSR_int64_sparse_cpu_float64
TestSparseAnyCPU.test_gradcheck_to_dense_SparseCOO_int64_masked_cpu_complex128
TestSparseAnyCPU.test_gradcheck_to_dense_SparseCOO_int64_masked_cpu_float64
TestSparseAnyCPU.test_gradcheck_to_dense_SparseCOO_int64_sparse_cpu_complex128
TestSparseAnyCPU.test_gradcheck_to_dense_SparseCOO_int64_sparse_cpu_float64
TestSparseAnyCPU.test_gradcheck_to_dense_SparseCSC_int64_masked_cpu_complex128
TestSparseAnyCPU.test_gradcheck_to_dense_SparseCSC_int64_masked_cpu_float64
TestSparseAnyCPU.test_gradcheck_to_dense_SparseCSC_int64_sparse_cpu_complex128
TestSparseAnyCPU.test_gradcheck_to_dense_SparseCSC_int64_sparse_cpu_float64
TestSparseAnyCPU.test_gradcheck_to_dense_SparseCSR_int64_masked_cpu_complex128
TestSparseAnyCPU.test_gradcheck_to_dense_SparseCSR_int64_masked_cpu_float64
TestSparseAnyCPU.test_gradcheck_to_dense_SparseCSR_int64_sparse_cpu_complex128
TestSparseAnyCPU.test_gradcheck_to_dense_SparseCSR_int64_sparse_cpu_float64
TestSparseCPU.test_log_softmax_zero_nnz_cpu_float32
TestSparseCPU.test_log_softmax_zero_nnz_cpu_float64
TestSparseCPU.test_softmax_zero_nnz_cpu_float32
TestSparseCPU.test_softmax_zero_nnz_cpu_float64
TestSparseCPU.test_sparse_mask_backward_cpu_complex128
TestSparseCPU.test_sparse_mask_backward_cpu_float64
TestSparseCPU.test_sparse_matmul_cpu_complex128
TestSparseCPU.test_sparse_matmul_cpu_complex64
TestSparseCPU.test_sparse_matmul_cpu_float32
TestSparseCPU.test_sparse_matmul_cpu_float64
TestSparseCPU.test_sparse_mul_masked_cpu_float64
TestSparseCPU.test_sparse_mul_sparse_cpu_float64
TestSparseCPU.test_Sparse_to_Sparse_copy__cpu_bfloat16
TestSparseCPU.test_Sparse_to_Sparse_copy__cpu_complex128
TestSparseCPU.test_Sparse_to_Sparse_copy__cpu_float64
TestSparseCSRCPU.test_autograd_dense_output_addmm_cpu_float64
TestSparseCSRCPU.test_autograd_dense_output_addmv_cpu_float64
TestSparseCSRCPU.test_mul_cpu_float32
TestSparseCSRCPU.test_mul_cpu_float64
TestSparseCSRCPU.test_sampled_addmm_autograd_cpu_complex128
TestSparseCSRCPU.test_sampled_addmm_autograd_cpu_complex64
TestSparseCSRCPU.test_sampled_addmm_autograd_cpu_float32
TestSparseCSRCPU.test_sampled_addmm_autograd_cpu_float64
TestSparseCSRCPU.test_sparse_mm_reduce_sum_cpu_bfloat16
TestSparseCSRCPU.test_sparse_mm_reduce_sum_cpu_float16
TestSparseCSRCPU.test_sparse_mm_reduce_sum_cpu_float32
TestSparseCSRCPU.test_sparse_mm_reduce_sum_cpu_float64
TestSparseCSRCPU.test_sum_cpu_float32
TestSparseCSRCPU.test_sum_cpu_float64
TestTEFuserDynamic.test_addcmul
TestTEFuserDynamic.test_clamp
TestTEFuserDynamic.test_comparison_eq_ne
TestTEFuserDynamic.test_comparison_ge_le
TestTEFuserDynamic.test_comparison_gt_lt
TestTEFuserDynamic.test_concat_invariant
TestTEFuserDynamic.test_exp
TestTEFuserDynamic.test_hardsigmoid_fwd_bwd
TestTEFuserDynamic.test_hardswish_fwd_bwd
TestTEFuserDynamic.test_inlined_optimized_graph
TestTEFuserDynamic.test_lerp
TestTEFuserDynamic.test_lstm_concat
TestTEFuserDynamic.test_lstm_traced
TestTEFuserDynamic.test_relu
TestTEFuserDynamic.test_relu_fwd_bwd
TestTEFuserDynamic.test_small_constant
TestTEFuserStatic.test_addcmul
TestTEFuserStatic.test_chunk_distributes
TestTEFuserStatic.test_clamp
TestTEFuserStatic.test_comparison_eq_ne
TestTEFuserStatic.test_comparison_ge_le
TestTEFuserStatic.test_comparison_gt_lt
TestTEFuserStatic.test_concat_invariant
TestTEFuserStatic.test_constant_chunk_shapes
TestTEFuserStatic.test_exp
TestTEFuserStatic.test_hardsigmoid_fwd_bwd
TestTEFuserStatic.test_hardswish_fwd_bwd
TestTEFuserStatic.test_lerp
TestTEFuserStatic.test_lstm_concat
TestTEFuserStatic.test_lstm_traced
TestTEFuserStatic.test_milstm
TestTEFuserStatic.test_relu
TestTEFuserStatic.test_relu_fwd_bwd
TestTEFuserStatic.test_small_constant
TestViewOpsCPU.test_as_strided_gradients_cpu
TestViewOpsLAZY.test_as_strided_gradients_lazy
TestViewOpsLAZY.test_view_copy_lazy
TestVmapAPILegacy.test_batched_gradient_basic
TestVmapAPILegacy.test_fallback_with_undefined_grad
TestVmapBatchedGradientLegacyCPU.test_add_cpu
TestVmapBatchedGradientLegacyCPU.test_binary_cross_entropy_cpu
TestVmapBatchedGradientLegacyCPU.test_diagonal_cpu
TestVmapBatchedGradientLegacyCPU.test_div_cpu
TestVmapBatchedGradientLegacyCPU.test_expand_cpu
TestVmapBatchedGradientLegacyCPU.test_index_cpu
TestVmapBatchedGradientLegacyCPU.test_inplace_manyview_cpu
TestVmapBatchedGradientLegacyCPU.test_inplace_on_view_cpu
TestVmapBatchedGradientLegacyCPU.test_lgamma_cpu
TestVmapBatchedGradientLegacyCPU.test_log1p_cpu
TestVmapBatchedGradientLegacyCPU.test_log_cpu
TestVmapBatchedGradientLegacyCPU.test_logsumexp_cpu
TestVmapBatchedGradientLegacyCPU.test_max_cpu
TestVmapBatchedGradientLegacyCPU.test_median_cpu
TestVmapBatchedGradientLegacyCPU.test_min_cpu
TestVmapBatchedGradientLegacyCPU.test_mul_cpu
TestVmapBatchedGradientLegacyCPU.test_permute_cpu
TestVmapBatchedGradientLegacyCPU.test_reshape_cpu
TestVmapBatchedGradientLegacyCPU.test_select_cpu
TestVmapBatchedGradientLegacyCPU.test_sigmoid_cpu
TestVmapBatchedGradientLegacyCPU.test_slice_cpu
TestVmapBatchedGradientLegacyCPU.test_stack_cpu
TestVmapBatchedGradientLegacyCPU.test_sub_cpu
TestVmapBatchedGradientLegacyCPU.test_threshold_cpu
TestVmapBatchedGradientLegacyCPU.test_trace_cpu
TestVmapBatchedGradientLegacyCPU.test_unrelated_output_cpu
TestVmapBatchedGradientLegacyCPU.test_unrelated_output_multiple_grad_cpu
TestWithEffects.test_effectful_custom_op_with_subclasses
TestWithEffects.test_effects_and_aliased_outputs
TestWithEffects.test_regular_effectful_op_in_forward_and_backward
TestWithEffects.test_regular_effectful_op_only_in_backward