Searched refs:unpacked_tensor (Results 1 – 2 of 2) sorted by relevance
74 …auto unpacked_tensor = at::empty({rgba_size, packed_tensor.size(1), packed_tensor.size(2)}, at::CP… in unpack_rgb() local75 uint8_t* unpacked = (uint8_t*) unpacked_tensor.data_ptr<uint8_t>(); in unpack_rgb()85 return unpacked_tensor; in unpack_rgb()89 const at::Tensor& unpacked_tensor, // IN in pack_rgb() argument94 uint8_t* unpacked = (uint8_t*)unpacked_tensor.data_ptr<uint8_t>(); in pack_rgb()99 auto unpacked_increment = unpacked_tensor.size(0); in pack_rgb()
160 for accumulator, unpacked_tensor in zip(162 accumulator.append(unpacked_tensor)