| /aosp_15_r20/external/tensorflow/tensorflow/lite/c/ |
| H A D | common.h | 81 typedef struct TfLiteIntArray { struct
97 } TfLiteIntArray; typedef
106 TfLiteIntArray* TfLiteIntArrayCreate(int size);
110 int TfLiteIntArrayEqual(const TfLiteIntArray* a, const TfLiteIntArray* b);
113 int TfLiteIntArrayEqualsArray(const TfLiteIntArray* a, int b_size,
119 TfLiteIntArray* TfLiteIntArrayCopy(const TfLiteIntArray* src);
122 void TfLiteIntArrayFree(TfLiteIntArray* a);
299 TfLiteIntArray* zero_point;
367 TfLiteIntArray* array_segments;
368 TfLiteIntArray* array_indices;
[all …]
|
| H A D | common.cc | 31 static TfLiteIntArray dummy; in TfLiteIntArrayGetSizeInBytes()
41 int TfLiteIntArrayEqual(const TfLiteIntArray* a, const TfLiteIntArray* b) { in TfLiteIntArrayEqual()
47 int TfLiteIntArrayEqualsArray(const TfLiteIntArray* a, int b_size, in TfLiteIntArrayEqualsArray()
59 TfLiteIntArray* TfLiteIntArrayCreate(int size) { in TfLiteIntArrayCreate()
62 TfLiteIntArray* ret = (TfLiteIntArray*)malloc(alloc_size); in TfLiteIntArrayCreate()
68 TfLiteIntArray* TfLiteIntArrayCopy(const TfLiteIntArray* src) { in TfLiteIntArrayCopy()
70 TfLiteIntArray* ret = TfLiteIntArrayCreate(src->size); in TfLiteIntArrayCopy()
77 void TfLiteIntArrayFree(TfLiteIntArray* a) { free(a); } in TfLiteIntArrayFree()
173 TfLiteIntArrayFree((TfLiteIntArray*)t->dims_signature); in TfLiteTensorFree()
182 void TfLiteTensorReset(TfLiteType type, const char* name, TfLiteIntArray* dims, in TfLiteTensorReset()
|
| H A D | common_test.cc | 31 TfLiteIntArray* a = TfLiteIntArrayCreate(0); in TEST()
32 TfLiteIntArray* b = TfLiteIntArrayCreate(3); in TEST()
38 TfLiteIntArray* a = TfLiteIntArrayCreate(2); in TEST()
41 TfLiteIntArray* b = TfLiteIntArrayCopy(a); in TEST()
51 TfLiteIntArray* a = TfLiteIntArrayCreate(1); in TEST()
53 TfLiteIntArray* b = TfLiteIntArrayCreate(2); in TEST()
56 TfLiteIntArray* c = TfLiteIntArrayCreate(2); in TEST()
59 TfLiteIntArray* d = TfLiteIntArrayCreate(2); in TEST()
|
| /aosp_15_r20/external/mesa3d/include/tensorflow/lite/core/c/ |
| H A D | common.h | 90 typedef struct TfLiteIntArray { struct
106 } TfLiteIntArray; typedef
115 TfLiteIntArray* TfLiteIntArrayCreate(int size);
119 int TfLiteIntArrayEqual(const TfLiteIntArray* a, const TfLiteIntArray* b);
122 int TfLiteIntArrayEqualsArray(const TfLiteIntArray* a, int b_size,
128 TfLiteIntArray* TfLiteIntArrayCopy(const TfLiteIntArray* src);
131 void TfLiteIntArrayFree(TfLiteIntArray* a);
312 TfLiteIntArray* zero_point;
380 TfLiteIntArray* array_segments;
381 TfLiteIntArray* array_indices;
[all …]
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/ |
| H A D | conv_2d_builder.h | 47 TfLiteStatus PopulateSubGraph(const TfLiteIntArray* inputs,
48 const TfLiteIntArray* outputs,
51 TfLiteStatus RegisterOutputs(const TfLiteIntArray* outputs,
57 TfLiteStatus InitializeWeightsNodes(const TfLiteIntArray* inputs,
58 const TfLiteIntArray* outputs,
62 TfLiteStatus InitializeBiasNodes(const TfLiteIntArray* inputs,
63 const TfLiteIntArray* outputs,
66 void BuildStandardConv(const TfLiteIntArray* inputs,
75 void BuildDilatedDwConv(const TfLiteIntArray* inputs,
86 const TfLiteIntArray* inputs, const TfLiteTensor& data_tensor,
|
| H A D | matmul_builder.h | 32 TfLiteStatus PopulateSubGraph(const TfLiteIntArray* inputs,
33 const TfLiteIntArray* outputs,
36 TfLiteStatus RegisterOutputs(const TfLiteIntArray* outputs,
51 TfLiteStatus PopulateSubGraph(const TfLiteIntArray* inputs,
52 const TfLiteIntArray* outputs,
55 TfLiteStatus RegisterOutputs(const TfLiteIntArray* outputs,
60 TfLiteStatus AddFullyConnected(const TfLiteIntArray* inputs,
61 const TfLiteIntArray* outputs,
|
| H A D | batch_seq_builder.h | 29 TfLiteStatus PopulateSubGraph(const TfLiteIntArray* inputs,
30 const TfLiteIntArray* outputs,
33 TfLiteStatus RegisterOutputs(const TfLiteIntArray* outputs, in RegisterOutputs()
43 void SetInputBatchDimensions(TfLiteIntArray* input_batch_dimensions) { in SetInputBatchDimensions()
47 void SetOutputBatchDimensions(TfLiteIntArray* output_batch_dimensions) { in SetOutputBatchDimensions()
59 TfLiteIntArray* input_batch_dims_;
62 TfLiteIntArray* output_batch_dims_;
|
| H A D | matmul_builder.cc | 33 int* depth_size, const TfLiteIntArray* dims) { in GetDims()
43 TfLiteStatus AddFullyConnectedHelper(const TfLiteIntArray* inputs, in AddFullyConnectedHelper()
44 const TfLiteIntArray* outputs, in AddFullyConnectedHelper()
155 const TfLiteIntArray* inputs, const TfLiteIntArray* outputs, in PopulateSubGraph()
216 const TfLiteIntArray* outputs, TfLiteContext* context) { in RegisterOutputs()
223 TfLiteStatus MatMulOpBuilder::PopulateSubGraph(const TfLiteIntArray* inputs, in PopulateSubGraph()
224 const TfLiteIntArray* outputs, in PopulateSubGraph()
263 TfLiteStatus MatMulOpBuilder::AddFullyConnected(const TfLiteIntArray* inputs, in AddFullyConnected()
264 const TfLiteIntArray* outputs, in AddFullyConnected()
275 TfLiteStatus MatMulOpBuilder::RegisterOutputs(const TfLiteIntArray* outputs, in RegisterOutputs()
|
| H A D | op_builder.h | 106 virtual TfLiteStatus PopulateSubGraph(const TfLiteIntArray* inputs, in PopulateSubGraph()
107 const TfLiteIntArray* outputs, in PopulateSubGraph()
114 virtual TfLiteStatus RegisterOutputs(const TfLiteIntArray* outputs, in RegisterOutputs()
137 TensorID AddOutput(const TfLiteIntArray* dims, int element_size);
178 int* depth_size, const TfLiteIntArray* dims) { in GetDims()
255 TfLiteStatus AddInputTensors(const TfLiteIntArray* input_tensors,
259 TfLiteStatus AddOutputTensors(const TfLiteIntArray* output_tensors,
266 TfLiteIntArray* input_batch_dimensions,
267 TfLiteIntArray* output_batch_dimensions);
374 int* depth_size, const TfLiteIntArray* dims) { in GetDims()
|
| H A D | squared_difference.cc | 25 TfLiteStatus PopulateSubGraph(const TfLiteIntArray* inputs,
26 const TfLiteIntArray* outputs,
29 TfLiteStatus RegisterOutputs(const TfLiteIntArray* outputs,
37 const TfLiteIntArray* inputs, const TfLiteIntArray* outputs, in PopulateSubGraph()
91 const TfLiteIntArray* outputs, TfLiteContext* context) { in RegisterOutputs()
|
| H A D | rsqrt_builder.cc | 29 TfLiteStatus PopulateSubGraph(const TfLiteIntArray* inputs,
30 const TfLiteIntArray* outputs,
33 TfLiteStatus RegisterOutputs(const TfLiteIntArray* outputs,
72 TfLiteStatus RsqrtOpBuilder::RegisterOutputs(const TfLiteIntArray* outputs, in RegisterOutputs()
79 TfLiteStatus RsqrtOpBuilder::PopulateSubGraph(const TfLiteIntArray* inputs, in PopulateSubGraph()
80 const TfLiteIntArray* outputs, in PopulateSubGraph()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| H A D | quant_lstm_sup.cc | 27 void ExtractQuantLstmWeightsSubmatrix(const TfLiteIntArray* submatrix_dims, in ExtractQuantLstmWeightsSubmatrix()
30 const TfLiteIntArray* weight_dims, in ExtractQuantLstmWeightsSubmatrix()
47 inline int OutputDepth(const TfLiteIntArray* weight_dims) { in OutputDepth()
51 inline int InputDepth(const TfLiteIntArray* weight_dims) { in InputDepth()
55 void SetWeightSubmatrixDims(const TfLiteIntArray* weight_dims, in SetWeightSubmatrixDims()
56 TfLiteIntArray* recurrent_submatrix_dims, in SetWeightSubmatrixDims()
57 TfLiteIntArray* input_submatrix_dims) { in SetWeightSubmatrixDims()
82 const TfLiteIntArray* weight_dims, in DecomposeQuantLstmWeightsTensor()
93 TfLiteIntArray* recurrent_submatrix_dims = TfLiteIntArrayCreate(2); in DecomposeQuantLstmWeightsTensor()
94 TfLiteIntArray* input_submatrix_dims = TfLiteIntArrayCreate(2); in DecomposeQuantLstmWeightsTensor()
|
| H A D | quant_lstm_sup_test.cc | 37 for (TfLiteIntArray* dim : allocated_dims_) { in ~DimsAllocatingTest()
42 TfLiteIntArray* CreateDimArray(int size, in CreateDimArray()
44 TfLiteIntArray* dims = TfLiteIntArrayCreate(size); in CreateDimArray()
56 std::vector<TfLiteIntArray*> allocated_dims_;
72 const TfLiteIntArray* weight_dims = CreateDimArray(2, {8, 5}); in TEST_F()
75 const TfLiteIntArray* submatrix_dims = CreateDimArray(2, {2, 3}); in TEST_F()
93 const TfLiteIntArray* weight_dims = CreateDimArray(2, {8, 5}); in TEST_F()
96 const TfLiteIntArray* submatrix_dims = CreateDimArray(2, {2, 2}); in TEST_F()
114 const TfLiteIntArray* weight_dims = CreateDimArray(2, {8, 5}); in TEST_F()
117 const TfLiteIntArray* submatrix_dims = CreateDimArray(2, {2, 2}); in TEST_F()
[all …]
|
| H A D | quant_lstm_sup.h | 26 void ExtractQuantLstmWeightsSubmatrix(const TfLiteIntArray* submatrix_dims,
29 const TfLiteIntArray* weight_dims,
34 const TfLiteIntArray* weight_dims,
44 void SetWeightSubmatrixDims(const TfLiteIntArray* weight_dims,
45 TfLiteIntArray* recurrent_submatrix_dims,
46 TfLiteIntArray* input_submatrix_dims);
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ |
| H A D | reshape.cc | 34 TfLiteIntArray* GetOutputShape(TfLiteContext*, TfLiteNode*);
37 TfLiteIntArray* output_shape = GetOutputShape(context, node); in ResizeOutput()
38 std::unique_ptr<TfLiteIntArray, void (*)(TfLiteIntArray*)> in ResizeOutput()
89 inline TfLiteIntArray* GetOutputShapeFromTensor(TfLiteContext* context, in GetOutputShapeFromTensor()
94 TfLiteIntArray* output_shape = TfLiteIntArrayCreate(shape->dims->data[0]); in GetOutputShapeFromTensor()
102 inline TfLiteIntArray* GetOutputShapeFromParam(TfLiteContext* context, in GetOutputShapeFromParam()
115 TfLiteIntArray* output_shape = TfLiteIntArrayCreate(num_dimensions); in GetOutputShapeFromParam()
130 TfLiteIntArray* GetOutputShape(TfLiteContext* context, TfLiteNode* node) { in GetOutputShape()
|
| H A D | kernel_util_test.cc | 84 std::vector<int> GetShape(TfLiteIntArray* dims) { in GetShape()
122 TfLiteIntArray* output = nullptr; in TEST_F()
133 TfLiteIntArray* output = nullptr; in TEST_F()
144 TfLiteIntArray* output = nullptr; in TEST_F()
159 TfLiteIntArray* output = nullptr; in TEST_F()
176 TfLiteIntArray* output = nullptr; in TEST_F()
193 TfLiteIntArray* output = nullptr; in TEST_F()
210 TfLiteIntArray* output = nullptr; in TEST_F()
223 TfLiteIntArray* output = nullptr; in TEST_F()
236 TfLiteIntArray* output = nullptr; in TEST_F()
[all …]
|
| H A D | bidirectional_sequence_lstm.cc | 550 TfLiteIntArray* fw_output_size = TfLiteIntArrayCreate(3); in Prepare()
584 TfLiteIntArray* fw_scratch_buffer_size = TfLiteIntArrayCreate(2); in Prepare()
617 TfLiteIntArray* bw_output_size = TfLiteIntArrayCreate(3); in Prepare()
648 TfLiteIntArray* bw_scratch_buffer_size = TfLiteIntArrayCreate(2); in Prepare()
675 TfLiteIntArray* input_quantized_size = TfLiteIntArrayCopy(input->dims); in Prepare()
690 TfLiteIntArray* fw_activation_state_quantized_size = in Prepare()
706 TfLiteIntArray* bw_activation_state_quantized_size = in Prepare()
722 TfLiteIntArray* fw_cell_state_quantized_size = in Prepare()
738 TfLiteIntArray* bw_cell_state_quantized_size = in Prepare()
760 TfLiteIntArray* input_sf_size = TfLiteIntArrayCreate(1); in Prepare()
[all …]
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/coreml/builders/ |
| H A D | reshape_op_builder.cc | 55 const TfLiteIntArray* input_shape) { in SetShapeFromTensor()
56 TfLiteIntArray* shape = TfLiteIntArrayCreate(output_shape->dims->data[0]); in SetShapeFromTensor()
64 void ReshapeOpBuilder::SetShapeFromIntArray(const TfLiteIntArray* output_shape, in SetShapeFromIntArray()
65 const TfLiteIntArray* input_shape) { in SetShapeFromIntArray()
95 TfLiteStatus ReshapeOpBuilder::RegisterInputs(const TfLiteIntArray* inputs, in RegisterInputs()
104 TfLiteIntArray* output_shape = TfLiteIntArrayCreate(params->num_dimensions); in RegisterInputs()
114 TfLiteStatus ReshapeOpBuilder::RegisterOutputs(const TfLiteIntArray* outputs, in RegisterOutputs()
|
| H A D | reshape_op_builder.h | 35 TfLiteStatus RegisterInputs(const TfLiteIntArray* inputs,
37 TfLiteStatus RegisterOutputs(const TfLiteIntArray* outputs,
43 const TfLiteIntArray* input_shape);
44 void SetShapeFromIntArray(const TfLiteIntArray* output_shape,
45 const TfLiteIntArray* input_shape);
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/ |
| H A D | model_builder_helper.h | 117 absl::Status SetAllDimensions(const TfLiteIntArray* dimensions, Scalar* shape);
119 absl::Status CheckIfLinearConvertible(const TfLiteIntArray* dimensions);
121 absl::Status SetAllDimensions(const TfLiteIntArray* dimensions, Linear* shape);
123 absl::Status SetAllDimensions(const TfLiteIntArray* dimensions, HWC* shape);
125 absl::Status SetAllDimensions(const TfLiteIntArray* dimensions, HW* shape);
127 absl::Status SetAllDimensions(const TfLiteIntArray* dimensions, OHWI* shape);
129 absl::Status SetAllDimensions(const TfLiteIntArray* dimensions, BHWC* shape);
|
| H A D | model_builder_helper.cc | 99 const TfLiteIntArray* dims = tflite_tensor.dims; in ExtractTensorShape()
126 const TfLiteIntArray* dims = tflite_tensor.dims; in ExtractAxisFromIndex()
287 const std::string GetDimensionString(const TfLiteIntArray* dimensions) { in GetDimensionString()
291 absl::Status SetAllDimensions(const TfLiteIntArray* dimensions, Scalar* shape) { in SetAllDimensions()
305 absl::Status CheckIfLinearConvertible(const TfLiteIntArray* dimensions) { in CheckIfLinearConvertible()
318 absl::Status SetAllDimensions(const TfLiteIntArray* dimensions, Linear* shape) { in SetAllDimensions()
324 absl::Status SetAllDimensions(const TfLiteIntArray* dimensions, HWC* shape) { in SetAllDimensions()
346 absl::Status SetAllDimensions(const TfLiteIntArray* dimensions, HW* shape) { in SetAllDimensions()
357 absl::Status SetAllDimensions(const TfLiteIntArray* dimensions, OHWI* shape) { in SetAllDimensions()
370 absl::Status SetAllDimensions(const TfLiteIntArray* dimensions, BHWC* shape) { in SetAllDimensions()
|
| H A D | quantization_util_test.cc | 40 std::unique_ptr<TfLiteIntArray, TfLiteIntArrayDeleter> BuildTfLiteIntArray( in BuildTfLiteIntArray()
42 std::unique_ptr<TfLiteIntArray, TfLiteIntArrayDeleter> result( in BuildTfLiteIntArray()
59 int ElementCount(const TfLiteIntArray& dims) { in ElementCount()
85 TfLiteTensor CreateQuantizedTensor(const int8_t* data, TfLiteIntArray* dims, in CreateQuantizedTensor()
104 TfLiteTensor CreateQuantizedTensor(const uint8_t* data, TfLiteIntArray* dims, in CreateQuantizedTensor()
123 TfLiteTensor CreateTensor(TfLiteIntArray* dims, const char* name, in CreateTensor()
138 TfLiteTensor CreateFloatTensor(const float* data, TfLiteIntArray* dims, in CreateFloatTensor()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/tools/evaluation/stages/ |
| H A D | topk_accuracy_eval_stage_test.cc | 75 TfLiteIntArray* model_output_shape = TfLiteIntArrayCreate(2); in TEST()
91 TfLiteIntArray* model_output_shape = TfLiteIntArrayCreate(2); in TEST()
109 TfLiteIntArray* model_output_shape = TfLiteIntArrayCreate(2); in TEST()
125 TfLiteIntArray* model_output_shape = TfLiteIntArrayCreate(2); in TEST()
141 TfLiteIntArray* model_output_shape = TfLiteIntArrayCreate(2); in TEST()
156 TfLiteIntArray* model_output_shape = TfLiteIntArrayCreate(2); in TEST()
173 TfLiteIntArray* model_output_shape = TfLiteIntArrayCreate(2); in TEST()
194 TfLiteIntArray* model_output_shape = TfLiteIntArrayCreate(2); in TEST()
251 TfLiteIntArray* model_output_shape = TfLiteIntArrayCreate(2); in VerifyCorrectBehaviorForType()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/ |
| H A D | util.h | 53 TfLiteIntArray* ConvertVectorToTfLiteIntArray(const std::vector<int>& input);
58 TfLiteIntArray* ConvertArrayToTfLiteIntArray(const int ndims, const int* dims);
62 bool EqualArrayAndTfLiteIntArray(const TfLiteIntArray* a, const int b_size,
68 void operator()(TfLiteIntArray* a) { in operator()
75 std::unique_ptr<TfLiteIntArray, TfLiteIntArrayDeleter> BuildTfLiteIntArray(
|
| H A D | util.cc | 50 std::unique_ptr<TfLiteIntArray, TfLiteIntArrayDeleter> BuildTfLiteIntArray( in BuildTfLiteIntArray()
52 std::unique_ptr<TfLiteIntArray, TfLiteIntArrayDeleter> result( in BuildTfLiteIntArray()
58 TfLiteIntArray* ConvertVectorToTfLiteIntArray(const std::vector<int>& input) { in ConvertVectorToTfLiteIntArray()
63 TfLiteIntArray* ConvertArrayToTfLiteIntArray(const int ndims, const int* dims) { in ConvertArrayToTfLiteIntArray()
64 TfLiteIntArray* output = TfLiteIntArrayCreate(ndims); in ConvertArrayToTfLiteIntArray()
71 bool EqualArrayAndTfLiteIntArray(const TfLiteIntArray* a, const int b_size, in EqualArrayAndTfLiteIntArray()
|