1 #include "caffe2/operators/lengths_reducer_fused_8bit_rowwise_ops.h" 2 #include "caffe2/core/registry.h" 7 SparseLengthsSumFused8BitRowwise,
8 SparseLengthsFused8BitRowwiseOp<CPUContext>);
9 OPERATOR_SCHEMA(SparseLengthsSumFused8BitRowwise)
13 Performs the same operation as SparseLengthsSum, but operating on 14 8-bit rowwise quantized matrices with fused storage (where each row 15 stores quantized values, and then 4-byte scale and 4-byte bias). 20 "uint8 tensor obtained with " 21 "operator FloatToFused8BitRowwiseQuantized")
25 "Integer vector containing indices of the first " 26 "dimension of DATA for the slices that are being aggregated")
30 "Vector with the same sum of elements as the first dimension of DATA")
31 .Output(0,
"output",
"output");
32 NO_GRADIENT(SparseLengthsSumFused8BitRowwise);
34 REGISTER_CPU_OPERATOR(
35 SparseLengthsWeightedSumFused8BitRowwise,
36 SparseLengthsFused8BitRowwiseOp<CPUContext, /*with_weights=*/true>);
37 OPERATOR_SCHEMA(SparseLengthsWeightedSumFused8BitRowwise)
41 Performs the same operation as SparseLengthsWeightedSum, 42 but operating on 8-bit rowwise quantized matrices with fused storage 43 (where each row stores quantized values, and then 4-byte scale and 4-byte bias). 48 "uint8 tensor obtained with " 49 "operator FloatToFused8BitRowwiseQuantized")
53 "Integer vector containing indices of the first " 54 "dimension of DATA for the slices that are being aggregated")
58 "Vector with the same sum of elements as the first dimension of DATA")
62 "Vector of weights to scale rows of DATA with before reduction")
63 .Output(0,
"output",
"output");
65 NO_GRADIENT(SparseLengthsWeightedSumFused8BitRowwise);
67 REGISTER_CPU_OPERATOR(
68 SparseLengthsMeanFused8BitRowwise,
69 SparseLengthsFused8BitRowwiseOp<
73 OPERATOR_SCHEMA(SparseLengthsMeanFused8BitRowwise)
77 Performs the same operation as SparseLengthsMean, but 78 operating on 8-bit rowwise quantized matrices with fused storage 79 (where each row stores quantized values, and then 4-byte scale and 4-byte bias). 84 "uint8 tensor obtained with " 85 "operator FloatToFused8BitRowwiseQuantized")
89 "Integer vector containing indices of the first " 90 "dimension of DATA for the slices that are being aggregated")
94 "Vector with the same sum of elements as the first dimension of DATA")
95 .Output(0,
"output",
"output");
96 NO_GRADIENT(SparseLengthsMeanFused8BitRowwise);
A global dictionary that holds information about what Caffe2 modules have been loaded in the current ...