Searched refs:model_executor (Results 1 – 4 of 4) sorted by relevance
| /external/libtextclassifier/native/actions/ |
| D | lua-actions.cc | 35 const TfLiteModelExecutor* model_executor, in GetTensorViewForOutput() argument
37 if (output < 0 || model_executor == nullptr || interpreter == nullptr) { in GetTensorViewForOutput()
40 return model_executor->OutputView<float>(output, interpreter); in GetTensorViewForOutput()
44 const TfLiteModelExecutor* model_executor, in GetStringTensorForOutput() argument
46 if (output < 0 || model_executor == nullptr || interpreter == nullptr) { in GetStringTensorForOutput()
49 return model_executor->Output<std::string>(output, interpreter); in GetStringTensorForOutput()
57 const TfLiteModelExecutor* model_executor, in CreateLuaActionsSuggestions() argument
64 snippet, conversation, model_executor, model_spec, interpreter, in CreateLuaActionsSuggestions()
76 const TfLiteModelExecutor* model_executor, in LuaActionsSuggestions() argument
86 : GetTensorViewForOutput(model_executor, interpreter, in LuaActionsSuggestions()
[all …]
|
| D | lua-actions.h | 33 const TfLiteModelExecutor* model_executor,
44 const TfLiteModelExecutor* model_executor,
|
| D | actions-suggestions.cc | 99 const std::unique_ptr<const TfLiteModelExecutor>& model_executor) { in SetVectorOrScalarAsModelInput() argument
101 model_executor->SetInput<T>( in SetVectorOrScalarAsModelInput()
104 model_executor->SetInput<float>(param_index, param_value.Value<T>(), in SetVectorOrScalarAsModelInput()
1272 const Conversation& conversation, const TfLiteModelExecutor* model_executor, in SuggestActionsFromLua() argument
1281 lua_bytecode_, conversation, model_executor, model_->tflite_model_spec(), in SuggestActionsFromLua()
1291 const Conversation& conversation, const TfLiteModelExecutor* model_executor, in SuggestActionsFromLua() argument
|
| D | actions-suggestions.h | 225 const TfLiteModelExecutor* model_executor,
|