Lines Matching refs:model_executor_
284 model_executor_ = TfLiteModelExecutor::FromBuffer( in ValidateAndInitialize()
286 if (!model_executor_) { in ValidateAndInitialize()
691 model_executor_->SetInput<std::string>( in SetupModelInput()
698 model_executor_->SetInput<std::string>( in SetupModelInput()
703 model_executor_->SetInput<int>( in SetupModelInput()
709 model_executor_->SetInput<int>( in SetupModelInput()
715 model_executor_->SetInput<int>( in SetupModelInput()
720 model_executor_->SetInput<int>( in SetupModelInput()
725 model_executor_->SetInput<float>( in SetupModelInput()
734 model_executor_->SetInput<int>( in SetupModelInput()
739 model_executor_->SetInput<float>( in SetupModelInput()
744 model_executor_->SetInput<float>( in SetupModelInput()
766 interpreter, model_executor_); in SetupModelInput()
768 model_executor_->SetInput<float>(param_index, kDefaultFloat, in SetupModelInput()
776 model_executor_); in SetupModelInput()
778 model_executor_->SetInput<int32_t>(param_index, kDefaultInt, in SetupModelInput()
783 model_executor_->SetInput<int64_t>( in SetupModelInput()
789 model_executor_->SetInput<uint8_t>( in SetupModelInput()
795 model_executor_->SetInput<int8_t>( in SetupModelInput()
801 model_executor_->SetInput<bool>( in SetupModelInput()
823 model_executor_->Output<tflite::StringRef>(suggestion_index, interpreter); in PopulateTextReplies()
825 model_executor_->OutputView<float>(score_index, interpreter); in PopulateTextReplies()
868 model_executor_->OutputView<bool>(suggestion_index, interpreter); in PopulateIntentTriggering()
870 model_executor_->OutputView<float>(score_index, interpreter); in PopulateIntentTriggering()
896 model_executor_->OutputView<float>( in ReadModelOutput()
910 model_executor_->OutputView<float>( in ReadModelOutput()
941 const TensorView<float> actions_scores = model_executor_->OutputView<float>( in ReadModelOutput()
1034 if (!model_executor_) { in SuggestActionsFromModel()
1037 *interpreter = model_executor_->CreateInterpreter(); in SuggestActionsFromModel()
1439 annotated_conversation, model_executor_.get(), interpreter.get(), in GatherActionsSuggestions()