Home
last modified time | relevance | path

Searched defs:flat_size (Results 1 – 25 of 53) sorted by relevance

123

/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/reference/
H A Dreference_ops.h97 const int flat_size = MatchingFlatSize(input_shape, output_shape); in Relu() local
110 const int flat_size = MatchingFlatSize(input_shape, output_shape); in Relu0To1() local
124 const int flat_size = MatchingFlatSize(input_shape, output_shape); in Relu1() local
137 const int flat_size = MatchingFlatSize(input_shape, output_shape); in Relu6() local
152 const int flat_size = MatchingFlatSize(input_shape, output_shape); in ReluX() local
170 const int flat_size = MatchingFlatSize(input_shape, output_shape); in ReluX() local
244 const int flat_size = in Mul() local
267 const int flat_size = in Mul() local
292 const int flat_size = in Sub16() local
533 const int flat_size = MatchingFlatSize(input_shape, output_shape); in Dequantize() local
[all …]
H A Dsub.h39 const int flat_size = in SubNonBroadcast() local
55 const int flat_size = in SubNonBroadcast() local
359 const int flat_size = in Sub() local
376 const int flat_size = in Sub() local
395 const int flat_size = in Sub() local
465 const int flat_size = in SubWithActivation() local
H A Dlogistic.h35 const int flat_size = MatchingFlatSize(input_shape, output_shape); in Logistic() local
71 const int flat_size = MatchingFlatSize(input_shape, output_shape); in Logistic() local
100 const int flat_size = MatchingFlatSize(input_shape, output_shape); in Logistic() local
H A Dtanh.h31 const int flat_size = MatchingFlatSize(input_shape, output_shape); in Tanh() local
58 const int flat_size = MatchingFlatSize(input_shape, output_shape); in Tanh() local
91 const int flat_size = MatchingFlatSize(input_shape, output_shape); in Tanh() local
H A Dadd.h36 const int flat_size = in Add() local
122 const int flat_size = in Add() local
141 const int flat_size = in AddGeneralParamScale() local
168 const int flat_size = variable
H A Dmul.h56 const int flat_size = in Mul() local
72 const int flat_size = in Mul() local
85 const int flat_size = in Mul() local
H A Ddiv.h84 const int flat_size = in Div() local
96 const int flat_size = in Div() local
235 const int flat_size = in Div() local
H A Dleaky_relu.h29 const int flat_size = MatchingFlatSize(input_shape, output_shape); in LeakyRelu() local
43 const int flat_size = MatchingFlatSize(input_shape, output_shape); in QuantizeLeakyRelu() local
H A Dfill.h29 const int flat_size = output_shape.FlatSize(); in Fill() local
H A Dneg.h27 const int flat_size = MatchingFlatSize(input_shape, output_shape); in Negate() local
H A Dceil.h28 const int flat_size = MatchingFlatSize(input_shape, output_shape); in Ceil() local
H A Delu.h27 const int flat_size = MatchingFlatSize(input_shape, output_shape); in Elu() local
H A Dcast.h28 const int flat_size = MatchingFlatSize(input_shape, output_shape); in Cast() local
H A Dfloor.h28 const int flat_size = MatchingFlatSize(input_shape, output_shape); in Floor() local
H A Dround.h39 const int flat_size = MatchingFlatSize(input_shape, output_shape); in Round() local
H A Dmaximum_minimum.h33 const int flat_size = in MaximumMinimumBroadcastSlow() local
H A Ddequantize.h37 const int flat_size = MatchingFlatSize(input_shape, output_shape); in Dequantize() local
H A Dhard_swish.h66 const int flat_size = MatchingFlatSize(input_shape, output_shape); in HardSwish() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/
H A Dtanh.h38 const int flat_size = MatchingFlatSize(input_shape, output_shape); in Tanh() local
81 int flat_size = MatchingFlatSize(input_shape, output_shape); in Tanh() local
H A Dmul.h54 const int flat_size = in Mul() local
71 const int flat_size = in Mul() local
H A Ddequantize.h31 const int flat_size = MatchingFlatSize(input_shape, output_shape); in Dequantize() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
H A Doptimized_ops.h1522 const int flat_size = in Add() local
1761 const int flat_size = in Add() local
1780 const int flat_size = in Add() local
1948 const int flat_size = in Mul() local
1959 const int flat_size = in Mul() local
2005 const int flat_size = in Mul() local
2030 const int flat_size = in Mul() local
2234 const int flat_size = in Mul() local
2907 const int flat_size = output_shape.FlatSize(); in AveragePool() local
3102 const int flat_size = output_shape.FlatSize(); in MaxPool() local
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/
H A Dtypes.h279 int flat_size = 1; in FlatSize() local
421 const int flat_size = MatchingExtendedShapeFlatSize(shape, check_shape_0); in MatchingExtendedShapeFlatSize() local
431 const int flat_size = MatchingExtendedShapeFlatSize(shape, check_shape_0); in MatchingExtendedShapeFlatSize() local
443 const int flat_size = MatchingExtendedShapeFlatSize(shape, check_shape_0); in MatchingExtendedShapeFlatSize() local
456 int flat_size = 1; in FlatSizeSkipDim() local
522 int flat_size = 1; in FlatSizeSkipDim() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/optimized/integer_ops/
H A Dadd.h465 const int flat_size = in Add() local
482 const int flat_size = in Add() local
H A Dleaky_relu.h32 const int flat_size = MatchingFlatSize(input_shape, output_shape); in QuantizeLeakyRelu() local

123