Searched refs:dst_weights (Results 1 – 2 of 2) sorted by relevance
113 Tensor<OHWI, DataType::FLOAT32>* dst_weights) { in RearrangeWeightsToWinograd4x4To6x6Weights() argument119 dst_weights->shape = dst_shape; in RearrangeWeightsToWinograd4x4To6x6Weights()120 dst_weights->data.resize(dst_shape.DimensionsProduct()); in RearrangeWeightsToWinograd4x4To6x6Weights()145 dst_weights->data[f_index] = out_vals[y * 6 + x]; in RearrangeWeightsToWinograd4x4To6x6Weights()
40 Tensor<OHWI, DataType::FLOAT32>* dst_weights);