Searched refs:block_to_tensor (Results 1 – 6 of 6) sorted by relevance
45 unsigned char block_to_tensor[depth_to_max_blocks[n - 1]]; member55 unsigned char block_to_tensor[depth_to_max_blocks[n - 1]]; member69 unsigned char block_to_tensor[depth_to_max_blocks[1 - 1]]; member81 unsigned char block_to_tensor[depth_to_max_blocks[2 - 1]]; member93 unsigned char block_to_tensor[depth_to_max_blocks[n - 1]]; member163 tensorListMeta.block_to_tensor[loc_block_info] = loc_tensor_info - 1; in multi_tensor_apply()248 tensorListMeta.block_to_tensor[loc_block_info] = loc_tensor_info - 1; in multi_tensor_apply()331 tensorListMeta.block_to_tensor[loc_block_info] = loc_tensor_info - 1; in multi_tensor_apply_for_fused_optimizer()
215 const int tensor_loc = tl.block_to_tensor[blockIdx.x]; in operator ()()238 const auto tensor_loc = tl.block_to_tensor[blockIdx.x]; in operator ()()263 const auto tensor_loc = tl.block_to_tensor[blockIdx.x]; in operator ()()316 const int tensor_loc = tl.block_to_tensor[blockIdx.x]; in operator ()()369 const auto tensor_loc = tl.block_to_tensor[blockIdx.x]; in operator ()()412 const auto tensor_loc = tl.block_to_tensor[blockIdx.x]; in operator ()()465 const auto tensor_loc = tl.block_to_tensor[blockIdx.x]; in operator ()()488 const auto tensor_loc = tl.block_to_tensor[blockIdx.x]; in operator ()()512 const auto tensor_loc = tl.block_to_tensor[blockIdx.x]; in operator ()()566 const auto tensor_loc = tl.block_to_tensor[blockIdx.x]; in operator ()()[all …]
60 const auto tensor_loc = tl.block_to_tensor[blockIdx.x]; in operator ()()250 const auto tensor_loc = tl.block_to_tensor[blockIdx.x]; in operator ()()
123 const auto tensor_loc = tl.block_to_tensor[blockIdx.x]; in operator ()()
80 const auto tensor_loc = tl.block_to_tensor[blockIdx.x]; in operator ()()
342 const auto tensor_loc = tl.block_to_tensor[blockIdx.x]; in operator ()()