Searched refs:aligned_channels (Results 1 – 4 of 4) sorted by relevance
236 int aligned_channels = AlignByN(bias.shape.v, 4 * conv_params_.block_size.w); in UploadBias() local237 desc.size = float_size * aligned_channels; in UploadBias()241 for (int i = 0; i < aligned_channels; ++i) { in UploadBias()246 for (int i = 0; i < aligned_channels; ++i) { in UploadBias()
385 const int aligned_channels = GetAlignedChannels(); in WriteDataBHWDC() local387 shape_.b * shape_.w * shape_.h * shape_.d * aligned_channels; in WriteDataBHWDC()451 const int aligned_channels = GetAlignedChannels(); in ReadDataBHWDC() local453 shape_.b * shape_.w * shape_.h * shape_.d * aligned_channels; in ReadDataBHWDC()
489 const int aligned_channels = GetAlignedChannels(); in WriteDataBHWDC() local491 shape_.b * shape_.w * shape_.h * shape_.d * aligned_channels; in WriteDataBHWDC()551 const int aligned_channels = GetAlignedChannels(); in ReadDataBHWDC() local553 shape_.b * shape_.w * shape_.h * shape_.d * aligned_channels; in ReadDataBHWDC()
905 int aligned_channels = storage_type == TensorStorageType::SINGLE_TEXTURE_2D in UploadData() local908 int elements_count = shape.b * shape.w * shape.h * shape.d * aligned_channels; in UploadData()