Home
last modified time | relevance | path

Searched refs:HardSwish (Results 1 – 13 of 13) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/xnnpack/
Dhard_swish_test.cc28 TEST(HardSwish, 4D) {
47 TEST(HardSwish, 3D) {
65 TEST(HardSwish, 2D) {
82 TEST(HardSwish, 1D) {
97 TEST(HardSwish, MultiThreading) { in TEST() argument
/external/tensorflow/tensorflow/lite/micro/kernels/
Dhard_swish.cc96 tflite::reference_ops::HardSwish<float>( in HardSwishEval()
103 tflite::reference_ops::HardSwish<uint8_t>( in HardSwishEval()
110 tflite::reference_ops::HardSwish<int8_t>( in HardSwishEval()
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dhard_swish.h45 inline void HardSwish(const RuntimeShape& input_shape, const T* input_data, in HardSwish() function
59 inline void HardSwish(const HardSwishParams& params, in HardSwish() function
/external/tensorflow/tensorflow/lite/kernels/
Dactivations.cc818 reference_ops::HardSwish( in HardSwishEval()
822 optimized_ops::HardSwish( in HardSwishEval()
831 reference_ops::HardSwish( in HardSwishEval()
835 optimized_ops::HardSwish( in HardSwishEval()
844 reference_ops::HardSwish( in HardSwishEval()
848 optimized_ops::HardSwish( in HardSwishEval()
Dactivations_test.cc460 TEST(FloatActivationsOpTest, HardSwish) { in TEST() argument
467 TEST(QuantizedActivationsOpTest, HardSwish) { in TEST() argument
/external/tensorflow/tensorflow/lite/delegates/gpu/cl/kernels/
Delementwise_test.cc55 TEST_F(OpenCLOperationTest, HardSwish) { in TEST_F() argument
/external/XNNPACK/
DREADME.md42 - HardSwish
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Doptimize_patterns.td252 // Matching HardSwish
278 // Matching HardSwish with extra FakeQuant. These FakeQuant ops were due to
/external/tensorflow/tensorflow/lite/delegates/gpu/gl/kernels/
Delementwise_test.cc106 TEST(ElementwiseOneArgumentTest, HardSwish) { in TEST() argument
/external/tensorflow/tensorflow/lite/toco/
Dtooling_util.cc341 HANDLE_OPERATORTYPENAME_CASE(HardSwish) in OperatorTypeName()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Doptimized_ops.h6435 inline void HardSwish(const RuntimeShape& input_shape, const float* input_data, in HardSwish() function
6512 inline void HardSwish(const HardSwishParams& params, in HardSwish() function
6671 reference_ops::HardSwish(params, leftover_shape, input_data + i, in HardSwish()
/external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/
Dlegalize_tfl.cc117 DECL_CONVERT_OP(HardSwish);
/external/tensorflow/
DRELEASE.md1202 * Fix error checking supported operations in a model containing `HardSwish`.