| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | logistic.h | 35 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
|
| D | leaky_relu.h | 29 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
|
| D | tanh.h | 31 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
|
| D | sub.h | 39 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
|
| D | reference_ops.h | 97 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 …]
|
| D | mul.h | 56 const int flat_size = in Mul() local
72 const int flat_size = in Mul() local
85 const int flat_size = in Mul() local
|
| D | ceil.h | 28 const int flat_size = MatchingFlatSize(input_shape, output_shape); in Ceil() local
|
| D | neg.h | 27 const int flat_size = MatchingFlatSize(input_shape, output_shape); in Negate() local
|
| D | fill.h | 29 const int flat_size = output_shape.FlatSize(); in Fill() local
|
| D | floor.h | 28 const int flat_size = MatchingFlatSize(input_shape, output_shape); in Floor() local
|
| D | cast.h | 28 const int flat_size = MatchingFlatSize(input_shape, output_shape); in Cast() local
|
| D | elu.h | 27 const int flat_size = MatchingFlatSize(input_shape, output_shape); in Elu() local
|
| D | div.h | 84 const int flat_size = in Div() local
96 const int flat_size = in Div() local
235 const int flat_size = in Div() local
|
| D | add.h | 36 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
|
| D | round.h | 39 const int flat_size = MatchingFlatSize(input_shape, output_shape); in Round() local
|
| D | maximum_minimum.h | 33 const int flat_size = in MaximumMinimumBroadcastSlow() local
|
| D | dequantize.h | 37 const int flat_size = MatchingFlatSize(input_shape, output_shape); in Dequantize() local
|
| D | hard_swish.h | 66 const int flat_size = MatchingFlatSize(input_shape, output_shape); in HardSwish() local
|
| D | quantize.h | 39 const int flat_size = MatchingFlatSize(input_shape, output_shape); in AffineQuantize() local
|
| D | prelu.h | 85 const int flat_size = in Prelu() local
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/ |
| D | tanh.h | 38 const int flat_size = MatchingFlatSize(input_shape, output_shape); in Tanh() local
81 int flat_size = MatchingFlatSize(input_shape, output_shape); in Tanh() local
|
| D | mul.h | 54 const int flat_size = in Mul() local
71 const int flat_size = in Mul() local
|
| D | dequantize.h | 31 const int flat_size = MatchingFlatSize(input_shape, output_shape); in Dequantize() local
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/integer_ops/ |
| D | leaky_relu.h | 32 const int flat_size = MatchingFlatSize(input_shape, output_shape); in QuantizeLeakyRelu() local
|
| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | transpose_utils.cc | 127 size_t flat_size = input_shape.FlatSize(); in Flatten() local
|