From 04c487dcbc72834ce1f519a06752073aeafa7b97 Mon Sep 17 00:00:00 2001 From: Song Jiaming Date: Fri, 19 Jun 2020 13:31:28 +0800 Subject: [PATCH] Cluster Serving Inference Model Tensorflow String inputs and UT (#2482) --- .../analytics/bigdl/dllib/inference/InferenceModel.scala | 6 ------ 1 file changed, 6 deletions(-) diff --git a/bigdl/dllib/src/main/scala/com/intel/analytics/bigdl/dllib/inference/InferenceModel.scala b/bigdl/dllib/src/main/scala/com/intel/analytics/bigdl/dllib/inference/InferenceModel.scala index b3b4eb273c6..80af7d44a7d 100644 --- a/bigdl/dllib/src/main/scala/com/intel/analytics/bigdl/dllib/inference/InferenceModel.scala +++ b/bigdl/dllib/src/main/scala/com/intel/analytics/bigdl/dllib/inference/InferenceModel.scala @@ -744,12 +744,6 @@ class InferenceModel(private var autoScalingEnabled: Boolean = true, val model: AbstractModel = retrieveModel() try { val begin = System.nanoTime() -// val batchSize = if (inputActivity.isTensor) { -// inputActivity.toTensor[T].size(1) -// } else { -// val sampleKey = inputActivity.toTable.keySet.head -// inputActivity.toTable(sampleKey).asInstanceOf[Tensor[T]].size(1) -// } val result = model.predict(inputActivity) val end = System.nanoTime()