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