Searched defs:output_layer (Results 1 - 5 of 5) sorted by relevance
/external/tensorflow/tensorflow/contrib/lite/testing/ |
H A D | tf_driver.cc | 65 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 D | label_image.py | 85 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 D | main.cc | 290 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 D | label_image.cc | 320 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 D | camera.cc | 410 string output_layer = "softmax"; local 423 Flag("output_layer", &output_layer, "output layer name"), 509 {output_layer}, {}, &outputs);
|
Completed in 168 milliseconds