1# Copyright (c) Meta Platforms, Inc. and affiliates. 2# 3# This yaml file contains operators that are defined by the ATen library. 4 5- op: _cdist_forward.out 6 7- op: _fake_quantize_per_tensor_affine_cachemask_tensor_qparams.out 8 9- op: _linalg_det.result 10 11- op: _linalg_svd.U 12 13- op: _log_softmax.out 14 15- op: _native_batch_norm_legit.out 16 17- op: _native_batch_norm_legit.no_stats_out 18 19- op: _native_batch_norm_legit_no_training.out 20 21- op: _pdist_forward.out 22 23- op: _reshape_alias_copy.out 24 25- op: _softmax.out 26 27- op: copy_ 28 29- op: _to_copy.out 30 31- op: _unique2.out 32 33- op: _upsample_bilinear2d_aa.out 34 35- op: abs.out 36 37- op: acos.out 38 39- op: acosh.out 40 41- op: add.out 42 43- op: add.Scalar_out 44 45- op: addmm.out 46 47- op: alias_copy.out 48 49- op: amax.out 50 51- op: amin.out 52 53- op: any.all_out 54 55- op: any.dims_out 56 57- op: any.out 58 59- op: arange.out 60 61- op: arange.start_out 62 63- op: argmax.out 64 65- op: argmin.out 66 67- op: as_strided_copy.out 68 69- op: asin.out 70 71- op: asinh.out 72 73- op: atan.out 74 75- op: atan2.out 76 77- op: atanh.out 78 79- op: avg_pool2d.out 80 81- op: baddbmm.out 82 83- op: bitwise_and.Tensor_out 84 85- op: bitwise_and.Scalar_out 86 87- op: bitwise_or.Tensor_out 88 89- op: bitwise_or.Scalar_out 90 91- op: bitwise_not.out 92 93- op: bitwise_xor.Tensor_out 94 95- op: bitwise_xor.Scalar_out 96 97- op: bmm.out 98 99- op: cat.out 100 101- op: ceil.out 102 103- op: clamp_min.out 104 105- op: clamp.out 106 cpp_no_default_args: ['min'] 107 108- op: clamp.Tensor_out 109 110- op: clone.out 111 112- op: constant_pad_nd.out 113 114- op: conv1d.out 115 116- op: convolution.out 117 118- op: convolution_backward.out 119 120- op: copy.out 121 122- op: cos.out 123 124- op: cosh.out 125 126- op: cumsum.out 127 128- op: dequantize.self_out 129 130- op: detach_copy.out 131 132- op: diagonal_copy.out 133 134- op: div.out 135 136- op: div.Scalar_mode_out 137 138- op: div.Scalar_out 139 140- op: div.out_mode 141 142- op: embedding.out 143 144- op: empty.out 145 146- op: eq.Scalar_out 147 148- op: eq.Tensor_out 149 150- op: erf.out 151 152- op: exp.out 153 154- op: expand_copy.out 155 156- op: expm1.out 157 158- op: eye.out 159 160- op: fill.Scalar_out 161 162- op: fill.Tensor_out 163 164- op: flip.out 165 166- op: floor_divide.out 167 168- op: floor.out 169 170- op: fmod.Tensor_out 171 172- op: fmod.Scalar_out 173 174- op: full_like.out 175 176- op: full.out 177 178- op: gather.out 179 180- op: ge.Scalar_out 181 182- op: ge.Tensor_out 183 184- op: gelu.out 185 186- op: glu.out 187 188- op: grid_sampler_2d.out 189 190- op: gt.Scalar_out 191 192- op: gt.Tensor_out 193 194- op: hardtanh.out 195 196- op: index_put.out 197 198- op: index_select.out 199 200- op: index.Tensor_out 201 202- op: isinf.out 203 204- op: isnan.out 205 206- op: le.Scalar_out 207 208- op: le.Tensor_out 209 210- op: leaky_relu.out 211 212- op: lift_fresh_copy.out 213 214- op: linalg_inv_ex.inverse 215 216- op: linalg_vector_norm.out 217 218- op: linear.out 219 220- op: log.out 221 222- op: log10.out 223 224- op: log1p.out 225 226- op: log2.out 227 228- op: logical_and.out 229 230- op: logical_not.out 231 232- op: logical_or.out 233 234- op: logical_xor.out 235 236- op: logit.out 237 238- op: lt.Scalar_out 239 240- op: lt.Tensor_out 241 242- op: masked_fill.Scalar_out 243 244- op: masked_scatter.out 245 246- op: masked_select.out 247 248- op: max_pool2d_with_indices.out 249 250- op: max.dim_max 251 252- op: max.unary_out 253 254- op: max.unary_out 255 256- op: maximum.out 257 258- op: mean.out 259 260- op: min.dim_min 261 262- op: min.unary_out 263 264- op: minimum.out 265 266- op: mm.out 267 268- op: mul.out 269 270- op: mul.Scalar_out 271 272- op: narrow_copy.out 273 274- op: native_batch_norm.out 275 276- op: native_group_norm.out 277 278- op: native_layer_norm.out 279 280- op: ne.Scalar_out 281 282- op: ne.Tensor_out 283 284- op: neg.out 285 286- op: nonzero.out 287 288- op: ones.out 289 290- op: permute_copy.out 291 292- op: pixel_shuffle.out 293 294- op: pixel_unshuffle.out 295 296- op: pow.Scalar_out 297 298- op: pow.Tensor_Tensor_out 299 300- op: pow.Tensor_Scalar_out 301 302- op: prod.int_out 303 304- op: prod.out 305 306- op: reciprocal.out 307 308- op: relu.out 309 310- op: remainder.Tensor_out 311 312- op: remainder.Scalar_out 313 314- op: repeat.out 315 316- op: reflection_pad1d.out 317 318- op: reflection_pad2d.out 319 320- op: reflection_pad3d.out 321 322- op: replication_pad1d.out 323 324- op: replication_pad2d.out 325 326- op: replication_pad3d.out 327 328- op: roll.out 329 330- op: round.out 331 332- op: rsqrt.out 333 334- op: rsub.Scalar_out 335 336- op: scalar_tensor.out 337 338- op: scatter.src_out 339 340- op: scatter.value_out 341 342- op: scatter_add.out 343 344- op: select_copy.int_out 345 346- op: select_scatter.out 347 348- op: sigmoid.out 349 350- op: sign.out 351 352- op: sin.out 353 354- op: sinh.out 355 356- op: slice_copy.Tensor_out 357 358- op: slice_scatter.out 359 360- op: softplus.out 361 362- op: sort.values 363 364- op: split_copy.Tensor_out 365 366- op: split_with_sizes_copy.out 367 368- op: sqrt.out 369 370- op: squeeze_copy.dim_out 371 372- op: squeeze_copy.dims_out 373 374- op: squeeze_copy.out 375 376- op: stack.out 377 378- op: sub.out 379 380- op: sub.Scalar_out 381 382- op: sum.IntList_out 383 384- op: t_copy.out 385 386- op: tan.out 387 388- op: tanh.out 389 390- op: topk.values 391 392- op: transpose_copy.int_out 393 394- op: tril.out 395 396- op: trunc.out 397 398- op: unbind_copy.int_out 399 400- op: unsafe_split.Tensor_out 401 402- op: unsqueeze_copy.dim_out 403 404- op: unsqueeze_copy.out 405 406- op: upsample_nearest2d.out 407 408- op: upsample_nearest2d.vec_out 409 410- op: var.correction_out 411 412- op: var.out 413 414- op: view_copy.out 415 416- op: where.self_out 417 418- op: zeros_like.out 419 420- op: zeros.out 421 422- op: gather.out 423 424- op: scatter.value_out 425 426- op: aten::native_dropout.out 427