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