Home
last modified time | relevance | path

Searched refs:flat_size (Results 1 – 25 of 27) sorted by relevance

12

/external/tensorflow/tensorflow/lite/micro/kernels/
Dactivations.cc37 const int flat_size = MatchingFlatSize(input_shape, output_shape); in ReluQuantized() local
38 for (int i = 0; i < flat_size; ++i) { in ReluQuantized()
47 const int flat_size = MatchingFlatSize(input_shape, output_shape); in ReluFloat() local
48 for (int i = 0; i < flat_size; ++i) { in ReluFloat()
58 const int flat_size = MatchingFlatSize(input_shape, output_shape); in Relu6Float() local
59 for (int i = 0; i < flat_size; ++i) { in Relu6Float()
72 const int flat_size = MatchingFlatSize(input_shape, output_shape); in Relu6Quantized() local
73 for (int i = 0; i < flat_size; ++i) { in Relu6Quantized()
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dlogistic.h35 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()
Ddequantize.h35 const int flat_size = MatchingFlatSize(input_shape, output_shape); in Dequantize() local
37 for (int i = 0; i < flat_size; i++) { in Dequantize()
53 const int flat_size = MatchingFlatSize(input_shape, output_shape); in DequantizeInteger() local
55 for (int i = 0; i < flat_size; i++) { in DequantizeInteger()
Dreference_ops.h174 const int flat_size = MatchingFlatSize(input_shape, output_shape); in Elu() local
175 for (int i = 0; i < flat_size; ++i) { in Elu()
184 const int flat_size = MatchingFlatSize(input_shape, output_shape); in Relu() local
185 for (int i = 0; i < flat_size; ++i) { in Relu()
197 const int flat_size = MatchingFlatSize(input_shape, output_shape); in Relu1() local
198 for (int i = 0; i < flat_size; ++i) { in Relu1()
210 const int flat_size = MatchingFlatSize(input_shape, output_shape); in Relu6() local
211 for (int i = 0; i < flat_size; ++i) { in Relu6()
225 const int flat_size = MatchingFlatSize(input_shape, output_shape); in ReluX() local
226 for (int i = 0; i < flat_size; ++i) { in ReluX()
[all …]
Dadd.h30 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()
125 const int flat_size = in Add() local
132 AddElementwise(flat_size, params, input1_data, input2_data, output_data); in Add()
143 const int flat_size = in Add() local
156 for (int i = 0; i < flat_size; i++) { in Add()
Dceil.h28 const int flat_size = MatchingFlatSize(input_shape, output_shape); in Ceil() local
30 for (int i = 0; i < flat_size; ++i) { in Ceil()
Dfloor.h28 const int flat_size = MatchingFlatSize(input_shape, output_shape); in Floor() local
30 for (int i = 0; i < flat_size; i++) { in Floor()
Dneg.h27 const int flat_size = MatchingFlatSize(input_shape, output_shape); in Negate() local
29 for (int i = 0; i < flat_size; ++i) { in Negate()
Dmul.h53 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()
Dround.h39 const int flat_size = MatchingFlatSize(input_shape, output_shape); in Round() local
40 for (int i = 0; i < flat_size; ++i) { in Round()
Dquantize.h34 const int flat_size = MatchingFlatSize(input_shape, output_shape); in AffineQuantize() local
38 for (int i = 0; i < flat_size; i++) { in AffineQuantize()
Dbinary_function.h74 const int flat_size = in BinaryFunction() local
76 for (int i = 0; i < flat_size; ++i) { in BinaryFunction()
/external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/
Dmul.h50 const int flat_size = in Mul() local
53 MulElementwise(flat_size, params, input1_data, input2_data, output_data); in Mul()
67 const int flat_size = in Mul() local
70 for (int i = 0; i < flat_size; i++) { in Mul()
Ddequantize.h31 const int flat_size = MatchingFlatSize(input_shape, output_shape); in Dequantize() local
33 for (int i = 0; i < flat_size; i++) { in Dequantize()
Dadd.h66 const int flat_size = in Add() local
74 AddElementwise(flat_size, params, input1_data, input2_data, output_data); in Add()
/external/tensorflow/tensorflow/lite/kernels/internal/
Ddepthwiseconv_per_channel_quantized_test.cc144 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-5f, 1.01f / flat_size * std::sqrt(1.f * depth_multiplier)); in CompareRoundingResults()
Dtranspose_utils.cc127 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()
Dtypes.h448 int flat_size = 1; in FlatSize() local
450 flat_size *= dims.sizes[i]; in FlatSize()
452 return flat_size; in FlatSize()
573 int flat_size = 1; in FlatSizeSkipDim() local
575 flat_size *= (i == skip_dim) ? 1 : dims.sizes[i]; in FlatSizeSkipDim()
577 return flat_size; in FlatSizeSkipDim()
639 int flat_size = 1; in FlatSizeSkipDim() local
641 flat_size *= (i == skip_dim) ? 1 : dims_data[i]; in FlatSizeSkipDim()
643 return flat_size; in FlatSizeSkipDim()
/external/tensorflow/tensorflow/lite/micro/kernels/xtensa-hifimini/
Dquantize.cc39 const int flat_size = MatchingFlatSize(input_shape, output_shape); in AffineQuantize() local
48 int iters = flat_size / 2; in AffineQuantize()
/external/tensorflow/tensorflow/lite/kernels/
Dpow_test.cc123 int flat_size, std::vector<T>* output_data) { in CalculateTrueResults() argument
124 for (int i = 0; i < flat_size; ++i) { in CalculateTrueResults()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Doptimized_ops.h1720 const int flat_size = in Add() local
1722 AddElementwise(flat_size, params, input1_data, input2_data, output_data); in Add()
1959 const int flat_size = in Add() local
1966 AddElementwise(flat_size, params, input1_data, input2_data, output_data); in Add()
1978 const int flat_size = in Add() local
1991 for (int i = 0; i < flat_size; i++) { in Add()
2290 const int flat_size = in Mul() local
2292 MulElementwise(flat_size, params, input1_data, input2_data, output_data); in Mul()
2301 const int flat_size = in Mul() local
2305 for (int i = 0; i < flat_size; ++i) { in Mul()
[all …]
/external/u-boot/tools/
Dzynqmpbif.c547 static char *elf2flat64(char *elf, size_t *flat_size, size_t *load_addr) in elf2flat64() argument
574 *flat_size = max_addr - min_addr; in elf2flat64()
575 flat = calloc(1, *flat_size); in elf2flat64()
595 static char *elf2flat32(char *elf, size_t *flat_size, size_t *load_addr) in elf2flat32() argument
622 *flat_size = max_addr - min_addr; in elf2flat32()
623 flat = calloc(1, *flat_size); in elf2flat32()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/integer_ops/
Dmul.h178 const int flat_size = in Mul() local
181 MulElementwise(flat_size, params, input1_data, input2_data, output_data); in Mul()
Dadd.h224 const int flat_size = in Add() local
231 AddElementwise(flat_size, params, input1_data, input2_data, output_data); in Add()
/external/tensorflow/tensorflow/python/ops/ragged/
Dragged_tensor.py2544 flat_size = _prod(old_shape[1:shape_split])
2548 flat_size = math_ops.cast(
2551 new_splits = new_splits * flat_size

12