Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/base/
Dminimal_filtering_generator.c189 float inverse_diagonal_matrix[MAX_LEN]; // input_unit * input_unit in CookToomFilter() local
205 inverse_diagonal_matrix[j] = 1.0 / diagonal_matrix[j]; in CookToomFilter()
207 inverse_diagonal_matrix[j] = 0; in CookToomFilter()
228 MatrixMultiply(tmp_g, inverse_diagonal_matrix, matrix_gt, filter_size, in_unit, in_unit); in CookToomFilter()