Searched refs:channel_shuffle_op (Results 1 – 4 of 4) sorted by relevance
| /external/XNNPACK/src/operators/ |
| D | channel-shuffle-nc.c | 31 xnn_operator_t channel_shuffle_op = NULL; in create_channel_shuffle_nc() local
75 channel_shuffle_op = xnn_allocate_zero_simd_memory(sizeof(struct xnn_operator)); in create_channel_shuffle_nc()
76 if (channel_shuffle_op == NULL) { in create_channel_shuffle_nc()
83 channel_shuffle_op->groups = groups; in create_channel_shuffle_nc()
84 channel_shuffle_op->group_channels = group_channels; in create_channel_shuffle_nc()
85 channel_shuffle_op->input_pixel_stride = input_stride; in create_channel_shuffle_nc()
86 channel_shuffle_op->output_pixel_stride = output_stride; in create_channel_shuffle_nc()
88 channel_shuffle_op->type = operator_type; in create_channel_shuffle_nc()
90 channel_shuffle_op->state = xnn_run_state_invalid; in create_channel_shuffle_nc()
92 *channel_shuffle_op_out = channel_shuffle_op; in create_channel_shuffle_nc()
[all …]
|
| /external/XNNPACK/test/ |
| D | channel-shuffle-operator-tester.h | 113 xnn_operator_t channel_shuffle_op = nullptr; in TestX8() local
119 0, &channel_shuffle_op)); in TestX8()
120 ASSERT_NE(nullptr, channel_shuffle_op); in TestX8()
123 …_operator, decltype(&xnn_delete_operator)> auto_channel_shuffle_op(channel_shuffle_op, xnn_delete_… in TestX8()
127 channel_shuffle_op, in TestX8()
133 xnn_run_operator(channel_shuffle_op, nullptr /* thread pool */)); in TestX8()
161 xnn_operator_t channel_shuffle_op = nullptr; in TestX32() local
167 0, &channel_shuffle_op)); in TestX32()
168 ASSERT_NE(nullptr, channel_shuffle_op); in TestX32()
171 …_operator, decltype(&xnn_delete_operator)> auto_channel_shuffle_op(channel_shuffle_op, xnn_delete_… in TestX32()
[all …]
|
| /external/XNNPACK/bench/ |
| D | channel-shuffle.cc | 41 xnn_operator_t channel_shuffle_op = nullptr; in channel_shuffle_x8() local
46 0 /* flags */, &channel_shuffle_op); in channel_shuffle_x8()
47 if (status != xnn_status_success || channel_shuffle_op == nullptr) { in channel_shuffle_x8()
53 channel_shuffle_op, in channel_shuffle_x8()
63 status = xnn_run_operator(channel_shuffle_op, nullptr /* thread pool */); in channel_shuffle_x8()
70 status = xnn_delete_operator(channel_shuffle_op); in channel_shuffle_x8()
109 xnn_operator_t channel_shuffle_op = nullptr; in channel_shuffle_x32() local
114 0 /* flags */, &channel_shuffle_op); in channel_shuffle_x32()
115 if (status != xnn_status_success || channel_shuffle_op == nullptr) { in channel_shuffle_x32()
121 channel_shuffle_op, in channel_shuffle_x32()
[all …]
|
| /external/XNNPACK/include/ |
| D | xnnpack.h | 1709 xnn_operator_t channel_shuffle_op,
2264 xnn_operator_t channel_shuffle_op,
|