From efc56c8852225527ac0896fe1504ef99c06be6ae Mon Sep 17 00:00:00 2001 From: yejianwu Date: Tue, 22 May 2018 20:05:45 +0800 Subject: [PATCH] format code --- mace/python/tools/tensor_util.py | 1 + 1 file changed, 1 insertion(+) diff --git a/mace/python/tools/tensor_util.py b/mace/python/tools/tensor_util.py index 62d33052..61a5e90c 100644 --- a/mace/python/tools/tensor_util.py +++ b/mace/python/tools/tensor_util.py @@ -165,6 +165,7 @@ def del_tensor_data(net_def, runtime, gpu_data_type): elif t.data_type == mace_pb2.DT_UINT8: del t.int32_data[:] + def update_tensor_data_type(net_def, runtime, gpu_data_type): for t in net_def.tensors: if t.data_type == mace_pb2.DT_FLOAT and runtime == 'gpu' \ -- GitLab