Searched refs:output_values_ (Results 1 – 6 of 6) sorted by relevance
48 output_values_ = AddOutput(TensorType_STRING); in WhitespaceTokenizerModel()62 return GetTensorShape(output_values_); in GetValuesTensorShape()67 TfLiteTensor* tensor = interpreter_->tensor(output_values_); in ExtractValuesTensorVector()107 int output_values_; member in tflite::ops::custom::whitespace_tokenizer::test::WhitespaceTokenizerModel
47 output_values_ = AddOutput(TensorType_STRING); in NgramsModel()67 output_values_ = AddOutput(TensorType_STRING); in NgramsModel()97 return GetTensorShape(output_values_); in GetValuesTensorShape()102 TfLiteTensor* tensor = interpreter_->tensor(output_values_); in ExtractValuesTensorVector()149 int output_values_; member in tflite::ops::custom::ngrams::test::NgramsModel
54 output_values_ = AddOutput(value_type); in RaggedRangeOpModel()67 std::vector<T> GetValues() const { return ExtractVector<T>(output_values_); } in GetValues()75 int output_values_ = -1; member in tflite::__anon7bee03b80111::RaggedRangeOpModel
49 output_values_ = AddOutput(GetTensorType<InputType>()); in TopKV2OpModel()65 return ExtractVector<InputType>(output_values_); in GetValues()72 int output_values_; member in tflite::__anonc19024f40111::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