Home
last modified time | relevance | path

Searched refs:builtin_data (Results 1 – 25 of 141) sorted by relevance

123456

/external/tensorflow/tensorflow/lite/core/api/
Dflatbuffer_conversions.h65 BuiltinDataAllocator* allocator, void** builtin_data);
73 BuiltinDataAllocator* allocator, void** builtin_data);
76 BuiltinDataAllocator* allocator, void** builtin_data);
79 BuiltinDataAllocator* allocator, void** builtin_data);
82 BuiltinDataAllocator* allocator, void** builtin_data);
85 BuiltinDataAllocator* allocator, void** builtin_data);
89 void** builtin_data);
94 void** builtin_data);
97 BuiltinDataAllocator* allocator, void** builtin_data);
100 BuiltinDataAllocator* allocator, void** builtin_data);
[all …]
Dflatbuffer_conversions.cc69 void** builtin_data) { in CheckParsePointerParams() argument
73 TFLITE_DCHECK(builtin_data != nullptr); in CheckParsePointerParams()
138 void** builtin_data) { in ParseOpDataTfLite() argument
162 *builtin_data = nullptr; in ParseOpDataTfLite()
165 return ParseAbs(op, error_reporter, allocator, builtin_data); in ParseOpDataTfLite()
169 return ParseAdd(op, error_reporter, allocator, builtin_data); in ParseOpDataTfLite()
173 return ParseAddN(op, error_reporter, allocator, builtin_data); in ParseOpDataTfLite()
177 return ParseArgMax(op, error_reporter, allocator, builtin_data); in ParseOpDataTfLite()
181 return ParseArgMin(op, error_reporter, allocator, builtin_data); in ParseOpDataTfLite()
185 return ParsePool(op, error_reporter, allocator, builtin_data); in ParseOpDataTfLite()
[all …]
/external/tensorflow/tensorflow/lite/micro/kernels/
Dstrided_slice_test.cc61 TfLiteStridedSliceParams* builtin_data, in TestStridedSliceFloat() argument
85 builtin_data, expect_prepare_err, num_invoke, in TestStridedSliceFloat()
92 const int* strides_shape, TfLiteStridedSliceParams* builtin_data, in TestStridedSliceQuantized() argument
116 builtin_data, expect_prepare_err, num_invoke, in TestStridedSliceQuantized()
139 TfLiteStridedSliceParams builtin_data = {}; in TF_LITE_MICRO_TEST() local
142 input_shape, begin_shape, end_shape, strides_shape, &builtin_data, in TF_LITE_MICRO_TEST()
160 TfLiteStridedSliceParams builtin_data = {}; in TF_LITE_MICRO_TEST() local
163 input_shape, begin_shape, end_shape, strides_shape, &builtin_data, in TF_LITE_MICRO_TEST()
181 TfLiteStridedSliceParams builtin_data = {}; in TF_LITE_MICRO_TEST() local
184 input_shape, begin_shape, end_shape, strides_shape, &builtin_data, in TF_LITE_MICRO_TEST()
[all …]
Dpack_test.cc67 TfLitePackParams builtin_data = { in TestPackTwoInputsFloat() local
76 ValidatePackGoldens(tensors, tensors_size, builtin_data, inputs_array, in TestPackTwoInputsFloat()
101 TfLitePackParams builtin_data = { in TestPackThreeInputsFloat() local
110 ValidatePackGoldens(tensors, tensors_size, builtin_data, inputs_array, in TestPackThreeInputsFloat()
138 TfLitePackParams builtin_data = { in TestPackTwoInputsQuantized() local
147 ValidatePackGoldens(tensors, tensors_size, builtin_data, inputs_array, in TestPackTwoInputsQuantized()
171 TfLitePackParams builtin_data = { in TestPackTwoInputsQuantized32() local
180 ValidatePackGoldens(tensors, tensors_size, builtin_data, inputs_array, in TestPackTwoInputsQuantized32()
Dunpack_test.cc62 TfLiteUnpackParams builtin_data = { in TestUnpackThreeOutputsFloat() local
75 reinterpret_cast<void*>(&builtin_data)); in TestUnpackThreeOutputsFloat()
113 TfLiteUnpackParams builtin_data = { in TestUnpackOneOutputFloat() local
126 reinterpret_cast<void*>(&builtin_data)); in TestUnpackOneOutputFloat()
175 TfLiteUnpackParams builtin_data = { in TestUnpackThreeOutputsQuantized() local
188 reinterpret_cast<void*>(&builtin_data)); in TestUnpackThreeOutputsQuantized()
242 TfLiteUnpackParams builtin_data = { in TestUnpackThreeOutputsQuantized32() local
255 reinterpret_cast<void*>(&builtin_data)); in TestUnpackThreeOutputsQuantized32()
Dfully_connected.cc39 TFLITE_DCHECK(node->builtin_data != nullptr); in Prepare()
43 static_cast<const TfLiteFullyConnectedParams*>(node->builtin_data); in Prepare()
65 TFLITE_DCHECK(node->builtin_data != nullptr); in Eval()
67 static_cast<const TfLiteFullyConnectedParams*>(node->builtin_data); in Eval()
Dconcatenation_test.cc50 TfLiteConcatenationParams builtin_data = { in TestConcatenateTwoInputs() local
59 reinterpret_cast<void*>(&builtin_data)); in TestConcatenateTwoInputs()
97 TfLiteConcatenationParams builtin_data = { in TestConcatenateQuantizedTwoInputs() local
106 reinterpret_cast<void*>(&builtin_data)); in TestConcatenateQuantizedTwoInputs()
Dpooling.cc156 TFLITE_DCHECK(node->builtin_data != nullptr); in AverageEval()
157 auto* params = reinterpret_cast<TfLitePoolParams*>(node->builtin_data); in AverageEval()
185 TFLITE_DCHECK(node->builtin_data != nullptr); in MaxEval()
186 auto* params = reinterpret_cast<TfLitePoolParams*>(node->builtin_data); in MaxEval()
218 TFLITE_DCHECK(node->builtin_data != nullptr); in Prepare()
219 auto* params = reinterpret_cast<TfLitePoolParams*>(node->builtin_data); in Prepare()
Dkernel_runner.cc35 void* builtin_data) in KernelRunner() argument
54 node_.builtin_data = builtin_data; in KernelRunner()
Dmul.cc182 TFLITE_DCHECK(node->builtin_data != nullptr); in Prepare()
183 auto* params = reinterpret_cast<TfLiteMulParams*>(node->builtin_data); in Prepare()
192 TFLITE_DCHECK(node->builtin_data != nullptr); in Eval()
193 auto* params = reinterpret_cast<TfLiteMulParams*>(node->builtin_data); in Eval()
Dconv.cc49 TFLITE_DCHECK(node->builtin_data != nullptr); in Eval()
51 *(reinterpret_cast<TfLiteConvParams*>(node->builtin_data)); in Eval()
/external/tensorflow/tensorflow/lite/tools/serialization/
Dwriter_lib_test.cc74 TfLiteAddParams* builtin_data = in TEST_P() local
76 builtin_data->activation = kTfLiteActNone; in TEST_P()
77 builtin_data->pot_scale_int16 = false; in TEST_P()
80 reinterpret_cast<void*>(builtin_data), reg); in TEST_P()
117 TfLiteAddParams* builtin_data = in TEST_P() local
119 builtin_data->activation = kTfLiteActNone; in TEST_P()
120 builtin_data->pot_scale_int16 = false; in TEST_P()
123 reinterpret_cast<void*>(builtin_data), reg); in TEST_P()
155 TfLiteAddParams* builtin_data = in TEST_P() local
157 builtin_data->activation = kTfLiteActNone; in TEST_P()
[all …]
/external/tensorflow/tensorflow/lite/delegates/hexagon/
Dutils.cc148 reinterpret_cast<const TfLiteAddParams*>(node->builtin_data); in IsNodeSupportedByHexagon()
157 reinterpret_cast<const TfLiteMulParams*>(node->builtin_data); in IsNodeSupportedByHexagon()
167 reinterpret_cast<const TfLiteSubParams*>(node->builtin_data); in IsNodeSupportedByHexagon()
188 node->builtin_data); in IsNodeSupportedByHexagon()
217 node->builtin_data); in IsNodeSupportedByHexagon()
238 reinterpret_cast<const TfLitePoolParams*>(node->builtin_data); in IsNodeSupportedByHexagon()
245 reinterpret_cast<const TfLitePoolParams*>(node->builtin_data); in IsNodeSupportedByHexagon()
268 node->builtin_data); in IsNodeSupportedByHexagon()
280 reinterpret_cast<const TfLiteConvParams*>(node->builtin_data); in IsNodeSupportedByHexagon()
297 node->builtin_data); in IsNodeSupportedByHexagon()
[all …]
/external/tensorflow/tensorflow/lite/micro/kernels/arc_mli/
Ddepthwise_conv_slicing_test.cc66 TfLiteDepthwiseConvParams builtin_data; in ValidateDepthwiseConvGoldens() local
67 builtin_data.padding = kTfLitePaddingValid; in ValidateDepthwiseConvGoldens()
68 builtin_data.activation = activation; in ValidateDepthwiseConvGoldens()
69 builtin_data.stride_height = 1; in ValidateDepthwiseConvGoldens()
70 builtin_data.stride_width = 1; in ValidateDepthwiseConvGoldens()
71 builtin_data.dilation_height_factor = 1; in ValidateDepthwiseConvGoldens()
72 builtin_data.dilation_width_factor = 1; in ValidateDepthwiseConvGoldens()
73 builtin_data.depth_multiplier = depth_mul; in ValidateDepthwiseConvGoldens()
75 const char* init_data = reinterpret_cast<const char*>(&builtin_data); in ValidateDepthwiseConvGoldens()
93 node.builtin_data = reinterpret_cast<void*>(&builtin_data); in ValidateDepthwiseConvGoldens()
Dpooling_slicing_test.cc68 TfLitePoolParams builtin_data = {padding, stride_width, stride_height, in TestAveragePoolingQuantized() local
70 const char* init_data = reinterpret_cast<const char*>(&builtin_data); in TestAveragePoolingQuantized()
88 node.builtin_data = reinterpret_cast<void*>(&builtin_data); in TestAveragePoolingQuantized()
139 TfLitePoolParams builtin_data = { in TestMaxPoolQuantized() local
144 const char* init_data = reinterpret_cast<const char*>(&builtin_data); in TestMaxPoolQuantized()
163 node.builtin_data = reinterpret_cast<void*>(&builtin_data); in TestMaxPoolQuantized()
Dfully_connected_slicing_test.cc77 TfLiteFullyConnectedParams builtin_data = { in TestFullyConnectedQuantized() local
81 const char* init_data = reinterpret_cast<const char*>(&builtin_data); in TestFullyConnectedQuantized()
100 node.builtin_data = reinterpret_cast<void*>(&builtin_data); in TestFullyConnectedQuantized()
Dpooling.cc87 TFLITE_DCHECK(node->builtin_data != nullptr); in Prepare()
88 auto* params = reinterpret_cast<TfLitePoolParams*>(node->builtin_data); in Prepare()
334 TFLITE_DCHECK(node->builtin_data != nullptr); in AverageEval()
335 auto* params = reinterpret_cast<TfLitePoolParams*>(node->builtin_data); in AverageEval()
367 auto* params = reinterpret_cast<TfLitePoolParams*>(node->builtin_data); in MaxEval()
/external/tensorflow/tensorflow/lite/delegates/gpu/common/
Dmodel_builder_test.cc193 void* builtin_data = malloc(sizeof(int)); in InterpreterFp16() local
231 /*builtin_data=*/builtin_data, in InterpreterFp16()
466 void* builtin_data = malloc(sizeof(int)); in InterpreterFp32() local
499 /*builtin_data=*/builtin_data, in InterpreterFp32()
581 void* builtin_data = malloc(sizeof(int)); in Interpreter2Fp32() local
614 /*builtin_data=*/builtin_data, in Interpreter2Fp32()
645 /*builtin_data=*/builtin_data, in Interpreter2Fp32()
761 void* builtin_data = malloc(sizeof(int)); in InterpreterMultiNode() local
797 /*builtin_data=*/builtin_data, in InterpreterMultiNode()
815 /*builtin_data=*/builtin_data, in InterpreterMultiNode()
[all …]
/external/tensorflow/tensorflow/lite/delegates/coreml/builders/
Dconcatenation_op_builder.cc64 if (node->builtin_data == nullptr) return false; in IsConcatenationOpSupported()
66 reinterpret_cast<TfLiteConcatenationParams*>(node->builtin_data); in IsConcatenationOpSupported()
Dop_builder.cc103 op->SetBuiltinData(node->builtin_data); in AddBuilder()
183 void OpBuilder::SetBuiltinData(void* builtin_data) { in SetBuiltinData() argument
184 builtin_data_ = builtin_data; in SetBuiltinData()
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate.cc1691 Expect(reinterpret_cast<TfLiteAddParams*>(node->builtin_data) in Validate()
1729 reinterpret_cast<TfLiteArgMaxParams*>(node->builtin_data); in Validate()
1735 reinterpret_cast<TfLiteArgMinParams*>(node->builtin_data); in Validate()
1746 Expect(reinterpret_cast<TfLiteMulParams*>(node->builtin_data) in Validate()
1758 auto builtin = reinterpret_cast<TfLitePoolParams*>(node->builtin_data); in Validate()
1778 auto builtin = reinterpret_cast<TfLitePoolParams*>(node->builtin_data); in Validate()
1811 auto builtin = reinterpret_cast<TfLiteConvParams*>(node->builtin_data); in Validate()
1835 reinterpret_cast<TfLiteDepthwiseConvParams*>(node->builtin_data); in Validate()
1867 reinterpret_cast<TfLiteFullyConnectedParams*>(node->builtin_data); in Validate()
1912 reinterpret_cast<TfLiteReshapeParams*>(node->builtin_data); in Validate()
[all …]
/external/tensorflow/tensorflow/lite/kernels/
Dfake_quant.cc47 reinterpret_cast<TfLiteFakeQuantParams*>(node->builtin_data); in Prepare()
68 reinterpret_cast<TfLiteFakeQuantParams*>(node->builtin_data); in Eval()
/external/tensorflow/tensorflow/lite/micro/kernels/cmsis_nn/
Dpooling.cc268 TFLITE_DCHECK(node->builtin_data != nullptr); in MaxPrepare()
271 auto* params = reinterpret_cast<TfLitePoolParams*>(node->builtin_data); in MaxPrepare()
283 TFLITE_DCHECK(node->builtin_data != nullptr); in AveragePrepare()
286 auto* params = reinterpret_cast<TfLitePoolParams*>(node->builtin_data); in AveragePrepare()
317 auto* params = reinterpret_cast<TfLitePoolParams*>(node->builtin_data); in AverageEval()
344 auto* params = reinterpret_cast<TfLitePoolParams*>(node->builtin_data); in MaxEval()
Dfully_connected.cc51 TFLITE_DCHECK(node->builtin_data != nullptr); in Prepare()
55 static_cast<const TfLiteFullyConnectedParams*>(node->builtin_data); in Prepare()
185 TFLITE_DCHECK(node->builtin_data != nullptr); in Eval()
187 static_cast<const TfLiteFullyConnectedParams*>(node->builtin_data); in Eval()
Dmul.cc89 TFLITE_DCHECK(node->builtin_data != nullptr); in Prepare()
90 auto* params = reinterpret_cast<TfLiteMulParams*>(node->builtin_data); in Prepare()
183 auto* params = reinterpret_cast<TfLiteMulParams*>(node->builtin_data); in Eval()

123456