Home
last modified time | relevance | path

Searched refs:TensorView (Results 1 – 25 of 40) sorted by relevance

12

/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/shim/
H A Dtensor_view.h46 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 …]
H A Dtflite_tensor_view.cc43 : 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()
H A Dtf_tensor_view.cc40 : 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()
H A Dtflite_tensor_view_test.cc43 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()
H A Dtf_tensor_view_test.cc49 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()
H A Dtflite_tensor_view.h32 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>(
H A Dtf_tensor_view.h30 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(),
H A Dtf_op_shim.cc88 TensorView::New(&tf_tensor)); in GetInput()
103 TensorView::New(output_t)); in GetOutput()
148 TensorView::New(tf_tensor)); in GetInputTensor()
H A Dtflite_op_shim.cc88 TensorView::New(tflite_tensor)); in GetInput()
111 TensorView::New(tflite_tensor)); in GetOutput()
160 TensorView::New(tflite_tensor)); in GetInputTensor()
H A DREADME.md31 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);
H A Dop_kernel.h54 using TensorViewOr = absl::StatusOr<std::unique_ptr<TensorView>>;
55 using ConstTensorViewOr = absl::StatusOr<std::unique_ptr<const TensorView>>;
/aosp_15_r20/external/libtextclassifier/native/actions/
H A Dlua-actions.cc34 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()
H A Dlua-actions.h53 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_;
H A Dactions-suggestions.cc824 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()
H A Dfeature-processor_test.cc35 bool AddEmbedding(const TensorView<int>& sparse_features, float* dest, in AddEmbedding()
H A Dtflite-sensitive-model.cc108 const TensorView<float> scores = model_executor_->OutputView<float>( in EvalConversation()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/transformers/hip/
H A Daotriton_adapter.h103 aotriton::TensorView<Rank> mk_aotensor(const at::Tensor& q, c10::string_view tensor_name)
112 return aotriton::TensorView<Rank>(reinterpret_cast<intptr_t>(q.data_ptr()),
118 inline aotriton::TensorView<0> mk_aoscalartensor(const at::Tensor& q)
120 return aotriton::TensorView<0>(reinterpret_cast<intptr_t>(q.data_ptr()),
124 inline aotriton::TensorView<0> mk_philoxtensor(const int64_t* ptr)
126 return aotriton::TensorView<0>(reinterpret_cast<intptr_t>(ptr),
/aosp_15_r20/external/libtextclassifier/native/annotator/
H A Dmodel-executor.cc24 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()
H A Dmodel-executor.h52 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,
/aosp_15_r20/external/pytorch/aten/src/ATen/native/transformers/cuda/mem_eff_attention/iterators/
H A Dpredicated_tile_access_iterator_residual_last.h118 using TensorView = TensorView<Element, Layout>; variable
477 using TensorView = TensorView<Element, Layout>; variable
702 using TensorView = TensorView<Element, Layout>; variable
924 using TensorView = TensorView<Element, Layout>; variable
1258 using TensorView = TensorView<Element, Layout>; variable
1477 using TensorView = TensorView<Element, Layout>; variable
1699 using TensorView = TensorView<Element, Layout>; variable
1926 using TensorView = TensorView<Element, Layout>; variable
H A Dpredicated_tile_iterator_residual_last.h193 using TensorView = TensorView<Element, Layout>; variable
489 using TensorView = TensorView<Element, Layout>; variable
721 using TensorView = TensorView<Element, Layout>; variable
950 using TensorView = TensorView<Element, Layout>; variable
1241 using TensorView = TensorView<Element, Layout>; variable
1468 using TensorView = TensorView<Element, Layout>; variable
1698 using TensorView = TensorView<Element, Layout>; variable
1927 using TensorView = TensorView<Element, Layout>; variable
/aosp_15_r20/external/libtextclassifier/native/utils/
H A Dtensor-view.h33 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()
H A Dtflite-model-executor.h79 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()
H A Dtensor-view_test.cc27 const TensorView<float> tensor(data.data(), {3, 1, 2}); in TEST()
47 const TensorView<float> invalid_tensor = TensorView<float>::Invalid(); in TEST()
/aosp_15_r20/external/angle/third_party/glslang/src/Test/baseResults/
H A Dspv.coopmat2_tensor.comp.out195 102: 78 CooperativeMatrixLoadTensorNV 98 101 99 None TensorView 100
203 CooperativeMatrixStoreTensorNV 105 104 106 None TensorView 107
219 …121: 78 CooperativeMatrixLoadTensorNV 117 120 118 None TensorView DecodeFunc 119 16(deco…
229 …136: 78 CooperativeMatrixLoadTensorNV 132 135 133 None TensorView DecodeFunc 134 16(deco…

12