Home
last modified time | relevance | path

Searched refs:model_outputs (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/c/eager/
Dunified_api_testutil.cc69 std::vector<AbstractTensorHandle*> model_outputs; in RunModel() local
70 model_outputs.resize(outputs.size()); in RunModel()
72 absl::MakeSpan(model_outputs))); in RunModel()
80 for (int i = 0; i < model_outputs.size(); i++) { in RunModel()
81 if (model_outputs[i]) { in RunModel()
82 output_list.outputs.emplace_back(model_outputs[i]); in RunModel()
Dgradients_util.cc255 vector<AbstractTensorHandle*> model_outputs; in RunModel() local
256 model_outputs.resize(outputs.size()); in RunModel()
258 absl::MakeSpan(model_outputs), registry)); in RunModel()
266 for (int i = 0; i < model_outputs.size(); i++) { in RunModel()
267 if (model_outputs[i]) { in RunModel()
268 output_list.outputs.emplace_back(model_outputs[i]); in RunModel()
Dgradient_checker.cc52 std::vector<AbstractTensorHandle*> model_outputs(1); in RunAndMaybeSum() local
56 absl::MakeSpan(model_outputs), use_function)); in RunAndMaybeSum()
57 AbstractTensorHandlePtr model_out(model_outputs[0]); in RunAndMaybeSum()
/external/tensorflow/tensorflow/python/keras/engine/
Dfunctional.py1385 model_outputs = []
1392 model_outputs.append(
1394 model_outputs = nest.pack_sequence_as(network._nested_outputs, model_outputs)
1396 if not nest.is_nested(model_outputs):
1397 model_outputs = [model_outputs]
1398 model_outputs = tf_utils.convert_inner_node_data(model_outputs)
1399 config['output_layers'] = model_outputs
/external/tensorflow/tensorflow/lite/testing/
Dparse_testdata.cc192 int model_outputs = interpreter->outputs().size(); in CheckOutputs() local
193 TF_LITE_ENSURE_EQ(context, model_outputs, example.outputs.size()); in CheckOutputs()
/external/tensorflow/tensorflow/lite/testing/nnapi_tflite_zip_tests/
Dparse_testdata.cc195 int model_outputs = interpreter->outputs().size(); in CheckOutputs() local
196 TF_LITE_ENSURE_EQ(context, model_outputs, example.outputs.size()); in CheckOutputs()