Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/delegates/gpu/common/
Dwinograd_util.cc122 auto gt_mat = GetTransposedMatrixForWinograd(6, 3); in RearrangeWeightsToWinograd4x4To6x6Weights() local
123 std::vector<float> g_mat(gt_mat.size()); in RearrangeWeightsToWinograd4x4To6x6Weights()
126 g_mat[x * 3 + y] = gt_mat[y * 6 + x]; in RearrangeWeightsToWinograd4x4To6x6Weights()
141 auto out_vals = Multiply(temp_vals, gt_mat, 6, 3, 6); in RearrangeWeightsToWinograd4x4To6x6Weights()