Searched defs:FusedOptimizerTensorListMetadata (Results 1 – 1 of 1) sorted by relevance
89 struct FusedOptimizerTensorListMetadata { struct90 const void* addresses[n][depth_to_max_tensors[n - 1]];91 int64_t numel_for_tensor[depth_to_max_tensors[n - 1]];92 const void* state_steps_addresses[depth_to_max_tensors_scalarlist[n - 1]];93 unsigned char block_to_tensor[depth_to_max_blocks[n - 1]];94 int block_to_chunk[depth_to_max_blocks[n - 1]];95 int start_tensor_this_launch;