def tfs_command()

in docker/build_artifacts/sagemaker/tfs_utils.py [0:0]

• 20 lines of code
• 1 McCabe index (conditional complexity)

def tfs_command(tfs_grpc_port,
                tfs_rest_port,
                tfs_config_path,
                tfs_enable_batching,
                tfs_batching_config_file,
                tfs_intra_op_parallelism=None,
                tfs_inter_op_parallelism=None,
                tfs_enable_gpu_memory_fraction=False,
                tfs_gpu_memory_fraction=None):
    cmd = "tensorflow_model_server " \
          "--port={} " \
          "--rest_api_port={} " \
          "--model_config_file={} " \
          "--max_num_load_retries=0 {} {} {} {}"\
        .format(tfs_grpc_port, tfs_rest_port, tfs_config_path,
                get_tfs_batching_args(tfs_enable_batching, tfs_batching_config_file),
                get_tensorflow_intra_op_parallelism_args(tfs_intra_op_parallelism),
                get_tensorflow_inter_op_parallelism_args(tfs_inter_op_parallelism),
                get_tfs_gpu_mem_args(tfs_enable_gpu_memory_fraction, tfs_gpu_memory_fraction))
    return cmd