Searched refs:RearrangeWeightsForDWConv2D (Results 1 – 1 of 1) sorted by relevance
117 void RearrangeWeightsForDWConv2D(const tflite::gpu::Tensor<OHWI, S>& weights, in RearrangeWeightsForDWConv2D() function162 RearrangeWeightsForDWConv2D(weights, absl::MakeSpan(ptr, elements_count)); in UploadWeightsForDWConv2D()165 RearrangeWeightsForDWConv2D(weights, absl::MakeSpan(ptr, elements_count)); in UploadWeightsForDWConv2D()