Searched refs:copy_channels (Results 1 – 1 of 1) sorted by relevance
75 static void copy_tensor(const TENSOR *src, int copy_channels, int dst_offset, in copy_tensor() argument79 assert(copy_channels <= src->channels); in copy_tensor()81 for (int c = 0; c < copy_channels; ++c) { in copy_tensor()86 for (int c = 0; c < copy_channels; ++c) { in copy_tensor()336 int copy_channels = branch_config->channels_to_copy > 0 in copy_active_tensor_to_branches() local339 realloc_tensor(&branch_output[b], copy_channels, in copy_active_tensor_to_branches()341 copy_tensor(layer_active_tensor, copy_channels, 0, &branch_output[b]); in copy_active_tensor_to_branches()