Searched refs:out_refs (Results 1 – 3 of 3) sorted by relevance
143 std::vector<int64_t> out_refs(outputs.size()); in RunModelSampleWithInternalAPI() local148 out_refs[i] = outputs[i]->tensor.ref; in RunModelSampleWithInternalAPI()182 for (int i = 0; i < out_refs.size(); ++i) { in RunModelSampleWithInternalAPI()206 std::vector<std::vector<float>> output_tensors(out_refs.size()); in RunModelSampleWithInternalAPI()207 for (int i = 0; i < out_refs.size(); ++i) { in RunModelSampleWithInternalAPI()208 TfLiteTensor* tensor_ptr = cpu_inference->tensor(out_refs[i]); in RunModelSampleWithInternalAPI()216 CompareCPUGPUResults(cpu_inference.get(), out_refs, output_tensors, 1e-4f); in RunModelSampleWithInternalAPI()267 std::vector<int64_t> out_refs(outputs.size()); in RunModelSampleWithInternalAPISerializedKernels() local272 out_refs[i] = outputs[i]->tensor.ref; in RunModelSampleWithInternalAPISerializedKernels()311 for (int i = 0; i < out_refs.size(); ++i) { in RunModelSampleWithInternalAPISerializedKernels()[all …]
94 std::vector<int64_t>* out_refs) = 0;
689 std::vector<int64_t>* out_refs = nullptr) { in Initialize() argument712 if (out_refs) { in Initialize()713 *out_refs = context_->GetOutputRefs(); in Initialize()962 std::vector<int64_t>* out_refs) final { in NewInferenceBuilder() argument974 in_refs, out_refs)); in NewInferenceBuilder()