onnx_legalizer.py File Reference

Go to the source code of this file.

Data Structures
class	onnx_legalizer.LegalizeOptions
class	onnx_legalizer._ModelTransformerHelper
class	onnx_legalizer._TensorInfo

Namespaces
namespace	onnx_legalizer

Functions
	onnx_legalizer._reverse_str (s)
	onnx_legalizer._parse_tensor_name (name)
	onnx_legalizer._get_tensor_infos (model)
	onnx_legalizer._dtype_to_np (dtype)
	onnx_legalizer._generate_one_direction_RNN (transformer, X, W, R, B, initial_h, clip, activation_name)
	onnx_legalizer._transform_unidirectional_RNN (transformer, original_node, x, tensor_infos, activation, clip, direction, hidden_size, layout)
	onnx_legalizer._transform_bidirectional_RNN (transformer, original_node, x, tensor_infos, activations, clip, hidden_size, layout)
	onnx_legalizer._legalize_RNN (transformer, tensor_infos, node)
	onnx_legalizer._generate_one_direction_LSTM (transformer, X, W, R, B, initial_h, initial_c, P, clip, act, dtype, hidden_size, batch_size)
	onnx_legalizer._transform_unidirectional_LSTM (transformer, original_node, x, tensor_infos, activations, clip, direction, hidden_size, layout)
	onnx_legalizer._transform_bidirectional_LSTM (transformer, original_node, x, tensor_infos, activations, clip, hidden_size, layout)
	onnx_legalizer._legalize_LSTM (transformer, tensor_infos, node)
	onnx_legalizer.legalize (model, options)

Variables
	onnx_legalizer.options = LegalizeOptions()
	onnx_legalizer.unroll_lstm
	onnx_legalizer.unroll_rnn
	onnx_legalizer.model = onnx.load(sys.argv[1])