| | pattern | |
| | test | |
| | util | |
| | math_constants.h | 1.5 KB |
| | op__empty_dim_order.cpp | 2.3 KB |
| | op__to_dim_order_copy.cpp | 3.9 KB |
| | op_abs.cpp | 2.3 KB |
| | op_acos.cpp | 650 B |
| | op_acosh.cpp | 652 B |
| | op_add.cpp | 3.4 KB |
| | op_addmm.cpp | 3.2 KB |
| | op_alias_copy.cpp | 1.2 KB |
| | op_allclose.cpp | 5.5 KB |
| | op_amax.cpp | 1.8 KB |
| | op_amin.cpp | 1.8 KB |
| | op_any.cpp | 5.6 KB |
| | op_arange.cpp | 2.7 KB |
| | op_argmax.cpp | 2.2 KB |
| | op_argmin.cpp | 2.7 KB |
| | op_as_strided_copy.cpp | 1.4 KB |
| | op_asin.cpp | 650 B |
| | op_asinh.cpp | 652 B |
| | op_atan.cpp | 650 B |
| | op_atan2.cpp | 1.9 KB |
| | op_atanh.cpp | 652 B |
| | op_avg_pool2d.cpp | 3.6 KB |
| | op_bitwise_and.cpp | 1.1 KB |
| | op_bitwise_not.cpp | 1.9 KB |
| | op_bitwise_or.cpp | 1.0 KB |
| | op_bitwise_xor.cpp | 1.1 KB |
| | op_bmm.cpp | 1.5 KB |
| | op_cat.cpp | 2.4 KB |
| | op_cdist_forward.cpp | 5.0 KB |
| | op_ceil.cpp | 661 B |
| | op_clamp.cpp | 7.1 KB |
| | op_clone.cpp | 1.8 KB |
| | op_constant_pad_nd.cpp | 5.1 KB |
| | op_convolution_backward.cpp | 9.8 KB |
| | op_convolution.cpp | 13.3 KB |
| | op_copy.cpp | 3.6 KB |
| | op_cos.cpp | 641 B |
| | op_cosh.cpp | 650 B |
| | op_cumsum.cpp | 4.0 KB |
| | op_detach_copy.cpp | 1.4 KB |
| | op_diagonal_copy.cpp | 2.7 KB |
| | op_div.cpp | 7.7 KB |
| | op_elu.cpp | 2.1 KB |
| | op_embedding.cpp | 3.5 KB |
| | op_empty.cpp | 1012 B |
| | op_eq.cpp | 1.0 KB |
| | op_erf.cpp | 641 B |
| | op_exp.cpp | 641 B |
| | op_expand_copy.cpp | 2.8 KB |
| | op_expm1.cpp | 652 B |
| | op_fill.cpp | 2.9 KB |
| | op_flip.cpp | 2.3 KB |
| | op_floor_divide.cpp | 2.4 KB |
| | op_floor.cpp | 663 B |
| | op_fmod.cpp | 4.2 KB |
| | op_full_like.cpp | 2.0 KB |
| | op_full.cpp | 1.5 KB |
| | op_gather.cpp | 2.2 KB |
| | op_ge.cpp | 1.1 KB |
| | op_gelu.cpp | 2.7 KB |
| | op_glu.cpp | 5.5 KB |
| | op_gt.cpp | 1.0 KB |
| | op_hardtanh.cpp | 2.2 KB |
| | op_index_put.cpp | 4.9 KB |
| | op_index_select.cpp | 2.5 KB |
| | op_index.cpp | 3.1 KB |
| | op_isinf.cpp | 830 B |
| | op_isnan.cpp | 830 B |
| | op_le.cpp | 1.0 KB |
| | op_leaky_relu.cpp | 2.0 KB |
| | op_lift_fresh_copy.cpp | 1.2 KB |
| | op_linear_scratch_example.cpp | 4.4 KB |
| | op_log_softmax.cpp | 2.8 KB |
| | op_log.cpp | 641 B |
| | op_log10.cpp | 652 B |
| | op_log1p.cpp | 652 B |
| | op_log2.cpp | 650 B |
| | op_logical_and.cpp | 873 B |
| | op_logical_not.cpp | 1.5 KB |
| | op_logical_or.cpp | 869 B |
| | op_logical_xor.cpp | 873 B |
| | op_logit.cpp | 1.8 KB |
| | op_lt.cpp | 1.0 KB |
| | op_masked_fill.cpp | 1.9 KB |
| | op_masked_scatter.cpp | 2.0 KB |
| | op_masked_select.cpp | 4.4 KB |
| | op_max_pool2d_with_indices_backward.cpp | 5.2 KB |
| | op_max_pool2d_with_indices.cpp | 2.7 KB |
| | op_max.cpp | 4.3 KB |
| | op_maximum.cpp | 1.8 KB |
| | op_mean.cpp | 2.6 KB |
| | op_min.cpp | 4.3 KB |
| | op_minimum.cpp | 1.9 KB |
| | op_mm.cpp | 1.6 KB |
| | op_mul.cpp | 3.7 KB |
| | op_narrow_copy.cpp | 1.3 KB |
| | op_native_batch_norm.cpp | 8.3 KB |
| | op_native_dropout.cpp | 2.7 KB |
| | op_native_group_norm.cpp | 5.1 KB |
| | op_native_layer_norm.cpp | 5.2 KB |
| | op_ne.cpp | 1.0 KB |
| | op_neg.cpp | 1.3 KB |
| | op_nonzero.cpp | 2.6 KB |
| | op_ones.cpp | 932 B |
| | op_pdist_forward.cpp | 1.3 KB |
| | op_permute_copy.cpp | 2.4 KB |
| | op_pixel_shuffle.cpp | 3.0 KB |
| | op_pixel_unshuffle.cpp | 2.9 KB |
| | op_pow.cpp | 4.8 KB |
| | op_prod.cpp | 3.2 KB |
| | op_reciprocal.cpp | 719 B |
| | op_reflection_pad1d.cpp | 1.5 KB |
| | op_reflection_pad2d.cpp | 1.5 KB |
| | op_reflection_pad3d.cpp | 1.5 KB |
| | op_relu.cpp | 1.5 KB |
| | op_remainder.cpp | 4.1 KB |
| | op_repeat_interleave.cpp | 3.7 KB |
| | op_repeat.cpp | 2.3 KB |
| | op_replication_pad1d.cpp | 1.3 KB |
| | op_replication_pad2d.cpp | 1.3 KB |
| | op_replication_pad3d.cpp | 1.3 KB |
| | op_roll.cpp | 2.6 KB |
| | op_round.cpp | 2.0 KB |
| | op_rsqrt.cpp | 708 B |
| | op_rsub.cpp | 2.0 KB |
| | op_scalar_tensor.cpp | 1.1 KB |
| | op_scatter_add.cpp | 2.8 KB |
| | op_scatter.cpp | 4.2 KB |
| | op_select_copy.cpp | 805 B |
| | op_select_scatter.cpp | 2.7 KB |
| | op_sigmoid.cpp | 2.0 KB |
| | op_sign.cpp | 1.6 KB |
| | op_sin.cpp | 641 B |
| | op_sinh.cpp | 650 B |
| | op_slice_copy.cpp | 1.8 KB |
| | op_slice_scatter.cpp | 3.1 KB |
| | op_softmax.cpp | 2.8 KB |
| | op_split_copy.cpp | 2.6 KB |
| | op_split_with_sizes_copy.cpp | 4.9 KB |
| | op_sqrt.cpp | 650 B |
| | op_squeeze_copy.cpp | 2.7 KB |
| | op_stack.cpp | 2.2 KB |
| | op_sub.cpp | 3.6 KB |
| | op_sum.cpp | 3.6 KB |
| | op_t_copy.cpp | 2.1 KB |
| | op_tan.cpp | 641 B |
| | op_tanh.cpp | 650 B |
| | op_to_copy.cpp | 1.9 KB |
| | op_topk.cpp | 5.7 KB |
| | op_transpose_copy.cpp | 1.8 KB |
| | op_tril.cpp | 4.5 KB |
| | op_trunc.cpp | 630 B |
| | op_unbind_copy.cpp | 2.5 KB |
| | op_unfold_copy.cpp | 2.4 KB |
| | op_unsqueeze_copy.cpp | 2.1 KB |
| | op_upsample_bilinear2d.cpp | 7.0 KB |
| | op_upsample_nearest2d.cpp | 4.2 KB |
| | op_var.cpp | 4.5 KB |
| | op_view_as_real_copy.cpp | 2.1 KB |
| | op_view_copy.cpp | 1.6 KB |
| | op_where.cpp | 1.8 KB |
| | op_zeros.cpp | 1.6 KB |
| | scalar_utils.h | 9.1 KB |
| | selective_build.h | 1.5 KB |
| | TARGETS | 232 B |
| | targets.bzl | 3.2 KB |
| | vec_ops.h | 6.8 KB |