Searched refs:run_model (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/python/keras/engine/ |
| D | input_layer_test.py | 149 def run_model(inp): function
160 self.assertAllEqual(run_model(array_ops.ones((10, 8))),
182 def run_model(inp): function
192 self.assertAllEqual(run_model(array_ops.ones((10, 16))),
217 def run_model(inp): function
231 self.assertAllEqual(run_model(rt), rt * 3)
289 def run_model(inp): function
300 self.assertAllEqual(run_model(array_ops.ones((10, 16))),
331 def run_model(inp): function
345 self.assertAllEqual(run_model(rt), rt * 3)
|
| /external/tensorflow/tensorflow/python/keras/benchmarks/ |
| D | model_memory_profile.py | 70 run_model = models[FLAGS.model]
71 run_model()
|
| /external/tensorflow/tensorflow/python/compiler/tensorrt/model_tests/ |
| D | model_handler.py | 630 def run_model(model, **kwargs): function
636 cpu_base_result = run_model(self._ori_model, enable_gpu=False)
641 gpu_base_result = run_model(self._ori_model, enable_gpu=True)
642 trt_results = list(map(run_model, self._trt_models))
|
| /external/tensorflow/tensorflow/lite/examples/experimental_new_converter/ |
| D | Keras_LSTM_fusion_Codelab.ipynb | 164 "run_model = tf.function(lambda x: model(x))\n",
169 "concrete_func = run_model.get_concrete_function(\n",
|