Searched refs:k_scaled (Results 1 – 4 of 4) sorted by relevance
32 const size_t k_scaled = context->k_scaled; in xnn_compute_grouped_gemm() local39 k_scaled, in xnn_compute_grouped_gemm()40 (const void*) ((uintptr_t) context->a + mr_block_start * a_stride + group_index * k_scaled), in xnn_compute_grouped_gemm()62 context->k_scaled, in xnn_compute_gemm()908 const size_t k_scaled = context->k_scaled; in xnn_compute_hmp_grouped_gemm() local915 k_scaled, in xnn_compute_hmp_grouped_gemm()916 (const void*) ((uintptr_t) context->a + mr_block_start * a_stride + group_index * k_scaled), in xnn_compute_hmp_grouped_gemm()939 context->k_scaled, in xnn_compute_hmp_gemm()
206 .k_scaled = input_channels << log2_input_element_size, in setup_fully_connected_nc()
866 .k_scaled = group_input_channels << log2_input_element_size, in setup_convolution2d_nhwc()
65 size_t k_scaled; member