Searched refs:src_batch (Results 1 – 3 of 3) sorted by relevance
32 int src_channels, int src_batch) { in GetOffset() argument70 offset.w = src_batch + attr.ends.b; in GetOffset()
193 const unsigned int src_batch = src.shape()[3]; in compute_reduction_operation() local258 …const int in_offset = du * src_batch * src_depth * src_height * src_width + z * src_width * src_… in compute_reduction_operation()
324 const VkSubmitInfo *src_batch = &submit->submit_batches[batch_index]; in vn_fix_device_group_cmd_count() local331 vk_foreach_struct_const(src, src_batch->pNext) { in vn_fix_device_group_cmd_count()