Searched refs:argmax_pooling_op (Results 1 – 3 of 3) sorted by relevance
55 xnn_operator_t argmax_pooling_op = NULL; in xnn_create_argmax_pooling2d_nhwc_f32() local119 argmax_pooling_op = xnn_allocate_zero_simd_memory(sizeof(struct xnn_operator)); in xnn_create_argmax_pooling2d_nhwc_f32()120 if (argmax_pooling_op == NULL) { in xnn_create_argmax_pooling2d_nhwc_f32()127 argmax_pooling_op->padding_top = input_padding_top; in xnn_create_argmax_pooling2d_nhwc_f32()128 argmax_pooling_op->padding_right = input_padding_right; in xnn_create_argmax_pooling2d_nhwc_f32()129 argmax_pooling_op->padding_bottom = input_padding_bottom; in xnn_create_argmax_pooling2d_nhwc_f32()130 argmax_pooling_op->padding_left = input_padding_left; in xnn_create_argmax_pooling2d_nhwc_f32()132 argmax_pooling_op->kernel_height = pooling_height; in xnn_create_argmax_pooling2d_nhwc_f32()133 argmax_pooling_op->kernel_width = pooling_width; in xnn_create_argmax_pooling2d_nhwc_f32()134 argmax_pooling_op->stride_height = pooling_height; in xnn_create_argmax_pooling2d_nhwc_f32()[all …]
376 xnn_operator_t argmax_pooling_op = nullptr; in TestF32() local385 &argmax_pooling_op)); in TestF32()386 ASSERT_NE(nullptr, argmax_pooling_op); in TestF32()389 …nn_operator, decltype(&xnn_delete_operator)> auto_argmax_pooling_op(argmax_pooling_op, xnn_delete_… in TestF32()393 argmax_pooling_op, in TestF32()399 xnn_run_operator(argmax_pooling_op, nullptr /* thread pool */)); in TestF32()473 xnn_operator_t argmax_pooling_op = nullptr; in TestSetupF32() local480 0, &argmax_pooling_op)); in TestSetupF32()481 ASSERT_NE(nullptr, argmax_pooling_op); in TestSetupF32()485 argmax_pooling_op, in TestSetupF32()[all …]
1558 xnn_operator_t argmax_pooling_op,