| /external/tensorflow/tensorflow/lite/kernels/shim/ |
| D | tensor_view.h | 46 class TensorView {
76 explicit Tensor(TensorView *t) in Tensor()
82 explicit Tensor(const TensorView *t) in Tensor()
153 TensorView(TensorView &&o) = default;
155 TensorView(const TensorView &o) = default;
157 TensorView &operator=(TensorView &&o) = default;
159 TensorView &operator=(const TensorView &) = default;
163 virtual ~TensorView() = default;
206 TensorView(const absl::Span<int> shape, void *data, in TensorView() function
240 DType &TensorView::AsScalar() { in AsScalar()
[all …]
|
| D | tflite_tensor_view.cc | 43 : TensorView(absl::Span<int>(wrapped_tensor->dims->data, in TfLiteTensorView()
53 : TensorView(absl::Span<int>(wrapped_tensor->dims->data, in TfLiteTensorView()
61 : TensorView(std::move(o)), in TfLiteTensorView()
68 : TensorView(o), in TfLiteTensorView()
78 TensorView::operator=(std::move(o)); in operator =()
84 TensorView::operator=(o); in operator =()
146 absl::StatusOr<TfLiteTensorView> TensorView::New<::TfLiteTensor>( in New()
152 absl::StatusOr<const TfLiteTensorView> TensorView::New<const ::TfLiteTensor>( in New()
|
| D | tf_tensor_view.cc | 40 : TensorView(std::move(o)), shape_data_(std::move(o.shape_data_)) { in TfTensorView()
45 : TensorView(o), shape_data_(o.shape_data_) { in TfTensorView()
51 TensorView::operator=(std::move(o)); in operator =()
58 TensorView::operator=(o); in operator =()
87 absl::StatusOr<TfTensorView> TensorView::New<::tensorflow::Tensor>( in New()
93 absl::StatusOr<const TfTensorView> TensorView::New<const ::tensorflow::Tensor>( in New()
|
| D | tf_tensor_view_test.cc | 49 auto t_premove_or = TensorView::New(&tf_tensor); in TEST()
77 auto t_premove_or = TensorView::New(&tf_tensor); in TEST()
103 auto t_or = TensorView::New(&tf_tensor); in TEST()
129 auto t_or = TensorView::New(&tf_tensor); in TEST()
156 auto t_or = TensorView::New(&tf_tensor); in TEST()
183 auto t_or = TensorView::New(&tf_tensor); in TEST()
199 const auto const_t_or = TensorView::New(&const_tf_tensor); in TEST()
|
| D | tflite_tensor_view_test.cc | 43 auto t_premove_or = TensorView::New(tflite_tensor); in TEST()
65 auto t_premove_or = TensorView::New(tflite_tensor); in IntTest()
92 auto t_or = TensorView::New(tflite_tensor); in FloatTest()
116 auto t_or = TensorView::New(tflite_tensor); in TEST()
129 auto t_or = TensorView::New(tflite_tensor); in TEST()
138 const auto t_or = TensorView::New(const_tflite_tensor); in TEST()
160 auto t_or = TensorView::New(tflite_tensor); in TEST()
|
| D | tflite_tensor_view.h | 32 class TfLiteTensorView : public TensorView {
49 : TensorView(absl::Span<int>(wrapped_tensor->dims->data, in TfLiteTensorView()
62 : TensorView(absl::Span<int>(wrapped_tensor->dims->data, in TfLiteTensorView()
117 absl::StatusOr<TfLiteTensorView> TensorView::New<::TfLiteTensor>(
122 absl::StatusOr<const TfLiteTensorView> TensorView::New<const ::TfLiteTensor>(
|
| D | tf_tensor_view.h | 30 class TfTensorView : public TensorView {
72 absl::StatusOr<TfTensorView> TensorView::New<::tensorflow::Tensor>(
77 absl::StatusOr<const TfTensorView> TensorView::New<const ::tensorflow::Tensor>(
87 : TensorView({}, wrapped_tensor->data(),
|
| D | tf_op_shim.cc | 88 TensorView::New(&tf_tensor)); in GetInput()
103 TensorView::New(output_t)); in GetOutput()
148 TensorView::New(tf_tensor)); in GetInputTensor()
|
| D | tflite_op_shim.cc | 88 TensorView::New(tflite_tensor)); in GetInput()
111 TensorView::New(tflite_tensor)); in GetOutput()
160 TensorView::New(tflite_tensor)); in GetInputTensor()
|
| D | README.md | 31 1. `TensorView` as a shim over `::tensorflow::Tensor` and `TfLiteTensor`
35 ### TensorView subsection
45 auto t = TensorView::New(&tf_tensor);
59 auto t = TensorView::New(&tflite_tensor);
|
| D | op_kernel.h | 54 using TensorViewOr = absl::StatusOr<std::unique_ptr<TensorView>>;
55 using ConstTensorViewOr = absl::StatusOr<std::unique_ptr<const TensorView>>;
|
| /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 | actions-suggestions.cc | 824 const TensorView<float> scores = in PopulateTextReplies()
867 const TensorView<bool> intent_prediction = in PopulateIntentTriggering()
869 const TensorView<float> intent_scores = in PopulateIntentTriggering()
895 const TensorView<float> triggering_score = in ReadModelOutput()
909 const TensorView<float> sensitive_topic_score = in ReadModelOutput()
941 const TensorView<float> actions_scores = model_executor_->OutputView<float>( in ReadModelOutput()
|
| D | feature-processor.cc | 91 TensorView<int>(sparse_features.data(), in AppendFeatures()
|
| /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 | 1572 TensorView<float> logits = classification_executor_->ComputeLogits( in ModelClassifyText()
1573 TensorView<float>(features.data(), in ModelClassifyText()
3020 TensorView<float> logits = selection_executor_->ComputeLogits( in ModelClickContextScoreChunks()
3021 TensorView<float>(all_features.data(), {batch_size, features_size}), in ModelClickContextScoreChunks()
3129 TensorView<float> logits = selection_executor_->ComputeLogits( in ModelBoundsSensitiveScoreChunks()
3130 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()
|