Searched refs:hardswish_op (Results 1 – 5 of 5) sorted by relevance
| /external/XNNPACK/test/ |
| D | hardswish-operator-tester.h | 109 xnn_operator_t hardswish_op = nullptr; in TestF16() local
112 0, &hardswish_op); in TestF16()
116 ASSERT_NE(nullptr, hardswish_op); in TestF16()
119 …std::unique_ptr<xnn_operator, decltype(&xnn_delete_operator)> auto_hardswish_op(hardswish_op, xnn_… in TestF16()
123 hardswish_op, in TestF16()
129 xnn_run_operator(hardswish_op, nullptr /* thread pool */)); in TestF16()
165 xnn_operator_t hardswish_op = nullptr; in TestF32() local
170 0, &hardswish_op)); in TestF32()
171 ASSERT_NE(nullptr, hardswish_op); in TestF32()
174 …std::unique_ptr<xnn_operator, decltype(&xnn_delete_operator)> auto_hardswish_op(hardswish_op, xnn_… in TestF32()
[all …]
|
| /external/XNNPACK/bench/ |
| D | hardswish.cc | 49 xnn_operator_t hardswish_op = nullptr; in xnnpack_hardswish_f32() local
52 0 /* flags */, &hardswish_op); in xnnpack_hardswish_f32()
53 if (status != xnn_status_success || hardswish_op == nullptr) { in xnnpack_hardswish_f32()
59 hardswish_op, in xnnpack_hardswish_f32()
69 status = xnn_run_operator(hardswish_op, nullptr /* thread pool */); in xnnpack_hardswish_f32()
76 status = xnn_delete_operator(hardswish_op); in xnnpack_hardswish_f32()
230 xnn_operator_t hardswish_op = nullptr; in xnnpack_hardswish_f16() local
233 0 /* flags */, &hardswish_op); in xnnpack_hardswish_f16()
234 if (status != xnn_status_success || hardswish_op == nullptr) { in xnnpack_hardswish_f16()
240 hardswish_op, in xnnpack_hardswish_f16()
[all …]
|
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | identify_hardswish.cc | 102 auto* hardswish_op = new HardSwishOperator; in Run() local
103 hardswish_op->inputs = {input_tensor_name}; in Run()
104 hardswish_op->outputs = {output_tensor_name}; in Run()
105 model->operators.emplace(add_with_relu6_op_it, hardswish_op); in Run()
107 LogName(*hardswish_op)); in Run()
|
| /external/XNNPACK/src/operators/ |
| D | unary-elementwise-nc.c | 645 xnn_operator_t hardswish_op, in xnn_setup_hardswish_nc_f16() argument
651 if (hardswish_op->type != xnn_operator_type_hardswish_nc_f16) { in xnn_setup_hardswish_nc_f16()
654 xnn_operator_type_to_string(hardswish_op->type)); in xnn_setup_hardswish_nc_f16()
657 hardswish_op->state = xnn_run_state_invalid; in xnn_setup_hardswish_nc_f16()
660 hardswish_op, in xnn_setup_hardswish_nc_f16()
663 &hardswish_op->params.f16_hswish, sizeof(hardswish_op->params.f16_hswish)); in xnn_setup_hardswish_nc_f16()
667 xnn_operator_t hardswish_op, in xnn_setup_hardswish_nc_f32() argument
673 if (hardswish_op->type != xnn_operator_type_hardswish_nc_f32) { in xnn_setup_hardswish_nc_f32()
676 xnn_operator_type_to_string(hardswish_op->type)); in xnn_setup_hardswish_nc_f32()
679 hardswish_op->state = xnn_run_state_invalid; in xnn_setup_hardswish_nc_f32()
[all …]
|
| /external/XNNPACK/include/ |
| D | xnnpack.h | 1382 xnn_operator_t hardswish_op,
1879 xnn_operator_t hardswish_op,
|