Searched refs:flt4_count (Results 1 – 2 of 2) sorted by relevance
71 const int flt4_count = in UploadData() local81 desc.size = flt4_size * (flt4_count + 1); in UploadData()86 RearrangeWeightsData(weights, absl::MakeSpan(gpu_data, flt4_count)); in UploadData()91 gpu_data[flt4_count] = bias_value; in UploadData()94 RearrangeWeightsData(weights, absl::MakeSpan(gpu_data, flt4_count)); in UploadData()99 gpu_data[flt4_count] = bias_value; in UploadData()
34 const size_t flt4_count = shape.b * shape.w * shape.h * shape.d * slices; in CreateTextureBuffer() local35 const size_t data_size = flt4_count * 4 * SizeOf(descriptor.data_type); in CreateTextureBuffer()37 texture_desc.width = flt4_count; in CreateTextureBuffer()