Crate caffe2_export

source ·

Macros

Structs

  • C10OperatorWrapper
    | To make a c10 operator “C10Add” callable | from caffe2 as “C2MyAddOpName”, just | write | | To export the CPU kernel | | C10_EXPORT_C10_OP_TO_CAFFE2_CPU(C10Add, | C2MyAddOp) | | To export the CUDA kernel | | C10_EXPORT_C10_OP_TO_CAFFE2_CUDA(C10Add, | C2MyAddOp) |

Constants

Functions

  • call_caffe2_op
  • call_caffe2_op_from_c10
    | This function is inline in the hope that | compilers optimizing for speed will | inline it into call_caffe2_op_from_c10, | allowing call_op to be inlined and avoiding | the function pointer indirection, | while compilers optimizing for binary | size will keep it a separate function | instead of inlining it into a template | and will reuse the binary code of this | function between ops. | | We measured and confirmed that binary | size off the instagram ios app is reduced | when having _call_caffe2_op_from_c10 | separate from the templated call_caffe2_op_from_c10. |
  • call_caffe2_op_from_c10_default_schema
  • create_c10operator_wrapper
  • make_function_schema_for_c10

Type Definitions