Searched refs:total_dim_size (Results 1 – 1 of 1) sorted by relevance
176 size_t total_dim_size = var_first_dim_size_ * var_outer_dim_size_; in LaunchKernel() local184 MultiThreadCompute<T>(ComputeMomentum<T>, &input_params, total_dim_size); in LaunchKernel()198 MultiThreadCompute<T>(ComputeWeight<T>, &input_params, total_dim_size); in LaunchKernel()