Searched refs:TensorView (Results 1 – 14 of 14) sorted by relevance
| /external/libtextclassifier/actions/ |
| D | lua-actions.h | 45 : public LuaEnvironment::ItemIterator<TensorView<float>> {
48 int Item(const TensorView<float>* tensor, const int64 index,
66 TensorView<float> actions_scores_;
67 TensorView<float> smart_reply_scores_;
68 TensorView<float> sensitivity_score_;
69 TensorView<float> triggering_score_;
|
| D | lua-actions.cc | 32 TensorView<float> GetTensorViewForOutput( in GetTensorViewForOutput()
36 return TensorView<float>::Invalid(); in GetTensorViewForOutput()
43 const TensorView<float>* tensor, const int64 index, in Item()
81 ? TensorView<float>::Invalid() in LuaActionsSuggestions()
86 ? TensorView<float>::Invalid() in LuaActionsSuggestions()
90 ? TensorView<float>::Invalid() in LuaActionsSuggestions()
96 ? TensorView<float>::Invalid() in LuaActionsSuggestions()
|
| D | feature-processor_test.cc | 35 bool AddEmbedding(const TensorView<int>& sparse_features, float* dest, in AddEmbedding()
|
| D | feature-processor.cc | 91 TensorView<int>(sparse_features.data(), in AppendFeatures()
|
| D | actions-suggestions.cc | 866 const TensorView<float>& triggering_score = in ReadModelOutput()
880 const TensorView<float>& sensitive_topic_score = in ReadModelOutput()
907 TensorView<float> scores = model_executor_->OutputView<float>( in ReadModelOutput()
923 const TensorView<float> actions_scores = model_executor_->OutputView<float>( in ReadModelOutput()
|
| D | actions-suggestions_test.cc | 1037 bool AddEmbedding(const TensorView<int>& sparse_features, float* dest, in AddEmbedding()
|
| /external/libtextclassifier/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 | 68 void SetInput(const int input_index, const TensorView<T>& input_data, in SetInput()
114 TensorView<T> OutputView(const int output_index, in OutputView()
118 return TensorView<T>(interpreter->typed_output_tensor<T>(output_index), in OutputView()
127 TensorView<T> output_view = OutputView<T>(output_index, interpreter); in Output()
|
| /external/libtextclassifier/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 | 1302 TensorView<float> logits = classification_executor_->ComputeLogits( in ModelClassifyText()
1303 TensorView<float>(features.data(), in ModelClassifyText()
2184 TensorView<float> logits = selection_executor_->ComputeLogits( in ModelClickContextScoreChunks()
2185 TensorView<float>(all_features.data(), {batch_size, features_size}), in ModelClickContextScoreChunks()
2295 TensorView<float> logits = selection_executor_->ComputeLogits( in ModelBoundsSensitiveScoreChunks()
2296 TensorView<float>(all_features.data(), {batch_size, features_size}), in ModelBoundsSensitiveScoreChunks()
|
| D | feature-processor.cc | 842 TensorView<int>(sparse_features.data(), in AppendTokenFeaturesWithCache()
|
| D | feature-processor_test.cc | 65 bool AddEmbedding(const TensorView<int>& sparse_features, float* dest, in AddEmbedding()
|