Searched refs:TensorView (Results 1 – 15 of 15) sorted by relevance
| /external/libtextclassifier/native/actions/ |
| D | lua-actions.h | 53 void PushTensor(const TensorView<T>* tensor) const { in PushTensor()
63 TensorView<float> actions_scores_;
64 TensorView<float> smart_reply_scores_;
65 TensorView<float> sensitivity_score_;
66 TensorView<float> triggering_score_;
|
| D | lua-actions.cc | 34 TensorView<float> GetTensorViewForOutput( in GetTensorViewForOutput()
38 return TensorView<float>::Invalid(); in GetTensorViewForOutput()
85 ? TensorView<float>::Invalid() in LuaActionsSuggestions()
90 ? TensorView<float>::Invalid() in LuaActionsSuggestions()
94 ? TensorView<float>::Invalid() in LuaActionsSuggestions()
100 ? TensorView<float>::Invalid() in LuaActionsSuggestions()
|
| D | feature-processor_test.cc | 35 bool AddEmbedding(const TensorView<int>& sparse_features, float* dest, in AddEmbedding()
|
| D | tflite-sensitive-model.cc | 108 const TensorView<float> scores = model_executor_->OutputView<float>( in EvalConversation()
|
| D | feature-processor.cc | 91 TensorView<int>(sparse_features.data(), in AppendFeatures()
|
| D | actions-suggestions.cc | 816 const TensorView<float> scores = in PopulateTextReplies()
848 const TensorView<bool> intent_prediction = in PopulateIntentTriggering()
850 const TensorView<float> intent_scores = in PopulateIntentTriggering()
876 const TensorView<float> triggering_score = in ReadModelOutput()
890 const TensorView<float> sensitive_topic_score = in ReadModelOutput()
920 const TensorView<float> actions_scores = model_executor_->OutputView<float>( in ReadModelOutput()
|
| D | actions-suggestions_test.cc | 1386 bool AddEmbedding(const TensorView<int>& sparse_features, float* dest, in AddEmbedding()
|
| /external/libtextclassifier/native/utils/ |
| D | tensor-view.h | 33 class TensorView {
35 TensorView(const T* data, const std::vector<int>& shape) in TensorView() function
38 static TensorView Invalid() { in Invalid()
41 return TensorView(nullptr, invalid_shape); in Invalid()
|
| D | tensor-view_test.cc | 27 const TensorView<float> tensor(data.data(), {3, 1, 2}); in TEST()
47 const TensorView<float> invalid_tensor = TensorView<float>::Invalid(); in TEST()
|
| D | tflite-model-executor.h | 79 void SetInput(const int input_index, const TensorView<T>& input_data, in SetInput()
125 TensorView<T> OutputView(const int output_index, in OutputView()
129 return TensorView<T>(interpreter->typed_output_tensor<T>(output_index), in OutputView()
138 TensorView<T> output_view = OutputView<T>(output_index, interpreter); in Output()
|
| /external/libtextclassifier/native/annotator/ |
| D | model-executor.cc | 24 TensorView<float> ModelExecutor::ComputeLogits( in ComputeLogits()
25 const TensorView<float>& features, tflite::Interpreter* interpreter) const { in ComputeLogits()
27 return TensorView<float>::Invalid(); in ComputeLogits()
32 return TensorView<float>::Invalid(); in ComputeLogits()
39 return TensorView<float>::Invalid(); in ComputeLogits()
153 const TensorView<int>& sparse_features, float* dest, int dest_size) const { in AddEmbedding()
|
| D | model-executor.h | 52 TensorView<float> ComputeLogits(const TensorView<float>& features,
70 virtual bool AddEmbedding(const TensorView<int>& sparse_features, float* dest,
86 bool AddEmbedding(const TensorView<int>& sparse_features, float* dest,
|
| D | annotator.cc | 1546 TensorView<float> logits = classification_executor_->ComputeLogits( in ModelClassifyText()
1547 TensorView<float>(features.data(), in ModelClassifyText()
3003 TensorView<float> logits = selection_executor_->ComputeLogits( in ModelClickContextScoreChunks()
3004 TensorView<float>(all_features.data(), {batch_size, features_size}), in ModelClickContextScoreChunks()
3112 TensorView<float> logits = selection_executor_->ComputeLogits( in ModelBoundsSensitiveScoreChunks()
3113 TensorView<float>(all_features.data(), {batch_size, features_size}), in ModelBoundsSensitiveScoreChunks()
|
| D | feature-processor.cc | 872 TensorView<int>(sparse_features.data(), in AppendTokenFeaturesWithCache()
|
| D | feature-processor_test.cc | 65 bool AddEmbedding(const TensorView<int>& sparse_features, float* dest, in AddEmbedding()
|