Searched refs:batch_var_ptr (Results 1 – 1 of 1) sorted by relevance
917 auto batch_var_ptr = StreamExecutorUtil::AsDeviceMemory<U>(*batch_var); in operator ()() local948 ->ThenMemcpyD2D(&batch_var_ptr, estimated_variance_ptr, in operator ()()962 &batch_var_ptr, &saved_mean_ptr, &saved_inv_var_ptr, in operator ()()