Searched refs:RearrangeWeightsForDWConv3D (Results 1 – 1 of 1) sorted by relevance
185 void RearrangeWeightsForDWConv3D(const tflite::gpu::Tensor<OHWDI, S>& weights, in RearrangeWeightsForDWConv3D() function234 RearrangeWeightsForDWConv3D(weights, absl::MakeSpan(ptr, elements_count)); in UploadWeightsForDWConv3D()237 RearrangeWeightsForDWConv3D(weights, absl::MakeSpan(ptr, elements_count)); in UploadWeightsForDWConv3D()