Home
last modified time | relevance | path

Searched refs:AveragePool (Results 1 – 16 of 16) sorted by relevance

/external/tensorflow/tensorflow/lite/micro/kernels/
Dpooling.cc73 reference_ops::AveragePool(op_params, tflite::micro::GetTensorShape(input), in AverageEvalFloat()
96 reference_ops::AveragePool(op_params, tflite::micro::GetTensorShape(input), in AverageEvalQuantized()
101 reference_integer_ops::AveragePool( in AverageEvalQuantized()
/external/tensorflow/tensorflow/lite/kernels/internal/
Daveragepool_quantized_test.cc43 reference_integer_ops::AveragePool(params, input_shape, input_data, in RunOneAveragePoolTest()
46 optimized_integer_ops::AveragePool(params, input_shape, input_data, in RunOneAveragePoolTest()
/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/tests/
Dpool_test.cc53 TEST(QuantizedPoolingOpTest, AveragePool) { in TEST() argument
/external/tensorflow/tensorflow/lite/kernels/
Dpooling.cc132 type::AveragePool(op_params, GetTensorShape(input), \ in AverageEvalFloat()
162 type::AveragePool(op_params, GetTensorShape(input), \ in AverageEvalQuantizedUint8()
192 type::AveragePool(op_params, GetTensorShape(input), \ in AverageEvalQuantizedInt8()
222 type::AveragePool(op_params, GetTensorShape(input), \ in AverageEvalQuantizedInt16()
Dpooling_test.cc140 TEST(FloatPoolingOpTest, AveragePool) { in TEST() argument
238 TEST(QuantizedPoolingOpTest, AveragePool) { in TEST() argument
/external/tensorflow/tensorflow/lite/micro/kernels/arc_mli/
Dpooling.cc161 reference_ops::AveragePool(op_params, tflite::micro::GetTensorShape(input), in AverageEvalFloat()
255 reference_ops::AveragePool(op_params, tflite::micro::GetTensorShape(input), in AverageEvalQuantized()
260 reference_integer_ops::AveragePool( in AverageEvalQuantized()
/external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/
Dpooling.h24 inline void AveragePool(const PoolParams& params, in AveragePool() function
139 inline void AveragePool(const PoolParams& params, in AveragePool() function
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dpooling.h26 inline void AveragePool(const PoolParams& params, in AveragePool() function
79 inline void AveragePool(const PoolParams& params, in AveragePool() function
Dlegacy_reference_ops.h1489 inline void AveragePool(const float* input_data, const Dims<4>& input_dims, in AveragePool() function
1504 AveragePool(params, DimsToShape(input_dims), input_data, in AveragePool()
1564 void AveragePool(const float* input_data, const Dims<4>& input_dims, in AveragePool() function
1571 AveragePool(input_data, input_dims, stride_width, stride_height, pad_width, in AveragePool()
1578 void AveragePool(const float* input_data, const Dims<4>& input_dims, int stride, in AveragePool() function
1582 AveragePool<Ac>(input_data, input_dims, stride, stride, pad_width, pad_height, in AveragePool()
1586 inline void AveragePool(const uint8* input_data, const Dims<4>& input_dims, in AveragePool() function
1601 AveragePool(params, DimsToShape(input_dims), input_data, in AveragePool()
1607 void AveragePool(const uint8* input_data, const Dims<4>& input_dims, in AveragePool() function
1621 AveragePool(input_data, input_dims, stride_width, stride_height, pad_width, in AveragePool()
[all …]
/external/tensorflow/tensorflow/lite/micro/kernels/cmsis_nn/
Dpooling.cc90 reference_ops::AveragePool(op_params, tflite::micro::GetTensorShape(input), in AverageEvalFloat()
113 reference_ops::AveragePool(op_params, tflite::micro::GetTensorShape(input), in AverageEvalQuantized()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/integer_ops/
Dpooling.h147 inline void AveragePool(const PoolParams& params, in AveragePool() function
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Dlegacy_optimized_ops.h3765 inline void AveragePool(const float* input_data, const Dims<4>& input_dims, in AveragePool() function
3780 AveragePool(params, DimsToShape(input_dims), input_data, in AveragePool()
3786 void AveragePool(const float* input_data, const Dims<4>& input_dims, in AveragePool() function
3793 AveragePool(input_data, input_dims, stride_width, stride_height, pad_width, in AveragePool()
3800 void AveragePool(const float* input_data, const Dims<4>& input_dims, int stride, in AveragePool() function
3804 AveragePool<Ac>(input_data, input_dims, stride, stride, pad_width, pad_height, in AveragePool()
3808 inline void AveragePool(const uint8* input_data, const Dims<4>& input_dims, in AveragePool() function
3823 AveragePool(params, DimsToShape(input_dims), input_data, in AveragePool()
3829 void AveragePool(const uint8* input_data, const Dims<4>& input_dims, in AveragePool() function
3843 AveragePool(input_data, input_dims, stride_width, stride_height, pad_width, in AveragePool()
[all …]
Doptimized_ops.h3173 inline void AveragePool(const PoolParams& params, in AveragePool() function
3235 inline void AveragePool(const PoolParams& params, in AveragePool() function
/external/tensorflow/tensorflow/lite/g3doc/r1/convert/
Dcmdline_examples.md280 `AveragePool` node in the top right of each image.
/external/tensorflow/tensorflow/lite/toco/tflite/
Doperator.cc89 class AveragePool class
1845 ops.push_back(MakeUnique<AveragePool>( in BuildOperatorList()
/external/tensorflow/tensorflow/lite/toco/
Dtooling_util.cc332 HANDLE_OPERATORTYPENAME_CASE(AveragePool) in OperatorTypeName()