Source code for experimental_experiment.reference.ops.op_transpose_cast

import numpy as np
from onnx.reference.op_run import OpRun



[docs]
class Transpose2DCastFP16(OpRun):
    op_domain = "onnx_extended.ortops.optim.cuda"

    def _run(self, X):
        return (X.T.astype(np.float16),)




[docs]
class Transpose2DCastFP32(OpRun):
    op_domain = "onnx_extended.ortops.optim.cuda"

    def _run(self, X):
        return (X.T.astype(np.float32),)