Home
last modified time | relevance | path

Searched refs:filter_offset (Results 1 – 4 of 4) sorted by relevance

/frameworks/ml/nn/common/operations/internal/optimized/
Ddepthwiseconv_uint8.h40 int16 filter_offset, int32* acc_buffer_ptr) {
48 vdupq_n_s16(filter_offset));
87 int16 filter_offset, int32* acc_buffer_ptr) {
91 const int16x8_t filter = vaddq_s16(filter_s16, vdupq_n_s16(filter_offset));
155 int16 filter_offset, int32* acc_buffer_ptr) {
159 const int16x8_t filter = vaddq_s16(filter_s16, vdupq_n_s16(filter_offset));
225 int16 filter_offset, int32* acc_buffer_ptr) {
231 filter[i] = vaddq_s16(filter_s16, vdupq_n_s16(filter_offset));
302 int16 filter_offset, int32* acc_buffer_ptr) {
311 const int16x4_t filter = vadd_s16(filter_s16, vdup_n_s16(filter_offset));
[all …]
Doptimized_ops.h371 const Dims<4>& filter_dims, int32 filter_offset, in FullyConnectedAsGEMV() argument
408 const int16x8_t filter_offset_vec = vdupq_n_s16(filter_offset); in FullyConnectedAsGEMV()
479 filter_data[in + (out + k) * input_size] + filter_offset; in FullyConnectedAsGEMV()
591 const Dims<4>& filter_dims, int32 filter_offset,
616 filter_offset, bias_data, bias_dims, output_offset, output_multiplier,
643 gemm_context, filter_matrix, input_matrix, &output_matrix, filter_offset,
810 const Dims<4>& filter_dims, int32 filter_offset,
879 gemm_context, filter_matrix, input_matrix, &output_matrix, filter_offset,
949 const Dims<4>& filter_dims, int32 filter_offset,
988 gemm_context, filter_matrix, input_matrix, &output_matrix, filter_offset,
/frameworks/ml/nn/common/operations/internal/reference/
Ddepthwiseconv_uint8.h32 const Dims<4>& filter_dims, int32 filter_offset, in DepthwiseConv() argument
83 (filter_val + filter_offset) * (input_val + input_offset); in DepthwiseConv()
Dreference_ops.h221 const Dims<4>& filter_dims, int32 filter_offset, in Conv() argument
273 (filter_val + filter_offset) * (input_val + input_offset); in Conv()
412 const Dims<4>& filter_dims, int32 filter_offset, in FullyConnected() argument
445 acc += (filter_val + filter_offset) * (input_val + input_offset); in FullyConnected()