Google Git
Sign in
android / platform / external / pytorch / f85d3a022c / . / test / dynamo_expected_failures
tree: 5bba2622ac1ed46a2360e9b08a0305f3234342d8 [path history] [tgz]
  1. ActivationCheckpointingTests.test_cond_with_kwargs
  2. AotAutogradFallbackTests.test_aot_sequence_nr
  3. AutogradFunctionTests.test_graph_break_if_lifted_free_variable
  4. AutogradFunctionTests.test_print_in_bwd
  5. AutogradFunctionTests.test_stride_in_bwd
  6. ComptimeTests.test_graph_break
  7. CtxManagerTests.test_autograd_profiler_enabled
  8. CtxManagerTests.test_disable_saved_tensors_hooks
  9. CtxManagerTests.test_disable_saved_tensors_hooks_prev_disabled
  10. CtxManagerTests.test_disable_saved_tensors_hooks_prev_disabled_nested
  11. DecoratorTests.test_mark_static_address_guarded
  12. DecoratorTests.test_mark_static_address_unguarded
  13. DefaultsTests.test_cast_tensor_single_elem
  14. DefaultsTests.test_compare_constant_and_tensor
  15. DefaultsTests.test_func_default_torch_args
  16. DefaultsTests.test_in_set_inplace
  17. DefaultsTests.test_in_set_would_fail_broadcast
  18. DynamoProfilerTests.test_dynamo_timed_profiling_isolated
  19. DynamoProfilerTests.test_profiler_cache_lookup
  20. DynamoProfilerTests.test_profiler_dynamo_compiled_region
  21. End2EndTests.test_init_group
  22. ExampleTests.test_exportdb_supported_case_assume_constant_result
  23. ExampleTests.test_exportdb_supported_case_constrain_as_size_example
  24. ExcTests.test_backend_suppress_line
  25. ExcTests.test_graph_break_log
  26. ExcTests.test_internal_error_no_suppress
  27. ExcTests.test_internal_error_suppress_errors
  28. ExcTests.test_not_implemented_error
  29. ExcTests.test_trigger_bisect_on_error
  30. ExcTests.test_trigger_on_error
  31. ExportTests.test_cond_raise_user_error_on_non_list_operands
  32. ExportTests.test_cond_raise_user_error_on_non_tensor_operands
  33. ExportTests.test_cond_raise_user_error_on_unsupported_pred
  34. ExportTests.test_export_decomp
  35. ExportTests.test_export_dynamic_dim_cleanup
  36. ExportTests.test_export_dynamic_dim_range_constraint
  37. ExportTests.test_export_mark_dynamic_conflict_dynamic_dim
  38. ExportTests.test_export_multi_dynamic_dim_constraint
  39. ExportTests.test_export_multi_dynamic_dim_unsafe_relationship
  40. ExportTests.test_export_no_raise
  41. ExportTests.test_export_preserve_constraints_as_metadata_scalar
  42. ExportTests.test_export_raise_on_relationship
  43. ExportTests.test_export_with_constant_dict_values
  44. ExportTests.test_export_with_constant_free_function
  45. ExportTests.test_export_with_constant_free_function_and_class_method
  46. ExportTests.test_export_with_constant_free_function_and_class_method_multiarg
  47. ExportTests.test_export_with_constant_free_function_and_class_method_multiarg_diff
  48. ExportTests.test_export_with_constant_list_nonzero
  49. ExportTests.test_export_with_constant_list_nonzero_free_function
  50. ExportTests.test_export_with_constant_method_on_module
  51. ExportTests.test_export_with_constant_method_on_module_invoke_twice
  52. ExportTests.test_export_with_constant_none_control_flow
  53. ExportTests.test_export_with_constant_none_control_flow_free_func
  54. ExportTests.test_export_with_constant_not_none_control_flow
  55. ExportTests.test_export_with_constant_not_none_control_flow_free_func
  56. ExportTests.test_export_with_constant_not_none_control_flow_pos
  57. ExportTests.test_export_with_constant_not_return_const
  58. ExportTests.test_export_with_constant_tuple_nonzero
  59. ExportTests.test_export_with_map_cond
  60. ExportTests.test_exported_graph_serialization
  61. ExportTests.test_map_cond_param_buffer_lifted
  62. ExportTests.test_torch_inference_mode_ctx
  63. ExportTests.test_untracked_inputs_in_constraints
  64. FakeTensorOperatorInvariants.test_like_ops
  65. FakeTensorOperatorInvariants.test_non_kwarg_only_device
  66. FakeTensorOperatorInvariants.test_tensor_constructors_all_have_kwarg_device
  67. FunctionTests.test_default_dict
  68. FunctionTests.test_default_dict_closure
  69. FunctionTests.test_default_dict_lambda
  70. FunctionTests.test_is_contiguous_frame_counts
  71. FunctionTests.test_math_radians
  72. FunctionTests.test_partials_as_input_partials_lambda
  73. FunctionTests.test_partials_as_input_partials_mod
  74. FuncTorchHigherOrderOpTests.test_grad
  75. FuncTorchHigherOrderOpTests.test_grad_disable_capture
  76. FuncTorchHigherOrderOpTests.test_grad_fn_with_kwargs
  77. FuncTorchHigherOrderOpTests.test_grad_freevar_python_scalar
  78. FuncTorchHigherOrderOpTests.test_grad_has_aux
  79. FuncTorchHigherOrderOpTests.test_grad_non_tensor_input
  80. FuncTorchHigherOrderOpTests.test_grad_pytree
  81. FuncTorchHigherOrderOpTests.test_grad_two_tensor_all_grad_has_aux
  82. FuncTorchHigherOrderOpTests.test_grad_two_tensor_has_aux
  83. FuncTorchHigherOrderOpTests.test_grad_with_graph_break
  84. FuncTorchHigherOrderOpTests.test_grad_with_side_effect
  85. FuncTorchHigherOrderOpTests.test_vmap
  86. FuncTorchHigherOrderOpTests.test_vmap_disable_capture
  87. FuncTorchHigherOrderOpTests.test_vmap_free_const
  88. FuncTorchHigherOrderOpTests.test_vmap_free_tensor
  89. FuncTorchHigherOrderOpTests.test_vmap_kwargs
  90. FuncTorchHigherOrderOpTests.test_vmap_multiple_invocation_in_dims
  91. FuncTorchHigherOrderOpTests.test_vmap_multiple_invocation_out_dims
  92. FuncTorchHigherOrderOpTests.test_vmap_multiple_outputs
  93. FuncTorchHigherOrderOpTests.test_vmap_multiple_outputs_diff_dims
  94. FuncTorchHigherOrderOpTests.test_vmap_multiple_outputs_out_dims_tuple
  95. FuncTorchHigherOrderOpTests.test_vmap_over_vmap_captured
  96. FuncTorchHigherOrderOpTests.test_vmap_over_vmap_two_inputs
  97. FuncTorchHigherOrderOpTests.test_vmap_previous_illegal_op_no_graph_break
  98. FuncTorchHigherOrderOpTests.test_vmap_pytree_inputs
  99. FuncTorchHigherOrderOpTests.test_vmap_recompile_same_config
  100. FuncTorchHigherOrderOpTests.test_vmap_recompile_with_randomness
  101. FuncTorchHigherOrderOpTests.test_vmap_side_effects
  102. FuncTorchHigherOrderOpTests.test_vmap_two_inputs
  103. FuncTorchHigherOrderOpTests.test_vmap_two_inputs_tuple_in_dims
  104. FuncTorchHigherOrderOpTests.test_vmap_with_conditional_graph_break
  105. FuncTorchHigherOrderOpTests.test_vmap_with_graph_break
  106. FuncTorchHigherOrderOpTests.test_vmap_with_graph_break_2
  107. FuncTorchHigherOrderOpTests.test_vmap_with_graph_break_lambda
  108. HigherOrderOpTests.test_capture_untracked_global_nested
  109. HigherOrderOpTests.test_capture_value_created_in_subgraph
  110. HigherOrderOpTests.test_cond_branches_no_arguments
  111. HigherOrderOpTests.test_cond_branches_no_arguments_no_closure
  112. HigherOrderOpTests.test_cond_pytree_operands
  113. HigherOrderOpTests.test_cond_pytree_operands_with_non_tensor_leaves
  114. HigherOrderOpTests.test_cond_source_fn_stack
  115. HigherOrderOpTests.test_cond_subgraph_name_is_valid
  116. HigherOrderOpTests.test_fallback_on_python_primitives_output
  117. HigherOrderOpTests.test_map_lowers_to_graph
  118. HigherOrderOpTests.test_map_multi_return
  119. HigherOrderOpTests.test_map_pytree_return
  120. HigherOrderOpTests.test_map_source_fn_stack
  121. HigherOrderOpTests.test_map_subgraph_name_is_valid
  122. HigherOrderOpTests.test_map_symint_input
  123. HigherOrderOpTests.test_modules
  124. HigherOrderOpTests.test_nested_tuple_output
  125. HigherOrderOpTests.test_output_with_dict
  126. HigherOrderOpTests.test_side_effect_in_body
  127. HigherOrderOpTests.test_vmap_source_fn_stack
  128. HigherOrderOpTests.test_wrap_kwarg_recompile
  129. HigherOrderOpTests.test_wrap_subgraph_name_is_valid
  130. HooksTests.test_functools_arg_vary
  131. HooksTests.test_post_acc_grad_hook
  132. HooksTests.test_tensor_register_global_hook
  133. HooksTests.test_tensor_register_hook_in_graph_local
  134. HooksTests.test_tensor_register_multiple_hooks
  135. HooksTests.test_tensor_register_multiple_hooks_handles_in_list
  136. InteropTests.test_vmap_in_graph
  137. LoggingTests.test_aot
  138. LoggingTests.test_aot_graphs
  139. LoggingTests.test_aot_joint_graph
  140. LoggingTests.test_bytecode
  141. LoggingTests.test_custom_format
  142. LoggingTests.test_custom_format_exc
  143. LoggingTests.test_dynamo_debug
  144. LoggingTests.test_dynamo_error
  145. LoggingTests.test_dynamo_info
  146. LoggingTests.test_graph
  147. LoggingTests.test_graph_breaks
  148. LoggingTests.test_graph_code
  149. LoggingTests.test_graph_sizes
  150. LoggingTests.test_graph_sizes_dynamic
  151. LoggingTests.test_guards_recompiles
  152. LoggingTests.test_inductor_debug
  153. LoggingTests.test_inductor_error
  154. LoggingTests.test_inductor_info
  155. LoggingTests.test_multiline_format
  156. LoggingTests.test_output_code
  157. LoggingTests.test_recompiles
  158. LoggingTests.test_trace_source_cond
  159. LoggingTests.test_trace_source_nested
  160. MiscTests.test_add_to_set
  161. MiscTests.test_any_all_symnode
  162. MiscTests.test_backend_match_guard
  163. MiscTests.test_boolarg
  164. MiscTests.test_callpacked
  165. MiscTests.test_closure_out_of_scope_cell
  166. MiscTests.test_compare_shapes_with_constant
  167. MiscTests.test_cond_nested
  168. MiscTests.test_cond_side_effects
  169. MiscTests.test_dataclass_fields
  170. MiscTests.test_deque_append_left
  171. MiscTests.test_deque_input
  172. MiscTests.test_deterministic_algorithms_mutated
  173. MiscTests.test_dict_order_keys
  174. MiscTests.test_dict_order_keys_modules
  175. MiscTests.test_dict_order_keys_tensors
  176. MiscTests.test_dictcomp
  177. MiscTests.test_dtypes_no_graphbreaks
  178. MiscTests.test_dunder_new_function_inlining
  179. MiscTests.test_frozenset_torch_func_contains
  180. MiscTests.test_get_cache_entry
  181. MiscTests.test_guard_failure_fn
  182. MiscTests.test_guard_failure_fn2
  183. MiscTests.test_guard_failure_fn_shape_control
  184. MiscTests.test_guard_failure_fn_tensor_iter
  185. MiscTests.test_inference_mode
  186. MiscTests.test_inline_closure_not_loaded_by_parent
  187. MiscTests.test_inplace_param_update
  188. MiscTests.test_inplace_view_on_graph_input
  189. MiscTests.test_intermediary_tensor_grad_access
  190. MiscTests.test_is_compiling
  191. MiscTests.test_iter_set
  192. MiscTests.test_itertools_accumulate_symint_default_sum
  193. MiscTests.test_itertools_accumulate_tensors_default_sum
  194. MiscTests.test_itertools_groupby_pure_python_default_identify_func
  195. MiscTests.test_itertools_groupby_pure_python_key_func
  196. MiscTests.test_itertools_infinite_cycle
  197. MiscTests.test_itertools_infinite_repeat
  198. MiscTests.test_itertools_infinite_repeat_mutation
  199. MiscTests.test_itertools_repeat
  200. MiscTests.test_listcomp
  201. MiscTests.test_mandelbrot_numpy
  202. MiscTests.test_mark_static
  203. MiscTests.test_namedtuple1
  204. MiscTests.test_namedtuple2
  205. MiscTests.test_nan
  206. MiscTests.test_nested_closure
  207. MiscTests.test_nested_closure_mutation
  208. MiscTests.test_nested_optimize
  209. MiscTests.test_nested_optimize_decorator
  210. MiscTests.test_nested_optimize_run
  211. MiscTests.test_nn_module_getattr
  212. MiscTests.test_no_raise_guard_partial_constraint
  213. MiscTests.test_no_raise_guard_partial_constraint_across_break
  214. MiscTests.test_numpy_array_of_arrays
  215. MiscTests.test_numpy_force
  216. MiscTests.test_numpy_int_constant
  217. MiscTests.test_numpy_iter
  218. MiscTests.test_numpy_non_torch_dtype
  219. MiscTests.test_numpy_readonly
  220. MiscTests.test_numpy_recompilation_scalar
  221. MiscTests.test_numpy_size_attr
  222. MiscTests.test_numpy_subdtype
  223. MiscTests.test_numpy_tolist
  224. MiscTests.test_numpy_torch_operators
  225. MiscTests.test_numpy_with_builtin_type
  226. MiscTests.test_out_variants_with_resizing_on_graph_inputs
  227. MiscTests.test_pure_python_accumulate
  228. MiscTests.test_py_guards_mark_dynamic
  229. MiscTests.test_raise_guard_full_constraint
  230. MiscTests.test_raise_guard_partial_constraint_no_graph_break
  231. MiscTests.test_raise_on_backend_error
  232. MiscTests.test_recompile_on_global_state_change
  233. MiscTests.test_release_input_memory
  234. MiscTests.test_release_module_memory
  235. MiscTests.test_release_scope_memory
  236. MiscTests.test_repeat_interleave_graphbreaks
  237. MiscTests.test_return_nested_function
  238. MiscTests.test_set_aliasing_recompiles
  239. MiscTests.test_setattr_mutation1
  240. MiscTests.test_simple_set_usage
  241. MiscTests.test_size_input
  242. MiscTests.test_str_format_assert2
  243. MiscTests.test_tensor_build_list_unpack
  244. MiscTests.test_tensor_dict1
  245. MiscTests.test_tensor_dict2
  246. MiscTests.test_tensor_dict3
  247. MiscTests.test_tensor_item_capture
  248. MiscTests.test_tolist_0d
  249. MiscTests.test_tolist_1d
  250. MiscTests.test_tolist_kd
  251. MiscTests.test_tolist_kd_dynamic
  252. MiscTests.test_tolist_scalar
  253. MiscTests.test_torch_seed
  254. MiscTests.test_tracing_nested_py_tree
  255. MiscTests.test_tracing_nested_py_tree_dicts
  256. MiscTests.test_tracing_nested_py_tree_mixed_all
  257. MiscTests.test_tracing_nested_py_tree_tuples
  258. MiscTests.test_tracing_py_tree
  259. MiscTests.test_tracing_py_tree_tensor_subclass
  260. MiscTests.test_tracing_tree_map_only
  261. MiscTests.test_type_copy
  262. MiscTests.test_typing_typevar
  263. MiscTests.test_user_getattribute
  264. MiscTests.test_with_builtin_type
  265. MiscTests.test_yield_from
  266. MiscTests.test_yield_gen_and_from
  267. MiscTests.test_yield_send_to_subgenerator_graph_break
  268. NNModuleTests.test_lazy_module1
  269. NNModuleTests.test_lazy_module2
  270. NNModuleTests.test_lazy_module4
  271. NNModuleTests.test_lazy_module5
  272. NNModuleTests.test_lazy_module6
  273. NNModuleTests.test_lazy_module_no_cls_to_become
  274. NNModuleTests.test_self_mutating1
  275. NNModuleTests.test_unsupportedmethod
  276. NNModuleTests.test_unsupportedmodule
  277. NonStrictExportTestExport.test__scaled_dot_product_flash_attention_non_strict
  278. NonStrictExportTestExport.test_basic_non_strict_fake_tensor_non_strict
  279. NonStrictExportTestExport.test_basic_non_strict_real_tensor_non_strict
  280. NonStrictExportTestExport.test_buffer_util_non_strict
  281. NonStrictExportTestExport.test_cond_with_module_stack_export_with_non_strict
  282. NonStrictExportTestExport.test_export_decomps_dynamic_non_strict
  283. NonStrictExportTestExport.test_export_decomps_simple_non_strict
  284. NonStrictExportTestExport.test_export_with_wrong_inputs_non_strict
  285. NonStrictExportTestExport.test_external_call_non_strict_real_tensor_non_strict
  286. NonStrictExportTestExport.test_fqn_non_strict
  287. NonStrictExportTestExport.test_nn_module_stack_non_strict
  288. NonStrictExportTestExport.test_nn_module_stack_shared_submodule_non_strict
  289. NonStrictExportTestExport.test_non_strict_dynamic_shapes_non_strict
  290. NonStrictExportTestExport.test_non_strict_dynamic_shapes_suggested_fixes_non_strict
  291. NonStrictExportTestExport.test_param_util_non_strict
  292. NonStrictExportTestExport.test_raise_user_error_when_guard_on_data_dependent_operation_non_strict
  293. NonStrictExportTestExport.test_sym_sqrt_non_strict
  294. NonStrictExportTestExport.test_to_module_with_mutated_buffer_multiple_non_strict
  295. NonStrictExportTestExport.test_to_module_with_mutated_buffer_multiple_update_sub_later_non_strict
  296. NonStrictExportTestExport.test_to_module_with_mutated_buffer_non_strict
  297. NumpyTestsCPU.test_boolean_indexing_weirdness_cpu
  298. NumpyTestsCPU.test_boolean_shape_mismatch_cpu
  299. NumpyTestsCPU.test_empty_fancy_index_cpu
  300. NumpyTestsCPU.test_index_no_floats_cpu
  301. OptimizedModuleTest.test_backward_hooks
  302. OptimizedModuleTest.test_cache_size_limit_on_guarded_nn_modules
  303. OptimizedModuleTest.test_composition_with_opt_mod
  304. OptimizedModuleTest.test_hooks_inner
  305. OptimizedModuleTest.test_hooks_outer
  306. OptimizedModuleTest.test_hooks_skip_guards
  307. OptimizedModuleTest.test_module_dict_iter_name
  308. OptimizedModuleTest.test_nn_module
  309. OptimizedModuleTest.test_no_recompile_on_nn_guarded_modules
  310. OptimizedModuleTest.test_to
  311. OptimizerTests.test_adagrad
  312. OptimizerTests.test_adam
  313. OptimizerTests.test_adamax
  314. OptimizerTests.test_adamw
  315. OptimizerTests.test_asgd
  316. OptimizerTests.test_nadam
  317. OptimizerTests.test_rmsprop
  318. OptimizerTests.test_rprop
  319. OptimizerTests.test_sgd
  320. PackedSequenceTest.test_pack_sequence
  321. PackedSequenceTest.test_to
  322. PackedSequenceTest.test_total_length
  323. PackedSequenceTest.test_type_casts
  324. PublicTorchCompilerTests.test_dynamo_signatures
  325. RecompileTests.test_aliasing_guard_failures_with_globals
  326. RecompileTests.test_automatic_dynamic_reduce_recompiles
  327. RecompileTests.test_automatic_dynamic_tensor_scalar_change
  328. RecompileTests.test_dynamic_shape_parameter_recompile
  329. RecompileTests.test_recompiles_true_false_flop
  330. RecompileUxTests.test_mismatched_type
  331. RecompileUxTests.test_verbose_tensor_check
  332. ReproTests.test_add_sub_alpha_out
  333. ReproTests.test_addr_alpha_beta_out
  334. ReproTests.test_boxes_len
  335. ReproTests.test_chunk_reformer_ff
  336. ReproTests.test_convert_boxes_to_pooler_format
  337. ReproTests.test_create_rand_mask_from_inputs
  338. ReproTests.test_do_paste_mask
  339. ReproTests.test_dynamic_shapes_float_guard
  340. ReproTests.test_dynamic_shapes_implicit_guard
  341. ReproTests.test_empty_list_contains_with_jump
  342. ReproTests.test_function_in_skipfiles
  343. ReproTests.test_functools_wraps
  344. ReproTests.test_hf_t5_forward
  345. ReproTests.test_hf_xsoftmax_training
  346. ReproTests.test_issue175
  347. ReproTests.test_jit_trace_errors
  348. ReproTests.test_list_aliasing
  349. ReproTests.test_list_self_reference
  350. ReproTests.test_longformer_chunk
  351. ReproTests.test_merge_criteria_processor_list1
  352. ReproTests.test_merge_criteria_processor_list2
  353. ReproTests.test_module_in_skipfiles
  354. ReproTests.test_multi_import
  355. ReproTests.test_negative_shape_guard
  356. ReproTests.test_numpy_not_ndarray_recompiles
  357. ReproTests.test_recursive_map
  358. ReproTests.test_reformer_eval
  359. ReproTests.test_reformer_min_chunk_len
  360. ReproTests.test_reformer_sorting
  361. ReproTests.test_relative_import
  362. ReproTests.test_relative_import_no_modulename
  363. ReproTests.test_restricted_list_subclass1
  364. ReproTests.test_rewrite_assert_noop
  365. ReproTests.test_rewrite_assert_with_msg
  366. ReproTests.test_rewrite_assert_with_non_string_msg
  367. ReproTests.test_seq_append_list
  368. ReproTests.test_size_typematch
  369. ReproTests.test_tensor_data_kwarg
  370. ReproTests.test_threading_local
  371. ReproTests.test_validate_model_kwargs
  372. RetraceExportTestDynamismExpression.test_export_inline_constraints_retraceability
  373. RetraceExportTestExport.test_cond_with_module_stack_export_with_retraceability
  374. RetraceExportTestExport.test_constrain_size_in_eager_retraceability
  375. RetraceExportTestExport.test_constrain_size_with_constrain_value_retraceability
  376. RetraceExportTestExport.test_constrain_size_with_various_cases_retraceability
  377. RetraceExportTestExport.test_nn_module_stack_retraceability
  378. RetraceExportTestExport.test_nn_module_stack_shared_submodule_retraceability
  379. RetraceExportTestExport.test_non_strict_dynamic_shapes_retraceability
  380. RetraceExportTestExport.test_non_strict_dynamic_shapes_suggested_fixes_retraceability
  381. SerDesExportTestDynamismExpression.test_export_inline_constraints_serdes
  382. SerDesExportTestExport.test_basic_non_strict_fake_tensor_serdes
  383. SerDesExportTestExport.test_basic_non_strict_real_tensor_serdes
  384. SerDesExportTestExport.test_cond_with_module_stack_export_with_serdes
  385. SerDesExportTestExport.test_constrain_size_in_eager_serdes
  386. SerDesExportTestExport.test_constrain_size_with_constrain_value_serdes
  387. SerDesExportTestExport.test_constrain_size_with_various_cases_serdes
  388. SerDesExportTestExport.test_external_call_non_strict_real_tensor_serdes
  389. SerDesExportTestExport.test_nn_module_stack_serdes
  390. SerDesExportTestExport.test_nn_module_stack_shared_submodule_serdes
  391. SerDesExportTestExport.test_non_strict_dynamic_shapes_serdes
  392. SerDesExportTestExport.test_non_strict_dynamic_shapes_suggested_fixes_serdes
  393. SkipNonTensorTests.test_add_tensor1
  394. SkipNonTensorTests.test_add_tensor2
  395. SkipNonTensorTests.test_add_tensor_dict
  396. SkipNonTensorTests.test_add_tensor_list
  397. SkipNonTensorTests.test_do_not_skip_side_effects
  398. SkipNonTensorTests.test_recursive_list
  399. SubclassTests.test_compile_higher_order_with_functionalization
  400. SubclassTests.test_compile_with_functionalization
  401. SubclassTests.test_torch_function_state_guards
  402. SubclassTests.test_wrapper_subclass_guards_on_inner_tensor
  403. SubclassTests.test_wrapper_subclass_with_differently_sized_inner_tensor
  404. SubclassTests.test_wrapper_subclass_with_same_sized_inner_tensor
  405. SubGraphTests.test_dynamic_duck_size
  406. SubGraphTests.test_dynamic_getitem
  407. SubGraphTests.test_dynamic_kwarg
  408. SubGraphTests.test_dynamic_order_dependence
  409. SubGraphTests.test_dynamic_zero_inference
  410. SubGraphTests.test_enumerate_not_break_graph
  411. SubGraphTests.test_no_graph_break_on_item
  412. TestAnalyze.test_trace_dependencies
  413. TestAOTAutograd.test_default_partitioner_saves_symints_not_tensors_for_bw
  414. TestAOTAutograd.test_dupe_arg_torture
  415. TestAOTAutograd.test_inference_mode
  416. TestAOTAutograd.test_input_aliased_with_mutation_output_alias
  417. TestAOTAutograd.test_input_data_and_metadata_mutation
  418. TestAOTAutograd.test_input_data_and_metadata_mutation_aliases_other_input
  419. TestAOTAutograd.test_input_mutation_alias_everything
  420. TestAOTAutograd.test_input_mutation_aliases_and_none_require_gradients
  421. TestAOTAutograd.test_input_mutation_aliases_and_output_alias
  422. TestAOTAutograd.test_input_mutation_aliases_bases_out_of_order
  423. TestAOTAutograd.test_input_mutation_and_output_view
  424. TestAOTAutograd.test_input_mutation_false_aliasing
  425. TestAOTAutograd.test_input_mutation_is_output
  426. TestAOTAutograd.test_input_mutation_metadata
  427. TestAOTAutograd.test_input_mutation_noncontiguous
  428. TestAOTAutograd.test_input_mutation_output_view_multiple
  429. TestAOTAutograd.test_input_mutation_set__input_mutation
  430. TestAOTAutograd.test_input_mutation_set__nop
  431. TestAOTAutograd.test_input_output_view_metadata_mutate_multiple
  432. TestAOTAutograd.test_input_output_view_mutate_multiple
  433. TestAOTAutograd.test_input_output_view_simple
  434. TestAOTAutograd.test_invalid_dupe
  435. TestAOTAutograd.test_invalid_dupe_fake
  436. TestAOTAutograd.test_invalid_dupe_left_bias
  437. TestAOTAutograd.test_new_inp_requires_grad_now
  438. TestAOTAutograd.test_output_aliases_input_multi_output_view
  439. TestAOTAutograd.test_output_aliases_intermediate_and_returned
  440. TestAOTAutograd.test_output_aliases_intermediate_and_returned_different_grad
  441. TestAOTAutograd.test_output_aliases_intermediate_and_returned_flipped
  442. TestAOTAutograd.test_output_aliases_intermediate_multi_output_view
  443. TestAOTAutograd.test_output_aliases_intermediate_multiple
  444. TestAOTAutograd.test_output_aliases_intermediate_multiple_mixed
  445. TestAOTAutograd.test_output_aliases_intermediate_returned_multiple_times
  446. TestAOTAutograd.test_output_aliases_multiple_inputs_get_correct_one
  447. TestAOTAutograd.test_output_all_alias_types
  448. TestAOTAutograd.test_output_dict
  449. TestAOTAutograd.test_output_op_depending_on_symint
  450. TestAOTAutograd.test_outputs_are_aliased
  451. TestAOTAutograd.test_set__and_data_mutation_bad
  452. TestAOTAutograd.test_set__and_data_mutation_good
  453. TestAOTAutograd.test_some_output_requires_grad_input_doesnt
  454. TestAOTAutograd.test_some_outputs_dont_require_grad_non_view
  455. TestAOTAutograd.test_some_outputs_dont_require_grad_view
  456. TestAOTAutograd.test_view_and_inplace_view
  457. TestAOTAutograd.test_view_detach
  458. TestAOTDispatch.test_aot_dispatch_input_mutation
  459. TestAOTDispatch.test_aot_dispatch_input_mutation_and_output_alias
  460. TestAOTDispatch.test_aot_dispatch_output_alias
  461. TestAOTDispatch.test_aot_dispatch_simple
  462. TestApplyAlongAxis.test_0d_array
  463. TestApplyAlongAxis.test_axis_insertion
  464. TestApplyAlongAxis.test_scalar_array
  465. TestArange.test_infinite
  466. TestArange.test_nan_step
  467. TestArange.test_require_range
  468. TestArange.test_zero_step
  469. TestArgmax.test_combinations_data62
  470. TestArgmax.test_combinations_data63
  471. TestArgmax.test_combinations_data65
  472. TestArgmax.test_combinations_data66
  473. TestArrayAttributeDeletion.test_multiarray_not_writable_attributes_deletion
  474. TestArrayAttributeDeletion.test_multiarray_writable_attributes_deletion
  475. TestArrayConstruction.test_array_empty
  476. TestArrayCreationCopyArgument.test_scalars
  477. TestArrayCreationCopyArgument.test_striding_not_ok
  478. TestArrayFromScalar.test_integers_np_byte_np_longlong
  479. TestArrayFromScalar.test_integers_np_int__np_longlong
  480. TestArrayFromScalar.test_integers_np_intc_np_longlong
  481. TestArrayFromScalar.test_integers_np_longlong_np_longlong
  482. TestArrayFromScalar.test_integers_np_longlong_t26
  483. TestArrayFromScalar.test_integers_np_short_np_longlong
  484. TestArrayFromScalar.test_integers_t15_np_longlong
  485. TestArraySplit.test_integer_0_split
  486. TestAsArrayCPU.test_copy_list_cpu_bfloat16
  487. TestAsArrayCPU.test_copy_list_cpu_bool
  488. TestAsArrayCPU.test_copy_list_cpu_complex128
  489. TestAsArrayCPU.test_copy_list_cpu_complex64
  490. TestAsArrayCPU.test_copy_list_cpu_float16
  491. TestAsArrayCPU.test_copy_list_cpu_float32
  492. TestAsArrayCPU.test_copy_list_cpu_float64
  493. TestAsArrayCPU.test_copy_list_cpu_int16
  494. TestAsArrayCPU.test_copy_list_cpu_int32
  495. TestAsArrayCPU.test_copy_list_cpu_int64
  496. TestAsArrayCPU.test_copy_list_cpu_int8
  497. TestAsArrayCPU.test_copy_list_cpu_uint8
  498. TestAssertCloseSparseBSC.test_matching
  499. TestAssertCloseSparseBSC.test_mismatching_ccol_indices_msg
  500. TestAssertCloseSparseBSC.test_mismatching_row_indices_msg
  501. TestAssertCloseSparseBSC.test_mismatching_values_msg
  502. TestAssertCloseSparseBSR.test_matching
  503. TestAssertCloseSparseBSR.test_mismatching_col_indices_msg
  504. TestAssertCloseSparseBSR.test_mismatching_crow_indices_msg
  505. TestAssertCloseSparseBSR.test_mismatching_values_msg
  506. TestAssertCloseSparseCOO.test_matching_coalesced
  507. TestAssertCloseSparseCOO.test_matching_uncoalesced
  508. TestAssertCloseSparseCOO.test_mismatching_indices_msg
  509. TestAssertCloseSparseCOO.test_mismatching_nnz
  510. TestAssertCloseSparseCOO.test_mismatching_values_msg
  511. TestAssertCloseSparseCSC.test_matching
  512. TestAssertCloseSparseCSC.test_mismatching_ccol_indices_msg
  513. TestAssertCloseSparseCSC.test_mismatching_row_indices_msg
  514. TestAssertCloseSparseCSC.test_mismatching_values_msg
  515. TestAssertCloseSparseCSR.test_matching
  516. TestAssertCloseSparseCSR.test_mismatching_col_indices_msg
  517. TestAssertCloseSparseCSR.test_mismatching_crow_indices_msg
  518. TestAssertCloseSparseCSR.test_mismatching_values_msg
  519. TestAssignment.test_assignment_broadcasting
  520. TestAttributes.test_fill_max_uint64
  521. TestAttributes.test_fill_readonly
  522. TestAttributes.test_fill_struct_array
  523. TestAttributes.test_set_stridesattr
  524. TestAttributes.test_stridesattr
  525. TestAutograd.test_access_saved_tensor_twice_without_recomputation_works
  526. TestAutograd.test_accumulate_grad_tensor_reference
  527. TestAutograd.test_anomaly_mode_no_check_nan
  528. TestAutograd.test_autograd_simple_views_python
  529. TestAutograd.test_autograd_views_codegen
  530. TestAutograd.test_backward_with_inputs
  531. TestAutograd.test_checkpoint_detects_non_determinism
  532. TestAutograd.test_checkpointing_non_reentrant_autocast_cpu
  533. TestAutograd.test_checkpointing_without_reentrant_custom_function_works
  534. TestAutograd.test_checkpointing_without_reentrant_with_context_fn
  535. TestAutograd.test_custom_autograd_no_early_free
  536. TestAutograd.test_custom_autograd_repeated_grad_grad
  537. TestAutograd.test_custom_function_cycle
  538. TestAutograd.test_custom_function_forward_mode_forward_is_no_op
  539. TestAutograd.test_custom_function_forward_mode_inplace_checks
  540. TestAutograd.test_custom_function_forward_mode_view_checks
  541. TestAutograd.test_custom_function_forward_mode_wrong_formula
  542. TestAutograd.test_custom_function_non_tensor_inputs_outputs
  543. TestAutograd.test_custom_function_save_for_forward
  544. TestAutograd.test_custom_function_saved_tensors
  545. TestAutograd.test_gradcheck_backward_mul_by_grad_output
  546. TestAutograd.test_gradcheck_check_forward_or_backward_only
  547. TestAutograd.test_gradcheck_check_no_differentiable_outputs
  548. TestAutograd.test_gradcheck_forward_ad
  549. TestAutograd.test_gradcheck_forward_ad_batched_grad
  550. TestAutograd.test_gradcheck_nondeterministic
  551. TestAutograd.test_hook_closure_cycle_use_custom_function_True_use_tensor_hook_False
  552. TestAutograd.test_hook_none
  553. TestAutograd.test_increment_version
  554. TestAutograd.test_named_tensor_for_complex_views
  555. TestAutograd.test_nested_anomaly_detect_nan
  556. TestAutograd.test_nested_anomaly_printstack_cleanup
  557. TestAutograd.test_no_grad_copy_sparse
  558. TestAutograd.test_profiler
  559. TestAutograd.test_profiler_aggregation_table
  560. TestAutograd.test_profiler_propagation
  561. TestAutograd.test_profiler_seq_nr
  562. TestAutograd.test_profiler_shapes
  563. TestAutograd.test_record_function
  564. TestAutograd.test_record_function_callbacks
  565. TestAutograd.test_record_function_legacy
  566. TestAutograd.test_return_duplicate
  567. TestAutograd.test_return_duplicate_inplace
  568. TestAutograd.test_save_on_cpu_and_checkpoint
  569. TestAutograd.test_saved_tensor_hooks_custom_function_intermediates
  570. TestAutograd.test_set_data_tensorimpl_type
  571. TestAutograd.test_set_grad_coroutines
  572. TestAutograd.test_set_grad_coroutines_exit
  573. TestAutograd.test_set_grad_generator_functions
  574. TestAutograd.test_set_grad_generator_functions_recursive
  575. TestAutograd.test_setitem
  576. TestAutograd.test_setitem_mask
  577. TestAutograd.test_sparse_mm_backward
  578. TestAutograd.test_version_counter
  579. TestAutogradDeviceTypeCPU.test_inplace_on_view_modify_base_cpu
  580. TestAutogradDeviceTypeCPU.test_inplace_on_view_python_cpu
  581. TestAutogradDeviceTypeCPU.test_sparse_ctor_getter_backward_cpu_complex128
  582. TestAutogradDeviceTypeCPU.test_sparse_ctor_getter_backward_cpu_float64
  583. TestAutogradForwardMode.test_create_new_zeros_with_same_meta
  584. TestAutogradForwardMode.test_detach_view_tracking
  585. TestAutogradForwardMode.test_forward_level_cleanup
  586. TestAutogradFunctionCPU.test_once_differentiable_autograd_vjp_cpu
  587. TestAutogradInferenceMode.test_inference_mode_context_manager
  588. TestAutogradInferenceMode.test_inference_mode_decorator
  589. TestAutogradInferenceMode.test_inference_mode_inf_tensor_in_inf_mode_inplace_op
  590. TestAutogradInferenceMode.test_inference_mode_tensor_creation
  591. TestAutogradLogging.test_logging
  592. TestBaseMath.test_lower_align
  593. TestBaseSparsifier.test_state_dict
  594. TestBaseStructuredSparsifier.test_constructor
  595. TestBaseStructuredSparsifier.test_prepare_conv2d
  596. TestBaseStructuredSparsifier.test_prepare_linear
  597. TestBaseStructuredSparsifier.test_step_linear
  598. TestBincount.test_with_incorrect_minlength
  599. TestBitsCPU.test_cat_cpu
  600. TestBool.test_sum_2
  601. TestBooleanIndexing.test_bool_as_int_argument_errors
  602. TestBooleanIndexing.test_boolean_indexing_weirdness
  603. TestBroadcast.test_broadcast_error_kwargs
  604. TestBroadcast.test_broadcast_single_arg
  605. TestBufferProtocolCPU.test_byte_to_int_cpu
  606. TestClip.test_clip_func_takes_out
  607. TestClip.test_simple_complex
  608. TestClip.test_simple_int32_inout_casting_unsafe
  609. TestColumnStack.test_non_iterable
  610. TestCommonPass.test_correctness_CSEPass_Mutation_cpu
  611. TestCommonPass.test_correctness_CSEPass_MutationInput_cpu
  612. TestCommonPass.test_correctness_CSEPass_MutationMetadata_cpu
  613. TestCommonPass.test_correctness_CSEPass_MutationTorchTensorCall_cpu
  614. TestCommonPass.test_correctness_CSEPass_ReturnList_cpu
  615. TestCommonPass.test_correctness_CSEPass_TakeList_cpu
  616. TestCommonPass.test_correctness_factory_CSEPass_FactoryFunctionCall_cpu
  617. TestCommonPass.test_correctness_factory_CSEPass_MutationFactory_cpu
  618. TestComparatorOps.test_compare_tensor_scalar
  619. TestComparatorOps.test_compare_tensor_tensor
  620. TestCompileTransformsCPU.test_compile_vmap_hessian_cpu
  621. TestComposability.test_convert_without_squash_mask
  622. TestComposability.test_fusion_before_s_prep
  623. TestComposability.test_q_prep_before_s_prep
  624. TestComposability.test_qat_prep_before_s_prep
  625. TestComposability.test_s_prep_before_fusion
  626. TestComposability.test_s_prep_before_q_prep
  627. TestComposability.test_s_prep_before_qat_prep
  628. TestComposabilityCPU.test_make_fx_vjp_cpu
  629. TestComposabilityCPU.test_vjp_grad_cpu
  630. TestComposabilityCPU.test_vjp_vjp_cpu
  631. TestComposabilityCPU.test_vjp_vmap_cpu
  632. TestComposabilityCPU.test_vmap_vjp_cpu
  633. TestConcatenate.test_bad_out_shape
  634. TestConcatenate.test_concatenate
  635. TestConcatenate.test_exceptions
  636. TestConcatenate.test_large_concatenate_axis_None
  637. TestCond.test_empty_sq_cases
  638. TestCond.test_sq_cases
  639. TestConstant.test_check_constant_float3
  640. TestConstant.test_check_constant_pad_2d
  641. TestContentStoreCPU.test_repeated_hash_cpu
  642. TestControlFlow.test_map_autograd_nested_list
  643. TestControlFlow.test_map_autograd_no_grad_output
  644. TestControlFlow.test_map_dict_in_out
  645. TestControlFlow.test_map_list_in_out
  646. TestControlFlowTraced.test_map_functionalized
  647. TestControlFlowTraced.test_map_functionalized_aot_func
  648. TestControlFlowTraced.test_nested_cond_map_cond_symbolic
  649. TestControlFlowTraced.test_nested_map_cond_real
  650. TestControlFlowTraced.test_nested_map_cond_symbolic
  651. TestControlFlowTraced.test_tracing_map_autograd_aot_functionalized
  652. TestControlFlowTraced.test_tracing_map_autograd_symbolic_dict
  653. TestControlFlowTraced.test_tracing_map_autograd_symbolic_list
  654. TestControlFlowTraced.test_tracing_map_autograd_symbolic_simple
  655. TestControlFlowTraced.test_tracing_map_real
  656. TestControlFlowTraced.test_tracing_map_symbolic_dict
  657. TestControlFlowTraced.test_tracing_map_symbolic_list
  658. TestControlFlowTraced.test_tracing_map_symbolic_simple
  659. TestConversion.test_to_int_scalar
  660. TestConvolutionNN.test_Conv2d_missing_argument
  661. TestConvolutionNNDeviceTypeCPU.test_conv_backend_empty_batch_channel1d_has_bias_False_strided_False_contiguous_False_cpu
  662. TestConvolutionNNDeviceTypeCPU.test_conv_backend_empty_batch_channel1d_has_bias_False_strided_False_contiguous_True_cpu
  663. TestConvolutionNNDeviceTypeCPU.test_conv_backend_empty_batch_channel1d_has_bias_False_strided_True_contiguous_False_cpu
  664. TestConvolutionNNDeviceTypeCPU.test_conv_backend_empty_batch_channel1d_has_bias_False_strided_True_contiguous_True_cpu
  665. TestConvolutionNNDeviceTypeCPU.test_conv_backend_empty_batch_channel1d_has_bias_True_strided_False_contiguous_False_cpu
  666. TestConvolutionNNDeviceTypeCPU.test_conv_backend_empty_batch_channel1d_has_bias_True_strided_False_contiguous_True_cpu
  667. TestConvolutionNNDeviceTypeCPU.test_conv_backend_empty_batch_channel1d_has_bias_True_strided_True_contiguous_False_cpu
  668. TestConvolutionNNDeviceTypeCPU.test_conv_backend_empty_batch_channel1d_has_bias_True_strided_True_contiguous_True_cpu
  669. TestConvolutionNNDeviceTypeCPU.test_conv_backend_empty_batch_channel2d_has_bias_False_strided_False_contiguous_False_cpu
  670. TestConvolutionNNDeviceTypeCPU.test_conv_backend_empty_batch_channel2d_has_bias_False_strided_False_contiguous_True_cpu
  671. TestConvolutionNNDeviceTypeCPU.test_conv_backend_empty_batch_channel2d_has_bias_False_strided_True_contiguous_False_cpu
  672. TestConvolutionNNDeviceTypeCPU.test_conv_backend_empty_batch_channel2d_has_bias_False_strided_True_contiguous_True_cpu
  673. TestConvolutionNNDeviceTypeCPU.test_conv_backend_empty_batch_channel2d_has_bias_True_strided_False_contiguous_False_cpu
  674. TestConvolutionNNDeviceTypeCPU.test_conv_backend_empty_batch_channel2d_has_bias_True_strided_False_contiguous_True_cpu
  675. TestConvolutionNNDeviceTypeCPU.test_conv_backend_empty_batch_channel2d_has_bias_True_strided_True_contiguous_False_cpu
  676. TestConvolutionNNDeviceTypeCPU.test_conv_backend_empty_batch_channel2d_has_bias_True_strided_True_contiguous_True_cpu
  677. TestConvolutionNNDeviceTypeCPU.test_conv_backend_empty_batch_channel3d_has_bias_False_strided_False_contiguous_False_cpu
  678. TestConvolutionNNDeviceTypeCPU.test_conv_backend_empty_batch_channel3d_has_bias_False_strided_False_contiguous_True_cpu
  679. TestConvolutionNNDeviceTypeCPU.test_conv_backend_empty_batch_channel3d_has_bias_False_strided_True_contiguous_False_cpu
  680. TestConvolutionNNDeviceTypeCPU.test_conv_backend_empty_batch_channel3d_has_bias_False_strided_True_contiguous_True_cpu
  681. TestConvolutionNNDeviceTypeCPU.test_conv_backend_empty_batch_channel3d_has_bias_True_strided_False_contiguous_False_cpu
  682. TestConvolutionNNDeviceTypeCPU.test_conv_backend_empty_batch_channel3d_has_bias_True_strided_False_contiguous_True_cpu
  683. TestConvolutionNNDeviceTypeCPU.test_conv_backend_empty_batch_channel3d_has_bias_True_strided_True_contiguous_False_cpu
  684. TestConvolutionNNDeviceTypeCPU.test_conv_backend_empty_batch_channel3d_has_bias_True_strided_True_contiguous_True_cpu
  685. TestConvolutionNNDeviceTypeCPU.test_conv_backend_empty_channel1d_has_bias_False_strided_False_contiguous_False_cpu
  686. TestConvolutionNNDeviceTypeCPU.test_conv_backend_empty_channel1d_has_bias_False_strided_False_contiguous_True_cpu
  687. TestConvolutionNNDeviceTypeCPU.test_conv_backend_empty_channel1d_has_bias_False_strided_True_contiguous_False_cpu
  688. TestConvolutionNNDeviceTypeCPU.test_conv_backend_empty_channel1d_has_bias_False_strided_True_contiguous_True_cpu
  689. TestConvolutionNNDeviceTypeCPU.test_conv_backend_empty_channel1d_has_bias_True_strided_False_contiguous_False_cpu
  690. TestConvolutionNNDeviceTypeCPU.test_conv_backend_empty_channel1d_has_bias_True_strided_False_contiguous_True_cpu
  691. TestConvolutionNNDeviceTypeCPU.test_conv_backend_empty_channel1d_has_bias_True_strided_True_contiguous_False_cpu
  692. TestConvolutionNNDeviceTypeCPU.test_conv_backend_empty_channel1d_has_bias_True_strided_True_contiguous_True_cpu
  693. TestConvolutionNNDeviceTypeCPU.test_conv_backend_empty_channel2d_has_bias_False_strided_False_contiguous_False_cpu
  694. TestConvolutionNNDeviceTypeCPU.test_conv_backend_empty_channel2d_has_bias_False_strided_False_contiguous_True_cpu
  695. TestConvolutionNNDeviceTypeCPU.test_conv_backend_empty_channel2d_has_bias_False_strided_True_contiguous_False_cpu
  696. TestConvolutionNNDeviceTypeCPU.test_conv_backend_empty_channel2d_has_bias_False_strided_True_contiguous_True_cpu
  697. TestConvolutionNNDeviceTypeCPU.test_conv_backend_empty_channel2d_has_bias_True_strided_False_contiguous_False_cpu
  698. TestConvolutionNNDeviceTypeCPU.test_conv_backend_empty_channel2d_has_bias_True_strided_False_contiguous_True_cpu
  699. TestConvolutionNNDeviceTypeCPU.test_conv_backend_empty_channel2d_has_bias_True_strided_True_contiguous_False_cpu
  700. TestConvolutionNNDeviceTypeCPU.test_conv_backend_empty_channel2d_has_bias_True_strided_True_contiguous_True_cpu
  701. TestConvolutionNNDeviceTypeCPU.test_conv_backend_empty_channel3d_has_bias_False_strided_False_contiguous_False_cpu
  702. TestConvolutionNNDeviceTypeCPU.test_conv_backend_empty_channel3d_has_bias_False_strided_False_contiguous_True_cpu
  703. TestConvolutionNNDeviceTypeCPU.test_conv_backend_empty_channel3d_has_bias_False_strided_True_contiguous_False_cpu
  704. TestConvolutionNNDeviceTypeCPU.test_conv_backend_empty_channel3d_has_bias_False_strided_True_contiguous_True_cpu
  705. TestConvolutionNNDeviceTypeCPU.test_conv_backend_empty_channel3d_has_bias_True_strided_False_contiguous_False_cpu
  706. TestConvolutionNNDeviceTypeCPU.test_conv_backend_empty_channel3d_has_bias_True_strided_False_contiguous_True_cpu
  707. TestConvolutionNNDeviceTypeCPU.test_conv_backend_empty_channel3d_has_bias_True_strided_True_contiguous_False_cpu
  708. TestConvolutionNNDeviceTypeCPU.test_conv_backend_empty_channel3d_has_bias_True_strided_True_contiguous_True_cpu
  709. TestConvolutionNNDeviceTypeCPU.test_conv_backend_mkldnn_empty_batch_channel1d_has_bias_False_strided_False_contiguous_False_cpu
  710. TestConvolutionNNDeviceTypeCPU.test_conv_backend_mkldnn_empty_batch_channel1d_has_bias_False_strided_False_contiguous_True_cpu
  711. TestConvolutionNNDeviceTypeCPU.test_conv_backend_mkldnn_empty_batch_channel1d_has_bias_False_strided_True_contiguous_False_cpu
  712. TestConvolutionNNDeviceTypeCPU.test_conv_backend_mkldnn_empty_batch_channel1d_has_bias_False_strided_True_contiguous_True_cpu
  713. TestConvolutionNNDeviceTypeCPU.test_conv_backend_mkldnn_empty_batch_channel1d_has_bias_True_strided_False_contiguous_False_cpu
  714. TestConvolutionNNDeviceTypeCPU.test_conv_backend_mkldnn_empty_batch_channel1d_has_bias_True_strided_False_contiguous_True_cpu
  715. TestConvolutionNNDeviceTypeCPU.test_conv_backend_mkldnn_empty_batch_channel1d_has_bias_True_strided_True_contiguous_False_cpu
  716. TestConvolutionNNDeviceTypeCPU.test_conv_backend_mkldnn_empty_batch_channel1d_has_bias_True_strided_True_contiguous_True_cpu
  717. TestConvolutionNNDeviceTypeCPU.test_conv_backend_mkldnn_empty_batch_channel2d_has_bias_False_strided_False_contiguous_False_cpu
  718. TestConvolutionNNDeviceTypeCPU.test_conv_backend_mkldnn_empty_batch_channel2d_has_bias_False_strided_False_contiguous_True_cpu
  719. TestConvolutionNNDeviceTypeCPU.test_conv_backend_mkldnn_empty_batch_channel2d_has_bias_False_strided_True_contiguous_False_cpu
  720. TestConvolutionNNDeviceTypeCPU.test_conv_backend_mkldnn_empty_batch_channel2d_has_bias_False_strided_True_contiguous_True_cpu
  721. TestConvolutionNNDeviceTypeCPU.test_conv_backend_mkldnn_empty_batch_channel2d_has_bias_True_strided_False_contiguous_False_cpu
  722. TestConvolutionNNDeviceTypeCPU.test_conv_backend_mkldnn_empty_batch_channel2d_has_bias_True_strided_False_contiguous_True_cpu
  723. TestConvolutionNNDeviceTypeCPU.test_conv_backend_mkldnn_empty_batch_channel2d_has_bias_True_strided_True_contiguous_False_cpu
  724. TestConvolutionNNDeviceTypeCPU.test_conv_backend_mkldnn_empty_batch_channel2d_has_bias_True_strided_True_contiguous_True_cpu
  725. TestConvolutionNNDeviceTypeCPU.test_conv_backend_mkldnn_empty_batch_channel3d_has_bias_False_strided_False_contiguous_False_cpu
  726. TestConvolutionNNDeviceTypeCPU.test_conv_backend_mkldnn_empty_batch_channel3d_has_bias_False_strided_False_contiguous_True_cpu
  727. TestConvolutionNNDeviceTypeCPU.test_conv_backend_mkldnn_empty_batch_channel3d_has_bias_False_strided_True_contiguous_False_cpu
  728. TestConvolutionNNDeviceTypeCPU.test_conv_backend_mkldnn_empty_batch_channel3d_has_bias_False_strided_True_contiguous_True_cpu
  729. TestConvolutionNNDeviceTypeCPU.test_conv_backend_mkldnn_empty_batch_channel3d_has_bias_True_strided_False_contiguous_False_cpu
  730. TestConvolutionNNDeviceTypeCPU.test_conv_backend_mkldnn_empty_batch_channel3d_has_bias_True_strided_False_contiguous_True_cpu
  731. TestConvolutionNNDeviceTypeCPU.test_conv_backend_mkldnn_empty_batch_channel3d_has_bias_True_strided_True_contiguous_False_cpu
  732. TestConvolutionNNDeviceTypeCPU.test_conv_backend_mkldnn_empty_batch_channel3d_has_bias_True_strided_True_contiguous_True_cpu
  733. TestConvolutionNNDeviceTypeCPU.test_conv_backend_mkldnn_empty_channel1d_has_bias_False_strided_False_contiguous_False_cpu
  734. TestConvolutionNNDeviceTypeCPU.test_conv_backend_mkldnn_empty_channel1d_has_bias_False_strided_False_contiguous_True_cpu
  735. TestConvolutionNNDeviceTypeCPU.test_conv_backend_mkldnn_empty_channel1d_has_bias_False_strided_True_contiguous_False_cpu
  736. TestConvolutionNNDeviceTypeCPU.test_conv_backend_mkldnn_empty_channel1d_has_bias_False_strided_True_contiguous_True_cpu
  737. TestConvolutionNNDeviceTypeCPU.test_conv_backend_mkldnn_empty_channel1d_has_bias_True_strided_False_contiguous_False_cpu
  738. TestConvolutionNNDeviceTypeCPU.test_conv_backend_mkldnn_empty_channel1d_has_bias_True_strided_False_contiguous_True_cpu
  739. TestConvolutionNNDeviceTypeCPU.test_conv_backend_mkldnn_empty_channel1d_has_bias_True_strided_True_contiguous_False_cpu
  740. TestConvolutionNNDeviceTypeCPU.test_conv_backend_mkldnn_empty_channel1d_has_bias_True_strided_True_contiguous_True_cpu
  741. TestConvolutionNNDeviceTypeCPU.test_conv_backend_mkldnn_empty_channel2d_has_bias_False_strided_False_contiguous_False_cpu
  742. TestConvolutionNNDeviceTypeCPU.test_conv_backend_mkldnn_empty_channel2d_has_bias_False_strided_False_contiguous_True_cpu
  743. TestConvolutionNNDeviceTypeCPU.test_conv_backend_mkldnn_empty_channel2d_has_bias_False_strided_True_contiguous_False_cpu
  744. TestConvolutionNNDeviceTypeCPU.test_conv_backend_mkldnn_empty_channel2d_has_bias_False_strided_True_contiguous_True_cpu
  745. TestConvolutionNNDeviceTypeCPU.test_conv_backend_mkldnn_empty_channel2d_has_bias_True_strided_False_contiguous_False_cpu
  746. TestConvolutionNNDeviceTypeCPU.test_conv_backend_mkldnn_empty_channel2d_has_bias_True_strided_False_contiguous_True_cpu
  747. TestConvolutionNNDeviceTypeCPU.test_conv_backend_mkldnn_empty_channel2d_has_bias_True_strided_True_contiguous_False_cpu
  748. TestConvolutionNNDeviceTypeCPU.test_conv_backend_mkldnn_empty_channel2d_has_bias_True_strided_True_contiguous_True_cpu
  749. TestConvolutionNNDeviceTypeCPU.test_conv_backend_mkldnn_empty_channel3d_has_bias_False_strided_False_contiguous_False_cpu
  750. TestConvolutionNNDeviceTypeCPU.test_conv_backend_mkldnn_empty_channel3d_has_bias_False_strided_False_contiguous_True_cpu
  751. TestConvolutionNNDeviceTypeCPU.test_conv_backend_mkldnn_empty_channel3d_has_bias_False_strided_True_contiguous_False_cpu
  752. TestConvolutionNNDeviceTypeCPU.test_conv_backend_mkldnn_empty_channel3d_has_bias_False_strided_True_contiguous_True_cpu
  753. TestConvolutionNNDeviceTypeCPU.test_conv_backend_mkldnn_empty_channel3d_has_bias_True_strided_False_contiguous_False_cpu
  754. TestConvolutionNNDeviceTypeCPU.test_conv_backend_mkldnn_empty_channel3d_has_bias_True_strided_False_contiguous_True_cpu
  755. TestConvolutionNNDeviceTypeCPU.test_conv_backend_mkldnn_empty_channel3d_has_bias_True_strided_True_contiguous_False_cpu
  756. TestConvolutionNNDeviceTypeCPU.test_conv_backend_mkldnn_empty_channel3d_has_bias_True_strided_True_contiguous_True_cpu
  757. TestCov.test_aweights
  758. TestCov.test_fweights
  759. TestCppExtensionJIT.test_cpp_frontend_module_has_same_output_as_python
  760. TestCppExtensionJIT.test_cpp_frontend_module_has_up_to_date_attribute
  761. TestCppExtensionJIT.test_cpp_frontend_module_has_up_to_date_attributes
  762. TestCppExtensionOpenRgistration.test_open_device_registration
  763. TestCross.test_broadcasting_shapes
  764. TestCustomBackendAPI.test_aot_autograd_api
  765. TestCustomBackendAPI.test_lookup_backend
  766. TestCustomBackendAPI.test_register_backend_api
  767. TestCustomOp.test_impl_device_cpu
  768. TestCustomOp.test_impl_device_function
  769. TestCustomOp.test_legacy_define
  770. TestCustomOpTestingCPU.test_missing_functionalization_cpu
  771. TestCxxPytree.test_pytree_serialize_namedtuple
  772. TestCxxPytree.test_pytree_serialize_spec0
  773. TestCxxPytree.test_pytree_serialize_spec1
  774. TestCxxPytree.test_pytree_serialize_spec2
  775. TestCxxPytree.test_pytree_serialize_spec3
  776. TestCxxPytree.test_pytree_serialize_spec4
  777. TestCxxPytree.test_pytree_serialize_spec5
  778. TestCxxPytree.test_pytree_serialize_spec6
  779. TestCxxPytree.test_pytree_serialize_spec7
  780. TestCxxPytree.test_pytree_serialize_spec8
  781. TestCxxPytree.test_pytree_serialize_spec9
  782. TestDelete.test_slices
  783. TestDeserialize.test_tensor_tensor_list
  784. TestDiag.test_failure
  785. TestDiff.test_append
  786. TestDiff.test_axis
  787. TestDiff.test_n
  788. TestDiff.test_prepend
  789. TestDistributionShapes.test_mixture_same_family_shape
  790. TestDLPack.test_dlpack_device
  791. TestDLPack.test_dtype_passthrough_dtype0
  792. TestDLPack.test_dtype_passthrough_dtype1
  793. TestDLPack.test_dtype_passthrough_dtype2
  794. TestDLPack.test_dtype_passthrough_dtype3
  795. TestDLPack.test_dtype_passthrough_dtype4
  796. TestDLPack.test_dtype_passthrough_dtype5
  797. TestDLPack.test_dtype_passthrough_dtype6
  798. TestDLPack.test_dtype_passthrough_dtype7
  799. TestDLPack.test_dtype_passthrough_dtype8
  800. TestDLPack.test_dtype_passthrough_dtype9
  801. TestDLPack.test_dunder_dlpack_refcount
  802. TestDLPack.test_from_dlpack_refcount
  803. TestDLPack.test_from_torch
  804. TestDLPack.test_higher_dims_ndim_0
  805. TestDLPack.test_higher_dims_ndim_1
  806. TestDLPack.test_higher_dims_ndim_10
  807. TestDLPack.test_higher_dims_ndim_11
  808. TestDLPack.test_higher_dims_ndim_12
  809. TestDLPack.test_higher_dims_ndim_13
  810. TestDLPack.test_higher_dims_ndim_14
  811. TestDLPack.test_higher_dims_ndim_15
  812. TestDLPack.test_higher_dims_ndim_16
  813. TestDLPack.test_higher_dims_ndim_17
  814. TestDLPack.test_higher_dims_ndim_18
  815. TestDLPack.test_higher_dims_ndim_19
  816. TestDLPack.test_higher_dims_ndim_2
  817. TestDLPack.test_higher_dims_ndim_20
  818. TestDLPack.test_higher_dims_ndim_21
  819. TestDLPack.test_higher_dims_ndim_22
  820. TestDLPack.test_higher_dims_ndim_23
  821. TestDLPack.test_higher_dims_ndim_24
  822. TestDLPack.test_higher_dims_ndim_25
  823. TestDLPack.test_higher_dims_ndim_26
  824. TestDLPack.test_higher_dims_ndim_27
  825. TestDLPack.test_higher_dims_ndim_28
  826. TestDLPack.test_higher_dims_ndim_29
  827. TestDLPack.test_higher_dims_ndim_3
  828. TestDLPack.test_higher_dims_ndim_30
  829. TestDLPack.test_higher_dims_ndim_31
  830. TestDLPack.test_higher_dims_ndim_32
  831. TestDLPack.test_higher_dims_ndim_4
  832. TestDLPack.test_higher_dims_ndim_5
  833. TestDLPack.test_higher_dims_ndim_6
  834. TestDLPack.test_higher_dims_ndim_7
  835. TestDLPack.test_higher_dims_ndim_8
  836. TestDLPack.test_higher_dims_ndim_9
  837. TestDLPack.test_ndim0
  838. TestDLPack.test_non_contiguous
  839. TestDropoutNN.test_invalid_dropout_p
  840. TestDsplit.test_0D_array
  841. TestDsplit.test_1D_array
  842. TestDsplit.test_2D_array
  843. TestDsplit.test_non_iterable
  844. TestDstack.test_non_iterable
  845. TestDynamicQuantizedOps.test_qlinear
  846. TestDynamismExpression.test_export_inline_constraints
  847. TestEigh.test_invalid
  848. TestEigvalsh.test_invalid
  849. TestEmbeddingNNDeviceTypeCPU.test_embedding_bag_bfloat16_cpu_int32_int32
  850. TestEmbeddingNNDeviceTypeCPU.test_embedding_bag_bfloat16_cpu_int32_int64
  851. TestEmbeddingNNDeviceTypeCPU.test_embedding_bag_bfloat16_cpu_int64_int32
  852. TestEmbeddingNNDeviceTypeCPU.test_embedding_bag_bfloat16_cpu_int64_int64
  853. TestEmbeddingNNDeviceTypeCPU.test_embedding_bag_device_cpu_int32_int32_bfloat16
  854. TestEmbeddingNNDeviceTypeCPU.test_embedding_bag_device_cpu_int32_int32_float16
  855. TestEmbeddingNNDeviceTypeCPU.test_embedding_bag_device_cpu_int32_int32_float32
  856. TestEmbeddingNNDeviceTypeCPU.test_embedding_bag_device_cpu_int32_int32_float64
  857. TestEmbeddingNNDeviceTypeCPU.test_embedding_bag_device_cpu_int32_int64_bfloat16
  858. TestEmbeddingNNDeviceTypeCPU.test_embedding_bag_device_cpu_int32_int64_float16
  859. TestEmbeddingNNDeviceTypeCPU.test_embedding_bag_device_cpu_int32_int64_float32
  860. TestEmbeddingNNDeviceTypeCPU.test_embedding_bag_device_cpu_int32_int64_float64
  861. TestEmbeddingNNDeviceTypeCPU.test_embedding_bag_device_cpu_int64_int32_bfloat16
  862. TestEmbeddingNNDeviceTypeCPU.test_embedding_bag_device_cpu_int64_int32_float16
  863. TestEmbeddingNNDeviceTypeCPU.test_embedding_bag_device_cpu_int64_int32_float32
  864. TestEmbeddingNNDeviceTypeCPU.test_embedding_bag_device_cpu_int64_int32_float64
  865. TestEmbeddingNNDeviceTypeCPU.test_embedding_bag_device_cpu_int64_int64_bfloat16
  866. TestEmbeddingNNDeviceTypeCPU.test_embedding_bag_device_cpu_int64_int64_float16
  867. TestEmbeddingNNDeviceTypeCPU.test_embedding_bag_device_cpu_int64_int64_float32
  868. TestEmbeddingNNDeviceTypeCPU.test_embedding_bag_device_cpu_int64_int64_float64
  869. TestEmbeddingNNDeviceTypeCPU.test_embedding_bag_dimension_errors_cpu
  870. TestEmbeddingNNDeviceTypeCPU.test_embedding_bag_half_cpu_int32_int32
  871. TestEmbeddingNNDeviceTypeCPU.test_embedding_bag_half_cpu_int32_int64
  872. TestEmbeddingNNDeviceTypeCPU.test_embedding_bag_half_cpu_int64_int32
  873. TestEmbeddingNNDeviceTypeCPU.test_embedding_bag_half_cpu_int64_int64
  874. TestEmbeddingNNDeviceTypeCPU.test_EmbeddingBag_per_sample_weights_and_no_offsets_cpu_int32_float32
  875. TestEmbeddingNNDeviceTypeCPU.test_EmbeddingBag_per_sample_weights_and_no_offsets_cpu_int32_float64
  876. TestEmbeddingNNDeviceTypeCPU.test_EmbeddingBag_per_sample_weights_and_no_offsets_cpu_int64_float32
  877. TestEmbeddingNNDeviceTypeCPU.test_EmbeddingBag_per_sample_weights_and_no_offsets_cpu_int64_float64
  878. TestExamplesCorrectnessCPU.test_maml_regression_mechanism_functional_call_cpu
  879. TestExamplesCorrectnessCPU.test_maml_regression_mechanism_make_functional_cpu
  880. TestExamplesCorrectnessCPU.test_resnet18_per_sample_grads_mechanism_functional_call_cpu
  881. TestExamplesCorrectnessCPU.test_resnet18_per_sample_grads_mechanism_make_functional_cpu
  882. TestExpandDims.test_axis_out_of_range
  883. TestExpandDims.test_repeated_axis
  884. TestExpandedWeightFunctionalCPU.test_expanded_weight_per_sample_grad_mean_nn_functional_conv1d_cpu_float64
  885. TestExpandedWeightFunctionalCPU.test_expanded_weight_per_sample_grad_mean_nn_functional_conv2d_cpu_float64
  886. TestExpandedWeightFunctionalCPU.test_expanded_weight_per_sample_grad_mean_nn_functional_conv3d_cpu_float64
  887. TestExpandedWeightFunctionalCPU.test_expanded_weight_per_sample_grad_mean_nn_functional_group_norm_cpu_float64
  888. TestExpandedWeightFunctionalCPU.test_expanded_weight_per_sample_grad_mean_nn_functional_instance_norm_cpu_float64
  889. TestExpandedWeightFunctionalCPU.test_expanded_weight_per_sample_grad_mean_nn_functional_layer_norm_cpu_float64
  890. TestExpandedWeightFunctionalCPU.test_expanded_weight_per_sample_grad_sum_nn_functional_conv1d_cpu_float64
  891. TestExpandedWeightFunctionalCPU.test_expanded_weight_per_sample_grad_sum_nn_functional_conv2d_cpu_float64
  892. TestExpandedWeightFunctionalCPU.test_expanded_weight_per_sample_grad_sum_nn_functional_conv3d_cpu_float64
  893. TestExpandedWeightFunctionalCPU.test_expanded_weight_per_sample_grad_sum_nn_functional_group_norm_cpu_float64
  894. TestExpandedWeightFunctionalCPU.test_expanded_weight_per_sample_grad_sum_nn_functional_instance_norm_cpu_float64
  895. TestExpandedWeightFunctionalCPU.test_expanded_weight_per_sample_grad_sum_nn_functional_layer_norm_cpu_float64
  896. TestExpandedWeightFunctionalCPU.test_expanded_weights_per_sample_grad_input_no_grad_nn_functional_conv1d_cpu_float64
  897. TestExpandedWeightFunctionalCPU.test_expanded_weights_per_sample_grad_input_no_grad_nn_functional_conv2d_cpu_float64
  898. TestExpandedWeightFunctionalCPU.test_expanded_weights_per_sample_grad_input_no_grad_nn_functional_conv3d_cpu_float64
  899. TestExpandedWeightFunctionalCPU.test_expanded_weights_per_sample_grad_input_no_grad_nn_functional_group_norm_cpu_float64
  900. TestExpandedWeightFunctionalCPU.test_expanded_weights_per_sample_grad_input_no_grad_nn_functional_instance_norm_cpu_float64
  901. TestExpandedWeightFunctionalCPU.test_expanded_weights_per_sample_grad_input_no_grad_nn_functional_layer_norm_cpu_float64
  902. TestExperiment.test_with_buffer_as_submodule
  903. TestExperimentalUtils.test_profiler_name_pattern
  904. TestExperimentalUtils.test_profiler_optimizer_single_tensor_pattern
  905. TestExperimentalUtils.test_profiler_pattern_match_helper
  906. TestExperimentalUtils.test_profiler_pattern_matcher_json_report
  907. TestExperimentalUtils.test_profiler_synchronized_dataloader_pattern
  908. TestExperimentalUtils.test_utils_compute_queue_depth_when_no_cuda_events
  909. TestExperimentalUtils.test_utils_compute_self_time
  910. TestExport.test_basic_non_strict_fake_tensor
  911. TestExport.test_basic_non_strict_real_tensor
  912. TestExport.test_cond_with_module_stack_export_with
  913. TestExport.test_constrain_size_in_eager
  914. TestExport.test_constrain_size_with_constrain_value
  915. TestExport.test_constrain_size_with_various_cases
  916. TestExport.test_external_call_non_strict_real_tensor
  917. TestExport.test_nn_module_stack
  918. TestExport.test_nn_module_stack_shared_submodule
  919. TestExport.test_non_strict_dynamic_shapes
  920. TestExport.test_non_strict_dynamic_shapes_suggested_fixes
  921. TestFakeQuantize.test_fq_module_per_channel
  922. TestFakeQuantizeOps.test_backward_per_channel
  923. TestFakeQuantizeOps.test_backward_per_tensor
  924. TestFakeQuantizeOps.test_fake_quant_per_channel_qparam_range
  925. TestFakeQuantizeOps.test_fixed_qparams_fq_module
  926. TestFakeQuantizeOps.test_forward_per_channel
  927. TestFakeQuantizeOps.test_forward_per_tensor
  928. TestFakeQuantizeOps.test_fq_module_per_tensor
  929. TestFakeQuantizeOps.test_learnable_backward_per_channel_cpu
  930. TestFakeQuantizeOps.test_learnable_backward_per_channel_cuda
  931. TestFakeQuantizeOps.test_learnable_backward_per_tensor_cpu
  932. TestFakeQuantizeOps.test_learnable_backward_per_tensor_cuda
  933. TestFakeQuantizeOps.test_learnable_forward_per_channel_cpu
  934. TestFakeQuantizeOps.test_learnable_forward_per_channel_cuda
  935. TestFakeQuantizeOps.test_learnable_forward_per_tensor_cpu
  936. TestFakeQuantizeOps.test_learnable_forward_per_tensor_cuda
  937. TestFakeSparsity.test_jit_trace
  938. TestFakeSparsity.test_masking_logic
  939. TestFlag.test_writeable_from_readonly
  940. TestFlip.test_axes
  941. TestFliplr.test_basic
  942. TestFloatNonIntegerArgument.test_non_integer_argument_errors
  943. TestFloatNonIntegerArgument.test_reduce_axis_float_index
  944. TestFromBuffer.test_basic_little_dtype0
  945. TestFromBuffer.test_basic_little_dtype1
  946. TestFromBuffer.test_basic_little_dtype2
  947. TestFromBuffer.test_empty
  948. TestFunctionalizeCPU.test_multioutput_view_cpu
  949. TestFunctionalizeCPU.test_simple_view_cpu
  950. TestFunctors.test_cat_transform
  951. TestFunctors.test_cat_transform_non_uniform
  952. TestFusionEval.test_fuse_module_eval_numerics
  953. TestFX.test_custom_traceback_raised_when_exception_source_is_graphmodule
  954. TestFX.test_immutable_dict_pytree_ops
  955. TestFX.test_immutable_list_pytree_ops
  956. TestFX.test_pytree_concrete
  957. TestFXAPIBackwardCompatibility.test_public_api_surface
  958. TestFxComposability.test_q_prep_fx_before_s_prep
  959. TestFxComposability.test_q_prep_fx_s_prep_ref_conv
  960. TestFxComposability.test_s_prep_before_q_prep_fx
  961. TestFxComposability.test_s_prep_before_qat_prep_fx
  962. TestFxComposability.test_s_prep_q_prep_fx_ref
  963. TestFXGraphMatcherModels.test_mobilenet_v2_qat
  964. TestGenerateNumericDebugHandle.test_quantize_pt2e_preserve_handle
  965. TestGenerateOpcheckTests.test_opcheck_bad_op
  966. TestGenericCumSumProd.test_bad_axis_func0
  967. TestGenericCumSumProd.test_bad_axis_func1
  968. TestGenericPytree.test_flatten_unflatten_deque_cxx
  969. TestGenericPytree.test_flatten_unflatten_deque_py
  970. TestGenericReductions.test_bad_axis_func0
  971. TestGenericReductions.test_bad_axis_func1
  972. TestGenericReductions.test_bad_axis_func10
  973. TestGenericReductions.test_bad_axis_func11
  974. TestGenericReductions.test_bad_axis_func2
  975. TestGenericReductions.test_bad_axis_func3
  976. TestGenericReductions.test_bad_axis_func4
  977. TestGenericReductions.test_bad_axis_func5
  978. TestGenericReductions.test_bad_axis_func6
  979. TestGenericReductions.test_bad_axis_func7
  980. TestGenericReductions.test_bad_axis_func8
  981. TestGenericReductions.test_bad_axis_func9
  982. TestGradCheckOverride.test_gradcheck
  983. TestGradient.test_badargs
  984. TestGradient.test_second_order_accurate
  985. TestGradient.test_specific_axes
  986. TestGradient.test_values
  987. TestGradNewOnesOverride.test_newones
  988. TestGradTransformCPU.test_grad_of_vjp_composition_cpu
  989. TestGradTransformCPU.test_no_grad_outside_vjp_cpu
  990. TestGradTransformCPU.test_no_grad_outside_vjp_fn_cpu
  991. TestGradTransformCPU.test_no_grad_outside_vjp_only_cpu
  992. TestGradTransformCPU.test_unrelated_vjp_cpu
  993. TestGradTransformCPU.test_unrelated_vjp_multiple_inputs_outputs_cpu
  994. TestGradTransformCPU.test_vjp_aux_pytree_cpu
  995. TestGradTransformCPU.test_vjp_cpu
  996. TestGradTransformCPU.test_vjp_of_grad_composition_cpu
  997. TestGradTransformCPU.test_vjp_pytree_input_cpu
  998. TestGradTransformCPU.test_vjp_pytree_output_cpu
  999. TestGradTransformCPU.test_vjp_two_outputs_cpu
  1000. TestGraph.test_traverse_circular_datapipe
  1001. TestGraph.test_traverse_unhashable_datapipe
  1002. TestHistogram.test_error_binnum_type
  1003. TestHistogram.test_finite_range
  1004. TestHistogram.test_one_bin
  1005. TestHistogram.test_precision
  1006. TestHistogram.test_unsigned_monotonicity_check
  1007. TestHistogram2d.test_all_outliers
  1008. TestHistogram2d.test_asym
  1009. TestHistogram2d.test_binparameter_combination
  1010. TestHistogramdd.test_bins_error_2
  1011. TestHistogramdd.test_bins_errors
  1012. TestHistogramdd.test_equal_edges
  1013. TestHistogramdd.test_finite_range
  1014. TestHistogramdd.test_inf_edges
  1015. TestHistogramdd.test_simple
  1016. TestHistogramdd.test_weights
  1017. TestHsplit.test_0D_array
  1018. TestHsplit.test_non_iterable
  1019. TestHstack.test_empty_input
  1020. TestHstack.test_non_iterable
  1021. TestImports.test_circular_dependencies
  1022. TestIndexing.test_index_no_floats
  1023. TestIndexing.test_slicing_no_floats
  1024. TestIndexingCPU.test_empty_ndim_index_bool_cpu
  1025. TestIndexingCPU.test_index_cpu
  1026. TestIndexingCPU.test_index_limits_cpu
  1027. TestIndexingCPU.test_invalid_index_cpu
  1028. TestIndexingCPU.test_out_of_bound_index_cpu
  1029. TestIndexingCPU.test_zero_dim_index_cpu
  1030. TestInputAttrTracking.test_complex_attr_access_without_graph_breaks
  1031. TestInputAttrTracking.test_const_property_on_tensor
  1032. TestInputAttrTracking.test_set_data_on_input_tensor
  1033. TestInputAttrTracking.test_tensor_property_assigned_on_tensor
  1034. TestInputAttrTracking.test_tensor_property_on_tensor
  1035. TestIsScalar.test_is_not_scalar_value6
  1036. TestIterator.test_iterator
  1037. TestIterDataPipeGraphFastForward.test_simple_snapshot_custom_non_generator
  1038. TestIterDataPipeGraphFastForward.test_simple_snapshot_custom_self_next
  1039. TestJacCPU.test_against_reference_correctness_different_devices_cpu
  1040. TestJacCPU.test_against_reference_default_arg_cpu
  1041. TestJacCPU.test_against_reference_multi_input_cpu
  1042. TestJacCPU.test_against_reference_multi_input_multi_output_cpu
  1043. TestJacCPU.test_against_reference_simple_cpu
  1044. TestJacCPU.test_against_reference_unrelated_outputs_cpu
  1045. TestJacCPU.test_against_reference_zero_dim_cpu
  1046. TestJacCPU.test_argnums_defaults_to_zero_cpu
  1047. TestJacCPU.test_argnums_effect_on_return_jacfwd_cpu
  1048. TestJacCPU.test_argnums_effect_on_return_jacrev_cpu
  1049. TestJacCPU.test_argnums_tuple_jacfwd_cpu
  1050. TestJacCPU.test_argnums_tuple_jacrev_cpu
  1051. TestJacCPU.test_aux_pytree_cpu
  1052. TestJacCPU.test_dimensionality_cpu
  1053. TestJacCPU.test_empty_argnums_jacfwd_cpu
  1054. TestJacCPU.test_empty_argnums_jacrev_cpu
  1055. TestJacCPU.test_empty_output_cpu
  1056. TestJacCPU.test_float_argnums_jacfwd_cpu
  1057. TestJacCPU.test_float_argnums_jacrev_cpu
  1058. TestJacCPU.test_inplace_cpu
  1059. TestJacCPU.test_jac_with_non_tensor_args_cpu
  1060. TestJacCPU.test_multiple_args_jacfwd_cpu
  1061. TestJacCPU.test_multiple_args_jacrev_cpu
  1062. TestJacCPU.test_multiple_inputs_outputs_pytree_cpu
  1063. TestJacCPU.test_multiple_inputs_pytree_cpu
  1064. TestJacCPU.test_multiple_outputs_multiple_argnums_cpu
  1065. TestJacCPU.test_multiple_outputs_single_argnums_cpu
  1066. TestJacCPU.test_negative_argnums_jacfwd_cpu
  1067. TestJacCPU.test_negative_argnums_jacrev_cpu
  1068. TestJacCPU.test_out_of_bounds_argnums_jacfwd_cpu
  1069. TestJacCPU.test_out_of_bounds_argnums_jacrev_cpu
  1070. TestJacCPU.test_outputs_can_any_pytree_cpu
  1071. TestJacCPU.test_repeated_argnums_jacfwd_cpu
  1072. TestJacCPU.test_repeated_argnums_jacrev_cpu
  1073. TestJacCPU.test_simple_jacfwd_cpu
  1074. TestJacCPU.test_simple_jacrev_cpu
  1075. TestJacCPU.test_simple_not_flat_jacfwd_cpu
  1076. TestJacCPU.test_simple_not_flat_jacrev_cpu
  1077. TestJacCPU.test_unrelated_input_cpu
  1078. TestJacCPU.test_unrelated_output_cpu
  1079. TestJacCPU.test_vmap_on_jac_simple_jacfwd_cpu
  1080. TestJacCPU.test_vmap_on_jac_simple_jacrev_cpu
  1081. TestJit.test_batchnorm
  1082. TestJit.test_function_default_values
  1083. TestJvpCPU.test_disable_fwd_grad_outside_cpu
  1084. TestJvpCPU.test_inputs_are_tuples_of_tensors_cpu
  1085. TestJvpCPU.test_jvp_inside_autograd_function_cpu
  1086. TestJvpCPU.test_nonempty_primals_and_tangents_cpu
  1087. TestJvpCPU.test_primals_tangents_length_mismatch_cpu
  1088. TestJvpCPU.test_simple_cpu
  1089. TestLazyModules.test_lazy_batchnorm_with_dict_input
  1090. TestLazyModules.test_lazy_conv1d
  1091. TestLazyModules.test_lazy_conv1d_pickle
  1092. TestLazyModules.test_lazy_conv2d
  1093. TestLazyModules.test_lazy_conv2d_pickle
  1094. TestLazyModules.test_lazy_conv3d
  1095. TestLazyModules.test_lazy_conv3d_pickle
  1096. TestLazyModules.test_lazy_conv_transpose1d_pickle
  1097. TestLazyModules.test_lazy_conv_transpose2d
  1098. TestLazyModules.test_lazy_conv_transpose2d_pickle
  1099. TestLazyModules.test_lazy_conv_transpose3d
  1100. TestLazyModules.test_lazy_conv_transpose3d_pickle
  1101. TestLazyModules.test_lazy_conv_transposed1d
  1102. TestLazyModules.test_lazy_linear_pickle
  1103. TestLazyModules.test_linear
  1104. TestLexsort.test_datetime
  1105. TestLexsort.test_mixed
  1106. TestLinalgCPU.test_addmm_sizes_cpu_float32
  1107. TestLinalgCPU.test_addmm_sizes_cpu_float64
  1108. TestLinalgCPU.test_addr_integral_cpu_int16
  1109. TestLinalgCPU.test_addr_integral_cpu_int32
  1110. TestLinalgCPU.test_addr_integral_cpu_int64
  1111. TestLinalgCPU.test_addr_integral_cpu_int8
  1112. TestLinalgCPU.test_addr_integral_cpu_uint8
  1113. TestLinalgCPU.test_geqrf_cpu_complex128
  1114. TestLinalgCPU.test_geqrf_cpu_complex64
  1115. TestLinalgCPU.test_geqrf_cpu_float32
  1116. TestLinalgCPU.test_geqrf_cpu_float64
  1117. TestLinalgCPU.test_householder_product_cpu_complex128
  1118. TestLinalgCPU.test_householder_product_cpu_complex64
  1119. TestLinalgCPU.test_householder_product_cpu_float32
  1120. TestLinalgCPU.test_householder_product_cpu_float64
  1121. TestLinalgCPU.test_inverse_cpu_complex128
  1122. TestLinalgCPU.test_inverse_cpu_complex64
  1123. TestLinalgCPU.test_inverse_cpu_float32
  1124. TestLinalgCPU.test_inverse_cpu_float64
  1125. TestLinalgCPU.test_lobpcg_torchscript_cpu_float64
  1126. TestLinalgCPU.test_matmul_small_brute_force_3d_Nd_cpu_complex64
  1127. TestLinalgCPU.test_matmul_small_brute_force_3d_Nd_cpu_float32
  1128. TestLinalgCPU.test_matmul_small_brute_force_3d_Nd_cpu_int64
  1129. TestLinalgCPU.test_norm_bfloat16_and_half_cpu_bfloat16
  1130. TestLinalgCPU.test_norm_bfloat16_and_half_cpu_float16
  1131. TestLinalgCPU.test_norm_dtype_cpu_complex128
  1132. TestLinalgCPU.test_norm_dtype_cpu_float64
  1133. TestLinalgCPU.test_norm_vector_cpu_float32
  1134. TestLinalgCPU.test_norm_vector_cpu_float64
  1135. TestLinalgCPU.test_pinv_cpu_complex128
  1136. TestLinalgCPU.test_pinv_cpu_complex64
  1137. TestLinalgCPU.test_pinv_cpu_float32
  1138. TestLinalgCPU.test_pinv_cpu_float64
  1139. TestLinalgCPU.test_slogdet_errors_and_warnings_cpu_complex128
  1140. TestLinalgCPU.test_slogdet_errors_and_warnings_cpu_complex64
  1141. TestLinalgCPU.test_slogdet_errors_and_warnings_cpu_float32
  1142. TestLinalgCPU.test_slogdet_errors_and_warnings_cpu_float64
  1143. TestLinalgCPU.test_solve_cpu_complex128
  1144. TestLinalgCPU.test_solve_cpu_complex64
  1145. TestLinalgCPU.test_solve_cpu_float32
  1146. TestLinalgCPU.test_solve_cpu_float64
  1147. TestMatmul.test_exceptions
  1148. TestMedian.test_extended_axis_invalid
  1149. TestMeshgrid.test_indexing
  1150. TestMeshgrid.test_invalid_arguments
  1151. TestMetaKernel.test_addmm_invalid_dtype
  1152. TestMin.test_adapt
  1153. TestMin.test_attn
  1154. TestMin.test_big_split
  1155. TestMin.test_c
  1156. TestMin.test_compare_dims
  1157. TestMin.test_diag
  1158. TestMin.test_dim_args
  1159. TestMin.test_dims_with_size
  1160. TestMin.test_dir
  1161. TestMin.test_doc
  1162. TestMin.test_embed
  1163. TestMin.test_eq
  1164. TestMin.test_expand
  1165. TestMin.test_functorch
  1166. TestMin.test_hello
  1167. TestMin.test_index
  1168. TestMin.test_index_placement
  1169. TestMin.test_inplace
  1170. TestMin.test_manual_stuff
  1171. TestMin.test_mask
  1172. TestMin.test_max
  1173. TestMin.test_mm
  1174. TestMin.test_mm_fuse
  1175. TestMin.test_monkey
  1176. TestMin.test_network
  1177. TestMin.test_order
  1178. TestMin.test_order_keyword
  1179. TestMin.test_parse
  1180. TestMin.test_permute_orig
  1181. TestMin.test_seg
  1182. TestMin.test_simple
  1183. TestMin.test_softmax_split
  1184. TestMin.test_stack
  1185. TestMin.test_time_mm_fuse
  1186. TestMin.test_with_dims_split
  1187. TestMinFunctorchOnly.test_adapt
  1188. TestMinFunctorchOnly.test_attn
  1189. TestMinFunctorchOnly.test_big_split
  1190. TestMinFunctorchOnly.test_c
  1191. TestMinFunctorchOnly.test_compare_dims
  1192. TestMinFunctorchOnly.test_diag
  1193. TestMinFunctorchOnly.test_dim_args
  1194. TestMinFunctorchOnly.test_dims_with_size
  1195. TestMinFunctorchOnly.test_dir
  1196. TestMinFunctorchOnly.test_doc
  1197. TestMinFunctorchOnly.test_embed
  1198. TestMinFunctorchOnly.test_eq
  1199. TestMinFunctorchOnly.test_expand
  1200. TestMinFunctorchOnly.test_functorch
  1201. TestMinFunctorchOnly.test_hello
  1202. TestMinFunctorchOnly.test_index
  1203. TestMinFunctorchOnly.test_index_placement
  1204. TestMinFunctorchOnly.test_inplace
  1205. TestMinFunctorchOnly.test_manual_stuff
  1206. TestMinFunctorchOnly.test_mask
  1207. TestMinFunctorchOnly.test_max
  1208. TestMinFunctorchOnly.test_mm
  1209. TestMinFunctorchOnly.test_mm_fuse
  1210. TestMinFunctorchOnly.test_monkey
  1211. TestMinFunctorchOnly.test_network
  1212. TestMinFunctorchOnly.test_order
  1213. TestMinFunctorchOnly.test_order_keyword
  1214. TestMinFunctorchOnly.test_parse
  1215. TestMinFunctorchOnly.test_permute_orig
  1216. TestMinFunctorchOnly.test_seg
  1217. TestMinFunctorchOnly.test_simple
  1218. TestMinFunctorchOnly.test_softmax_split
  1219. TestMinFunctorchOnly.test_stack
  1220. TestMinFunctorchOnly.test_with_dims_split
  1221. TestMinMax.test_axis
  1222. TestMinMax.test_scalar
  1223. TestMisc.test_byteorder_check
  1224. TestMisc.test_generalized_raise_multiloop
  1225. TestMkldnnCPU.test_add_cpu
  1226. TestMkldnnCPU.test_batch_norm_2d_cpu
  1227. TestMkldnnCPU.test_batch_norm_3d_cpu
  1228. TestMkldnnCPU.test_conv1d_cpu
  1229. TestMkldnnCPU.test_conv2d_cpu
  1230. TestMkldnnCPU.test_conv3d_cpu
  1231. TestMkldnnCPU.test_copy_cpu
  1232. TestMkldnnCPU.test_linear_cpu
  1233. TestMkldnnCPU.test_linear_lowp_cpu_float16
  1234. TestMkldnnCPU.test_mul_cpu
  1235. TestMkldnnCPU.test_prelu_bf16_cpu
  1236. TestMkldnnCPU.test_prelu_cpu
  1237. TestMkldnnCPU.test_reshape_blocked_format_cpu
  1238. TestMkldnnCPU.test_resnet18_cpu
  1239. TestMkldnnCPU.test_resnext50_32x4d_cpu
  1240. TestMkldnnCPU.test_sigmoid_cpu
  1241. TestMkldnnCPU.test_tanh_cpu
  1242. TestModuleHookNN.test_hook_inplace
  1243. TestMoveaxis.test_errors
  1244. TestMultiDot.test_dynamic_programming_optimization_and_out
  1245. TestMultiDot.test_three_arguments_and_out
  1246. TestMultiDot.test_too_few_input_arrays
  1247. TestMultiDot.test_two_arguments_and_out
  1248. TestMultiprocessing.test_empty_shared
  1249. TestMultiprocessing.test_fs_is_shared
  1250. TestMultiprocessing.test_inherit_tensor
  1251. TestMultiprocessing.test_is_shared
  1252. TestNamedTensor.test_addcmul_addcdiv
  1253. TestNamedTensor.test_addmm
  1254. TestNamedTensor.test_addmv
  1255. TestNamedTensor.test_any_all
  1256. TestNamedTensor.test_as_strided
  1257. TestNamedTensor.test_autograd_ignores_names
  1258. TestNamedTensor.test_autograd_smoke
  1259. TestNamedTensor.test_autograd_warns_named_grad
  1260. TestNamedTensor.test_bernoulli
  1261. TestNamedTensor.test_big_tensor_repr_has_names
  1262. TestNamedTensor.test_binary_ops
  1263. TestNamedTensor.test_bitwise_not
  1264. TestNamedTensor.test_bmm
  1265. TestNamedTensor.test_cat
  1266. TestNamedTensor.test_cdist
  1267. TestNamedTensor.test_comparison_ops
  1268. TestNamedTensor.test_copy_transpose
  1269. TestNamedTensor.test_cummax_cummin
  1270. TestNamedTensor.test_detach
  1271. TestNamedTensor.test_diagonal
  1272. TestNamedTensor.test_dot
  1273. TestNamedTensor.test_equal
  1274. TestNamedTensor.test_expand
  1275. TestNamedTensor.test_factory_coverage
  1276. TestNamedTensor.test_factory_edge_cases
  1277. TestNamedTensor.test_flatten
  1278. TestNamedTensor.test_has_names
  1279. TestNamedTensor.test_index_fill
  1280. TestNamedTensor.test_info_smoke
  1281. TestNamedTensor.test_logcumsumexp
  1282. TestNamedTensor.test_logical_not
  1283. TestNamedTensor.test_logical_ops
  1284. TestNamedTensor.test_masked_fill
  1285. TestNamedTensor.test_masked_select
  1286. TestNamedTensor.test_matmul
  1287. TestNamedTensor.test_max_pooling
  1288. TestNamedTensor.test_mm
  1289. TestNamedTensor.test_mv
  1290. TestNamedTensor.test_no_multiprocessing_support
  1291. TestNamedTensor.test_no_pickle_support
  1292. TestNamedTensor.test_no_save_support
  1293. TestNamedTensor.test_noncontig_contiguous
  1294. TestNamedTensor.test_out_fn_semantics
  1295. TestNamedTensor.test_pow_special
  1296. TestNamedTensor.test_reduction_fns
  1297. TestNamedTensor.test_rename
  1298. TestNamedTensor.test_rename_
  1299. TestNamedTensor.test_rename_globber
  1300. TestNamedTensor.test_rename_rename_map
  1301. TestNamedTensor.test_resize
  1302. TestNamedTensor.test_select
  1303. TestNamedTensor.test_set_names_property
  1304. TestNamedTensor.test_size
  1305. TestNamedTensor.test_split_fns_propagates_names
  1306. TestNamedTensor.test_stride
  1307. TestNamedTensor.test_tensor_from_named_tensor
  1308. TestNamedTensor.test_tensor_grad_is_unnamed
  1309. TestNamedTensor.test_transpose_variants
  1310. TestNamedTensor.test_unary_propagate_names_fns
  1311. TestNamedTensor.test_unflatten
  1312. TestNamedTensor.test_unsupported_op_error_msg
  1313. TestNamedTensor.test_using_seen_interned_string_doesnt_bump_refcount
  1314. TestNamedTensor.test_using_unseen_interned_string_bumps_refcount_permanently
  1315. TestNamedTensor.test_using_unseen_uninterned_string_refcounts
  1316. TestNamedTuple.test_max
  1317. TestNamedTupleAPI.test_namedtuple_return
  1318. TestNativeFunctions.test_optional_floatlist_invalid
  1319. TestNativeFunctions.test_optional_intlist_invalid
  1320. TestNativeFunctions.test_symintlist_error
  1321. TestNativeFunctions.test_symintlist_error_with_overload
  1322. TestNativeFunctions.test_vararg_symintlist_error
  1323. TestNegative.test_exceptions
  1324. TestNestedCheckpoint.test_nested_checkpoint_early_stop_False
  1325. TestNestedCheckpoint.test_nested_checkpoint_early_stop_True
  1326. TestNestedCheckpoint.test_nested_checkpoint_kwargs_early_stop_False
  1327. TestNestedCheckpoint.test_nested_checkpoint_kwargs_early_stop_True
  1328. TestNestedCheckpoint.test_nested_checkpoint_non_tensor_inputs_and_outputs_early_stop_False
  1329. TestNestedCheckpoint.test_nested_checkpoint_non_tensor_inputs_and_outputs_early_stop_True
  1330. TestNestedCheckpoint.test_nested_checkpoint_reentrant_backwards_early_stop_False
  1331. TestNestedCheckpoint.test_nested_checkpoint_reentrant_backwards_early_stop_True
  1332. TestNestedCheckpoint.test_nested_checkpoint_same_graph_early_stop_False
  1333. TestNestedCheckpoint.test_nested_checkpoint_same_graph_early_stop_True
  1334. TestNestedCheckpoint.test_nested_checkpoint_set_early_stop
  1335. TestNestedCheckpoint.test_nested_checkpoint_set_early_stop_no_recompution_needed
  1336. TestNestedCheckpoint.test_nested_checkpoint_two_children_early_stop_False
  1337. TestNestedCheckpoint.test_nested_checkpoint_two_children_early_stop_True
  1338. TestNestedTensor.test_binary_recompiles
  1339. TestNestedTensor.test_unbind
  1340. TestNN.test_fb_fc_packed
  1341. TestNN.test_overwrite_module_params_on_conversion
  1342. TestNN.test_ParameterDict_replication
  1343. TestNN.test_ParameterList_replication
  1344. TestNN.test_Sequential_iadd
  1345. TestNNDeviceTypeCPU.test_invalid_reduction_strings_cpu
  1346. TestNNDeviceTypeCPU.test_module_to_empty_cpu_float32
  1347. TestNNDeviceTypeCPU.test_module_to_empty_cpu_float64
  1348. TestNNDeviceTypeCPU.test_nll_loss_all_ignored_cpu
  1349. TestNNDeviceTypeCPU.test_nll_loss_byte_target_matches_long_cpu
  1350. TestNNDeviceTypeCPU.test_nll_loss_empty_tensor_reduction_mean_cpu
  1351. TestNNDeviceTypeCPU.test_nll_loss_empty_tensor_reduction_none_cpu
  1352. TestNNDeviceTypeCPU.test_nll_loss_empty_tensor_reduction_sum_cpu
  1353. TestNNDeviceTypeCPU.test_nll_loss_total_weight_is_zero_cpu
  1354. TestNNDeviceTypeCPU.test_threshold_inplace_overlap_cpu
  1355. TestNNDeviceTypeCPU.test_triplet_margin_with_distance_loss_cpu
  1356. TestNNInit.test_xavier_uniform
  1357. TestNNParametrization.test_deepcopy_after_parametrization
  1358. TestNNParametrization.test_new_spectral_norm
  1359. TestNNParametrization.test_new_spectral_norm_dim
  1360. TestNNParametrization.test_new_spectral_norm_forward
  1361. TestNNParametrization.test_weight_norm_deepcopy
  1362. TestNNParametrization.test_weight_norm_state_dict_compat
  1363. TestNonarrayArgs.test_dunder_round_edgecases_val_2147483647_ndigits_-1
  1364. TestNonarrayArgs.test_dunder_round_edgecases_val_2147483647_ndigits_-10
  1365. TestNonarrayArgs.test_dunder_round_edgecases_val_2147483647_ndigits_-9
  1366. TestNonzeroAndCountNonzero.test_count_nonzero_axis
  1367. TestNormDouble.test_axis
  1368. TestNormDouble.test_bad_args
  1369. TestNormDouble.test_matrix_2x2
  1370. TestNormInt64.test_axis
  1371. TestNormInt64.test_bad_args
  1372. TestNormInt64.test_matrix_2x2
  1373. TestNormSingle.test_axis
  1374. TestNormSingle.test_bad_args
  1375. TestNormSingle.test_matrix_2x2
  1376. TestNumPyInteropCPU.test_numpy_non_writeable_cpu
  1377. TestObserver.test_per_channel_observers
  1378. TestObserver.test_per_tensor_observers
  1379. TestOldViewOpsCPU.test_crow_col_indices_cpu
  1380. TestOpCPU.test_cat_cpu_bfloat16
  1381. TestOpCPU.test_cat_cpu_float32
  1382. TestOptimizations.test_example_inputs
  1383. TestOptimizations.test_example_inputs_runtime_use
  1384. TestOutDtypeOp.test_out_dtype_non_op_overload
  1385. TestOutDtypeOp.test_out_dtype_wrong_output
  1386. TestPadding.test_constant_padNd
  1387. TestPartitioning.test_contiguous
  1388. TestPartitioning.test_default_partitioner_output_tensor_shape_tensor
  1389. TestPartitioning.test_min_cut_partitioner_output_tensor_shape_tensor
  1390. TestPasses.test_functionalize_inline_contraints
  1391. TestPasses.test_views_op_having_view_copy
  1392. TestPassManager.test_pass_manager
  1393. TestPercentile.test_extended_axis_invalid
  1394. TestPercentile.test_scalar_q
  1395. TestPickle.test_pickle
  1396. TestPoolingNN.test_MaxUnpool2d_output_size
  1397. TestPower.test_modular_power
  1398. TestProfiler.test_concrete_inputs_profiling
  1399. TestProfiler.test_export_stacks
  1400. TestProfiler.test_flops
  1401. TestProfiler.test_guarded_record_function_fast
  1402. TestProfiler.test_high_level_trace
  1403. TestProfiler.test_is_profiler_enabled
  1404. TestProfiler.test_profiler_correlation_id
  1405. TestProfiler.test_profiler_fwd_bwd_link
  1406. TestProfiler.test_profiler_op_event_args
  1407. TestProfiler.test_profiler_tracing
  1408. TestProfiler.test_source
  1409. TestProfiler.test_source_multithreaded_basic_work_in_main_thread_True
  1410. TestProfiler.test_source_multithreaded_close_in_scope_work_in_main_thread_True
  1411. TestProfiler.test_source_multithreaded_complex_work_in_main_thread_True
  1412. TestProfiler.test_source_multithreaded_multiple_preexisting_work_in_main_thread_True
  1413. TestProfiler.test_source_multithreaded_open_in_scope_work_in_main_thread_True
  1414. TestProfilerTree.test_profiler_experimental_tree_with_memory
  1415. TestProfilerTree.test_profiler_experimental_tree_with_memory_and_stack
  1416. TestProfilerTree.test_profiler_experimental_tree_with_record_function
  1417. TestProfilerTree.test_profiler_experimental_tree_with_stack_and_torch_dispatch
  1418. TestProfilerTree.test_profiler_experimental_tree_with_stack_and_torch_function
  1419. TestPT2ERepresentation.test_add
  1420. TestPT2ERepresentation.test_add_relu
  1421. TestPT2ERepresentation.test_conv2d
  1422. TestPT2ERepresentation.test_dynamic_linear
  1423. TestPT2ERepresentation.test_maxpool2d
  1424. TestPT2ERepresentation.test_qdq
  1425. TestPT2ERepresentation.test_qdq_per_channel
  1426. TestPT2ERepresentation.test_static_linear
  1427. TestPutAlongAxis.test_broadcast
  1428. TestPutmask.test_byteorder_greater_False
  1429. TestPutmask.test_byteorder_greater_True
  1430. TestPutmask.test_record_array
  1431. TestPythonAutograd.test_backwards2
  1432. TestPythonAutograd.test_forwards1
  1433. TestPythonAutograd.test_forwards2
  1434. TestPythonAutograd.test_split
  1435. TestPythonDispatch.test_make_subclass_with_modes
  1436. TestPythonPytree.test_treespec_equality
  1437. TestPythonRegistration.test_alias_analysis
  1438. TestPythonRegistration.test_finalizer
  1439. TestQNNPackOps.test_qnnpack_add
  1440. TestQNNPackOps.test_qnnpack_maxpool2d
  1441. TestQNNPackOps.test_qnnpack_mul
  1442. TestQNNPackOps.test_qnnpack_relu
  1443. TestQR.test_mode_raw
  1444. TestQR.test_qr_empty_m_0_n_0
  1445. TestQR.test_qr_empty_m_0_n_3
  1446. TestQR.test_qr_empty_m_3_n_0
  1447. TestQuantile.test_quantile_monotonic_method_averaged_inverted_cdf
  1448. TestQuantile.test_quantile_monotonic_method_closest_observation
  1449. TestQuantile.test_quantile_monotonic_method_hazen
  1450. TestQuantile.test_quantile_monotonic_method_interpolated_inverted_cdf
  1451. TestQuantile.test_quantile_monotonic_method_inverted_cdf
  1452. TestQuantile.test_quantile_monotonic_method_median_unbiased
  1453. TestQuantile.test_quantile_monotonic_method_normal_unbiased
  1454. TestQuantile.test_quantile_monotonic_method_weibull
  1455. TestQuantizationDocs.test_quantization_doc_fx
  1456. TestQuantizationDocs.test_quantization_doc_qat
  1457. TestQuantizedConv.test_qconv1d_unpack
  1458. TestQuantizedConv.test_qconv2d_unpack
  1459. TestQuantizedConv.test_qconv3d_unpack
  1460. TestQuantizedEmbeddingOps.test_embedding_bag_2bit
  1461. TestQuantizedEmbeddingOps.test_embedding_bag_2d_indices
  1462. TestQuantizedEmbeddingOps.test_embedding_bag_4bit
  1463. TestQuantizedEmbeddingOps.test_embedding_bag_byte
  1464. TestQuantizedLinear.test_qlinear_qnnpack_free_memory_and_unpack
  1465. TestQuantizedLinear.test_qlinear_unpack
  1466. TestQuantizedOps.test_add_scalar_relu
  1467. TestQuantizedOps.test_avg_pool2d
  1468. TestQuantizedOps.test_avg_pool2d_nhwc
  1469. TestQuantizedOps.test_avg_pool3d
  1470. TestQuantizedOps.test_avg_pool3d_nhwc
  1471. TestQuantizedOps.test_cat
  1472. TestQuantizedOps.test_cat_nhwc
  1473. TestQuantizedOps.test_channel_shuffle
  1474. TestQuantizedOps.test_custom_module_lstm
  1475. TestQuantizedOps.test_equal
  1476. TestQuantizedOps.test_hardtanh
  1477. TestQuantizedOps.test_interpolate
  1478. TestQuantizedOps.test_interpolate3d
  1479. TestQuantizedOps.test_leaky_relu_observed_output
  1480. TestQuantizedOps.test_max_pool1d
  1481. TestQuantizedOps.test_max_pool2d
  1482. TestQuantizedOps.test_max_pool2d_cudnn
  1483. TestQuantizedOps.test_max_pool2d_nhwc
  1484. TestQuantizedOps.test_mul_scalar_relu
  1485. TestQuantizedOps.test_qcelu
  1486. TestQuantizedOps.test_qclamp
  1487. TestQuantizedOps.test_qelu
  1488. TestQuantizedOps.test_qtanh
  1489. TestQuantizedOps.test_qthreshold
  1490. TestQuantizedOps.test_sigmoid
  1491. TestQuantizedOps.test_sigmoid_non_observed
  1492. TestQuantizedTensor.test_choose_qparams
  1493. TestQuantizedTensor.test_choose_qparams_optimized
  1494. TestQuantizedTensor.test_decomposed_dequantize_per_tensor
  1495. TestQuantizedTensor.test_decomposed_quantize_per_tensor
  1496. TestQuantizedTensor.test_decomposed_quantize_per_tensor_bfloat16_input
  1497. TestQuantizedTensor.test_fp16_saturate_op
  1498. TestQuantizedTensor.test_per_channel_qtensor_to_memory_format
  1499. TestQuantizedTensor.test_per_tensor_qtensor_to_memory_format
  1500. TestQuantizedTensor.test_qtensor_cpu
  1501. TestQuantizedTensor.test_quantize_per_channel_sub_byte
  1502. TestQuantizeEagerQATNumerics.test_conv_bn_folded_vs_unfolded
  1503. TestQuantizeFx.test_prepare_custom_config_set_standalone_module_class
  1504. TestQuantizeFx.test_static_lstm_with_custom_fixed_qparams
  1505. TestQuantizePT2E.test_composable_quantizer_linear_conv
  1506. TestQuantizePT2E.test_constant_prop_preserve_metadata
  1507. TestQuantizePT2E.test_derived_qspec
  1508. TestQuantizePT2E.test_embedding_conv_linear_quantization
  1509. TestQuantizePT2E.test_fold_all_ops_before_quantize
  1510. TestQuantizePT2E.test_fold_quantize
  1511. TestQuantizePT2E.test_fold_quantize_per_channel
  1512. TestQuantizePT2E.test_groupwise_per_channel_quant
  1513. TestQuantizePT2E.test_reentrant
  1514. TestQuantizePT2E.test_save_load
  1515. TestQuantizePT2E.test_speed
  1516. TestQuantizePT2EQAT_ConvBn1d.test_prepare_qat_conv_bn_fusion_getitem_placeholder
  1517. TestQuantizePT2EQAT_ConvBn1d.test_qat_conv_bn_bias_derived_qspec
  1518. TestQuantizePT2EQAT_ConvBn1d.test_qat_conv_bn_fusion
  1519. TestQuantizePT2EQAT_ConvBn1d.test_qat_conv_bn_fusion_literal_args
  1520. TestQuantizePT2EQAT_ConvBn1d.test_qat_conv_bn_fusion_no_conv_bias
  1521. TestQuantizePT2EQAT_ConvBn1d.test_qat_conv_bn_relu_fusion
  1522. TestQuantizePT2EQAT_ConvBn1d.test_qat_conv_bn_relu_fusion_no_conv_bias
  1523. TestQuantizePT2EQAT_ConvBn1d.test_qat_conv_no_bias
  1524. TestQuantizePT2EQAT_ConvBn1d.test_qat_inplace_add_relu
  1525. TestQuantizePT2EQAT_ConvBn1d.test_qat_preserve_source_fn_stack
  1526. TestQuantizePT2EQAT_ConvBn1d.test_qat_update_shared_qspec
  1527. TestQuantizePT2EQAT_ConvBn2d.test_prepare_qat_conv_bn_fusion_getitem_placeholder
  1528. TestQuantizePT2EQAT_ConvBn2d.test_qat_conv_bn_bias_derived_qspec
  1529. TestQuantizePT2EQAT_ConvBn2d.test_qat_conv_bn_fusion
  1530. TestQuantizePT2EQAT_ConvBn2d.test_qat_conv_bn_fusion_literal_args
  1531. TestQuantizePT2EQAT_ConvBn2d.test_qat_conv_bn_fusion_no_conv_bias
  1532. TestQuantizePT2EQAT_ConvBn2d.test_qat_conv_bn_relu_fusion
  1533. TestQuantizePT2EQAT_ConvBn2d.test_qat_conv_bn_relu_fusion_no_conv_bias
  1534. TestQuantizePT2EQAT_ConvBn2d.test_qat_conv_no_bias
  1535. TestQuantizePT2EQAT_ConvBn2d.test_qat_inplace_add_relu
  1536. TestQuantizePT2EQAT_ConvBn2d.test_qat_preserve_source_fn_stack
  1537. TestQuantizePT2EQAT_ConvBn2d.test_qat_update_shared_qspec
  1538. TestQuantizePT2EQATModels.test_qat_mobilenet_v2
  1539. TestQuantizePT2EQATModels.test_qat_resnet18
  1540. TestRecordFunction.test_datapipe_with_record_function
  1541. TestRecordFunction.test_datapipe_with_record_function_fork
  1542. TestRecordFunction.test_record_function
  1543. TestRequire.test_C_and_F_simul
  1544. TestRequire.test_require_each
  1545. TestRequire.test_unknown_requirement
  1546. TestRollaxis.test_exceptions
  1547. TestRot90.test_basic
  1548. TestScalarIndexing.test_invalid_newaxis
  1549. TestScalarIndexing.test_invalid_subscript
  1550. TestScalarIndexing.test_invalid_subscript_assignment
  1551. TestScalarTypeNames.test_names_reflect_attributes_t0
  1552. TestScalarTypeNames.test_names_reflect_attributes_t1
  1553. TestScalarTypeNames.test_names_reflect_attributes_t2
  1554. TestScalarTypeNames.test_names_reflect_attributes_t3
  1555. TestScalarTypeNames.test_names_reflect_attributes_t4
  1556. TestScalarTypeNames.test_names_reflect_attributes_t5
  1557. TestScalarTypeNames.test_names_reflect_attributes_t6
  1558. TestScalarTypeNames.test_names_reflect_attributes_t7
  1559. TestScalarTypeNames.test_names_reflect_attributes_t8
  1560. TestScalarTypeNames.test_names_reflect_attributes_t9
  1561. TestScript.test_conv_error
  1562. TestScript.test_dict_str
  1563. TestScript.test_empty_tuple_str
  1564. TestScript.test_error_stacktrace_interface
  1565. TestScript.test_function_overloading_isinstance
  1566. TestScript.test_function_overloads
  1567. TestScript.test_ignored_as_value
  1568. TestScript.test_infer_size
  1569. TestScript.test_is_after_use
  1570. TestScript.test_is_scripting
  1571. TestScript.test_isinstance_dynamic
  1572. TestScript.test_method_overloading
  1573. TestScript.test_namedtuple_default_values_using_factory_constructor
  1574. TestScript.test_namedtuple_python
  1575. TestScript.test_nested_breaks
  1576. TestScript.test_nn_GRU
  1577. TestScript.test_nn_LSTM_with_layers
  1578. TestScript.test_no_self_arg_ignore_function
  1579. TestScript.test_none_type_str
  1580. TestScript.test_pack_unpack_state
  1581. TestScript.test_parse_nested_names
  1582. TestScript.test_parse_tensor_constants
  1583. TestScript.test_python_call
  1584. TestScript.test_python_call_non_tensor_wrong
  1585. TestScript.test_python_op_builtins
  1586. TestScript.test_script_optional_none
  1587. TestScript.test_string_device_implicit_conversion
  1588. TestScript.test_torch_functional_tensordot_int
  1589. TestScript.test_torchscript_multi_head_attn
  1590. TestScript.test_tuple_str
  1591. TestScript.test_type_annotation_module
  1592. TestScript.test_unspecialized_any_binding
  1593. TestScript.test_unused_decorator
  1594. TestScript.test_wrong_return_type
  1595. TestSDPACPU.test_fused_sdp_choice_cpu_type_dense_dropout_0_0_bfloat16_cpu_bfloat16
  1596. TestSDPACPU.test_fused_sdp_choice_cpu_type_dense_dropout_0_0_float16_cpu_float16
  1597. TestSDPACPU.test_fused_sdp_choice_cpu_type_dense_dropout_0_0_float32_cpu_float32
  1598. TestSDPACPU.test_fused_sdp_choice_cpu_type_dense_dropout_0_0_float64_cpu_float64
  1599. TestSDPACPU.test_fused_sdp_choice_cpu_type_dense_dropout_0_7_bfloat16_cpu_bfloat16
  1600. TestSDPACPU.test_fused_sdp_choice_cpu_type_dense_dropout_0_7_float16_cpu_float16
  1601. TestSDPACPU.test_fused_sdp_choice_cpu_type_dense_dropout_0_7_float32_cpu_float32
  1602. TestSDPACPU.test_fused_sdp_choice_cpu_type_dense_dropout_0_7_float64_cpu_float64
  1603. TestSetOps.test_in1d_table_timedelta_fails
  1604. TestSetOps.test_in1d_timedelta_kind0
  1605. TestSetOps.test_in1d_timedelta_kind_sort
  1606. TestSetOps.test_setdiff1d
  1607. TestShapeOpsCUDA.test_flip_cuda_float32
  1608. TestSortAndSelectCPU.test_isin_cpu_float32
  1609. TestSortAndSelectCPU.test_isin_cpu_float64
  1610. TestSortAndSelectCPU.test_isin_cpu_int16
  1611. TestSortAndSelectCPU.test_isin_cpu_int32
  1612. TestSortAndSelectCPU.test_isin_cpu_int64
  1613. TestSortAndSelectCPU.test_isin_cpu_int8
  1614. TestSortAndSelectCPU.test_isin_cpu_uint8
  1615. TestSortAndSelectCPU.test_sort_overflow_cpu_int16
  1616. TestSortAndSelectCPU.test_topk_arguments_cpu
  1617. TestSortComplex.test_sort_real_type_in_g_type_out_G
  1618. TestSortComplex.test_sort_real_type_in_H_type_out_F
  1619. TestSplit.test_unequal_split
  1620. TestStackMisc.test_stack
  1621. TestStatelessFunctionalAPI.test_reparametrize_module_fail_reset_to_original_stateless
  1622. TestStatelessFunctionalAPI.test_reparametrize_module_fail_reset_to_original_torch_func
  1623. TestStatelessFunctionalAPI.test_reparametrized_module_change_parametrization_original_stateless
  1624. TestStatelessFunctionalAPI.test_reparametrized_module_change_parametrization_original_torch_func
  1625. TestStaticQuantizedModule.test_embedding_api
  1626. TestStaticQuantizedModule.test_embedding_bag_api
  1627. TestStats.test_dtype_from_dtype
  1628. TestStats.test_out
  1629. TestSubclass.test_parametrization_base_tensor_leave_parametrized_False
  1630. TestSubclass.test_parametrization_base_tensor_leave_parametrized_True
  1631. TestTake.test_raise
  1632. TestTakeAlongAxis.test_invalid
  1633. TestTEFuserDynamic.test_gelu
  1634. TestTEFuserDynamic.test_profiler
  1635. TestTEFuserDynamic.test_remove_output_used_only_in_size
  1636. TestTEFuserDynamic.test_to_dtype
  1637. TestTEFuserDynamic.test_torch_to
  1638. TestTEFuserStatic.test_gelu
  1639. TestTEFuserStatic.test_inlined_optimized_graph
  1640. TestTEFuserStatic.test_profiler
  1641. TestTEFuserStatic.test_remove_output_used_only_in_size
  1642. TestTEFuserStatic.test_skip_grad_in_check
  1643. TestTEFuserStatic.test_to_dtype
  1644. TestTEFuserStatic.test_torch_to
  1645. TestTensorBoardEmbedding.test_embedding
  1646. TestTensorBoardEmbedding.test_embedding_64
  1647. TestTensorBoardSummary.test_image_with_one_channel
  1648. TestTensorBoardSummary.test_image_without_channel
  1649. TestTensorBoardSummary.test_uint8_image
  1650. TestTensorBoardSummary.test_video
  1651. TestTensorBoardUtils.test_numpy_vid_uint8
  1652. TestTensorBoardUtils.test_to_HWC
  1653. TestTensorCreationCPU.test_block_diag_cpu
  1654. TestTensorCreationCPU.test_cartesian_prod_cpu
  1655. TestTensorCreationCPU.test_constructor_dtypes_cpu
  1656. TestTorch.test_contains
  1657. TestTorch.test_cuda_not_built
  1658. TestTorch.test_map
  1659. TestTorch.test_new
  1660. TestTorch.test_parsing_double
  1661. TestTorch.test_parsing_int64
  1662. TestTorch.test_parsing_intlist
  1663. TestTorch.test_pin_memory
  1664. TestTorch.test_type
  1665. TestTorch.test_upsample_nearest2d_meta
  1666. TestTorchDeviceTypeCPU.test_broadcast_fn_add_cpu
  1667. TestTorchDeviceTypeCPU.test_broadcast_fn_addcdiv_cpu
  1668. TestTorchDeviceTypeCPU.test_broadcast_fn_addcmul_cpu
  1669. TestTorchDeviceTypeCPU.test_broadcast_fn_atan2_cpu
  1670. TestTorchDeviceTypeCPU.test_broadcast_fn_copy_cpu
  1671. TestTorchDeviceTypeCPU.test_broadcast_fn_div_cpu
  1672. TestTorchDeviceTypeCPU.test_broadcast_fn_eq_cpu
  1673. TestTorchDeviceTypeCPU.test_broadcast_fn_fmod_cpu
  1674. TestTorchDeviceTypeCPU.test_broadcast_fn_ge_cpu
  1675. TestTorchDeviceTypeCPU.test_broadcast_fn_gt_cpu
  1676. TestTorchDeviceTypeCPU.test_broadcast_fn_le_cpu
  1677. TestTorchDeviceTypeCPU.test_broadcast_fn_lerp_cpu
  1678. TestTorchDeviceTypeCPU.test_broadcast_fn_lt_cpu
  1679. TestTorchDeviceTypeCPU.test_broadcast_fn_map2_cpu
  1680. TestTorchDeviceTypeCPU.test_broadcast_fn_map_cpu
  1681. TestTorchDeviceTypeCPU.test_broadcast_fn_masked_fill_cpu
  1682. TestTorchDeviceTypeCPU.test_broadcast_fn_masked_scatter_cpu
  1683. TestTorchDeviceTypeCPU.test_broadcast_fn_mul_cpu
  1684. TestTorchDeviceTypeCPU.test_broadcast_fn_ne_cpu
  1685. TestTorchDeviceTypeCPU.test_broadcast_fn_pow_cpu
  1686. TestTorchDeviceTypeCPU.test_broadcast_fn_remainder_cpu
  1687. TestTorchDeviceTypeCPU.test_broadcast_fn_sub_cpu
  1688. TestTorchDeviceTypeCPU.test_deterministic_empty_cpu_uint64
  1689. TestTorchDeviceTypeCPU.test_exponential_kstest_cpu_bfloat16
  1690. TestTorchDeviceTypeCPU.test_exponential_kstest_cpu_float16
  1691. TestTorchDeviceTypeCPU.test_exponential_kstest_cpu_float32
  1692. TestTorchDeviceTypeCPU.test_exponential_kstest_cpu_float64
  1693. TestTorchDeviceTypeCPU.test_nondeterministic_alert_MaxUnpool1d_cpu_float32
  1694. TestTorchDeviceTypeCPU.test_nondeterministic_alert_MaxUnpool1d_cpu_float64
  1695. TestTorchDeviceTypeCPU.test_nondeterministic_alert_MaxUnpool2d_cpu_float32
  1696. TestTorchDeviceTypeCPU.test_nondeterministic_alert_MaxUnpool2d_cpu_float64
  1697. TestTorchDeviceTypeCPU.test_nondeterministic_alert_MaxUnpool3d_cpu_float32
  1698. TestTorchDeviceTypeCPU.test_nondeterministic_alert_MaxUnpool3d_cpu_float64
  1699. TestTorchDeviceTypeCPU.test_normal_kstest_cpu_float16
  1700. TestTorchDeviceTypeCPU.test_normal_kstest_cpu_float32
  1701. TestTorchDeviceTypeCPU.test_normal_kstest_cpu_float64
  1702. TestTorchDeviceTypeCPU.test_uniform_kstest_cpu_bfloat16
  1703. TestTorchDeviceTypeCPU.test_uniform_kstest_cpu_float16
  1704. TestTorchDeviceTypeCPU.test_uniform_kstest_cpu_float32
  1705. TestTorchDeviceTypeCPU.test_uniform_kstest_cpu_float64
  1706. TestTorchDlPackCPU.test_dlpack_export_non_strided_cpu
  1707. TestTorchFunctionMode.test_mode_notimplemented_loop
  1708. TestTorchFunctionMode.test_modes_return_notimplemented
  1709. TestTorchFunctionMode.test_nested_modes_with_python_has_torch_function
  1710. TestTorchFunctionOverride.test_mean_semantics
  1711. TestTorchFunctionOverride.test_Tensor___cuda_array_interface_____get__
  1712. TestTorchTidyProfiler.test_allocation_id_uniqueness
  1713. TestTorchTidyProfiler.test_allocation_ids
  1714. TestTorchTidyProfiler.test_allocation_ids_with_other_ops
  1715. TestTorchTidyProfiler.test_allocations
  1716. TestTorchTidyProfiler.test_extra_fields
  1717. TestTorchTidyProfiler.test_impl_reuse
  1718. TestTorchTidyProfiler.test_mkldnn_tensors
  1719. TestTorchTidyProfiler.test_module_and_optimizer_ids
  1720. TestTorchTidyProfiler.test_nnmodule_params
  1721. TestTorchTidyProfiler.test_optimizer
  1722. TestTorchTidyProfiler.test_optimizer_parameters_adam
  1723. TestTorchTidyProfiler.test_optimizer_parameters_sgd
  1724. TestTorchTidyProfiler.test_pointers_and_ids
  1725. TestTorchTidyProfiler.test_scalar_ins
  1726. TestTorchTidyProfiler.test_sparse_tensors
  1727. TestTorchTidyProfiler.test_tensor_lists
  1728. TestTorchTidyProfiler.test_tensor_properties
  1729. TestTorchTidyProfiler.test_tensorimpl_invalidation_full
  1730. TestTorchTidyProfiler.test_tensorimpl_invalidation_keep_alive
  1731. TestTorchTidyProfiler.test_tensorimpl_invalidation_scalar_args
  1732. TestTorchTidyProfiler.test_tensorimpl_invalidation_set
  1733. TestTrilIndicesFrom.test_exceptions
  1734. TestTriuIndicesFrom.test_exceptions
  1735. TestTypeHints.test_doc_examples
  1736. TestTypePromotionCPU.test_alpha_mismatch_cpu
  1737. TestTypePromotionCPU.test_alternate_result_cpu
  1738. TestUnflatten.test_unflatten_container_type
  1739. TestUnique.test_unique_axis
  1740. TestUnique.test_unique_axis_errors
  1741. TestUtils.test_get_fqn_to_example_inputs_complex_args
  1742. TestUtils.test_get_fqn_to_example_inputs_default_kwargs
  1743. TestUtils.test_get_fqn_to_example_inputs_simple
  1744. TestVerifyCorrectness.test_incorrect_verify_true
  1745. TestVerifyCorrectness.test_torchscript
  1746. TestVmapAPILegacy.test_accepts_nested_inputs
  1747. TestVmapAPILegacy.test_backward_unsupported_interaction
  1748. TestVmapAPILegacy.test_batched_gradient_basic
  1749. TestVmapAPILegacy.test_fallback_with_undefined_grad
  1750. TestVmapAPILegacy.test_grad_unsupported_interaction
  1751. TestVmapAPILegacy.test_in_dim_not_in_tensor_err_msg
  1752. TestVmapAPILegacy.test_multiple_out_dims
  1753. TestVmapAPILegacy.test_multiple_outputs
  1754. TestVmapAPILegacy.test_multiple_outputs_error_cases
  1755. TestVmapAPILegacy.test_nested_out_dims
  1756. TestVmapAPILegacy.test_nested_with_different_map_dim
  1757. TestVmapAPILegacy.test_non_default_in_dims_out_dims
  1758. TestVmapAPILegacy.test_non_tensor_output_raises
  1759. TestVmapAPILegacy.test_non_zero_in_dims
  1760. TestVmapAPILegacy.test_nonzero_out_dims
  1761. TestVmapAPILegacy.test_noop_in_inner_vmap
  1762. TestVmapAPILegacy.test_out_dim_out_of_bounds_err_msg
  1763. TestVmapAPILegacy.test_out_dims_and_num_outputs_mismatch_err_msg
  1764. TestVmapAPILegacy.test_out_dims_edge_case
  1765. TestVmapAPILegacy.test_single_input
  1766. TestVmapAPILegacy.test_unsupported_op_err_msg
  1767. TestVmapBatchedGradientLegacyCPU.test_binary_cross_entropy_cpu
  1768. TestVmapBatchedGradientLegacyCPU.test_div_cpu
  1769. TestVmapBatchedGradientLegacyCPU.test_index_cpu
  1770. TestVmapBatchedGradientLegacyCPU.test_inplace_manyview_cpu
  1771. TestVmapBatchedGradientLegacyCPU.test_inplace_on_view_cpu
  1772. TestVmapBatchedGradientLegacyCPU.test_lgamma_cpu
  1773. TestVmapBatchedGradientLegacyCPU.test_log1p_cpu
  1774. TestVmapBatchedGradientLegacyCPU.test_log_cpu
  1775. TestVmapBatchedGradientLegacyCPU.test_logsumexp_cpu
  1776. TestVmapBatchedGradientLegacyCPU.test_mul_cpu
  1777. TestVmapBatchedGradientLegacyCPU.test_sigmoid_cpu
  1778. TestVmapBatchedGradientLegacyCPU.test_unrelated_output_cpu
  1779. TestVmapBatchedGradientLegacyCPU.test_unrelated_output_multiple_grad_cpu
  1780. TestVmapOperatorsLegacy.test_as_strided
  1781. TestVmapOperatorsLegacy.test_binary_pointwise_ops
  1782. TestVmapOperatorsLegacy.test_chunk
  1783. TestVmapOperatorsLegacy.test_clamp
  1784. TestVmapOperatorsLegacy.test_clone
  1785. TestVmapOperatorsLegacy.test_comparison_ops
  1786. TestVmapOperatorsLegacy.test_diagonal
  1787. TestVmapOperatorsLegacy.test_fill_and_zero_inplace
  1788. TestVmapOperatorsLegacy.test_is_complex
  1789. TestVmapOperatorsLegacy.test_is_contiguous
  1790. TestVmapOperatorsLegacy.test_is_floating_point
  1791. TestVmapOperatorsLegacy.test_new_empty
  1792. TestVmapOperatorsLegacy.test_new_empty_strided
  1793. TestVmapOperatorsLegacy.test_new_zeros
  1794. TestVmapOperatorsLegacy.test_no_random_op_support
  1795. TestVmapOperatorsLegacy.test_reshape
  1796. TestVmapOperatorsLegacy.test_result_type
  1797. TestVmapOperatorsLegacy.test_select
  1798. TestVmapOperatorsLegacy.test_slice
  1799. TestVmapOperatorsLegacy.test_split
  1800. TestVmapOperatorsLegacy.test_stride
  1801. TestVmapOperatorsLegacy.test_sum_dim
  1802. TestVmapOperatorsLegacy.test_T_numpy
  1803. TestVmapOperatorsLegacy.test_to
  1804. TestVmapOperatorsLegacy.test_transpose
  1805. TestVmapOperatorsLegacy.test_unbind
  1806. TestVmapOperatorsLegacy.test_view
  1807. TestVsplit.test_0D_array
  1808. TestVsplit.test_1D_array
  1809. TestVsplit.test_non_iterable
  1810. TestVstack.test_empty_input
  1811. TestVstack.test_non_iterable
  1812. TestWhere.test_error
  1813. TestWrapperSubclassAliasingCPU.test_wrapper_subclass_aliasing_conv2d_cpu
  1814. TestXNNPACKQuantizer.test_add_and_inplace_add
  1815. TestXNNPACKQuantizer.test_add_mul_long
  1816. TestXNNPACKQuantizer.test_add_mul_scalar
  1817. TestXNNPACKQuantizer.test_conv1d
  1818. TestXNNPACKQuantizer.test_conv1d_with_conv2d
  1819. TestXNNPACKQuantizer.test_conv2d
  1820. TestXNNPACKQuantizer.test_conv_linear
  1821. TestXNNPACKQuantizer.test_conv_linear_no_permute
  1822. TestXNNPACKQuantizer.test_dynamic_linear
  1823. TestXNNPACKQuantizer.test_dynamic_linear_int4_weight
  1824. TestXNNPACKQuantizer.test_dynamic_linear_with_conv
  1825. TestXNNPACKQuantizer.test_gru
  1826. TestXNNPACKQuantizer.test_linear
  1827. TestXNNPACKQuantizer.test_linear_gru
  1828. TestXNNPACKQuantizer.test_linear_relu
  1829. TestXNNPACKQuantizer.test_linear_with_dynamic_shape
  1830. TestXNNPACKQuantizer.test_mul_and_inplace_mul
  1831. TestXNNPACKQuantizer.test_mul_float32_max
  1832. TestXNNPACKQuantizer.test_obs_sharing_ops
  1833. TestXNNPACKQuantizer.test_propagate_annotation
  1834. TestXNNPACKQuantizer.test_qat_dynamic_linear
  1835. TestXNNPACKQuantizerModels.test_resnet18
  1836. TraceRuleTests.test_skipfiles_inlinelist
  1837. UnspecTests.test_builtin_max_min
  1838. UnspecTests.test_conv1d_symint_padding
  1839. UnspecTests.test_isinstance_symint
  1840. UnspecTests.test_mark_01_dynamic
  1841. UnspecTests.test_no_recompilations
  1842. UnspecTests.test_no_recompiles
  1843. UnspecTests.test_propagate_dynamic_dim
  1844. UnspecTests.test_use_and_specialize