INNER CODE UNIT · Python
output_shape
onnx/onnx-tensorrt · onnx_tensorrt/backend.py:164
if output_shape == (-99,):
# WAR for TRT requiring at least 2 dims (NC)
min_dims = 2
if array.ndim == min_dims:
npadding_dims = count_trailing_ones(array.shape)
if npadding_dims > 0:
outputs[i] = array.reshape(
array.shape[:-npadding_dims])
else:
# HACK WAR replace fixed batch dim with variable
if self._output_dtype[name] == onnx.TensorProto.INT64 and array.dtype == np.int32:
casted_output = np.array(outputs[i], dtype=np.int64)
if np.equal(outputs[i], casted_output).all():
outputs[i] = np.array(outputs[i], dtype=np.int64)
if self._output_dtype[name] == onnx.TensorProto.DOUBLE and array.dtype == np.float32:
casted_output = np.array(outputs[i], dtype=np.double)
if np.equal(outputs[i], casted_output).all():
outputs[i] = np.array(outputs[i], dtype=np.double)