1 #pragma once 2 #include <torch/csrc/jit/mobile/prim_ops_registery.h> 3 #include <torch/csrc/jit/mobile/register_ops_common_utils.h> 4 5 namespace torch::jit { 6 7 void tupleIndex(Stack& stack); 8 9 void raiseException(Stack& stack); 10 11 void is(Stack& stack); 12 13 void unInitialized(Stack& stack); 14 15 void isNot(Stack& stack); 16 17 void aten_format(Stack& stack); 18 19 void size(Stack& stack); 20 21 void sym_size(Stack& stack); 22 23 void sym_size_int(Stack& stack); 24 25 void sym_stride_int(Stack& stack); 26 27 void sym_numel(Stack& stack); 28 29 void sym_storage_offset(Stack& stack); 30 31 void sym_stride(Stack& stack); 32 33 void device(Stack& stack); 34 35 void device_with_index(Stack& stack); 36 37 void dtype(Stack& stack); 38 39 void layout(Stack& stack); 40 41 void toPrimDType(Stack& stack); 42 43 void dim(Stack& stack); 44 45 void _not(Stack& stack); 46 47 void boolTensor(Stack& stack); 48 49 void toList(Stack& stack); 50 51 void numToTensorScalar(Stack& stack); 52 53 void isCuda(Stack& stack); 54 55 void numToTensorBool(Stack& stack); 56 57 void dictIndex(Stack& stack); 58 59 void raiseExceptionWithMessage(Stack& stack); 60 61 } // namespace torch::jit 62