Searched refs:tpu_model_ops (Results 1 – 1 of 1) sorted by relevance
1121 def _test_model_compiles(self, tpu_model_ops): argument1126 result = K.get_session().run(tpu_model_ops.compile_op)1276 tpu_model_ops = self._tpu_model_ops_for_input_specs(input_specs,1278 infeed_dict = infeed_instance.make_feed_dict(tpu_model_ops)1284 tpu_model_ops.infeed_op, tpu_model_ops.execute_op,1285 tpu_model_ops.outfeed_op