def create_ensemble_modelconfig(models_dir, model_version, max_batch, dtype, input_shapes, input_model_shapes, output_shapes, output_model_shapes): assert len(input_shapes) == len(input_model_shapes) assert len(output_shapes) == len(output_model_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 # No reason to reshape ensemble inputs / outputs to empty as the inner models # have to have non-empty shapes for inputs / outputs. input_model_shapes_list = [] output_model_shapes_list = [] for idx in range(len(input_shapes)): if len(input_model_shapes[idx]) == 0: input_model_shapes_list.append(input_shapes[idx]) else: input_model_shapes_list.append(input_model_shapes[idx]) if len(output_model_shapes[idx]) == 0: output_model_shapes_list.append(output_shapes[idx]) else: output_model_shapes_list.append(output_model_shapes[idx]) emu.create_identity_ensemble_modelconfig("reshape", models_dir, model_version, max_batch, dtype, input_shapes, tuple(input_model_shapes_list), output_shapes, tuple(output_model_shapes_list))
def create_ensemble_modelconfig(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_modelconfig( "zero", models_dir, model_version, max_batch, dtype, [shape] * io_cnt, [shape] * io_cnt, [shape] * io_cnt, [shape] * io_cnt)
def create_ensemble_modelconfig( models_dir, model_version, max_batch, dtype, input_shapes, input_model_shapes, output_shapes, output_model_shapes): assert len(input_shapes) == len(input_model_shapes) assert len(output_shapes) == len(output_model_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_modelconfig( "reshape", models_dir, model_version, max_batch, dtype, input_shapes, input_model_shapes, output_shapes, output_model_shapes)