Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/lite/kernels/internal/reference/
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
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
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
Dadd.h30 const int flat_size = in Add() local
43 const int flat_size = in Add() local
130 const int flat_size = in Add() local
149 const int flat_size = in AddGeneralParamScale() local
176 const int flat_size = variable
Dfill.h29 const int flat_size = output_shape.FlatSize(); in Fill() local
Dceil.h28 const int flat_size = MatchingFlatSize(input_shape, output_shape); in Ceil() local
Dneg.h27 const int flat_size = MatchingFlatSize(input_shape, output_shape); in Negate() local
Dfloor.h28 const int flat_size = MatchingFlatSize(input_shape, output_shape); in Floor() local
Delu.h28 const int flat_size = MatchingFlatSize(input_shape, output_shape); in Elu() local
Dcast.h28 const int flat_size = MatchingFlatSize(input_shape, output_shape); in Cast() local
Dround.h39 const int flat_size = MatchingFlatSize(input_shape, output_shape); in Round() local
Dmaximum_minimum.h33 const int flat_size = in MaximumMinimumBroadcastSlow() local
Dsub.h39 const int flat_size = in SubNonBroadcast() local
55 const int flat_size = in SubNonBroadcast() local
452 const int flat_size = in Sub() local
469 const int flat_size = in Sub() local
542 const int flat_size = in SubWithActivation() local
Dquantize.h38 const int flat_size = MatchingFlatSize(input_shape, output_shape); in AffineQuantize() local
Dmul.h53 const int flat_size = in Mul() local
68 const int flat_size = in Mul() local
Ddiv.h67 const int flat_size = in Div() local
182 const int flat_size = in Div() local
Ddequantize.h37 const int flat_size = MatchingFlatSize(input_shape, output_shape); in Dequantize() local
Dbinary_function.h70 const int flat_size = in BinaryFunction() local
/external/tensorflow/tensorflow/lite/micro/kernels/
Dactivations.cc54 const int flat_size = MatchingFlatSize(input_shape, output_shape); in ReluQuantized() local
99 const int flat_size = MatchingFlatSize(input_shape, output_shape); in ReluFloat() local
110 const int flat_size = MatchingFlatSize(input_shape, output_shape); in Relu6Float() local
124 const int flat_size = MatchingFlatSize(input_shape, output_shape); in Relu6Quantized() local
Dexp.cc51 int flat_size = MatchingFlatSize(tflite::micro::GetTensorShape(input), in Eval() local
Dzeros_like.cc52 int flat_size = MatchingFlatSize(tflite::micro::GetTensorShape(input), in Eval() local
Ddequantize.cc116 int flat_size = MatchingFlatSize(tflite::micro::GetTensorShape(input), in Eval() local
/external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/
Dtanh.h37 const int flat_size = MatchingFlatSize(input_shape, output_shape); in Tanh() local
80 int flat_size = MatchingFlatSize(input_shape, output_shape); in Tanh() local
Dmul.h52 const int flat_size = in Mul() local
69 const int flat_size = in Mul() local
Ddequantize.h31 const int flat_size = MatchingFlatSize(input_shape, output_shape); in Dequantize() local

12