def create_ensemble_modelfile(create_savedmodel, models_dir, model_version, io_cnt, max_batch, dtype, shape): if not tu.validate_for_ensemble_model("zero", dtype, dtype, dtype, shape, shape, shape): return emu.create_identity_ensemble_modelfile("zero", models_dir, model_version, max_batch, dtype, [shape] * io_cnt, [shape] * io_cnt)
def create_ensemble_modelfile(models_dir, model_version, max_batch, dtype, input_shapes, output_shapes): assert len(input_shapes) == len(output_shapes) if not tu.validate_for_ensemble_model("reshape", dtype, dtype, dtype, input_shapes[0], input_shapes[0], input_shapes[0]): return emu.create_identity_ensemble_modelfile("reshape", models_dir, model_version, max_batch, dtype, input_shapes, output_shapes)