| /external/tensorflow/tensorflow/lite/micro/kernels/ceva/ |
| D | ceva_tflm_lib.h | 40 int flat_size);
42 int flat_size);
44 int flat_size);
46 int flat_size);
49 int flat_size);
51 int flat_size);
53 int flat_size);
56 int flat_size);
58 int flat_size);
61 int flat_size);
[all …]
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | logistic.h | 35 const int flat_size = MatchingFlatSize(input_shape, output_shape); in Logistic() local
45 for (int i = 0; i < flat_size; i++) { in Logistic()
71 const int flat_size = MatchingFlatSize(input_shape, output_shape); in Logistic() local
73 for (int i = 0; i < flat_size; i++) { in Logistic()
100 const int flat_size = MatchingFlatSize(input_shape, output_shape); in Logistic() local
110 for (int i = 0; i < flat_size; i++) { in Logistic()
|
| D | tanh.h | 31 const int flat_size = MatchingFlatSize(input_shape, output_shape); in Tanh() local
33 for (int i = 0; i < flat_size; i++) { in Tanh()
58 const int flat_size = MatchingFlatSize(input_shape, output_shape); in Tanh() local
68 for (int i = 0; i < flat_size; i++) { in Tanh()
74 for (int i = 0; i < flat_size; i++) { in Tanh()
91 const int flat_size = MatchingFlatSize(input_shape, output_shape); in Tanh() local
93 for (int i = 0; i < flat_size; i++) { in Tanh()
|
| D | leaky_relu.h | 29 const int flat_size = MatchingFlatSize(input_shape, output_shape); in LeakyRelu() local
30 for (int i = 0; i < flat_size; ++i) { in LeakyRelu()
43 const int flat_size = MatchingFlatSize(input_shape, output_shape); in QuantizeLeakyRelu() local
46 for (int i = 0; i < flat_size; ++i) { in QuantizeLeakyRelu()
|
| D | add.h | 30 const int flat_size = in Add() local
32 for (int i = 0; i < flat_size; ++i) { in Add()
43 const int flat_size = in Add() local
45 for (int i = 0; i < flat_size; i++) { in Add()
130 const int flat_size = in Add() local
137 AddElementwise(flat_size, params, input1_data, input2_data, output_data); in Add()
149 const int flat_size = in AddGeneralParamScale() local
158 AddElementwise(flat_size, params, input1_data, input2_data, output_data); in AddGeneralParamScale()
176 const int flat_size = variable
190 for (int i = 0; i < flat_size; i++) {
|
| D | ceil.h | 28 const int flat_size = MatchingFlatSize(input_shape, output_shape); in Ceil() local
30 for (int i = 0; i < flat_size; ++i) { in Ceil()
|
| D | floor.h | 28 const int flat_size = MatchingFlatSize(input_shape, output_shape); in Floor() local
30 for (int i = 0; i < flat_size; i++) { in Floor()
|
| D | neg.h | 27 const int flat_size = MatchingFlatSize(input_shape, output_shape); in Negate() local
29 for (int i = 0; i < flat_size; ++i) { in Negate()
|
| D | fill.h | 29 const int flat_size = output_shape.FlatSize(); in Fill() local
30 for (int i = 0; i < flat_size; ++i) { in Fill()
|
| D | elu.h | 28 const int flat_size = MatchingFlatSize(input_shape, output_shape); in Elu() local
29 for (int i = 0; i < flat_size; ++i) { in Elu()
|
| D | cast.h | 28 const int flat_size = MatchingFlatSize(input_shape, output_shape); in Cast() local
30 for (int i = 0; i < flat_size; i++) { in Cast()
|
| D | round.h | 39 const int flat_size = MatchingFlatSize(input_shape, output_shape); in Round() local
40 for (int i = 0; i < flat_size; ++i) { in Round()
|
| D | mul.h | 53 const int flat_size = in Mul() local
55 for (int i = 0; i < flat_size; ++i) { in Mul()
68 const int flat_size = in Mul() local
71 MulElementwise(flat_size, params, input1_data, input2_data, output_data); in Mul()
|
| D | sub.h | 39 const int flat_size = in SubNonBroadcast() local
41 for (int i = 0; i < flat_size; ++i) { in SubNonBroadcast()
55 const int flat_size = in SubNonBroadcast() local
57 for (int i = 0; i < flat_size; ++i) { in SubNonBroadcast()
452 const int flat_size = in Sub() local
459 SubElementwise(flat_size, params, input1_data, input2_data, output_data); in Sub()
469 const int flat_size = in Sub() local
477 SubElementwise(flat_size, params, input1_data, input2_data, output_data); in Sub()
542 const int flat_size = in SubWithActivation() local
547 for (int i = 0; i < flat_size; ++i) { in SubWithActivation()
|
| D | quantize.h | 38 const int flat_size = MatchingFlatSize(input_shape, output_shape); in AffineQuantize() local
42 for (int i = 0; i < flat_size; i++) { in AffineQuantize()
|
| D | div.h | 67 const int flat_size = in Div() local
70 DivElementwise(flat_size, params, input1_data, input2_data, output_data); in Div()
182 const int flat_size = in Div() local
184 for (int i = 0; i < flat_size; ++i) { in Div()
|
| D | maximum_minimum.h | 33 const int flat_size = in MaximumMinimumBroadcastSlow() local
36 for (int i = 0; i < flat_size; ++i) { in MaximumMinimumBroadcastSlow()
|
| D | reference_ops.h | 93 const int flat_size = MatchingFlatSize(input_shape, output_shape); in Relu() local
94 for (int i = 0; i < flat_size; ++i) { in Relu()
106 const int flat_size = MatchingFlatSize(input_shape, output_shape); in Relu1() local
107 for (int i = 0; i < flat_size; ++i) { in Relu1()
119 const int flat_size = MatchingFlatSize(input_shape, output_shape); in Relu6() local
120 for (int i = 0; i < flat_size; ++i) { in Relu6()
134 const int flat_size = MatchingFlatSize(input_shape, output_shape); in ReluX() local
135 for (int i = 0; i < flat_size; ++i) { in ReluX()
152 const int flat_size = MatchingFlatSize(input_shape, output_shape); in ReluX() local
155 for (int i = 0; i < flat_size; ++i) { in ReluX()
[all …]
|
| /external/tensorflow/tensorflow/lite/micro/kernels/ |
| D | zeros_like.cc | 52 int flat_size = MatchingFlatSize(tflite::micro::GetTensorShape(input), in Eval() local
56 resetZeros(tflite::micro::GetTensorData<int64_t>(output), flat_size); in Eval()
59 resetZeros(tflite::micro::GetTensorData<int32_t>(output), flat_size); in Eval()
62 resetZeros(tflite::micro::GetTensorData<float>(output), flat_size); in Eval()
|
| D | activations.cc | 54 const int flat_size = MatchingFlatSize(input_shape, output_shape); in ReluQuantized() local
55 for (int i = 0; i < flat_size; ++i) { in ReluQuantized()
99 const int flat_size = MatchingFlatSize(input_shape, output_shape); in ReluFloat() local
100 for (int i = 0; i < flat_size; ++i) { in ReluFloat()
110 const int flat_size = MatchingFlatSize(input_shape, output_shape); in Relu6Float() local
111 for (int i = 0; i < flat_size; ++i) { in Relu6Float()
124 const int flat_size = MatchingFlatSize(input_shape, output_shape); in Relu6Quantized() local
125 for (int i = 0; i < flat_size; ++i) { in Relu6Quantized()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/ |
| D | tanh.h | 37 const int flat_size = MatchingFlatSize(input_shape, output_shape); in Tanh() local
39 for (int i = 0; i < flat_size; ++i) { in Tanh()
80 int flat_size = MatchingFlatSize(input_shape, output_shape); in Tanh() local
82 for (int i = 0; i < flat_size; ++i, ptr_input_data++, ptr_output_data++) { in Tanh()
|
| D | mul.h | 52 const int flat_size = in Mul() local
55 MulElementwise(flat_size, params, input1_data, input2_data, output_data); in Mul()
69 const int flat_size = in Mul() local
72 for (int i = 0; i < flat_size; i++) { in Mul()
|
| D | dequantize.h | 31 const int flat_size = MatchingFlatSize(input_shape, output_shape); in Dequantize() local
33 for (int i = 0; i < flat_size; i++) { in Dequantize()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | depthwiseconv_per_channel_quantized_test.cc | 144 void CompareRoundingResults(int flat_size, const int depth_multiplier, in CompareRoundingResults() argument
147 std::vector<int> diff(flat_size); in CompareRoundingResults()
150 for (int i = 0; i < flat_size; i++) { in CompareRoundingResults()
161 const float mean_diff = static_cast<float>(sum_diff) / flat_size; in CompareRoundingResults()
162 const float mean_abs_diff = static_cast<float>(sum_abs_diff) / flat_size; in CompareRoundingResults()
167 std::max(1e-2f, 1.01f / flat_size * std::sqrt(1.f * depth_multiplier)); in CompareRoundingResults()
|
| D | transpose_utils.cc | 127 size_t flat_size = input_shape.FlatSize(); in Flatten() local
130 flat_size /= input_shape.Dims(i); in Flatten()
160 return flat_size; in Flatten()
|