def load_kaldi_nnet3_model(graph, file_descr, nnet_name):
    file_descr.read(1)
    component_layer_map = load_topology_map(file_descr, graph)
    for node in graph.get_op_nodes(**{'op': 'Parameter'}):
        for (o_n_name, params) in node.get_outputs():
            o_n = Node(graph, o_n_name)
            if (o_n['op'] == 'MemoryOffset'):
                o_n['parameters']['element_size'] = int64_array([1, node.shape[1]])
    load_components(file_descr, graph, component_layer_map)
    load_priors(file_descr, graph)
def load_kaldi_nnet3_model(graph, file_descr, nnet_name):
    file_descr.read(1)
    component_layer_map = load_topology_map(file_descr, graph)
    for node in graph.get_op_nodes(**{'op': 'Parameter'}):
        for (o_n_name, params) in node.get_outputs():
            o_n = Node(graph, o_n_name)
            if (o_n['op'] == 'MemoryOffset'):
                o_n['parameters']['element_size'] = int64_array([1, node.shape[1]])
    load_components(file_descr, graph, component_layer_map)
    load_priors(file_descr, graph)