Home
last modified time | relevance | path

Searched refs:MatchingFlatSize (Results 1 – 25 of 34) sorted by relevance

12

/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dlogistic.h35 const int flat_size = MatchingFlatSize(input_shape, output_shape); in Logistic()
71 const int flat_size = MatchingFlatSize(input_shape, output_shape); in Logistic()
100 const int flat_size = MatchingFlatSize(input_shape, output_shape); in Logistic()
Dtanh.h31 const int flat_size = MatchingFlatSize(input_shape, output_shape); in Tanh()
58 const int flat_size = MatchingFlatSize(input_shape, output_shape); in Tanh()
91 const int flat_size = MatchingFlatSize(input_shape, output_shape); in Tanh()
Dleaky_relu.h29 const int flat_size = MatchingFlatSize(input_shape, output_shape); in LeakyRelu()
43 const int flat_size = MatchingFlatSize(input_shape, output_shape); in QuantizeLeakyRelu()
Ddequantize.h37 const int flat_size = MatchingFlatSize(input_shape, output_shape); in Dequantize()
53 MatchingFlatSize(input_shape, output_shape); in PerChannelDequantize()
Dhard_swish.h48 auto matching_size = MatchingFlatSize(input_shape, output_shape); in HardSwish()
64 const int flat_size = MatchingFlatSize(input_shape, output_shape); in HardSwish()
Dceil.h28 const int flat_size = MatchingFlatSize(input_shape, output_shape); in Ceil()
Dfloor.h28 const int flat_size = MatchingFlatSize(input_shape, output_shape); in Floor()
Dneg.h27 const int flat_size = MatchingFlatSize(input_shape, output_shape); in Negate()
Delu.h28 const int flat_size = MatchingFlatSize(input_shape, output_shape); in Elu()
Dcast.h28 const int flat_size = MatchingFlatSize(input_shape, output_shape); in Cast()
Dmul.h54 MatchingFlatSize(input1_shape, input2_shape, output_shape); in Mul()
69 MatchingFlatSize(input1_shape, input2_shape, output_shape); in Mul()
Dround.h39 const int flat_size = MatchingFlatSize(input_shape, output_shape); in Round()
Dquantize.h38 const int flat_size = MatchingFlatSize(input_shape, output_shape); in AffineQuantize()
Dreference_ops.h93 const int flat_size = MatchingFlatSize(input_shape, output_shape); in Relu()
106 const int flat_size = MatchingFlatSize(input_shape, output_shape); in Relu1()
119 const int flat_size = MatchingFlatSize(input_shape, output_shape); in Relu6()
134 const int flat_size = MatchingFlatSize(input_shape, output_shape); in ReluX()
152 const int flat_size = MatchingFlatSize(input_shape, output_shape); in ReluX()
1029 const int flat_size = MatchingFlatSize(input_shape, output_shape); in Dequantize()
1054 const int flat_size = MatchingFlatSize(input_shape, output_shape); in FakeQuant()
1425 const int flat_size = MatchingFlatSize(input1_shape, output_shape); in Minimum()
1447 const int flat_size = MatchingFlatSize(input1_shape, output_shape); in Maximum()
1489 const int64_t flatsize = MatchingFlatSize( in Select()
[all …]
Dbinary_function.h71 MatchingFlatSize(input1_shape, input2_shape, output_shape); in BinaryFunction()
Dstring_comparisons.h48 MatchingFlatSize(input1_shape, input2_shape, output_shape); in ComparisonStringImpl()
/external/tensorflow/tensorflow/lite/kernels/internal/
Dtypes.h519 inline int MatchingFlatSize(const RuntimeShape& shape, in MatchingFlatSize() function
529 inline int MatchingFlatSize(const RuntimeShape& shape, in MatchingFlatSize() function
537 return MatchingFlatSize(shape, check_shape_1); in MatchingFlatSize()
540 inline int MatchingFlatSize(const RuntimeShape& shape, in MatchingFlatSize() function
549 return MatchingFlatSize(shape, check_shape_1, check_shape_2); in MatchingFlatSize()
552 inline int MatchingFlatSize(const RuntimeShape& shape, in MatchingFlatSize() function
562 return MatchingFlatSize(shape, check_shape_1, check_shape_2, check_shape_3); in MatchingFlatSize()
568 inline int MatchingFlatSize(const Dims<N>& dims, const Dims<N>& check_dims_0) { in MatchingFlatSize() function
576 inline int MatchingFlatSize(const Dims<N>& dims, const Dims<N>& check_dims_0, in MatchingFlatSize() function
581 return MatchingFlatSize(dims, check_dims_1); in MatchingFlatSize()
[all …]
/external/tensorflow/tensorflow/lite/micro/kernels/ceva/
Dtypes.h668 inline int MatchingFlatSize(const RuntimeShape& shape,
678 inline int MatchingFlatSize(const RuntimeShape& shape,
686 return MatchingFlatSize(shape, check_shape_1);
689 inline int MatchingFlatSize(const RuntimeShape& shape,
698 return MatchingFlatSize(shape, check_shape_1, check_shape_2);
701 inline int MatchingFlatSize(const RuntimeShape& shape,
711 return MatchingFlatSize(shape, check_shape_1, check_shape_2, check_shape_3);
717 inline int MatchingFlatSize(const Dims<N>& dims, const Dims<N>& check_dims_0) {
725 inline int MatchingFlatSize(const Dims<N>& dims, const Dims<N>& check_dims_0,
730 return MatchingFlatSize(dims, check_dims_1);
[all …]
/external/tensorflow/tensorflow/lite/kernels/
Dquantize.cc195 MatchingFlatSize(input_shape, output_shape), in Eval()
203 MatchingFlatSize(input_shape, output_shape), in Eval()
214 MatchingFlatSize(input_shape, output_shape), in Eval()
227 const int32_t size = MatchingFlatSize(input_shape, output_shape); in Eval()
249 const int32_t size = MatchingFlatSize(input_shape, output_shape); in Eval()
/external/tensorflow/tensorflow/lite/experimental/resource/
Dstatic_hashtable.cc35 MatchingFlatSize(GetTensorShape(keys), GetTensorShape(values)); in Lookup()
71 MatchingFlatSize(GetTensorShape(keys), GetTensorShape(values)); in Import()
/external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/
Dtanh.h37 const int flat_size = MatchingFlatSize(input_shape, output_shape); in Tanh()
80 int flat_size = MatchingFlatSize(input_shape, output_shape); in Tanh()
Ddequantize.h31 const int flat_size = MatchingFlatSize(input_shape, output_shape); in Dequantize()
/external/tensorflow/tensorflow/lite/micro/kernels/
Dactivations.cc54 const int flat_size = MatchingFlatSize(input_shape, output_shape); in ReluQuantized()
99 const int flat_size = MatchingFlatSize(input_shape, output_shape); in ReluFloat()
110 const int flat_size = MatchingFlatSize(input_shape, output_shape); in Relu6Float()
124 const int flat_size = MatchingFlatSize(input_shape, output_shape); in Relu6Quantized()
Dzeros_like.cc52 int flat_size = MatchingFlatSize(tflite::micro::GetTensorShape(input), in Eval()
Dexp.cc51 int flat_size = MatchingFlatSize(tflite::micro::GetTensorShape(input), in Eval()

12