mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 12:21:27 +01:00
a9b0a921d5
Summary: As GoogleTest `TEST` macro is non-compliant with it as well as `DEFINE_DISPATCH` All changes but the ones to `.clang-tidy` are generated using following script: ``` for i in `find . -type f -iname "*.c*" -or -iname "*.h"|xargs grep cppcoreguidelines-avoid-non-const-global-variables|cut -f1 -d:|sort|uniq`; do sed -i "/\/\/ NOLINTNEXTLINE(cppcoreguidelines-avoid-non-const-global-variables)/d" $i; done ``` Pull Request resolved: https://github.com/pytorch/pytorch/pull/62008 Reviewed By: driazati, r-barnes Differential Revision: D29838584 Pulled By: malfet fbshipit-source-id: 1b2f8602c945bd4ce50a9bfdd204755556e31d13
29 lines
922 B
C++
29 lines
922 B
C++
#include "caffe2/operators/quantized/int8_transpose_op.h"
|
|
|
|
namespace caffe2 {
|
|
|
|
REGISTER_CPU_OPERATOR(Int8Transpose, int8::Int8TransposeOp);
|
|
|
|
OPERATOR_SCHEMA(Int8Transpose)
|
|
.NumInputs(1)
|
|
.NumOutputs(1)
|
|
.SetDoc(R"DOC(
|
|
Transpose the input tensor by permuting the axes of the input according
|
|
to the `axes` argument. Similar to numpy's
|
|
[transpose](https://docs.scipy.org/doc/numpy/reference/generated/numpy.transpose.html)
|
|
function.
|
|
|
|
For example, when axes=(1, 0, 2), given an input tensor of shape
|
|
(1, 2, 3), the output shape will be (2, 1, 3).
|
|
)DOC")
|
|
.Arg(
|
|
"axes",
|
|
"*(type: Tuple(int))* Order to permute axes of input tensor. Reverses "
|
|
"the dimensions by default.")
|
|
.Arg("Y_scale", "Output tensor quantization scale")
|
|
.Arg("Y_zero_point", "Output tensor quantization offset")
|
|
.Input(0, "X", "Input tensor")
|
|
.Output(0, "Y", "Transposed output");
|
|
|
|
} // namespace caffe2
|