OpInvAlign

class OpInvAlign.InvAlignOp(tdps, nstates)[source]
itypes = [TensorType(int8, matrix), TensorType(int8, matrix), TensorType(float32, 3D), TensorType(int32, matrix)][source]
otypes = [TensorType(int32, matrix), TensorType(int32, matrix), TensorType(int8, matrix)][source]
perform(node, inputs_storage, output_storage)[source]
grad(inputs, output_grads)[source]
infer_shape(node, input_shapes)[source]
class OpInvAlign.InvFullAlignOp(tdps, nstates)[source]
itypes = [TensorType(int8, matrix), TensorType(int8, matrix), TensorType(float32, 3D), TensorType(int32, matrix)][source]
otypes = [TensorType(float32, 3D), TensorType(int8, matrix)][source]
perform(node, inputs_storage, output_storage)[source]
grad(inputs, output_grads)[source]
infer_shape(node, input_shapes)[source]
class OpInvAlign.InvBacktrackOp(tdps, nstates, penalty)[source]
itypes = [TensorType(int8, matrix), TensorType(float32, 3D), TensorType(float32, 3D)][source]
otypes = [TensorType(int32, matrix), TensorType(int32, matrix), TensorType(int8, matrix)][source]
perform(node, inputs_storage, output_storage)[source]
grad(inputs, output_grads)[source]
infer_shape(node, input_shapes)[source]
class OpInvAlign.InvDecodeOp(tdps, nstates, penalty)[source]
itypes = [TensorType(int8, matrix), TensorType(float32, 3D)][source]
otypes = [TensorType(int32, matrix), TensorType(int32, matrix), TensorType(int8, matrix)][source]
perform(node, inputs_storage, output_storage)[source]
grad(inputs, output_grads)[source]
infer_shape(node, input_shapes)[source]