1 /* Copyright 2022 The TensorFlow Authors. All Rights Reserved.
2
3 Licensed under the Apache License, Version 2.0 (the "License");
4 you may not use this file except in compliance with the License.
5 You may obtain a copy of the License at
6
7 http://www.apache.org/licenses/LICENSE-2.0
8
9 Unless required by applicable law or agreed to in writing, software
10 distributed under the License is distributed on an "AS IS" BASIS,
11 WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12 See the License for the specific language governing permissions and
13 limitations under the License.
14 ==============================================================================*/
15
16 #include "tensorflow/lite/c/c_api_opaque.h"
17
18 #include "tensorflow/lite/kernels/kernel_util.h"
19
TfLiteOpaqueTensorType(const TfLiteOpaqueTensor * opaque_tensor)20 TfLiteType TfLiteOpaqueTensorType(const TfLiteOpaqueTensor* opaque_tensor) {
21 return TfLiteTensorType(reinterpret_cast<const TfLiteTensor*>(opaque_tensor));
22 }
23
TfLiteOpaqueTensorNumDims(const TfLiteOpaqueTensor * opaque_tensor)24 int32_t TfLiteOpaqueTensorNumDims(const TfLiteOpaqueTensor* opaque_tensor) {
25 return TfLiteTensorNumDims(
26 reinterpret_cast<const TfLiteTensor*>(opaque_tensor));
27 }
28
TfLiteOpaqueTensorDim(const TfLiteOpaqueTensor * opaque_tensor,int32_t dim_index)29 int32_t TfLiteOpaqueTensorDim(const TfLiteOpaqueTensor* opaque_tensor,
30 int32_t dim_index) {
31 return TfLiteTensorDim(reinterpret_cast<const TfLiteTensor*>(opaque_tensor),
32 dim_index);
33 }
34
TfLiteOpaqueTensorByteSize(const TfLiteOpaqueTensor * opaque_tensor)35 size_t TfLiteOpaqueTensorByteSize(const TfLiteOpaqueTensor* opaque_tensor) {
36 return TfLiteTensorByteSize(
37 reinterpret_cast<const TfLiteTensor*>(opaque_tensor));
38 }
39
TfLiteOpaqueTensorData(const TfLiteOpaqueTensor * opaque_tensor)40 void* TfLiteOpaqueTensorData(const TfLiteOpaqueTensor* opaque_tensor) {
41 return TfLiteTensorData(reinterpret_cast<const TfLiteTensor*>(opaque_tensor));
42 }
43
TfLiteOpaqueTensorName(const TfLiteOpaqueTensor * opaque_tensor)44 const char* TfLiteOpaqueTensorName(const TfLiteOpaqueTensor* opaque_tensor) {
45 return TfLiteTensorName(reinterpret_cast<const TfLiteTensor*>(opaque_tensor));
46 }
47
TfLiteOpaqueTensorCopyFromBuffer(TfLiteOpaqueTensor * opaque_tensor,const void * input_data,size_t input_data_size)48 TfLiteStatus TfLiteOpaqueTensorCopyFromBuffer(TfLiteOpaqueTensor* opaque_tensor,
49 const void* input_data,
50 size_t input_data_size) {
51 return TfLiteTensorCopyFromBuffer(
52 reinterpret_cast<TfLiteTensor*>(opaque_tensor), input_data,
53 input_data_size);
54 }
55
TfLiteOpaqueTensorCopyToBuffer(const TfLiteOpaqueTensor * opaque_tensor,void * output_data,size_t output_data_size)56 TfLiteStatus TfLiteOpaqueTensorCopyToBuffer(
57 const TfLiteOpaqueTensor* opaque_tensor, void* output_data,
58 size_t output_data_size) {
59 return TfLiteTensorCopyToBuffer(
60 reinterpret_cast<const TfLiteTensor*>(opaque_tensor), output_data,
61 output_data_size);
62 }
63
TfLiteOpaqueNodeGetInput(TfLiteOpaqueContext * opaque_context,const TfLiteOpaqueNode * opaque_node,int index)64 const TfLiteOpaqueTensor* TfLiteOpaqueNodeGetInput(
65 TfLiteOpaqueContext* opaque_context, const TfLiteOpaqueNode* opaque_node,
66 int index) {
67 const TfLiteTensor* tensor =
68 tflite::GetInput(reinterpret_cast<TfLiteContext*>(opaque_context),
69 reinterpret_cast<const TfLiteNode*>(opaque_node), index);
70 return reinterpret_cast<const TfLiteOpaqueTensor*>(tensor);
71 }
72
TfLiteOpaqueNodeGetOutput(TfLiteOpaqueContext * opaque_context,const TfLiteOpaqueNode * opaque_node,int index)73 TfLiteOpaqueTensor* TfLiteOpaqueNodeGetOutput(
74 TfLiteOpaqueContext* opaque_context, const TfLiteOpaqueNode* opaque_node,
75 int index) {
76 TfLiteTensor* tensor = tflite::GetOutput(
77 reinterpret_cast<TfLiteContext*>(opaque_context),
78 reinterpret_cast<const TfLiteNode*>(opaque_node), index);
79 return reinterpret_cast<TfLiteOpaqueTensor*>(tensor);
80 }
81
TfLiteOpaqueNodeGetUserData(const TfLiteOpaqueNode * opaque_node)82 void* TfLiteOpaqueNodeGetUserData(const TfLiteOpaqueNode* opaque_node) {
83 return reinterpret_cast<const TfLiteNode*>(opaque_node)->user_data;
84 }
85