Searched defs:HardSwish (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | hard_swish.h | 45 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/delegates/xnnpack/ |
| D | hard_swish_test.cc | 97 TEST(HardSwish, MultiThreading) { in TEST() argument
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/cl/kernels/ |
| D | elementwise_test.cc | 55 TEST_F(OpenCLOperationTest, HardSwish) { in TEST_F() argument
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | activations_test.cc | 460 TEST(FloatActivationsOpTest, HardSwish) { in TEST() argument
467 TEST(QuantizedActivationsOpTest, HardSwish) { in TEST() argument
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/gl/kernels/ |
| D | elementwise_test.cc | 106 TEST(ElementwiseOneArgumentTest, HardSwish) { in TEST() argument
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 6435 inline void HardSwish(const RuntimeShape& input_shape, const float* input_data, in HardSwish() function
6512 inline void HardSwish(const HardSwishParams& params, in HardSwish() function
|