Searched refs:packed_channels (Results 1 – 3 of 3) sorted by relevance
77 const size_t packed_channels = round_up_po2(channels, XNN_EXTRA_BYTES / sizeof(float)); in xnn_create_prelu_nc_f32() local78 prelu_op->packed_weights = xnn_allocate_simd_memory(packed_channels * sizeof(float)); in xnn_create_prelu_nc_f32()81 packed_channels * sizeof(float)); in xnn_create_prelu_nc_f32()
51 inline size_t packed_channels() const { in packed_channels() function131 std::vector<float, AlignedAllocator<float, 64>> packed_w(packed_channels() * 2);
88 inline uint32_t packed_channels() const { in packed_channels() function162 …cator<uint8_t, 64>> packed_weights((kr() + sizeof(int32_t) / sizeof(uint8_t)) * packed_channels());266 std::vector<float, AlignedAllocator<float, 64>> packed_weights((kr() + 1) * packed_channels());