Home
last modified time | relevance | path

Searched refs:input_refs (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/gpu/
Ddelegate.cc128 std::vector<uint32_t> input_refs; in Prepare() local
131 &input_refs, &output_refs)); in Prepare()
153 &input_refs, &output_refs)); in Prepare()
162 input_indices_.reserve(input_refs.size()); in Prepare()
163 for (uint32_t tensor_index : input_refs) { in Prepare()
260 std::vector<uint32_t>* input_refs, in InitializeGraph() argument
270 input_refs->clear(); in InitializeGraph()
273 input_refs->reserve(inputs.size()); in InitializeGraph()
275 input_refs->push_back(input->tensor.ref); in InitializeGraph()
/external/tensorflow/tensorflow/lite/delegates/gpu/cl/
Dgpu_api_delegate.cc130 std::vector<uint32_t> input_refs; in Prepare() local
133 input_refs.reserve(inputs.size()); in Prepare()
135 input_refs.push_back(input->tensor.ref); in Prepare()
196 input_indices_.reserve(input_refs.size()); in Prepare()
197 for (auto tensor_index : input_refs) { in Prepare()
Dserialization_generated.h296 const flatbuffers::Vector<int64_t> *input_refs() const { in input_refs() function
326 verifier.VerifyVector(input_refs()) && in Verify()
375 void add_input_refs(flatbuffers::Offset<flatbuffers::Vector<int64_t>> input_refs) { in add_input_refs()
376 fbb_.AddOffset(InferenceContext::VT_INPUT_REFS, input_refs); in add_input_refs()
414 flatbuffers::Offset<flatbuffers::Vector<int64_t>> input_refs = 0,
418 builder_.add_input_refs(input_refs);
454 const std::vector<int64_t> *input_refs = nullptr,
467 auto input_refs__ = input_refs ? _fbb.CreateVector<int64_t>(*input_refs) : 0;
Dserialization.fbs49 input_refs:[int64];
Dserialization.cc1061 for (auto in_fb : *fb_inference->input_refs()) { in Decode()