Searched refs:input_matrix (Results 1 – 2 of 2) sorted by relevance
| /third_party/mindspore/mindspore/nn/optim/ |
| D | thor.py | 883 def _process_batch_matmul(self, input_matrix): argument
885 input_matrix_shape = self.shape(input_matrix)
887 input_matrix = self.vector_matmul(input_matrix, input_matrix)
889 input_matrix = self.tbe_batch_matmul(input_matrix, input_matrix)
890 return input_matrix
892 def _process_cholesky_pad(self, pad_dim, input_matrix, matrix_shape0): argument
897 input_matrix = P.Pad(((0, 0), (0, pad_dim)))(input_matrix)
898 input_matrix = self.concat((input_matrix, matrix_sup))
899 return input_matrix
|
| /third_party/ffmpeg/libavfilter/ |
| D | vf_convolution_opencl.c | 109 float input_matrix[4][49]; in convolution_opencl_make_filter_params() local
123 sscanf_err = sscanf(arg, "%f", &input_matrix[i][ctx->matrix_sizes[i]]); in convolution_opencl_make_filter_params()
152 matrix[i] = input_matrix[j][i]; in convolution_opencl_make_filter_params()
|