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