Home
last modified time | relevance | path

Searched refs:OutputView (Results 1 – 4 of 4) sorted by relevance

/external/libtextclassifier/utils/
Dtflite-model-executor.h114 TensorView<T> OutputView(const int output_index, in OutputView() function
127 TensorView<T> output_view = OutputView<T>(output_index, interpreter); in Output()
/external/libtextclassifier/actions/
Dlua-actions.cc38 return model_executor->OutputView<float>(output, interpreter); in GetTensorViewForOutput()
Dactions-suggestions.cc867 model_executor_->OutputView<float>( in ReadModelOutput()
881 model_executor_->OutputView<float>( in ReadModelOutput()
907 TensorView<float> scores = model_executor_->OutputView<float>( in ReadModelOutput()
923 const TensorView<float> actions_scores = model_executor_->OutputView<float>( in ReadModelOutput()
/external/libtextclassifier/annotator/
Dmodel-executor.cc42 return OutputView<float>(kOutputIndexLogits, interpreter); in ComputeLogits()