Lines Matching refs:model_executor_
279 model_executor_ = TfLiteModelExecutor::FromBuffer( in ValidateAndInitialize()
281 if (!model_executor_) { in ValidateAndInitialize()
686 model_executor_->SetInput<std::string>( in SetupModelInput()
693 model_executor_->SetInput<std::string>( in SetupModelInput()
698 model_executor_->SetInput<int>( in SetupModelInput()
704 model_executor_->SetInput<int>( in SetupModelInput()
710 model_executor_->SetInput<int>( in SetupModelInput()
715 model_executor_->SetInput<int>( in SetupModelInput()
720 model_executor_->SetInput<float>( in SetupModelInput()
729 model_executor_->SetInput<int>( in SetupModelInput()
734 model_executor_->SetInput<float>( in SetupModelInput()
739 model_executor_->SetInput<float>( in SetupModelInput()
761 interpreter, model_executor_); in SetupModelInput()
763 model_executor_->SetInput<float>(param_index, kDefaultFloat, in SetupModelInput()
771 model_executor_); in SetupModelInput()
773 model_executor_->SetInput<int32_t>(param_index, kDefaultInt, in SetupModelInput()
778 model_executor_->SetInput<int64_t>( in SetupModelInput()
784 model_executor_->SetInput<uint8_t>( in SetupModelInput()
790 model_executor_->SetInput<int8_t>( in SetupModelInput()
796 model_executor_->SetInput<bool>( in SetupModelInput()
815 model_executor_->Output<tflite::StringRef>(suggestion_index, interpreter); in PopulateTextReplies()
817 model_executor_->OutputView<float>(score_index, interpreter); in PopulateTextReplies()
849 model_executor_->OutputView<bool>(suggestion_index, interpreter); in PopulateIntentTriggering()
851 model_executor_->OutputView<float>(score_index, interpreter); in PopulateIntentTriggering()
877 model_executor_->OutputView<float>( in ReadModelOutput()
891 model_executor_->OutputView<float>( in ReadModelOutput()
920 const TensorView<float> actions_scores = model_executor_->OutputView<float>( in ReadModelOutput()
993 if (!model_executor_) { in SuggestActionsFromModel()
996 *interpreter = model_executor_->CreateInterpreter(); in SuggestActionsFromModel()
1398 annotated_conversation, model_executor_.get(), interpreter.get(), in GatherActionsSuggestions()