Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/testing/
Dtf_driver.h47 const std::vector<int>& GetOutputs() override { return output_ids_; } in GetOutputs()
73 std::vector<int> output_ids_; variable
Dtf_driver.cc124 output_ids_.resize(output_layer.size()); in TfDriver()
127 output_ids_[i] = i; in TfDriver()
/external/tensorflow/tensorflow/lite/delegates/gpu/cl/
Dinference_context.h96 const std::vector<ValueId>& GetOutputIds() const { return output_ids_; } in GetOutputIds()
234 std::vector<ValueId> output_ids_; variable
Dinference_context.cc250 output_ids_.push_back(output->id); in CopyInAndOutIds()
465 for (ValueId out_id : output_ids_) { in GetUsages()
Dserialization.cc960 std::vector<int32_t> out_ids(inference.output_ids_.size()); in Encode()
962 out_ids[i] = inference.output_ids_[i]; in Encode()
1053 inference->output_ids_.push_back(out_fb); in Decode()
/external/tensorflow/tensorflow/lite/delegates/gpu/metal/
Dinference_context.h209 std::vector<ValueId> output_ids_; variable
Dinference_context.cc146 output_ids_.push_back(output->id); in InitFromGraph()
461 for (ValueId out_id : output_ids_) { in GetUsages()