Searched defs:output_layer (Results 1 - 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/contrib/lite/testing/
H A Dtf_driver.cc65 const std::vector<string>& output_layer)
66 : input_names_(input_layer), output_names_(output_layer) {
81 output_ids_.resize(output_layer.size());
82 output_tensors_.reserve(output_layer.size());
83 for (int i = 0; i < output_layer.size(); i++) {
62 TfDriver(const std::vector<string>& input_layer, const std::vector<string>& input_layer_type, const std::vector<string>& input_layer_shape, const std::vector<string>& output_layer) argument
/external/tensorflow/tensorflow/examples/label_image/
H A Dlabel_image.py85 output_layer = "InceptionV3/Predictions/Reshape_1" variable
96 parser.add_argument("--output_layer", help="name of output layer")
115 if args.output_layer:
116 output_layer = args.output_layer variable
127 output_name = "import/" + output_layer
H A Dmain.cc290 string output_layer = "InceptionV3/Predictions/Reshape_1"; local
303 Flag("output_layer", &output_layer, "name of output layer"),
347 {output_layer}, {}, &outputs);
/external/tensorflow/tensorflow/contrib/pi_examples/label_image/
H A Dlabel_image.cc320 string output_layer = "softmax"; local
333 Flag("output_layer", &output_layer, "name of output layer"),
377 {output_layer}, {}, &outputs);
/external/tensorflow/tensorflow/contrib/pi_examples/camera/
H A Dcamera.cc410 string output_layer = "softmax"; local
423 Flag("output_layer", &output_layer, "output layer name"),
509 {output_layer}, {}, &outputs);

Completed in 168 milliseconds