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)
