Home
last modified time | relevance | path

Searched refs:input_ids_ (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/lite/testing/
Dtf_driver.h46 const std::vector<int>& GetInputs() override { return input_ids_; } in GetInputs()
67 std::vector<int> input_ids_; variable
Dtf_driver.cc113 input_ids_.resize(input_layer.size()); in TfDriver()
118 input_ids_[i] = i; in TfDriver()
/external/tensorflow/tensorflow/lite/delegates/gpu/cl/
Dinference_context.h95 const std::vector<ValueId>& GetInputIds() const { return input_ids_; } in GetInputIds()
232 std::vector<ValueId> input_ids_; variable
Dinference_context.cc240 input_ids_.push_back(input->id); in CopyInAndOutIds()
303 for (const auto& input_id : input_ids_) { in ConvertOperations()
406 for (const auto& input_id : input_ids_) { in Merge()
452 for (ValueId in_id : input_ids_) { in GetUsages()
Dserialization.cc956 std::vector<int32_t> in_ids(inference.input_ids_.size()); in Encode()
958 in_ids[i] = inference.input_ids_[i]; in Encode()
1050 inference->input_ids_.push_back(in_fb); in Decode()
/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/
Dop_builder.h132 void AddInput(const TensorID& tensor_id) { input_ids_.push_back(tensor_id); } in AddInput()
218 std::vector<TensorID> input_ids_; variable
Dop_builder.cc362 for (const auto& id : input_ids_) { in Build()
/external/tensorflow/tensorflow/lite/delegates/gpu/metal/
Dinference_context.cc140 input_ids_.push_back(input->id); in InitFromGraph()
213 for (const auto& input_id : input_ids_) { in Compile()
313 for (const auto& input_id : input_ids_) { in Merge()
444 for (ValueId in_id : input_ids_) { in GetUsages()
Dinference_context.h208 std::vector<ValueId> input_ids_; variable