Searched refs:output_values_ (Results 1 – 3 of 3) sorted by relevance
34 output_values_ = AddOutput(input_type); in TopKV2OpModel()66 return ExtractVector<float>(output_values_); in GetValuesFloat()70 return ExtractVector<uint8_t>(output_values_); in GetValuesUInt8()74 return ExtractVector<int8_t>(output_values_); in GetValuesInt8()78 return ExtractVector<int32_t>(output_values_); in GetValuesInt32()82 return ExtractVector<int64_t>(output_values_); in GetValuesInt64()89 int output_values_; member in tflite::__anondc2192160111::TopKV2OpModel
485 output_values_.resize(outputs_.size()); in SetOutputs()491 output_values_.resize(outputs_.size()); in SetOutputs()507 output_values_.resize(outputs_.size(), nullptr); in Run()515 output_values_.empty() ? nullptr : &output_values_[0]; in Run()545 for (size_t i = 0; i < output_values_.size(); ++i) { in ResetOutputValues()546 if (output_values_[i] != nullptr) TF_DeleteTensor(output_values_[i]); in ResetOutputValues()548 output_values_.clear(); in ResetOutputValues()
149 TF_Tensor* output_tensor(int i) { return output_values_[i]; } in output_tensor()161 std::vector<TF_Tensor*> output_values_; variable