pyspark/bigdl/util/tf_utils.py
def convert(input_ops, output_ops, byte_order, bigdl_type):
    """
    Convert tensorflow model to bigdl model
    :param input_ops: operation list used for input, should be placeholders
    :param output_ops: operations list used for output
    :return: bigdl model
    """

    input_names = map(lambda x: x.name.split(":")[0], input_ops)
    output_names = map(lambda x: x.name.split(":")[0], output_ops)
    temp = tempfile.mkdtemp()

    dump_model(path=temp)
    model_path = temp + '/model.pb'
    bin_path = temp + '/model.bin'

    model = Model.load_tensorflow(model_path, input_names, output_names,
                                  byte_order, bin_path, bigdl_type)

    try:
        shutil.rmtree(temp)
    except OSError as e:
        if e.errno != errno.ENOENT:
            raise

    return model
def convert(input_ops, output_ops, byte_order, bigdl_type):
    """
    Convert tensorflow model to bigdl model
    :param input_ops: operation list used for input, should be placeholders
    :param output_ops: operations list used for output
    :return: bigdl model
    """

    input_names = map(lambda x: x.name.split(":")[0], input_ops)
    output_names = map(lambda x: x.name.split(":")[0], output_ops)
    temp = tempfile.mkdtemp()

    dump_model(path=temp)
    model_path = temp + '/model.pb'
    bin_path = temp + '/model.bin'

    model = Model.load_tensorflow(model_path, input_names, output_names,
                                  byte_order, bin_path, bigdl_type)

    try:
        shutil.rmtree(temp)
    except OSError as e:
        if e.errno != errno.ENOENT:
            raise

    return model
def convert ( input_ops , output_ops , byte_order , bigdl_type ) : input_names = map ( lambda x : x . name . split ( ":" ) [ 0 ] , input_ops ) output_names = map ( lambda x : x . name . split ( ":" ) [ 0 ] , output_ops ) temp = tempfile . mkdtemp ( ) dump_model ( path = temp ) model_path = temp + '/model.pb' bin_path = temp + '/model.bin' model = Model . load_tensorflow ( model_path , input_names , output_names , byte_order , bin_path , bigdl_type ) try : shutil . rmtree ( temp ) except OSError as e : if e . errno != errno . ENOENT : raise return model
Convert tensorflow model to bigdl model
    :param input_ops: operation list used for input, should be placeholders
    :param output_ops: operations list used for output
    :return: bigdl model
Convert tensorflow model to bigdl model : param input_ops : operation list used for input should be placeholders : param output_ops : operations list used for output : return : bigdl model