Home
last modified time | relevance | path

Searched refs:TfTensorView (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/shim/
H A Dtf_tensor_view.h30 class TfTensorView : public TensorView {
33 TfTensorView(TfTensorView &&o) noexcept;
35 TfTensorView(const TfTensorView &o);
37 TfTensorView &operator=(TfTensorView &&o) noexcept;
39 TfTensorView &operator=(const TfTensorView &);
46 TfTensorView(const ::tensorflow::Tensor *wrapped_tensor, const DType &dtype);
51 typename MatchConstNess<TfTensorType, TfTensorView>::Type>
61 using Type = TfTensorView;
67 using Type = const TfTensorView;
72 absl::StatusOr<TfTensorView> TensorView::New<::tensorflow::Tensor>(
[all …]
H A Dtf_tensor_view.cc27 return TfTensorView(wrapped_tensor, DType()); \
39 TfTensorView::TfTensorView(TfTensorView &&o) noexcept in TfTensorView() function in tflite::shim::TfTensorView
44 TfTensorView::TfTensorView(const TfTensorView &o) in TfTensorView() function in tflite::shim::TfTensorView
49 TfTensorView &TfTensorView::operator=(TfTensorView &&o) noexcept { in operator =()
56 TfTensorView &TfTensorView::operator=(const TfTensorView &o) { in operator =()
65 absl::StatusOr<typename MatchConstNess<TfTensorType, TfTensorView>::Type>
87 absl::StatusOr<TfTensorView> TensorView::New<::tensorflow::Tensor>( in New()
93 absl::StatusOr<const TfTensorView> TensorView::New<const ::tensorflow::Tensor>( in New()
H A Dtf_op_shim.cc87 SH_ASSIGN_OR_RETURN(const TfTensorView& tensor_view, in GetInput()
89 return std::make_unique<const TfTensorView>(tensor_view); in GetInput()
102 SH_ASSIGN_OR_RETURN(const TfTensorView& tensor_view, in GetOutput()
104 return std::make_unique<TfTensorView>(std::move(tensor_view)); in GetOutput()
147 SH_ASSIGN_OR_RETURN(const TfTensorView& tensor_view, in GetInputTensor()
149 return std::make_unique<const TfTensorView>(tensor_view); in GetInputTensor()
H A Dtf_tensor_view_test.cc32 TEST(TfTensorView, Bool) { in TEST() argument
60 TEST(TfTensorView, Int32) { in TEST() argument
88 TEST(TfTensorView, Int64) { in TEST() argument
114 TEST(TfTensorView, Float) { in TEST() argument
141 TEST(TfTensorView, Double) { in TEST() argument
168 TEST(TfTensorView, Str) { in TEST() argument
H A DREADME.md71 either a `TfTensorView` or a `TfLiteTensorView`.