Searched refs:num_matrices (Results 1 – 1 of 1) sorted by relevance
129 const int64 num_matrices = output.size(); in operator ()() local131 CudaLaunchConfig config = GetCudaLaunchConfig(num_matrices, device); in operator ()()152 const int64 num_matrices = sign.size(); in operator ()() local154 CudaLaunchConfig config = GetCudaLaunchConfig(num_matrices, device); in operator ()()