Home
last modified time | relevance | path

Searched refs:prediction_size (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/lite/examples/label_image/
Dget_top_n_impl.h30 void get_top_n(T* prediction, int prediction_size, size_t num_results, in get_top_n() argument
38 const long count = prediction_size; // NOLINT(runtime/int) in get_top_n()
Dget_top_n.h25 void get_top_n(T* prediction, int prediction_size, size_t num_results,
/external/tensorflow/tensorflow/contrib/gan/python/estimator/python/
Dgan_estimator_test.py214 self, train_input_fn, eval_input_fn, predict_input_fn, prediction_size, argument
248 self.assertAllEqual(prediction_size, predictions.shape)
275 prediction_size=[batch_size, input_dim])
302 prediction_size=[batch_size, input_dim],
351 prediction_size=[batch_size, input_dim])
Dstargan_estimator_test.py211 prediction_size, argument
245 self.assertAllEqual(prediction_size, predictions.shape)
302 prediction_size=[batch_size, img_size, img_size, channel_size])
Dtpu_gan_estimator_test.py155 self, train_input_fn, eval_input_fn, predict_input_fn, prediction_size, argument
195 self.assertAllEqual(prediction_size, predictions.shape)
241 prediction_size=[predict_size, input_dim],
/external/tensorflow/tensorflow/lite/examples/ios/simple/
DRunModelViewController.mm49 static void GetTopN(const float* prediction, const int prediction_size, const int num_results,
56 const long count = prediction_size;
/external/tensorflow/tensorflow/lite/examples/ios/camera/
DCameraExampleViewController.mm88 const float* prediction, const int prediction_size, const int num_results,
95 const long count = prediction_size;