Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/
Dmul.h49 const int flat_size = in Mul() local
52 MulElementwise(flat_size, params, input1_data, input2_data, output_data); in Mul()
66 const int flat_size = in Mul() local
69 for (int i = 0; i < flat_size; i++) { in Mul()
Ddequantize.h30 const int flat_size = MatchingFlatSize(input_shape, output_shape); in Dequantize() local
32 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/reference/
Dreference_ops.h396 const int flat_size = MatchingFlatSize(input_shape, output_shape); in Elu() local
397 for (int i = 0; i < flat_size; ++i) { in Elu()
405 const int flat_size = MatchingFlatSize(input_shape, output_shape); in Relu() local
406 for (int i = 0; i < flat_size; ++i) { in Relu()
417 const int flat_size = MatchingFlatSize(input_shape, output_shape); in Relu1() local
418 for (int i = 0; i < flat_size; ++i) { in Relu1()
430 const int flat_size = MatchingFlatSize(input_shape, output_shape); in Relu6() local
431 for (int i = 0; i < flat_size; ++i) { in Relu6()
445 const int flat_size = MatchingFlatSize(input_shape, output_shape); in ReluX() local
448 for (int i = 0; i < flat_size; ++i) { in ReluX()
[all …]
/external/tensorflow/tensorflow/contrib/lite/kernels/internal/
Dtypes.h426 int flat_size = 1; in FlatSize() local
428 flat_size *= dims.sizes[i]; in FlatSize()
430 return flat_size; in FlatSize()
532 int flat_size = 1; in FlatSizeSkipDim() local
534 flat_size *= (i == skip_dim) ? 1 : dims.sizes[i]; in FlatSizeSkipDim()
536 return flat_size; in FlatSizeSkipDim()
598 int flat_size = 1; in FlatSizeSkipDim() local
600 flat_size *= (i == skip_dim) ? 1 : dims_data[i]; in FlatSizeSkipDim()
602 return flat_size; in FlatSizeSkipDim()
/external/tensorflow/tensorflow/lite/kernels/internal/
Dtypes.h442 int flat_size = 1; in FlatSize() local
444 flat_size *= dims.sizes[i]; in FlatSize()
446 return flat_size; in FlatSize()
548 int flat_size = 1; in FlatSizeSkipDim() local
550 flat_size *= (i == skip_dim) ? 1 : dims.sizes[i]; in FlatSizeSkipDim()
552 return flat_size; in FlatSizeSkipDim()
614 int flat_size = 1; in FlatSizeSkipDim() local
616 flat_size *= (i == skip_dim) ? 1 : dims_data[i]; in FlatSizeSkipDim()
618 return flat_size; in FlatSizeSkipDim()
/external/u-boot/tools/
Dzynqmpbif.c538 static char *elf2flat64(char *elf, size_t *flat_size, size_t *load_addr) in elf2flat64() argument
565 *flat_size = max_addr - min_addr; in elf2flat64()
566 flat = calloc(1, *flat_size); in elf2flat64()
586 static char *elf2flat32(char *elf, size_t *flat_size, size_t *load_addr) in elf2flat32() argument
613 *flat_size = max_addr - min_addr; in elf2flat32()
614 flat = calloc(1, *flat_size); in elf2flat32()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Doptimized_ops.h807 const int flat_size = output_shape.FlatSize(); in FullyConnected() local
808 for (int i = 0; i < flat_size; ++i) { in FullyConnected()
2878 const int flat_size = in Add() local
2885 AddElementwise(flat_size, params, input1_data, input2_data, output_data); in Add()
2897 const int flat_size = in Add() local
2910 for (int i = 0; i < flat_size; i++) { in Add()
3111 const int flat_size = in Mul() local
3115 for (int i = 0; i < flat_size; ++i) { in Mul()
3156 const int flat_size = in Mul() local
3159 for (int i = 0; i < flat_size; i++) { in Mul()
[all …]
/external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/
Doptimized_ops.h2991 const int flat_size = in Add() local
2998 AddElementwise(flat_size, params, input1_data, input2_data, output_data); in Add()
3010 const int flat_size = in Add() local
3023 for (int i = 0; i < flat_size; i++) { in Add()
3187 const int flat_size = in Mul() local
3191 for (int i = 0; i < flat_size; ++i) { in Mul()
3232 const int flat_size = in Mul() local
3235 for (int i = 0; i < flat_size; i++) { in Mul()
3257 const int flat_size = in Mul() local
3260 for (int i = 0; i < flat_size; i++) { in Mul()
[all …]