Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/contrib/lite/kernels/internal/
Dtypes.h440 inline int MatchingFlatSize(const RuntimeShape& shape, in MatchingFlatSize() function
450 inline int MatchingFlatSize(const RuntimeShape& shape, in MatchingFlatSize() function
458 return MatchingFlatSize(shape, check_shape_1); in MatchingFlatSize()
461 inline int MatchingFlatSize(const RuntimeShape& shape, in MatchingFlatSize() function
470 return MatchingFlatSize(shape, check_shape_1, check_shape_2); in MatchingFlatSize()
473 inline int MatchingFlatSize(const RuntimeShape& shape, in MatchingFlatSize() function
483 return MatchingFlatSize(shape, check_shape_1, check_shape_2, check_shape_3); in MatchingFlatSize()
489 inline int MatchingFlatSize(const Dims<N>& dims, const Dims<N>& check_dims_0) { in MatchingFlatSize() function
497 inline int MatchingFlatSize(const Dims<N>& dims, const Dims<N>& check_dims_0, in MatchingFlatSize() function
502 return MatchingFlatSize(dims, check_dims_1); in MatchingFlatSize()
[all …]
/external/tensorflow/tensorflow/lite/kernels/internal/
Dtypes.h456 inline int MatchingFlatSize(const RuntimeShape& shape, in MatchingFlatSize() function
466 inline int MatchingFlatSize(const RuntimeShape& shape, in MatchingFlatSize() function
474 return MatchingFlatSize(shape, check_shape_1); in MatchingFlatSize()
477 inline int MatchingFlatSize(const RuntimeShape& shape, in MatchingFlatSize() function
486 return MatchingFlatSize(shape, check_shape_1, check_shape_2); in MatchingFlatSize()
489 inline int MatchingFlatSize(const RuntimeShape& shape, in MatchingFlatSize() function
499 return MatchingFlatSize(shape, check_shape_1, check_shape_2, check_shape_3); in MatchingFlatSize()
505 inline int MatchingFlatSize(const Dims<N>& dims, const Dims<N>& check_dims_0) { in MatchingFlatSize() function
513 inline int MatchingFlatSize(const Dims<N>& dims, const Dims<N>& check_dims_0, in MatchingFlatSize() function
518 return MatchingFlatSize(dims, check_dims_1); in MatchingFlatSize()
[all …]
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dreference_ops.h396 const int flat_size = MatchingFlatSize(input_shape, output_shape); in Elu()
405 const int flat_size = MatchingFlatSize(input_shape, output_shape); in Relu()
417 const int flat_size = MatchingFlatSize(input_shape, output_shape); in Relu1()
430 const int flat_size = MatchingFlatSize(input_shape, output_shape); in Relu6()
445 const int flat_size = MatchingFlatSize(input_shape, output_shape); in ReluX()
460 const int flat_size = MatchingFlatSize(input_shape, output_shape); in LeakyRelu()
530 MatchingFlatSize(input1_shape, input2_shape, output_shape); in Add()
542 const int size = MatchingFlatSize(input1_shape, input2_shape, output_shape); in Add()
640 MatchingFlatSize(input1_shape, input2_shape, output_shape); in Add()
658 MatchingFlatSize(output_shape, input1_shape, input2_shape); in Add()
[all …]
/external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/
Dmul.h50 MatchingFlatSize(input1_shape, input2_shape, output_shape); in Mul()
67 MatchingFlatSize(input1_shape, input2_shape, output_shape); in Mul()
Ddequantize.h30 const int flat_size = MatchingFlatSize(input_shape, output_shape); in Dequantize()
Dadd.h67 MatchingFlatSize(input1_shape, input2_shape, output_shape); in Add()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Doptimized_ops.h2630 const int size = MatchingFlatSize(input1_shape, input2_shape, output_shape); in Add()
2879 MatchingFlatSize(input1_shape, input2_shape, output_shape); in Add()
2898 MatchingFlatSize(output_shape, input1_shape, input2_shape); in Add()
3054 const int size = MatchingFlatSize(input1_shape, input2_shape, output_shape); in Mul()
3112 MatchingFlatSize(input1_shape, input2_shape, output_shape); in Mul()
3157 MatchingFlatSize(input1_shape, input2_shape, output_shape); in Mul()
3182 MatchingFlatSize(input1_shape, input2_shape, output_shape); in Mul()
3343 MatchingFlatSize(input1_shape, input2_shape, output_shape); in Mul()
3487 MatchingFlatSize(input1_shape, input2_shape, output_shape); in SubNonBroadcast()
3504 MatchingFlatSize(input1_shape, input2_shape, input2_shape); in SubWithActivation()
[all …]
/external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/
Doptimized_ops.h2847 const int size = MatchingFlatSize(input1_shape, input2_shape, output_shape); in Add()
2992 MatchingFlatSize(input1_shape, input2_shape, output_shape); in Add()
3011 MatchingFlatSize(output_shape, input1_shape, input2_shape); in Add()
3130 const int size = MatchingFlatSize(input1_shape, input2_shape, output_shape); in Mul()
3188 MatchingFlatSize(input1_shape, input2_shape, output_shape); in Mul()
3233 MatchingFlatSize(input1_shape, input2_shape, output_shape); in Mul()
3258 MatchingFlatSize(input1_shape, input2_shape, output_shape); in Mul()
3419 MatchingFlatSize(input1_shape, input2_shape, output_shape); in Mul()
3578 MatchingFlatSize(input1_shape, input2_shape, output_shape); in SubNonBroadcast()
3595 MatchingFlatSize(input1_shape, input2_shape, input2_shape); in SubWithActivation()
[all …]
/external/tensorflow/tensorflow/lite/kernels/
Dactivations.cc468 const int size = MatchingFlatSize(input_shape, output_shape); in TanhEval()
534 MatchingFlatSize(GetTensorShape(input), GetTensorShape(output)); in SigmoidEval()