Searched refs:kernel_z (Results 1 – 2 of 2) sorted by relevance
114 const int kernel_z = weights.shape.d; in RearrangeWeightsForDWConv3D() local118 for (int z = 0; z < kernel_z; ++z) { in RearrangeWeightsForDWConv3D()148 const int kernel_z = weights.shape.d; in UploadWeightsForDWConv3D() local150 const int elements_count = kernel_x * kernel_y * kernel_z * dst_slices; in UploadWeightsForDWConv3D()176 desc.size = int2(kernel_x * kernel_y * kernel_z, dst_slices); in UploadWeightsForDWConv3D()
148 const int kernel_z = weights.shape.d; in UploadWeights() local151 kernel_x * kernel_y * kernel_z * src_depth * dst_depth * 4; in UploadWeights()187 int texture_height = src_depth * kernel_x * kernel_y * kernel_z; in UploadWeights()