Searched refs:aligned_channels (Results 1 – 2 of 2) sorted by relevance
244 int aligned_channels = AlignByN(bias.shape.v, 4 * conv_params_.block_size.w); in UploadBias() local245 desc.size = float_size * aligned_channels; in UploadBias()249 for (int i = 0; i < aligned_channels; ++i) { in UploadBias()254 for (int i = 0; i < aligned_channels; ++i) { in UploadBias()
1310 int aligned_channels = storage_type_ == TensorStorageType::SINGLE_TEXTURE_2D in GetSizeInBytesForShape() local1314 shape5d.b * shape5d.w * shape5d.h * shape5d.d * aligned_channels; in GetSizeInBytesForShape()