| /external/tensorflow/tensorflow/lite/kernels/ |
| D | pooling.cc | 122 template <KernelType kernel_type>
142 if (kernel_type == kReference) { in AverageEvalFloat()
150 template <KernelType kernel_type>
172 if (kernel_type == kReference) { in AverageEvalQuantizedUint8()
201 template <KernelType kernel_type>
220 if (kernel_type == kReference) { in MaxEvalFloat()
228 template <KernelType kernel_type>
249 if (kernel_type == kReference) { in MaxEvalQuantizedUInt8()
257 template <KernelType kernel_type>
282 template <KernelType kernel_type>
[all …]
|
| D | resize_bilinear.cc | 76 template <KernelType kernel_type>
99 if (kernel_type == kReference) { in Eval()
102 if (kernel_type == kGenericOptimized || kernel_type == kNeonOptimized) { in Eval()
106 if (kernel_type == kReference) { in Eval()
109 if (kernel_type == kGenericOptimized || kernel_type == kNeonOptimized) { in Eval()
|
| D | slice.cc | 135 template <KernelType kernel_type>
175 #define TF_LITE_SLICE(data_type, kernel_type) \ in Eval() argument
187 if (kernel_type == kGenericOptimized) { \ in Eval()
200 TF_LITE_SLICE(float, kernel_type); in Eval()
203 TF_LITE_SLICE(int32_t, kernel_type); in Eval()
206 TF_LITE_SLICE(int64_t, kernel_type); in Eval()
209 TF_LITE_SLICE(int8_t, kernel_type); in Eval()
212 TF_LITE_SLICE(uint8_t, kernel_type); in Eval()
215 TF_LITE_SLICE(bool, kernel_type); in Eval()
|
| D | mul.cc | 108 template <KernelType kernel_type>
125 if (kernel_type == kReference) { in EvalMul()
139 if (kernel_type == kReference) { in EvalMul()
156 template <KernelType kernel_type>
186 if (kernel_type == kReference) { in EvalQuantized()
209 if (kernel_type == kReference) { in EvalQuantized()
229 if (kernel_type == kReference) { in EvalQuantized()
244 template <KernelType kernel_type>
254 EvalMul<kernel_type>(context, node, params, data, input1, input2, output); in Eval()
258 context, EvalQuantized<kernel_type>(context, node, params, data, input1, in Eval()
|
| D | transpose.cc | 86 template <KernelType kernel_type>
111 if (kernel_type == kReference) { in Eval()
116 if (kernel_type == kReference) { in Eval()
121 if (kernel_type == kReference) { in Eval()
126 if (kernel_type == kReference) { in Eval()
131 if (kernel_type == kReference) { in Eval()
|
| D | space_to_depth.cc | 75 template <KernelType kernel_type>
91 if (kernel_type == kReference) { in Eval()
98 if (kernel_type == kReference) { in Eval()
105 if (kernel_type == kReference) { in Eval()
112 if (kernel_type == kReference) { in Eval()
119 if (kernel_type == kReference) { in Eval()
|
| D | pad.cc | 119 template <KernelType kernel_type>
170 if (kernel_type == kReference) { in Eval()
176 } else if (kernel_type == kGenericOptimized) { in Eval()
203 if (kernel_type == kReference) { in Eval()
209 } else if (kernel_type == kGenericOptimized) { in Eval()
247 if (kernel_type == kReference) { in Eval()
249 } else if (kernel_type == kGenericOptimized) { in Eval()
258 if (kernel_type == kReference) { in Eval()
260 } else if (kernel_type == kGenericOptimized) { in Eval()
|
| D | strided_slice.cc | 178 template <KernelType kernel_type>
210 #define TF_LITE_STRIDED_SLICE(kernel_type, data_type) \ in Eval() argument
211 kernel_type::StridedSlice(op_params, GetTensorShape(op_context.input), \ in Eval()
218 if (kernel_type == kReference) { in Eval()
223 if (kernel_type == kReference) { in Eval()
228 if (kernel_type == kReference) { in Eval()
233 if (kernel_type == kReference) { in Eval()
238 if (kernel_type == kReference) { in Eval()
|
| D | batch_to_space_nd.cc | 118 template <KernelType kernel_type>
138 if (kernel_type == kReference) { in Eval()
145 if (kernel_type == kReference) { in Eval()
152 if (kernel_type == kReference) { in Eval()
159 if (kernel_type == kReference) { in Eval()
166 if (kernel_type == kReference) { in Eval()
|
| D | space_to_batch_nd.cc | 107 template <KernelType kernel_type>
129 if (kernel_type == kReference) { in Eval()
136 if (kernel_type == kReference) { in Eval()
145 if (kernel_type == kReference) { in Eval()
154 if (kernel_type == kReference) { in Eval()
161 if (kernel_type == kReference) { in Eval()
|
| D | exp.cc | 53 template <KernelType kernel_type>
57 #define TF_LITE_EXP(kernel_type, data_type) \ in Eval() argument
58 kernel_type::Exp<data_type>(GetTensorData<data_type>(op_context.input), \ in Eval()
63 if (kernel_type == kReference) { in Eval()
|
| D | add.cc | 175 template <KernelType kernel_type>
191 if (kernel_type == kReference) { in EvalAdd()
205 if (kernel_type == kReference) { in EvalAdd()
222 template <KernelType kernel_type>
256 if (kernel_type == kReference) { in EvalAddQuantized()
287 if (kernel_type == kReference) { in EvalAddQuantized()
298 template <KernelType kernel_type>
308 EvalAdd<kernel_type>(context, node, params, data, input1, input2, output); in Eval()
312 EvalAddQuantized<kernel_type>(context, node, params, data, in Eval()
|
| D | l2norm.cc | 72 template <KernelType kernel_type>
85 if (kernel_type == kReference) { in Eval()
88 if (kernel_type == kGenericOptimized) { in Eval()
100 if (kernel_type == kReference) { in Eval()
103 if (kernel_type == kGenericOptimized) { in Eval()
|
| D | concatenation.cc | 103 template <KernelType kernel_type>
143 if (kernel_type == kReference) { in Eval()
150 if (kernel_type == kReference) { in Eval()
157 if (kernel_type == kReference) { in Eval()
164 if (kernel_type == kReference) { in Eval()
171 if (kernel_type == kReference) { in Eval()
|
| D | fully_connected.cc | 312 template <KernelType kernel_type>
346 } else if (kernel_type == kReference) { in EvalQuantized()
386 template <KernelType kernel_type>
419 if (kernel_type == kReference) { in EvalShuffledQuantized()
429 template <KernelType kernel_type>
448 if (kernel_type == kReference) { in EvalFloat()
450 } else if (kernel_type == kLegacyPie) { in EvalFloat()
462 template <KernelType kernel_type>
475 return EvalFloat<kernel_type>(context, node, params, data, input, filter, in Eval()
482 return EvalShuffledQuantized<kernel_type>(context, node, params, data, in Eval()
[all …]
|
| D | div.cc | 80 template <KernelType kernel_type>
96 if (kernel_type == kReference) { in EvalDiv()
110 if (kernel_type == kReference) { in EvalDiv()
127 template <KernelType kernel_type>
137 EvalDiv<kernel_type>(context, node, params, data, input1, input2, output); in Eval()
|
| D | sub.cc | 225 template <KernelType kernel_type>
241 if (kernel_type == kReference) { in EvalSub()
255 if (kernel_type == kReference) { in EvalSub()
272 template <KernelType kernel_type>
308 if (kernel_type == kReference) { in EvalQuantized()
325 if (kernel_type == kReference) { in EvalQuantized()
342 template <KernelType kernel_type>
352 EvalSub<kernel_type>(context, node, params, data, input1, input2, output); in Eval()
355 EvalQuantized<kernel_type>(context, node, params, data, input1, input2, in Eval()
|
| D | resize_nearest_neighbor.cc | 74 template <KernelType kernel_type>
97 if (kernel_type == kReference) { in Eval()
103 if (kernel_type == kGenericOptimized || kernel_type == kNeonOptimized) { in Eval()
|
| D | conv.cc | 419 template <KernelType kernel_type>
432 if ((kernel_type == kMultithreadOptimized || in EvalQuantized()
433 kernel_type == kCblasOptimized) && in EvalQuantized()
440 effective_kernel_type = kernel_type; in EvalQuantized()
521 template <KernelType kernel_type>
530 if ((kernel_type == kMultithreadOptimized) && in EvalFloat()
537 effective_kernel_type = kernel_type; in EvalFloat()
588 template <KernelType kernel_type>
633 switch (kernel_type) { in EvalHybrid()
661 template <KernelType kernel_type>
[all …]
|
| D | depthwise_conv.cc | 188 template <KernelType kernel_type>
201 if (kernel_type == kReference) { in EvalFloat()
224 template <KernelType kernel_type>
238 if (kernel_type == kReference) { in EvalQuantized()
294 template <KernelType kernel_type>
310 EvalFloat<kernel_type>(context, node, params, data, input, filter, bias, in Eval()
314 EvalQuantized<kernel_type>(context, node, params, data, input, filter, in Eval()
|
| D | local_response_norm.cc | 58 template <KernelType kernel_type>
76 if (kernel_type == kReference) { in Eval()
79 if (kernel_type == kGenericOptimized) { in Eval()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | depthwiseconv_quantized_test.cc | 150 DotProduct3x3KernelType kernel_type = in DispatchDepthwiseConv() local
155 kernel_type == DotProduct3x3KernelType::kPlain || in DispatchDepthwiseConv()
156 kernel_type == DotProduct3x3KernelType::kStride2 || in DispatchDepthwiseConv()
157 kernel_type == in DispatchDepthwiseConv()
159 kernel_type == in DispatchDepthwiseConv()
161 << "Kernel type = " << static_cast<int>(kernel_type); in DispatchDepthwiseConv()
172 DotProduct3x3KernelType kernel_type = in DispatchDepthwiseConv() local
177 kernel_type == DotProduct3x3KernelType::kPlain || in DispatchDepthwiseConv()
178 kernel_type == DotProduct3x3KernelType::kStride2 || in DispatchDepthwiseConv()
179 kernel_type == in DispatchDepthwiseConv()
[all …]
|
| /external/tensorflow/tensorflow/cc/gradients/ |
| D | image_grad_test.cc | 200 Input translation, const string& kernel_type, bool antialias, in MakeOp() argument
204 ScaleAndTranslate::KernelType(kernel_type) in MakeOp()
213 Input translation, const string& kernel_type, in TestScaleAndTranslate() argument
218 kernel_type, antialias, &x, &y); in TestScaleAndTranslate()
242 for (const std::string& kernel_type : kKernelTypes) { in TEST_F() local
244 kXShape, kOutHeight, kOutWidth, scale, translation, kernel_type, in TEST_F()
274 for (const std::string& kernel_type : kKernelTypes) { in TEST_F() local
276 kXShape, kOutHeight, kOutWidth, scale, translation, kernel_type, in TEST_F()
291 for (const std::string& kernel_type : kKernelTypes) { in TEST_F() local
293 kXShape, kOutHeight, kOutWidth, scale, translation, kernel_type, in TEST_F()
|
| D | image_grad.cc | 88 string kernel_type; in ScaleAndTranslateGradHelper() local
90 GetNodeAttr(op.node()->attrs(), "kernel_type", &kernel_type)); in ScaleAndTranslateGradHelper()
95 internal::ScaleAndTranslateGrad::KernelType(kernel_type) in ScaleAndTranslateGradHelper()
|
| /external/tensorflow/tensorflow/lite/testing/kernel_test/ |
| D | tflite_kernel_runner.cc | 21 const bool run_reference_kernel = options.kernel_type == "REFERENCE"; in main()
22 const bool use_nnapi = options.kernel_type == "NNAPI"; in main()
|