tree: abba940c4d1e57f939ff6a468dc4dc88ace95d9a [path history] [tgz]
  1. custom_kernel_example/
  2. CMakeLists.txt
  3. gen_supported_features.py
  4. gen_supported_features_test.py
  5. kernel_runtime_context_test.cpp
  6. op__to_dim_order_copy_test.cpp
  7. op_abs_test.cpp
  8. op_acos_test.cpp
  9. op_acosh_test.cpp
  10. op_add_test.cpp
  11. op_addmm_test.cpp
  12. op_alias_copy_test.cpp
  13. op_amax_test.cpp
  14. op_amin_test.cpp
  15. op_any_test.cpp
  16. op_arange_test.cpp
  17. op_argmax_test.cpp
  18. op_argmin_test.cpp
  19. op_as_strided_copy_test.cpp
  20. op_asin_test.cpp
  21. op_asinh_test.cpp
  22. op_atan2_test.cpp
  23. op_atan_test.cpp
  24. op_atanh_test.cpp
  25. op_avg_pool2d_test.cpp
  26. op_bitwise_and_test.cpp
  27. op_bitwise_not_test.cpp
  28. op_bitwise_or_test.cpp
  29. op_bitwise_xor_test.cpp
  30. op_bmm_test.cpp
  31. op_cat_test.cpp
  32. op_cdist_forward_test.cpp
  33. op_ceil_test.cpp
  34. op_clamp_test.cpp
  35. op_clone_test.cpp
  36. op_constant_pad_nd_test.cpp
  37. op_convolution_backward_test.cpp
  38. op_convolution_test.cpp
  39. op_copy_test.cpp
  40. op_cos_test.cpp
  41. op_cosh_test.cpp
  42. op_cumsum_test.cpp
  43. op_detach_copy_test.cpp
  44. op_diagonal_copy_test.cpp
  45. op_div_test.cpp
  46. op_embedding_test.cpp
  47. op_empty_test.cpp
  48. op_eq_test.cpp
  49. op_erf_test.cpp
  50. op_exp_test.cpp
  51. op_expand_copy_test.cpp
  52. op_expm1_test.cpp
  53. op_fill_test.cpp
  54. op_flip_test.cpp
  55. op_floor_divide_test.cpp
  56. op_floor_test.cpp
  57. op_fmod_test.cpp
  58. op_full_like_test.cpp
  59. op_full_test.cpp
  60. op_gather_test.cpp
  61. op_ge_test.cpp
  62. op_gelu_test.cpp
  63. op_glu_test.cpp
  64. op_gt_test.cpp
  65. op_hardtanh_test.cpp
  66. op_index_put_test.cpp
  67. op_index_select_test.cpp
  68. op_index_test.cpp
  69. op_isinf_test.cpp
  70. op_isnan_test.cpp
  71. op_le_test.cpp
  72. op_leaky_relu_test.cpp
  73. op_lift_fresh_copy_test.cpp
  74. op_linear_test.cpp
  75. op_log10_test.cpp
  76. op_log1p_test.cpp
  77. op_log2_test.cpp
  78. op_log_softmax_test.cpp
  79. op_log_test.cpp
  80. op_logical_and_test.cpp
  81. op_logical_not_test.cpp
  82. op_logical_or_test.cpp
  83. op_logical_xor_test.cpp
  84. op_logit_test.cpp
  85. op_lt_test.cpp
  86. op_masked_fill_test.cpp
  87. op_max_pool2d_with_indices_test.cpp
  88. op_max_test.cpp
  89. op_maximum_test.cpp
  90. op_mean_test.cpp
  91. op_min_test.cpp
  92. op_minimum_test.cpp
  93. op_mm_test.cpp
  94. op_mul_test.cpp
  95. op_narrow_copy_test.cpp
  96. op_native_batch_norm_test.cpp
  97. op_native_group_norm_test.cpp
  98. op_native_layer_norm_test.cpp
  99. op_ne_test.cpp
  100. op_neg_test.cpp
  101. op_nonzero_test.cpp
  102. op_ones_test.cpp
  103. op_pdist_forward_test.cpp
  104. op_permute_copy_test.cpp
  105. op_pixel_shuffle_test.cpp
  106. op_pixel_unshuffle_test.cpp
  107. op_pow_test.cpp
  108. op_prod_test.cpp
  109. op_reciprocal_test.cpp
  110. op_reflection_pad1d_test.cpp
  111. op_reflection_pad2d_test.cpp
  112. op_reflection_pad3d_test.cpp
  113. op_relu_test.cpp
  114. op_remainder_test.cpp
  115. op_repeat_test.cpp
  116. op_replication_pad1d_test.cpp
  117. op_replication_pad2d_test.cpp
  118. op_replication_pad3d_test.cpp
  119. op_roll_test.cpp
  120. op_round_test.cpp
  121. op_rsqrt_test.cpp
  122. op_rsub_test.cpp
  123. op_scalar_tensor_test.cpp
  124. op_scatter_add_test.cpp
  125. op_scatter_test.cpp
  126. op_select_copy_test.cpp
  127. op_select_scatter_test.cpp
  128. op_sigmoid_test.cpp
  129. op_sign_test.cpp
  130. op_sin_test.cpp
  131. op_sinh_test.cpp
  132. op_slice_copy_test.cpp
  133. op_slice_scatter_test.cpp
  134. op_softmax_test.cpp
  135. op_split_copy_test.cpp
  136. op_split_with_sizes_copy_test.cpp
  137. op_sqrt_test.cpp
  138. op_squeeze_copy_test.cpp
  139. op_stack_test.cpp
  140. op_sub_test.cpp
  141. op_sum_test.cpp
  142. op_t_copy_test.cpp
  143. op_tan_test.cpp
  144. op_tanh_test.cpp
  145. op_test.h
  146. op_to_copy_test.cpp
  147. op_topk_test.cpp
  148. op_transpose_copy_test.cpp
  149. op_tril_test.cpp
  150. op_trunc_test.cpp
  151. op_unbind_copy_test.cpp
  152. op_unsqueeze_copy_test.cpp
  153. op_var_test.cpp
  154. op_view_copy_test.cpp
  155. op_where_test.cpp
  156. op_zeros_test.cpp
  157. summarize_supported_features.py
  158. supported_features.yaml
  159. supported_features_def_aten.yaml
  160. supported_features_def_example.yaml
  161. supported_features_definition.ini
  162. supported_features_header.ini
  163. supported_features_summary.md
  164. TARGETS
  165. targets.bzl
  166. test_case_gen.py
  167. TestUtil.h
  168. update_supported_feature_summary.sh